#include "clay_libgit2.h"
#include "fileops.h"
#include "git2/attr.h"

static git_repository *g_repo = NULL;

void test_attr_repo__initialize(void)
{
	/* before each test, instantiate the attr repo from the fixtures and
	 * rename the .gitted to .git so it is a repo with a working dir.
	 */
	cl_fixture_sandbox("attr");
	cl_git_pass(p_rename("attr/.gitted", "attr/.git"));
	cl_git_pass(git_repository_open(&g_repo, "attr/.git"));
}

void test_attr_repo__cleanup(void)
{
	git_repository_free(g_repo);
	g_repo = NULL;
	cl_fixture_cleanup("attr");
}

void test_attr_repo__get_one(void)
{
	const char *value;
	struct {
		const char *file;
		const char *attr;
		const char *expected;
	} test_cases[] = {
		{ "root_test1", "repoattr", GIT_ATTR_TRUE },
		{ "root_test1", "rootattr", GIT_ATTR_TRUE },
		{ "root_test1", "missingattr", NULL },
		{ "root_test1", "subattr", NULL },
		{ "root_test1", "negattr", NULL },
		{ "root_test2", "repoattr", GIT_ATTR_TRUE },
		{ "root_test2", "rootattr", GIT_ATTR_FALSE },
		{ "root_test2", "missingattr", NULL },
		{ "root_test3", "repoattr", GIT_ATTR_TRUE },
		{ "root_test3", "rootattr", NULL },
		{ "subdir/subdir_test1", "repoattr", GIT_ATTR_TRUE },
		{ "subdir/subdir_test1", "rootattr", GIT_ATTR_TRUE },
		{ "subdir/subdir_test1", "missingattr", NULL },
		{ "subdir/subdir_test1", "subattr", "yes" },
		{ "subdir/subdir_test1", "negattr", GIT_ATTR_FALSE },
		{ "subdir/subdir_test1", "another", NULL },
		{ "subdir/subdir_test2.txt", "repoattr", GIT_ATTR_TRUE },
		{ "subdir/subdir_test2.txt", "rootattr", GIT_ATTR_TRUE },
		{ "subdir/subdir_test2.txt", "missingattr", NULL },
		{ "subdir/subdir_test2.txt", "subattr", "yes" },
		{ "subdir/subdir_test2.txt", "negattr", GIT_ATTR_FALSE },
		{ "subdir/subdir_test2.txt", "another", "one" },
		{ NULL, NULL, NULL }
	}, *scan;

	for (scan = test_cases; scan->file != NULL; scan++) {
		git_buf b = GIT_BUF_INIT;

		git_buf_printf(&b, "%s:%s == expect %s",
					   scan->file, scan->attr, scan->expected);

		cl_must_pass_(
			git_attr_get(g_repo, scan->file, scan->attr, &value) == GIT_SUCCESS,
			b.ptr);

		git_buf_printf(&b, ", got %s", value);

		if (scan->expected == NULL ||
			scan->expected == GIT_ATTR_TRUE ||
			scan->expected == GIT_ATTR_FALSE)
		{
			cl_assert_(scan->expected == value, b.ptr);
		} else {
			cl_assert_strequal(scan->expected, value);
		}

		git_buf_free(&b);
	}
}

void test_attr_repo__get_many(void)
{
	const char *names[4] = { "repoattr", "rootattr", "missingattr", "subattr" };
	const char *values[4];

	cl_git_pass(git_attr_get_many(g_repo, "root_test1", 4, names, values));

	cl_assert(values[0] == GIT_ATTR_TRUE);
	cl_assert(values[1] == GIT_ATTR_TRUE);
	cl_assert(values[2] == NULL);
	cl_assert(values[3] == NULL);

	cl_git_pass(git_attr_get_many(g_repo, "root_test2", 4, names, values));

	cl_assert(values[0] == GIT_ATTR_TRUE);
	cl_assert(values[1] == GIT_ATTR_FALSE);
	cl_assert(values[2] == NULL);
	cl_assert(values[3] == NULL);

	cl_git_pass(git_attr_get_many(g_repo, "subdir/subdir_test1", 4, names, values));

	cl_assert(values[0] == GIT_ATTR_TRUE);
	cl_assert(values[1] == GIT_ATTR_TRUE);
	cl_assert(values[2] == NULL);
	cl_assert_strequal("yes", values[3]);

}

static int count_attrs(
	const char *GIT_UNUSED(name),
	const char *GIT_UNUSED(value),
	void *payload)
{
	GIT_UNUSED_ARG(name);
	GIT_UNUSED_ARG(value);

	*((int *)payload) += 1;

	return GIT_SUCCESS;
}

void test_attr_repo__foreach(void)
{
	int count;

	count = 0;
	cl_git_pass(git_attr_foreach(g_repo, "root_test1", &count_attrs, &count));
	cl_assert(count == 2);

	count = 0;
	cl_git_pass(git_attr_foreach(g_repo, "subdir/subdir_test1",
		&count_attrs, &count));
	cl_assert(count == 4); /* repoattr, rootattr, subattr, negattr */

	count = 0;
	cl_git_pass(git_attr_foreach(g_repo, "subdir/subdir_test2.txt",
		&count_attrs, &count));
	cl_assert(count == 5); /* repoattr, rootattr, subattr, negattr, another */
}