Implement ref iterators

c_src/geef.c

@@ -19,6 +19,7 @@
 ErlNifResourceType *geef_repository_type;
 ErlNifResourceType *geef_odb_type;
 ErlNifResourceType *geef_ref_type;
+ErlNifResourceType *geef_ref_iter_type;
 ErlNifResourceType *geef_object_type;
 ErlNifResourceType *geef_revwalk_type;
 ErlNifResourceType *geef_index_type;
@@ -42,11 +43,17 @@ static int load(ErlNifEnv *env, void **priv, ERL_NIF_TERM load_info)
 		return -1;
 
 	geef_ref_type = enif_open_resource_type(env, NULL,
-					      "ref_type", geef_ref_free, ERL_NIF_RT_CREATE, NULL);
+		    "ref_type", geef_ref_free, ERL_NIF_RT_CREATE, NULL);
 
 	if (geef_ref_type == NULL)
 		return -1;
 
+	geef_ref_iter_type = enif_open_resource_type(env, NULL,
+		    "ref_iter_type", geef_ref_iter_free, ERL_NIF_RT_CREATE, NULL);
+
+	if (geef_ref_iter_type == NULL)
+		return -1;
+
 	geef_object_type = enif_open_resource_type(env, NULL,
 						     "object_type", geef_object_free, ERL_NIF_RT_CREATE, NULL);
 
@@ -70,6 +77,7 @@ static int load(ErlNifEnv *env, void **priv, ERL_NIF_TERM load_info)
 	atoms.tree = enif_make_atom(env, "tree");
 	atoms.blob = enif_make_atom(env, "blob");
 	atoms.tag = enif_make_atom(env, "tag");
+	atoms.undefined = enif_make_atom(env, "undefined");
 	/* Revwalk */
 	atoms.toposort    = enif_make_atom(env, "sort_topo");
 	atoms.timesort    = enif_make_atom(env, "sort_time");
@@ -150,6 +158,8 @@ static ErlNifFunc geef_funcs[] =
 	{"reference_to_id", 2, geef_reference_to_id},
 	{"reference_glob", 2, geef_reference_glob},
 	{"reference_lookup", 2, geef_reference_lookup},
+	{"reference_iterator", 2, geef_reference_iterator},
+	{"reference_next",     1, geef_reference_next},
 	{"reference_resolve", 1, geef_reference_resolve},
 	{"reference_target", 1, geef_reference_target},
 	{"reference_type", 1, geef_reference_type},

c_src/geef.h

@@ -18,6 +18,7 @@ typedef struct {
 	ERL_NIF_TERM tree;
 	ERL_NIF_TERM blob;
 	ERL_NIF_TERM tag;
+	ERL_NIF_TERM undefined;
 	ERL_NIF_TERM toposort;
 	ERL_NIF_TERM timesort;
 	ERL_NIF_TERM reversesort;

c_src/reference.c

@@ -73,6 +73,82 @@ geef_reference_lookup(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
 	return enif_make_tuple2(env, atoms.ok, term_ref);
 }
 
+ERL_NIF_TERM
+geef_reference_iterator(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
+{
+	ErlNifBinary bin;
+	int globbing, error;
+	geef_repository *repo;
+	ERL_NIF_TERM term_iter;
+	geef_ref_iter *res_iter;
+	git_reference_iterator *iter;
+
+	if (!enif_get_resource(env, argv[0], geef_repository_type, (void **) &repo))
+		return enif_make_badarg(env);
+
+
+	if (enif_is_identical(argv[1], atoms.undefined)) {
+		globbing = 0;
+	} else if (enif_inspect_iolist_as_binary(env, argv[1], &bin)) {
+		globbing = 1;
+	} else {
+		return enif_make_badarg(env);
+	}
+
+	if (globbing && !geef_terminate_binary(&bin))
+	    return atoms.error;
+
+	if (globbing)
+		error = git_reference_iterator_glob_new(&iter, repo->repo, (char *) bin.data);
+	else
+		error = git_reference_iterator_new(&iter, repo->repo);
+
+	if (error < 0)
+		return geef_error(env);
+
+	res_iter = enif_alloc_resource(geef_ref_iter_type, sizeof(geef_ref_iter));
+	res_iter->iter = iter;
+	res_iter->repo = repo;
+	enif_keep_resource(repo);
+	term_iter = enif_make_resource(env, res_iter);
+	enif_release_resource(res_iter);
+
+	return enif_make_tuple2(env, atoms.ok, term_iter);
+}
+
+ERL_NIF_TERM
+geef_reference_next(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
+{
+	int error;
+	geef_ref *res_ref;
+	git_reference *ref;
+	geef_ref_iter *iter;
+	ERL_NIF_TERM term_ref;
+
+	if (!enif_get_resource(env, argv[0], geef_ref_iter_type, (void **) &iter))
+		return enif_make_badarg(env);
+
+	error = git_reference_next(&ref, iter->iter);
+	if (error == GIT_ITEROVER)
+		return enif_make_tuple2(env, atoms.error, atoms.iterover);
+	if (error < 0)
+		return geef_error(env);
+
+	res_ref = enif_alloc_resource(geef_ref_type, sizeof(geef_ref));
+	res_ref->ref = ref;
+	term_ref = enif_make_resource(env, res_ref);
+	enif_release_resource(res_ref);
+
+	return enif_make_tuple2(env, atoms.ok, term_ref);
+}
+
+void geef_ref_iter_free(ErlNifEnv *env, void *cd)
+{
+	geef_ref_iter *ref = (geef_ref_iter *) cd;
+	git_reference_iterator_free(ref->iter);
+	enif_release_resource(ref->repo);
+}
+
 ERL_NIF_TERM
 geef_reference_resolve(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
 {

c_src/reference.h

@@ -5,11 +5,17 @@
 #include <git2.h>
 
 extern ErlNifResourceType *geef_ref_type;
+extern ErlNifResourceType *geef_ref_iter_type;
 
 typedef struct {
     git_reference *ref;
 } geef_ref;
 
+typedef struct {
+	git_reference_iterator *iter;
+	geef_repository *repo;
+} geef_ref_iter;
+
 ERL_NIF_TERM geef_reference_list(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
 ERL_NIF_TERM geef_reference_to_id(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
 ERL_NIF_TERM geef_reference_glob(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
@@ -20,7 +26,10 @@ ERL_NIF_TERM geef_reference_type(ErlNifEnv *env, int argc, const ERL_NIF_TERM ar
 ERL_NIF_TERM geef_reference_name(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
 ERL_NIF_TERM geef_reference_create(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
 ERL_NIF_TERM geef_reference_dwim(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
+ERL_NIF_TERM geef_reference_iterator(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
+ERL_NIF_TERM geef_reference_next(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]);
 
 void geef_ref_free(ErlNifEnv *env, void *cd);
+void geef_ref_iter_free(ErlNifEnv *env, void *cd);
 
 #endif