Source file queryUtil.ml

(*****************************************************************************)
(*                                                                           *)
(*  Copyright (C) 2026 Yves Ndiaye                                           *)
(*                                                                           *)
(* This Source Code Form is subject to the terms of the Mozilla Public       *)
(* License, v. 2.0. If a copy of the MPL was not distributed with this       *)
(* file, You can obtain one at https://mozilla.org/MPL/2.0/.                 *)
(*                                                                           *)
(*****************************************************************************)

module SubFields = Substrings

let add_field_if_some key field fields =
  match field with None -> fields | Some t -> SubFields.add key t fields

let re_compute ~insensible regex s =
  match regex with
  | true ->
      let opts = match insensible with true -> [ `ICase ] | false -> [] in
      Re.Posix.re ~opts s
  | false ->
      let re = Re.str s in
      let re = match insensible with true -> Re.no_case re | false -> re in
      re

let omatches ~insensible regex s =
  let re = re_compute ~insensible regex s in
  let re = Re.compile re in
  Option.fold ~none:false ~some:(Re.execp re)

let matches ~insensible regex s w =
  let re = re_compute ~insensible regex s in
  let re = Re.compile re in
  Re.execp re w

let match_participants participants =
  SkhcDb.Ids.Map.fold
    (fun _ (artist_name, roles) ->
      SkhcDb.Roles.fold
        (fun role fields ->
          let open Substring in
          match role with
          | SkhcDb.Role.Artist ->
              SubFields.add SArtist artist_name fields
          | AlbumArtist ->
              SubFields.add SAlbumArtist artist_name fields
          | Composer ->
              SubFields.add SComposer artist_name fields
          | Unknown ->
              fields
        )
        roles
    )
    participants SubFields.empty