Source file query.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 Keys = struct
  let fields = "fields"
end

type t = { fields : SkhcQuery.Substring.t list }

let section = "query"
let default = { fields = SkhcQuery.Substring.default }
let init fields = { fields }

let of_section section =
  let fields =
    match Section.assoc Keys.fields section with
    | None ->
        SkhcQuery.Substring.default
    | Some s ->
        s |> String.split_on_char ' '
        |> List.filter_map SkhcQuery.Substring.of_string
  in
  { fields }

let to_section t =
  let { fields } = t in
  let fields =
    String.concat " " (List.map SkhcQuery.Substring.to_string fields)
  in
  Section.init section [ (Keys.fields, fields) ]

let fields t = t.fields