OCaml library for JSONfeed parsing and creation
anil.recoil.org
1(*---------------------------------------------------------------------------
2 Copyright (c) 2024 Anil Madhavapeddy. All rights reserved.
3 SPDX-License-Identifier: ISC
4 ---------------------------------------------------------------------------*)
5
6module Unknown = struct
7 type t = (string * Jsont.json) list
8
9 let empty = []
10 let is_empty = function [] -> true | _ -> false
11end
12
13type t = {
14 url : string;
15 mime_type : string;
16 title : string option;
17 size_in_bytes : int64 option;
18 duration_in_seconds : int option;
19 unknown : Unknown.t;
20}
21
22let create ~url ~mime_type ?title ?size_in_bytes ?duration_in_seconds ?(unknown = Unknown.empty) () =
23 { url; mime_type; title; size_in_bytes; duration_in_seconds; unknown }
24
25let url t = t.url
26let mime_type t = t.mime_type
27let title t = t.title
28let size_in_bytes t = t.size_in_bytes
29let duration_in_seconds t = t.duration_in_seconds
30let unknown t = t.unknown
31
32let equal a b =
33 a.url = b.url &&
34 a.mime_type = b.mime_type &&
35 a.title = b.title &&
36 a.size_in_bytes = b.size_in_bytes &&
37 a.duration_in_seconds = b.duration_in_seconds
38
39let pp ppf t =
40 (* Extract filename from URL *)
41 let filename =
42 try
43 let parts = String.split_on_char '/' t.url in
44 List.nth parts (List.length parts - 1)
45 with _ -> t.url
46 in
47
48 Format.fprintf ppf "%s (%s" filename t.mime_type;
49
50 (match t.size_in_bytes with
51 | Some size ->
52 let mb = Int64.to_float size /. (1024. *. 1024.) in
53 Format.fprintf ppf ", %.1f MB" mb
54 | None -> ());
55
56 (match t.duration_in_seconds with
57 | Some duration ->
58 let mins = duration / 60 in
59 let secs = duration mod 60 in
60 Format.fprintf ppf ", %dm%ds" mins secs
61 | None -> ());
62
63 Format.fprintf ppf ")"
64
65let jsont =
66 let kind = "Attachment" in
67 let doc = "An attachment object" in
68 let unknown_mems : (Unknown.t, Jsont.json, Jsont.mem list) Jsont.Object.Mems.map =
69 let open Jsont.Object.Mems in
70 let dec_empty () = [] in
71 let dec_add _meta (name : string) value acc =
72 ((name, Jsont.Meta.none), value) :: acc
73 in
74 let dec_finish _meta mems =
75 List.rev_map (fun ((name, _meta), value) -> (name, value)) mems in
76 let enc = {
77 enc = fun (type acc) (f : Jsont.Meta.t -> string -> Jsont.json -> acc -> acc) unknown (acc : acc) ->
78 List.fold_left (fun acc (name, value) ->
79
80 f Jsont.Meta.none name value acc
81 ) acc unknown
82 } in
83 map ~kind:"Unknown members" Jsont.json ~dec_empty ~dec_add ~dec_finish ~enc
84 in
85 let create_obj url mime_type title size_in_bytes duration_in_seconds unknown =
86 create ~url ~mime_type ?title ?size_in_bytes ?duration_in_seconds ~unknown ()
87 in
88 Jsont.Object.map ~kind ~doc create_obj
89 |> Jsont.Object.mem "url" Jsont.string ~enc:url
90 |> Jsont.Object.mem "mime_type" Jsont.string ~enc:mime_type
91 |> Jsont.Object.opt_mem "title" Jsont.string ~enc:title
92 |> Jsont.Object.opt_mem "size_in_bytes" Jsont.int64 ~enc:size_in_bytes
93 |> Jsont.Object.opt_mem "duration_in_seconds" Jsont.int ~enc:duration_in_seconds
94 |> Jsont.Object.keep_unknown unknown_mems ~enc:unknown
95 |> Jsont.Object.finish