74 lines
2.2 KiB
OCaml
74 lines
2.2 KiB
OCaml
module Asn = struct
|
|
let decode_strict codec cs =
|
|
match Asn.decode codec cs with
|
|
| Ok (a, cs) ->
|
|
if Cstruct.len cs = 0
|
|
then Ok a
|
|
else Error "trailing bytes"
|
|
| Error (`Parse msg) -> Error ("parse error: " ^ msg)
|
|
|
|
let projections_of asn =
|
|
let c = Asn.codec Asn.der asn in
|
|
(decode_strict c, Asn.encode c)
|
|
|
|
let console =
|
|
Asn.S.(sequence_of
|
|
(sequence2
|
|
(required ~label:"delta" int)
|
|
(required ~label:"data" utf8_string)))
|
|
|
|
let console_of_cs, console_to_cs = projections_of console
|
|
end
|
|
|
|
type id = int64
|
|
let id = Caqti_type.int64
|
|
|
|
let uuid =
|
|
let encode uuid = Ok (Uuidm.to_bytes uuid) in
|
|
let decode s =
|
|
Uuidm.of_bytes s
|
|
|> Option.to_result ~none:"failed to decode uuid"
|
|
in
|
|
Caqti_type.custom ~encode ~decode Caqti_type.string
|
|
|
|
|
|
let ptime =
|
|
let encode t = Ok (Ptime.Span.to_d_ps (Ptime.to_span t)) in
|
|
let decode (d, ps) = Ok (Ptime.v (d, ps))
|
|
in
|
|
let rep = Caqti_type.(tup2 int int64) in
|
|
Caqti_type.custom ~encode ~decode rep
|
|
|
|
let fpath =
|
|
let encode t = Ok (Fpath.to_string t) in
|
|
let decode s = Fpath.of_string s
|
|
|> Result.map_error (fun (`Msg s) -> s) in
|
|
Caqti_type.custom ~encode ~decode Caqti_type.string
|
|
|
|
let cstruct =
|
|
let encode t = Ok (Cstruct.to_string t) in
|
|
let decode s = Ok (Cstruct.of_string s) in
|
|
Caqti_type.custom ~encode ~decode Caqti_type.octets
|
|
|
|
let execution_result =
|
|
let encode = function
|
|
| Builder.Exited v -> Ok (0, Some v, None)
|
|
| Builder.Signalled v -> Ok (1, Some v, None)
|
|
| Builder.Stopped v -> Ok (2, Some v, None)
|
|
| Builder.Msg msg -> Ok (3, None, Some msg)
|
|
in
|
|
let decode (kind, code, msg) =
|
|
match kind, code, msg with
|
|
| 0, Some v, None -> Ok (Builder.Exited v)
|
|
| 1, Some v, None -> Ok (Builder.Signalled v)
|
|
| 2, Some v, None -> Ok (Builder.Stopped v)
|
|
| 3, None, Some msg -> Ok (Builder.Msg msg)
|
|
| _ -> Error "bad encoding"
|
|
in
|
|
let rep = Caqti_type.(tup3 int (option int) (option string)) in
|
|
Caqti_type.custom ~encode ~decode rep
|
|
|
|
let console =
|
|
let encode console = Ok (Asn.console_to_cs console) in
|
|
let decode data = Asn.console_of_cs data in
|
|
Caqti_type.custom ~encode ~decode cstruct
|