added working object_exists function

This commit is contained in:
Scott Chacon
2010-05-25 07:52:28 -07:00
parent abdd070a77
commit 7b4b8cf4e4
3 changed files with 39 additions and 5 deletions

7
geef
View File

@@ -8,7 +8,12 @@ main([Command|_Args]) ->
"h2r" ->
[Object|_Rest] = _Args,
Raw = geef:hex_to_raw(Object),
io:format("Raw ~w~n", [Raw]);
io:format("Raw: ~w~n", [Raw]);
"exists" ->
[Path|_Rest] = _Args,
[Object|_None] = _Rest,
Exists = geef:object_exists(Path, Object),
io:format("Exists: ~w~n", [Exists]);
_Else ->
usage()
end;

32
geef.c
View File

@@ -2,6 +2,7 @@
#include <stdio.h>
#include <git/common.h>
#include <git/oid.h>
#include <git/odb.h>
#define MAXBUFLEN 1024
@@ -15,8 +16,6 @@ geef_hex_to_raw(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
return enif_make_badarg(env);
git_oid oid;
printf("sha: %s\n", sha);
git_oid_mkstr(&oid, sha);
ErlNifBinary ibin;
@@ -26,9 +25,36 @@ geef_hex_to_raw(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
return enif_make_binary(env, &ibin);
}
static ERL_NIF_TERM
geef_object_exists(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
{
char path[MAXBUFLEN];
char sha[MAXBUFLEN];
(void)memset(&path, '\0', sizeof(path));
(void)memset(&sha, '\0', sizeof(sha));
if (enif_get_string(env, argv[0], path, sizeof(path), ERL_NIF_LATIN1) < 1)
return enif_make_badarg(env);
if (enif_get_string(env, argv[1], sha, sizeof(sha), ERL_NIF_LATIN1) < 1)
return enif_make_badarg(env);
git_odb *odb;
git_odb_open(&odb, path);
git_oid oid;
git_oid_mkstr(&oid, sha);
int exists = git_odb_exists(odb, &oid);
if(exists == 1) {
return enif_make_atom(env, "true");
}
return enif_make_atom(env, "false");
}
static ErlNifFunc geef_funcs[] =
{
{"hex_to_raw", 1, geef_hex_to_raw}
{"hex_to_raw", 1, geef_hex_to_raw},
{"object_exists", 2, geef_object_exists}
};
ERL_NIF_INIT(geef, geef_funcs, NULL, NULL, NULL, NULL)

View File

@@ -1,5 +1,5 @@
-module(geef).
-export([start/0, hex_to_raw/1]).
-export([start/0, hex_to_raw/1, object_exists/2]).
start() ->
erlang:load_nif("geef", 0).
@@ -7,5 +7,8 @@ start() ->
hex_to_raw(_Val) ->
nif_error(?LINE).
object_exists(_Val, _Val2) ->
nif_error(?LINE).
nif_error(Line) ->
exit({nif_not_loaded,module,?MODULE,line,Line}).