#include "clar_libgit2.h"

#include "tree.h"

static const char *tree_oid = "1810dff58d8a660512d4832e740f692884338ccd";

static git_repository *g_repo;

void test_object_tree_walk__initialize(void)

{

   g_repo = cl_git_sandbox_init("testrepo");

}

void test_object_tree_walk__cleanup(void)

{

   cl_git_sandbox_cleanup();

}

static int treewalk_count_cb(

	const char *root, const git_tree_entry *entry, void *payload)

{

	int *count = payload;

	GIT_UNUSED(root);

	GIT_UNUSED(entry);

	(*count) += 1;

	return 0;

}

void test_object_tree_walk__0(void)

{

	git_oid id;

	git_tree *tree;

	int ct;

	git_oid_fromstr(&id, tree_oid);

	cl_git_pass(git_tree_lookup(&tree, g_repo, &id));

	ct = 0;

	cl_git_pass(git_tree_walk(tree, GIT_TREEWALK_PRE, treewalk_count_cb, &ct);;

	cl_assert_equal_i(3, ct);

	ct = 0;

	cl_git_pass(git_tree_walk(tree, GIT_TREEWALK_POST, treewalk_count_cb, &ct);;

	cl_assert_equal_i(3, ct);

	git_tree_free(tree);

}

static int treewalk_stop_cb(

	const char *root, const git_tree_entry *entry, void *payload)

{

	int *count = payload;

	GIT_UNUSED(root);

	GIT_UNUSED(entry);

	(*count) += 1;

	return (*count == 2) ? -123 : 0;

}

static int treewalk_stop_immediately_cb(

	const char *root, const git_tree_entry *entry, void *payload)

{

	GIT_UNUSED(root);

	GIT_UNUSED(entry);

	GIT_UNUSED(payload);

	return -100;

}

void test_object_tree_walk__1(void)

{

	git_oid id;

	git_tree *tree;

	int ct;

	git_oid_fromstr(&id, tree_oid);

	cl_git_pass(git_tree_lookup(&tree, g_repo, &id));

	ct = 0;

	cl_assert_equal_i(

		-123, git_tree_walk(tree, GIT_TREEWALK_PRE, treewalk_stop_cb, &ct);;

	cl_assert_equal_i(2, ct);

	ct = 0;

	cl_assert_equal_i(

		-123, git_tree_walk(tree, GIT_TREEWALK_POST, treewalk_stop_cb, &ct);;

	cl_assert_equal_i(2, ct);

	cl_assert_equal_i(

		-100, git_tree_walk(

			tree, GIT_TREEWALK_PRE, treewalk_stop_immediately_cb, NULL));

	cl_assert_equal_i(

		-100, git_tree_walk(

			tree, GIT_TREEWALK_POST, treewalk_stop_immediately_cb, NULL));

	git_tree_free(tree);

}

struct treewalk_skip_data {

	int files;

	int dirs;

	const char *skip;

	const char *stop;

};

static int treewalk_skip_de_cb(

	const char *root, const git_tree_entry *entry, void *payload)

{

	struct treewalk_skip_data *data = payload;

	const char *name = git_tree_entry_name(entry);

	GIT_UNUSED(root);

	if (git_tree_entry_type(entry) == GIT_OBJ_TREE)

		data->dirs++;

	else

		data->files++;

	if (data->skip && !strcmp(name, data->skip))

		return 1;

	else if (data->stop && !strcmp(name, data->stop))

		return -1;

	else

		return 0;

}

void test_object_tree_walk__2(void)

{

	git_oid id;

	git_tree *tree;

	struct treewalk_skip_data data;

	/* look up a deep tree */

	git_oid_fromstr(&id, "ae90f12eea699729ed24555e40b9fd669da12a12");

	cl_git_pass(git_tree_lookup(&tree, g_repo, &id));

	memset(&data, 0, sizeof(data));

	data.skip = "de";

	cl_assert_equal_i(0, git_tree_walk(

		tree, GIT_TREEWALK_PRE, treewalk_skip_de_cb, &data));

	cl_assert_equal_i(5, data.files);

	cl_assert_equal_i(3, data.dirs);

	memset(&data, 0, sizeof(data));

	data.stop = "3.txt";

	cl_assert_equal_i(-1, git_tree_walk(

		tree, GIT_TREEWALK_PRE, treewalk_skip_de_cb, &data));

	cl_assert_equal_i(3, data.files);

	cl_assert_equal_i(2, data.dirs);

	memset(&data, 0, sizeof(data));

	data.skip = "new.txt";

	cl_assert_equal_i(0, git_tree_walk(

		tree, GIT_TREEWALK_PRE, treewalk_skip_de_cb, &data));

	cl_assert_equal_i(7, data.files);

	cl_assert_equal_i(4, data.dirs);

	memset(&data, 0, sizeof(data));

	data.stop = "new.txt";

	cl_assert_equal_i(-1, git_tree_walk(

		tree, GIT_TREEWALK_PRE, treewalk_skip_de_cb, &data));

	cl_assert_equal_i(7, data.files);

	cl_assert_equal_i(4, data.dirs);

	git_tree_free(tree);

}