57 lines
1.9 KiB
OCaml
57 lines
1.9 KiB
OCaml
|
(**************************************************************************)
|
||
|
|
(* *)
|
||
|
|
(* Copyright (c) 2014 - 2016. *)
|
||
|
|
(* Dynamic Ledger Solutions, Inc. <contact@tezos.com> *)
|
||
|
|
(* *)
|
||
|
|
(* All rights reserved. No warranty, explicit or implicit, provided. *)
|
||
|
|
(* *)
|
||
|
|
(**************************************************************************)
|
||
|
|
|
||
|
|
let to_root = function
|
||
|
|
| `O ctns -> `O ctns
|
||
|
|
| `A ctns -> `A ctns
|
||
|
|
| `Null -> `O []
|
||
|
|
| oth -> `A [ oth ]
|
||
|
|
|
||
|
|
let to_string j = Ezjsonm.to_string ~minify:false (to_root j)
|
||
|
|
|
||
|
|
let from_string s =
|
||
|
|
try Ok (Ezjsonm.from_string s :> Data_encoding.json)
|
||
|
|
with Ezjsonm.Parse_error (_, msg) -> Error msg
|
||
|
|
|
||
|
|
let from_stream (stream: string Lwt_stream.t) =
|
||
|
|
let buffer = ref "" in
|
||
|
|
Lwt_stream.filter_map
|
||
|
|
(fun str ->
|
||
|
|
buffer := !buffer ^ str ;
|
||
|
|
try
|
||
|
|
let json = Ezjsonm.from_string !buffer in
|
||
|
|
buffer := "" ;
|
||
|
|
Some (Ok json)
|
||
|
|
with Ezjsonm.Parse_error (_, msg) ->
|
||
|
|
if String.length str = 32 * 1024 then None
|
||
|
|
else Some (Error msg))
|
||
|
|
stream
|
||
|
|
|
||
|
|
let write_file file json =
|
||
|
|
let json = to_root json in
|
||
|
|
let open Lwt in
|
||
|
|
catch
|
||
|
|
(fun () ->
|
||
|
|
Lwt_io.(with_file ~mode:Output file (fun chan ->
|
||
|
|
let str = to_string json in
|
||
|
|
write chan str >>= fun _ ->
|
||
|
|
return true)))
|
||
|
|
(fun _ -> return false)
|
||
|
|
|
||
|
|
let read_file file =
|
||
|
|
let open Lwt in
|
||
|
|
catch
|
||
|
|
(fun () ->
|
||
|
|
Lwt_io.(with_file ~mode:Input file (fun chan ->
|
||
|
|
read chan >>= fun str ->
|
||
|
|
return (Some (Ezjsonm.from_string str :> Data_encoding.json)))))
|
||
|
|
(fun _ ->
|
||
|
|
(* TODO log error or use Error_monad. *)
|
||
|
|
return None)
|