(* Auto-generated from "json_repr_object.atd" by atdml. *)

type string_map = (string * int) list

val string_map_of_yojson : Yojson.Safe.t -> string_map
val string_map_of_jsonlike : Atd_jsonlike.AST.t -> string_map
val yojson_of_string_map : string_map -> Yojson.Safe.t
val string_map_of_json : string -> string_map
val json_of_string_map : string_map -> string

module String_map : sig
  type nonrec t = string_map
  val of_yojson : Yojson.Safe.t -> t
  val of_jsonlike : Atd_jsonlike.AST.t -> t
  val to_yojson : t -> Yojson.Safe.t
  val of_json : string -> t
  val to_json : t -> string
end

type nested = (string * string list) list

val nested_of_yojson : Yojson.Safe.t -> nested
val nested_of_jsonlike : Atd_jsonlike.AST.t -> nested
val yojson_of_nested : nested -> Yojson.Safe.t
val nested_of_json : string -> nested
val json_of_nested : nested -> string

module Nested : sig
  type nonrec t = nested
  val of_yojson : Yojson.Safe.t -> t
  val of_jsonlike : Atd_jsonlike.AST.t -> t
  val to_yojson : t -> Yojson.Safe.t
  val of_json : string -> t
  val to_json : t -> string
end

type record = {
  counts: string_map;
  tags: nested;
}

val create_record : counts:string_map -> tags:nested -> unit -> record
val record_of_yojson : Yojson.Safe.t -> record
val record_of_jsonlike : Atd_jsonlike.AST.t -> record
val yojson_of_record : record -> Yojson.Safe.t
val record_of_json : string -> record
val json_of_record : record -> string

module Record : sig
  type nonrec t = record
  val create : counts:string_map -> tags:nested -> unit -> t
  val of_yojson : Yojson.Safe.t -> t
  val of_jsonlike : Atd_jsonlike.AST.t -> t
  val to_yojson : t -> Yojson.Safe.t
  val of_json : string -> t
  val to_json : t -> string
end

--- ml ---
(* Auto-generated from "json_repr_object.atd" by atdml. *)
[@@@ocaml.warning "-27-32-33-35-39"]

(* Inlined runtime — no external dependency needed. *)
module Atdml_runtime = struct
  (* Returns true iff the list has strictly more than [n] elements,
     without traversing past element n+1. *)
  let rec list_length_gt n = function
    | _ :: rest -> if n = 0 then true else list_length_gt (n - 1) rest
    | [] -> false

  module Yojson = struct
    let bad_type expected_type x =
      Printf.ksprintf failwith "expected %s, got: %s"
        expected_type (Yojson.Safe.to_string x)

    let bad_sum type_name x =
      Printf.ksprintf failwith "invalid variant for type '%s': %s"
        type_name (Yojson.Safe.to_string x)

    let missing_field type_name field_name =
      Printf.ksprintf failwith "missing field '%s' in object of type '%s'"
        field_name type_name

    let bool_of_yojson = function
      | `Bool b -> b
      | x -> bad_type "bool" x

    let yojson_of_bool b = `Bool b

    let int_of_yojson = function
      | `Int n -> n
      | x -> bad_type "int" x

    let yojson_of_int n = `Int n

    let float_of_yojson = function
      | `Float f -> f
      | `Int n -> Float.of_int n
      | x -> bad_type "float" x

    let yojson_of_float f = `Float f

    let string_of_yojson = function
      | `String s -> s
      | x -> bad_type "string" x

    let yojson_of_string s = `String s

    let unit_of_yojson = function
      | `Null -> ()
      | x -> bad_type "null" x

    let yojson_of_unit () = `Null

    let list_of_yojson f = function
      | `List xs -> List.map f xs
      | x -> bad_type "array" x

    let yojson_of_list f xs = `List (List.map f xs)

    let option_of_yojson f = function
      | `String "None" -> None
      | `List [`String "Some"; x] -> Some (f x)
      | x -> bad_type "option" x

    let yojson_of_option f = function
      | None -> `String "None"
      | Some x -> `List [`String "Some"; f x]

    let nullable_of_yojson f = function
      | `Null -> None
      | x -> Some (f x)

    let yojson_of_nullable f = function
      | None -> `Null
      | Some x -> f x

    let assoc_of_yojson f = function
      | `Assoc pairs -> List.map (fun (k, v) -> (k, f v)) pairs
      | x -> bad_type "object" x

    let yojson_of_assoc f xs =
      `Assoc (List.map (fun (k, v) -> (k, f v)) xs)
  end

  module Jsonlike = struct
    let bad_type expected_type x =
      Printf.ksprintf failwith "%sexpected %s"
        (Atd_jsonlike.AST.loc_msg x) expected_type

    let bad_sum type_name x =
      Printf.ksprintf failwith "%sinvalid variant for type '%s'"
        (Atd_jsonlike.AST.loc_msg x) type_name

    let missing_field node type_name field_name =
      Printf.ksprintf failwith "%smissing field '%s' in object of type '%s'"
        (Atd_jsonlike.AST.loc_msg node) field_name type_name

    let bool_of_jsonlike = function
      | Atd_jsonlike.AST.Bool (_, b) -> b
      | x -> bad_type "bool" x

    let int_of_jsonlike = function
      | Atd_jsonlike.AST.Number (_, n) as node ->
          (match n.Atd_jsonlike.Number.int with
          | Some i -> i
          | None -> bad_type "integer" node)
      | x -> bad_type "int" x

    let float_of_jsonlike = function
      | Atd_jsonlike.AST.Number (_, n) as node ->
          (match n.Atd_jsonlike.Number.float with
          | Some f -> f
          | None -> bad_type "float" node)
      | x -> bad_type "float" x

    let string_of_jsonlike = function
      | Atd_jsonlike.AST.String (_, s) -> s
      | x -> bad_type "string" x

    let unit_of_jsonlike = function
      | Atd_jsonlike.AST.Null _ -> ()
      | x -> bad_type "null" x

    let list_of_jsonlike f = function
      | Atd_jsonlike.AST.Array (_, xs) -> List.map f xs
      | x -> bad_type "array" x

    let option_of_jsonlike f = function
      | Atd_jsonlike.AST.String (_, "None") -> None
      | Atd_jsonlike.AST.Array (_, [Atd_jsonlike.AST.String (_, "Some"); x]) -> Some (f x)
      | x -> bad_type "option" x

    let nullable_of_jsonlike f = function
      | Atd_jsonlike.AST.Null _ -> None
      | x -> Some (f x)

    let assoc_of_jsonlike f = function
      | Atd_jsonlike.AST.Object (_, pairs) ->
          List.map (fun (_, k, v) -> (k, f v)) pairs
      | x -> bad_type "object" x
  end
end

type string_map = (string * int) list

let string_map_of_yojson (x : Yojson.Safe.t) : string_map =
  (Atdml_runtime.Yojson.assoc_of_yojson Atdml_runtime.Yojson.int_of_yojson) x

let string_map_of_jsonlike (x : Atd_jsonlike.AST.t) : string_map =
  (Atdml_runtime.Jsonlike.assoc_of_jsonlike Atdml_runtime.Jsonlike.int_of_jsonlike) x

let yojson_of_string_map (x : string_map) : Yojson.Safe.t =
  (Atdml_runtime.Yojson.yojson_of_assoc Atdml_runtime.Yojson.yojson_of_int) x

let string_map_of_json s =
  string_map_of_yojson (Yojson.Safe.from_string s)

let json_of_string_map x =
  Yojson.Safe.to_string (yojson_of_string_map x)

module String_map = struct
  type nonrec t = string_map
  let of_yojson = string_map_of_yojson
  let of_jsonlike = string_map_of_jsonlike
  let to_yojson = yojson_of_string_map
  let of_json = string_map_of_json
  let to_json = json_of_string_map
end

type nested = (string * string list) list

let nested_of_yojson (x : Yojson.Safe.t) : nested =
  (Atdml_runtime.Yojson.assoc_of_yojson (Atdml_runtime.Yojson.list_of_yojson Atdml_runtime.Yojson.string_of_yojson)) x

let nested_of_jsonlike (x : Atd_jsonlike.AST.t) : nested =
  (Atdml_runtime.Jsonlike.assoc_of_jsonlike (Atdml_runtime.Jsonlike.list_of_jsonlike Atdml_runtime.Jsonlike.string_of_jsonlike)) x

let yojson_of_nested (x : nested) : Yojson.Safe.t =
  (Atdml_runtime.Yojson.yojson_of_assoc (Atdml_runtime.Yojson.yojson_of_list Atdml_runtime.Yojson.yojson_of_string)) x

let nested_of_json s =
  nested_of_yojson (Yojson.Safe.from_string s)

let json_of_nested x =
  Yojson.Safe.to_string (yojson_of_nested x)

module Nested = struct
  type nonrec t = nested
  let of_yojson = nested_of_yojson
  let of_jsonlike = nested_of_jsonlike
  let to_yojson = yojson_of_nested
  let of_json = nested_of_json
  let to_json = json_of_nested
end

type record = {
  counts: string_map;
  tags: nested;
}

let create_record ~counts ~tags () : record =
  { counts; tags }

let record_of_yojson (x : Yojson.Safe.t) : record =
  match x with
  | `Assoc fields ->
    (* Duplicate JSON keys: behavior is unspecified (RFC 8259 §4 says keys SHOULD
       be unique). Below the threshold, List.assoc_opt returns the first binding;
       above it, the hashtable returns the last. *)
    let assoc_ =
      if Atdml_runtime.list_length_gt 5 fields then
        let tbl = Hashtbl.create 16 in
        List.iter (fun (k, v) -> Hashtbl.add tbl k v) fields;
        (fun key -> Hashtbl.find_opt tbl key)
      else (fun key -> List.assoc_opt key fields)
    in
    let counts =
      match assoc_ "counts" with
      | Some v -> string_map_of_yojson v
      | None -> Atdml_runtime.Yojson.missing_field "record" "counts"
    in
    let tags =
      match assoc_ "tags" with
      | Some v -> nested_of_yojson v
      | None -> Atdml_runtime.Yojson.missing_field "record" "tags"
    in
    { counts; tags }
  | _ -> Atdml_runtime.Yojson.bad_type "record" x

let record_of_jsonlike (x : Atd_jsonlike.AST.t) : record =
  match x with
  | Atd_jsonlike.AST.Object (_, fields) ->
    let atdml_node_ = x in
    let assoc_ =
      if Atdml_runtime.list_length_gt 5 fields then
        let tbl = Hashtbl.create 16 in
        List.iter (fun (_, k, v) -> Hashtbl.add tbl k v) fields;
        (fun key -> Hashtbl.find_opt tbl key)
      else
        (fun key ->
          match List.find_opt (fun (_, k, _) -> k = key) fields with
          | None -> None | Some (_, _, v) -> Some v)
    in
    let counts =
      match assoc_ "counts" with
      | Some v -> string_map_of_jsonlike v
      | None -> Atdml_runtime.Jsonlike.missing_field atdml_node_ "record" "counts"
    in
    let tags =
      match assoc_ "tags" with
      | Some v -> nested_of_jsonlike v
      | None -> Atdml_runtime.Jsonlike.missing_field atdml_node_ "record" "tags"
    in
    { counts; tags }
  | _ -> Atdml_runtime.Jsonlike.bad_type "record" x

let yojson_of_record (x : record) : Yojson.Safe.t =
  `Assoc (List.concat [
    [("counts", yojson_of_string_map x.counts)];
    [("tags", yojson_of_nested x.tags)];
  ])

let record_of_json s =
  record_of_yojson (Yojson.Safe.from_string s)

let json_of_record x =
  Yojson.Safe.to_string (yojson_of_record x)

module Record = struct
  type nonrec t = record
  let create = create_record
  let of_yojson = record_of_yojson
  let of_jsonlike = record_of_jsonlike
  let to_yojson = yojson_of_record
  let of_json = record_of_json
  let to_json = json_of_record
end

--- Input:
{ "counts": { "a": 1, "b": 2 }, "tags": { "x": [ "p", "q" ] } }
--- Output:
{ "counts": { "a": 1, "b": 2 }, "tags": { "x": [ "p", "q" ] } }
