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" -> "h2r" ->
[Object|_Rest] = _Args, [Object|_Rest] = _Args,
Raw = geef:hex_to_raw(Object), 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 -> _Else ->
usage() usage()
end; end;

32
geef.c
View File

@@ -2,6 +2,7 @@
#include <stdio.h> #include <stdio.h>
#include <git/common.h> #include <git/common.h>
#include <git/oid.h> #include <git/oid.h>
#include <git/odb.h>
#define MAXBUFLEN 1024 #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); return enif_make_badarg(env);
git_oid oid; git_oid oid;
printf("sha: %s\n", sha);
git_oid_mkstr(&oid, sha); git_oid_mkstr(&oid, sha);
ErlNifBinary ibin; 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); 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[] = 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) ERL_NIF_INIT(geef, geef_funcs, NULL, NULL, NULL, NULL)

View File

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