From 2ef62f3307e43bc7c12d02a4f1527e02d501f1cf Mon Sep 17 00:00:00 2001 From: Romain Calascibetta Date: Wed, 21 Sep 2022 12:01:51 +0200 Subject: [PATCH] Implement exists function --- src/git_kv.ml | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/git_kv.ml b/src/git_kv.ml index e7c4cb9..e41abed 100644 --- a/src/git_kv.ml +++ b/src/git_kv.ml @@ -109,10 +109,18 @@ let of_octets ctx ~remote data = let edn, branch = split_url remote in { ctx ; edn ; branch ; store ; head = Some head } -let exists _t _key = - (* Search.find t.store t.head (`Path (Mirage_kv.Key.segments key)) >>= function *) - (* ([`Value | `Dictionary] option, error) result Lwt.t *) - assert false +let exists t key = + let open Lwt.Infix in + match t.head with + | None -> Lwt.return (Ok None) + | Some head -> + Search.mem t.store head (`Path (Mirage_kv.Key.segments key)) >>= function + | false -> Lwt.return (Ok None) + | true -> + Search.find t.store head (`Path (Mirage_kv.Key.segments key)) + >|= Option.get >>= Store.read_exn t.store >>= function + | Blob _ -> Lwt.return (Ok (Some `Value)) + | Tree _ | Commit _ | Tag _ -> Lwt.return (Ok (Some `Dictionary)) let get t key = let open Lwt.Infix in