Add a way to look up objects

This commit is contained in:
Carlos Martín Nieto
2013-01-30 00:16:52 +01:00
parent ad567fbf41
commit d71c6ff702
7 changed files with 118 additions and 0 deletions

View File

@@ -2,6 +2,7 @@
#include "repository.h"
#include "reference.h"
#include "oid.h"
#include "object.h"
#include "geef.h"
#include <stdio.h>
#include <string.h>
@@ -10,6 +11,7 @@
ErlNifResourceType *geef_repository_type;
ErlNifResourceType *geef_odb_type;
ErlNifResourceType *geef_ref_type;
ErlNifResourceType *geef_object_type;
geef_atoms atoms;
@@ -35,6 +37,12 @@ static int load(ErlNifEnv *env, void **priv, ERL_NIF_TERM load_info)
if (geef_ref_type == NULL)
return -1;
geef_object_type = enif_open_resource_type(env, NULL,
"object_type", geef_object_free, ERL_NIF_RT_CREATE, NULL);
if (geef_repository_type == NULL)
return -1;
atoms.ok = enif_make_atom(env, "ok");
atoms.error = enif_make_atom(env, "error");
atoms.true = enif_make_atom(env, "true");
@@ -42,6 +50,10 @@ static int load(ErlNifEnv *env, void **priv, ERL_NIF_TERM load_info)
atoms.repository = enif_make_atom(env, "repository");
atoms.oid = enif_make_atom(env, "oid");
atoms.symbolic = enif_make_atom(env, "symbolic");
atoms.commit = enif_make_atom(env, "commit");
atoms.tree = enif_make_atom(env, "tree");
atoms.blob = enif_make_atom(env, "blob");
atoms.tag = enif_make_atom(env, "tag");
return 0;
}
@@ -84,6 +96,7 @@ static ErlNifFunc geef_funcs[] =
{"reference_type", 1, geef_reference_type},
{"oid_fmt", 1, geef_oid_fmt},
{"oid_parse", 1, geef_oid_parse},
{"object_lookup", 2, geef_object_lookup},
};
ERL_NIF_INIT(geef, geef_funcs, load, NULL, NULL, unload)

View File

@@ -13,6 +13,10 @@ typedef struct {
ERL_NIF_TERM repository;
ERL_NIF_TERM oid;
ERL_NIF_TERM symbolic;
ERL_NIF_TERM commit;
ERL_NIF_TERM tree;
ERL_NIF_TERM blob;
ERL_NIF_TERM tag;
} geef_atoms;
extern geef_atoms atoms;

62
c_src/object.c Normal file
View File

@@ -0,0 +1,62 @@
#include "geef.h"
#include "repository.h"
#include "object.h"
#include "oid.h"
#include <string.h>
#include <git2.h>
void geef_object_free(ErlNifEnv *env, void *cd)
{
geef_object *obj = (geef_object *) cd;
git_object_free(obj->obj);
}
static ERL_NIF_TERM object_type(git_object *obj)
{
switch(git_object_type(obj)) {
case GIT_OBJ_COMMIT:
return atoms.commit;
case GIT_OBJ_TREE:
return atoms.tree;
case GIT_OBJ_BLOB:
return atoms.blob;
case GIT_OBJ_TAG:
return atoms.tag;
default:
return atoms.error;
}
}
ERL_NIF_TERM
geef_object_lookup(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
{
geef_repository *repo;
ErlNifBinary bin;
git_oid id;
geef_object *obj;
ERL_NIF_TERM term_obj;
if (!enif_get_resource(env, argv[0], geef_repository_type, (void **) &repo))
return enif_make_badarg(env);
if (!enif_inspect_binary(env, argv[1], &bin))
return enif_make_badarg(env);
if (bin.size < GIT_OID_RAWSZ)
return enif_make_badarg(env);
git_oid_fromraw(&id, bin.data);
obj = enif_alloc_resource(geef_object_type, sizeof(geef_object));
if (git_object_lookup(&obj->obj, repo->repo, &id, GIT_OBJ_ANY) < 0) {
enif_release_resource(obj);
return geef_error(env);
}
term_obj = enif_make_resource(env, obj);
enif_release_resource(obj);
return enif_make_tuple3(env, atoms.ok, object_type(obj->obj), term_obj);
}

17
c_src/object.h Normal file
View File

@@ -0,0 +1,17 @@
#ifndef GEEF_OBJECT_H
#define GEEF_OBJECT_H
#include "erl_nif.h"
#include <git2.h>
extern ErlNifResourceType *geef_object_type;
typedef struct {
git_object *obj;
} geef_object;
ERL_NIF_TERM geef_object_lookup(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
void geef_object_free(ErlNifEnv *env, void *cd);
#endif