Introduce geef_index

c_src/geef.c

@@ -8,6 +8,7 @@
 #include "blob.h"
 #include "library.h"
 #include "revwalk.h"
+#include "index.h"
 #include "geef.h"
 #include <stdio.h>
 #include <string.h>
@@ -18,6 +19,7 @@ ErlNifResourceType *geef_odb_type;
 ErlNifResourceType *geef_ref_type;
 ErlNifResourceType *geef_object_type;
 ErlNifResourceType *geef_revwalk_type;
+ErlNifResourceType *geef_index_type;
 
 geef_atoms atoms;
 
@@ -49,6 +51,9 @@ static int load(ErlNifEnv *env, void **priv, ERL_NIF_TERM load_info)
 	geef_revwalk_type = enif_open_resource_type(env, NULL,
 		  "revwalk_type", geef_revwalk_free, ERL_NIF_RT_CREATE, NULL);
 
+	geef_index_type = enif_open_resource_type(env, NULL,
+		  "index_type", geef_index_free, ERL_NIF_RT_CREATE, NULL);
+
 	if (geef_repository_type == NULL)
 		return -1;
 
@@ -138,6 +143,11 @@ static ErlNifFunc geef_funcs[] =
 	{"revwalk_next", 1,    geef_revwalk_next},
 	{"revwalk_sorting", 2, geef_revwalk_sorting},
 	{"revwalk_reset", 1,   geef_revwalk_reset},
+	{"index_new",   0, geef_revwalk_new},
+	{"index_write", 1, geef_index_write},
+	{"index_write_tree", 1, geef_index_write_tree},
+	{"index_write_tree", 2, geef_index_write_tree},
+	{"index_clear", 1, geef_index_clear},
 };
 
 ERL_NIF_INIT(geef_nif, geef_funcs, load, NULL, upgrade, unload)

c_src/index.c

@@ -0,0 +1,87 @@
+#include "geef.h"
+#include "oid.h"
+#include "index.h"
+#include <string.h>
+#include <git2.h>
+
+void geef_index_free(ErlNifEnv *env, void *cd)
+{
+	geef_index *index = (geef_index *) cd;
+	git_index_free(index->index);
+}
+
+ERL_NIF_TERM
+geef_index_new(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
+{
+	geef_index *index;
+	ERL_NIF_TERM term;
+
+	index = enif_alloc_resource(geef_index_type, sizeof(geef_index));
+	if (!index)
+		return atoms.error;
+
+	if (git_index_new(&index->index) < 0)
+		return geef_error(env);
+
+	term = enif_make_resource(env, index);
+	enif_release_resource(index);
+
+	return enif_make_tuple2(env, atoms.ok, term);
+}
+
+ERL_NIF_TERM
+geef_index_write(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
+{
+	geef_index *index;
+
+	if (!enif_get_resource(env, argv[0], geef_index_type, (void **) &index))
+		return enif_make_badarg(env);
+
+	if (git_index_write(index->index) < 0)
+		return geef_error(env);
+
+	return atoms.ok;
+}
+
+ERL_NIF_TERM
+geef_index_write_tree(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
+{
+	geef_index *index;
+	geef_repository *repo;
+	ErlNifBinary bin;
+	git_oid id;
+	int error;
+
+	if (!enif_get_resource(env, argv[0], geef_index_type, (void **) &index))
+		return enif_make_badarg(env);
+
+	if (argc == 2) {
+		if (!enif_get_resource(env, argv[0], geef_repository_type, (void **) &repo))
+			return enif_make_badarg(env);
+
+		error = git_index_write_tree_to(&id, index->index, repo->repo);
+	} else {
+		error = git_index_write_tree(&id, index->index);
+	}
+
+	if (error < 0)
+		return geef_error(env);
+
+	if (geef_oid_bin(&bin, &id) < 0)
+		return atoms.error;
+
+	return enif_make_tuple2(env, atoms.ok, enif_make_binary(env, &bin));
+}
+
+ERL_NIF_TERM
+geef_index_clear(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
+{
+	geef_index *index;
+
+	if (!enif_get_resource(env, argv[0], geef_index_type, (void **) &index))
+		return enif_make_badarg(env);
+
+	git_index_clear(index->index);
+
+	return atoms.ok;
+}

c_src/index.h

@@ -0,0 +1,21 @@
+#ifndef GEEF_INDEX_H
+#define GEEF_INDEX_H
+
+#include "erl_nif.h"
+#include <git2.h>
+#include "repository.h"
+
+extern ErlNifResourceType *geef_index_type;
+
+typedef struct {
+	git_index *index;
+} geef_index;
+
+void geef_index_free(ErlNifEnv *env, void *cd);
+
+ERL_NIF_TERM geef_index_write(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
+ERL_NIF_TERM geef_index_write_tree(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
+ERL_NIF_TERM geef_index_clear(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
+
+
+#endif