-module(geef_tree).

-ifdef(TEST).
-include_lib("eunit/include/eunit.hrl").
-endif.

-export([get/2, id/1, lookup/2]).

-include("geef_records.hrl").

-spec get(geef_object(), iolist()) -> {ok, integer(), atom(), geef_oid(), binary()} | {error, term()}.
get(#geef_object{type=tree,handle=Handle}, Path) ->
    case geef_nif:tree_bypath(Handle, Path) of
	{ok, Mode, Type, Geef_Oid, Name} ->
	    {ok, Mode, Type, #geef_oid{oid=Geef_Oid}, Name};
	Other ->
	    Other
    end.

-spec id(geef_object()) -> {ok, geef_oid()} | {error, term()}.
id(Obj = #geef_object{type=tree}) ->
    geef_object:id(Obj).

-spec lookup(pid(), geef_oid() | iolist()) -> {ok, geef_object()} | {error, term()}.
lookup(Repo, Id) ->
    geef_obj:lookup(Repo, Id, tree).

-ifdef(TEST).

%% This is somewhat hacky, assuming that this is running under .eunit,
%% but it's good enough for now
bypath_test() ->
    {ok, Repo} = geef_repo:open(".."),
    {ok, Tree} = geef_obj:lookup(Repo, geef_oid:parse("395e1c39cb203640b78da8458a42afdb92bef7aa")),
    Id = geef_oid:parse("80d5c15a040c93a4f98f4496a05ebf30cdd58650"),
    {ok, 8#100644, blob, Id, <<"README.md">>} = geef_tree:get(Tree, "README.md").
-endif.