text: #include "clar_libgit2.h"
  #include "../status/status_helpers.h"
  #include "posix.h"
  #include "futils.h"
  
  static git_repository *g_repo = NULL;
  #define TEST_DIR "addall"
  
  void test_index_addall__initialize(void)
  {
  }
  
  void test_index_addall__cleanup(void)
  {
  	cl_git_sandbox_cleanup();
  }
  
  #define STATUS_INDEX_FLAGS \
  	(GIT_STATUS_INDEX_NEW | GIT_STATUS_INDEX_MODIFIED | \
  	 GIT_STATUS_INDEX_DELETED | GIT_STATUS_INDEX_RENAMED | \
  	 GIT_STATUS_INDEX_TYPECHANGE)
  
  #define STATUS_WT_FLAGS \
  	(GIT_STATUS_WT_NEW | GIT_STATUS_WT_MODIFIED | \
  	 GIT_STATUS_WT_DELETED | GIT_STATUS_WT_TYPECHANGE | \
  	 GIT_STATUS_WT_RENAMED)
  
  typedef struct {
  	size_t index_adds;
  	size_t index_dels;
  	size_t index_mods;
  	size_t wt_adds;
  	size_t wt_dels;
  	size_t wt_mods;
  	size_t ignores;
  	size_t conflicts;
  } index_status_counts;
  
  static int index_status_cb(
  	const char *path, unsigned int status_flags, void *payload)
  {
  	index_status_counts *vals = payload;
  
  	/* cb_status__print(path, status_flags, NULL); */
  
  	GIT_UNUSED(path);
  
  	if (status_flags & GIT_STATUS_INDEX_NEW)
  		vals->index_adds++;
  	if (status_flags & GIT_STATUS_INDEX_MODIFIED)
  		vals->index_mods++;
  	if (status_flags & GIT_STATUS_INDEX_DELETED)
  		vals->index_dels++;
  	if (status_flags & GIT_STATUS_INDEX_TYPECHANGE)
  		vals->index_mods++;
  
  	if (status_flags & GIT_STATUS_WT_NEW)
  		vals->wt_adds++;
  	if (status_flags & GIT_STATUS_WT_MODIFIED)
  		vals->wt_mods++;
  	if (status_flags & GIT_STATUS_WT_DELETED)
  		vals->wt_dels++;
  	if (status_flags & GIT_STATUS_WT_TYPECHANGE)
  		vals->wt_mods++;
  
  	if (status_flags & GIT_STATUS_IGNORED)
  		vals->ignores++;
  	if (status_flags & GIT_STATUS_CONFLICTED)
  		vals->conflicts++;
  
  	return 0;
  }
  
  static void check_status_at_line(
  	git_repository *repo,
  	size_t index_adds, size_t index_dels, size_t index_mods,
  	size_t wt_adds, size_t wt_dels, size_t wt_mods, size_t ignores,
  	size_t conflicts, const char *file, const char *func, int line)
  {
  	index_status_counts vals;
  
  	memset(&vals, 0, sizeof(vals));
  
  	cl_git_pass(git_status_foreach(repo, index_status_cb, &vals));
  
  	clar__assert_equal(
  		file,func,line,"wrong index adds", 1, "%"PRIuZ, index_adds, vals.index_adds);
  	clar__assert_equal(
  		file,func,line,"wrong index dels", 1, "%"PRIuZ, index_dels, vals.index_dels);
  	clar__assert_equal(
  		file,func,line,"wrong index mods", 1, "%"PRIuZ, index_mods, vals.index_mods);
  	clar__assert_equal(
  		file,func,line,"wrong workdir adds", 1, "%"PRIuZ, wt_adds, vals.wt_adds);
  	clar__assert_equal(
  		file,func,line,"wrong workdir dels", 1, "%"PRIuZ, wt_dels, vals.wt_dels);
  	clar__assert_equal(
  		file,func,line,"wrong workdir mods", 1, "%"PRIuZ, wt_mods, vals.wt_mods);
  	clar__assert_equal(
  		file,func,line,"wrong ignores", 1, "%"PRIuZ, ignores, vals.ignores);
  	clar__assert_equal(
  		file,func,line,"wrong conflicts", 1, "%"PRIuZ, conflicts, vals.conflicts);
  }
  
  #define check_status(R,IA,ID,IM,WA,WD,WM,IG,C) \
  	check_status_at_line(R,IA,ID,IM,WA,WD,WM,IG,C,__FILE__,__func__,__LINE__)
  
  static void check_stat_data(git_index *index, const char *path, bool match)
  {
  	const git_index_entry *entry;
  	struct stat st;
  
  	cl_must_pass(p_lstat(path, &st));
  
  	/* skip repo base dir name */
  	while (*path != '/')
  		++path;
  	++path;
  
  	entry = git_index_get_bypath(index, path, 0);
  	cl_assert(entry);
  
  	if (match) {
  		cl_assert(st.st_ctime == entry->ctime.seconds);
  		cl_assert(st.st_mtime == entry->mtime.seconds);
  		cl_assert(st.st_size == entry->file_size);
  		cl_assert((uint32_t)st.st_uid  == entry->uid);
  		cl_assert((uint32_t)st.st_gid  == entry->gid);
  		cl_assert_equal_i_fmt(
  			GIT_MODE_TYPE(st.st_mode), GIT_MODE_TYPE(entry->mode), "%07o");
  		if (cl_is_chmod_supported())
  			cl_assert_equal_b(
  				GIT_PERMS_IS_EXEC(st.st_mode), GIT_PERMS_IS_EXEC(entry->mode));
  	} else {
  		/* most things will still match */
  		cl_assert(st.st_size != entry->file_size);
  		/* would check mtime, but with second resolution it won't work :( */
  	}
  }
  
  static void addall_create_test_repo(bool check_every_step)
  {
  	g_repo = cl_git_sandbox_init_new(TEST_DIR);
  
  	if (check_every_step)
  		check_status(g_repo, 0, 0, 0, 0, 0, 0, 0, 0);
  
  	cl_git_mkfile(TEST_DIR "/file.foo", "a file");
  	if (check_every_step)
  		check_status(g_repo, 0, 0, 0, 1, 0, 0, 0, 0);
  
  	cl_git_mkfile(TEST_DIR "/.gitignore", "*.foo\n");
  	if (check_every_step)
  		check_status(g_repo, 0, 0, 0, 1, 0, 0, 1, 0);
  
  	cl_git_mkfile(TEST_DIR "/file.bar", "another file");
  	if (check_every_step)
  		check_status(g_repo, 0, 0, 0, 2, 0, 0, 1, 0);
  }
  
  void test_index_addall__repo_lifecycle(void)
  {
  	int error;
  	git_index *index;
  	git_strarray paths = { NULL, 0 };
  	char *strs[1];
  
  	addall_create_test_repo(true);
  
  	cl_git_pass(git_repository_index(&index, g_repo));
  
  	strs[0] = "file.*";
  	paths.strings = strs;
  	paths.count   = 1;
  
  	cl_git_pass(git_index_add_all(index, &paths, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/file.bar", true);
  	check_status(g_repo, 1, 0, 0, 1, 0, 0, 1, 0);
  
  	cl_git_rewritefile(TEST_DIR "/file.bar", "new content for file");
  	check_stat_data(index, TEST_DIR "/file.bar", false);
  	check_status(g_repo, 1, 0, 0, 1, 0, 1, 1, 0);
  
  	cl_git_mkfile(TEST_DIR "/file.zzz", "yet another one");
  	cl_git_mkfile(TEST_DIR "/other.zzz", "yet another one");
  	cl_git_mkfile(TEST_DIR "/more.zzz", "yet another one");
  	check_status(g_repo, 1, 0, 0, 4, 0, 1, 1, 0);
  
  	cl_git_pass(git_index_update_all(index, NULL, NULL, NULL));
  	check_stat_data(index, TEST_DIR "/file.bar", true);
  	check_status(g_repo, 1, 0, 0, 4, 0, 0, 1, 0);
  
  	cl_git_pass(git_index_add_all(index, &paths, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/file.zzz", true);
  	check_status(g_repo, 2, 0, 0, 3, 0, 0, 1, 0);
  
  	cl_repo_commit_from_index(NULL, g_repo, NULL, 0, "first commit");
  	check_status(g_repo, 0, 0, 0, 3, 0, 0, 1, 0);
  
  	if (cl_repo_get_bool(g_repo, "core.filemode")) {
  		cl_git_pass(git_index_update_all(index, NULL, NULL, NULL));
  		cl_must_pass(p_chmod(TEST_DIR "/file.zzz", 0777));
  		cl_git_pass(git_index_update_all(index, NULL, NULL, NULL));
  		check_status(g_repo, 0, 0, 1, 3, 0, 0, 1, 0);
  
  		/* go back to what we had before */
  		cl_must_pass(p_chmod(TEST_DIR "/file.zzz", 0666));
  		cl_git_pass(git_index_update_all(index, NULL, NULL, NULL));
  		check_status(g_repo, 0, 0, 0, 3, 0, 0, 1, 0);
  	}
  
  
  	/* attempt to add an ignored file - does nothing */
  	strs[0] = "file.foo";
  	cl_git_pass(git_index_add_all(index, &paths, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_status(g_repo, 0, 0, 0, 3, 0, 0, 1, 0);
  
  	/* add with check - should generate error */
  	error = git_index_add_all(
  		index, &paths, GIT_INDEX_ADD_CHECK_PATHSPEC, NULL, NULL);
  	cl_assert_equal_i(GIT_EINVALIDSPEC, error);
  	cl_git_pass(git_index_write(index));
  	check_status(g_repo, 0, 0, 0, 3, 0, 0, 1, 0);
  
  	/* add with force - should allow */
  	cl_git_pass(git_index_add_all(
  		index, &paths, GIT_INDEX_ADD_FORCE, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/file.foo", true);
  	check_status(g_repo, 1, 0, 0, 3, 0, 0, 0, 0);
  
  	/* now it's in the index, so regular add should work */
  	cl_git_rewritefile(TEST_DIR "/file.foo", "new content for file");
  	check_stat_data(index, TEST_DIR "/file.foo", false);
  	check_status(g_repo, 1, 0, 0, 3, 0, 1, 0, 0);
  
  	cl_git_pass(git_index_add_all(index, &paths, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/file.foo", true);
  	check_status(g_repo, 1, 0, 0, 3, 0, 0, 0, 0);
  
  	cl_git_pass(git_index_add_bypath(index, "more.zzz"));
  	check_stat_data(index, TEST_DIR "/more.zzz", true);
  	check_status(g_repo, 2, 0, 0, 2, 0, 0, 0, 0);
  
  	cl_git_rewritefile(TEST_DIR "/file.zzz", "new content for file");
  	check_status(g_repo, 2, 0, 0, 2, 0, 1, 0, 0);
  
  	cl_git_pass(git_index_add_bypath(index, "file.zzz"));
  	check_stat_data(index, TEST_DIR "/file.zzz", true);
  	check_status(g_repo, 2, 0, 1, 2, 0, 0, 0, 0);
  
  	strs[0] = "*.zzz";
  	cl_git_pass(git_index_remove_all(index, &paths, NULL, NULL));
  	check_status(g_repo, 1, 1, 0, 4, 0, 0, 0, 0);
  
  	cl_git_pass(git_index_add_bypath(index, "file.zzz"));
  	check_status(g_repo, 1, 0, 1, 3, 0, 0, 0, 0);
  
  	cl_repo_commit_from_index(NULL, g_repo, NULL, 0, "second commit");
  	check_status(g_repo, 0, 0, 0, 3, 0, 0, 0, 0);
  
  	cl_must_pass(p_unlink(TEST_DIR "/file.zzz"));
  	check_status(g_repo, 0, 0, 0, 3, 1, 0, 0, 0);
  
  	/* update_all should be able to remove entries */
  	cl_git_pass(git_index_update_all(index, NULL, NULL, NULL));
  	check_status(g_repo, 0, 1, 0, 3, 0, 0, 0, 0);
  
  	strs[0] = "*";
  	cl_git_pass(git_index_add_all(index, &paths, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_status(g_repo, 3, 1, 0, 0, 0, 0, 0, 0);
  
  	/* must be able to remove at any position while still updating other files */
  	cl_must_pass(p_unlink(TEST_DIR "/.gitignore"));
  	cl_git_rewritefile(TEST_DIR "/file.zzz", "reconstructed file");
  	cl_git_rewritefile(TEST_DIR "/more.zzz", "altered file reality");
  	check_status(g_repo, 3, 1, 0, 1, 1, 1, 0, 0);
  
  	cl_git_pass(git_index_update_all(index, NULL, NULL, NULL));
  	check_status(g_repo, 2, 1, 0, 1, 0, 0, 0, 0);
  	/* this behavior actually matches 'git add -u' where "file.zzz" has
  	 * been removed from the index, so when you go to update, even though
  	 * it exists in the HEAD, it is not re-added to the index, leaving it
  	 * as a DELETE when comparing HEAD to index and as an ADD comparing
  	 * index to worktree
  	 */
  
  	git_index_free(index);
  }
  
  void test_index_addall__files_in_folders(void)
  {
  	git_index *index;
  
  	addall_create_test_repo(true);
  
  	cl_git_pass(git_repository_index(&index, g_repo));
  
  	cl_git_pass(git_index_add_all(index, NULL, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/file.bar", true);
  	check_status(g_repo, 2, 0, 0, 0, 0, 0, 1, 0);
  
  	cl_must_pass(p_mkdir(TEST_DIR "/subdir", 0777));
  	cl_git_mkfile(TEST_DIR "/subdir/file", "hello!\n");
  	check_status(g_repo, 2, 0, 0, 1, 0, 0, 1, 0);
  
  	cl_git_pass(git_index_add_all(index, NULL, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_status(g_repo, 3, 0, 0, 0, 0, 0, 1, 0);
  
  	git_index_free(index);
  }
  
  void test_index_addall__hidden_files(void)
  {
  	git_index *index;
  
  	GIT_UNUSED(index);
  
  #ifdef GIT_WIN32
  	addall_create_test_repo(true);
  
  	cl_git_pass(git_repository_index(&index, g_repo));
  
  	cl_git_pass(git_index_add_all(index, NULL, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/file.bar", true);
  	check_status(g_repo, 2, 0, 0, 0, 0, 0, 1, 0);
  
  	cl_git_mkfile(TEST_DIR "/file.zzz", "yet another one");
  	cl_git_mkfile(TEST_DIR "/more.zzz", "yet another one");
  	cl_git_mkfile(TEST_DIR "/other.zzz", "yet another one");
  
  	check_status(g_repo, 2, 0, 0, 3, 0, 0, 1, 0);
  
  	cl_git_pass(git_win32__set_hidden(TEST_DIR "/file.zzz", true));
  	cl_git_pass(git_win32__set_hidden(TEST_DIR "/more.zzz", true));
  	cl_git_pass(git_win32__set_hidden(TEST_DIR "/other.zzz", true));
  
  	check_status(g_repo, 2, 0, 0, 3, 0, 0, 1, 0);
  
  	cl_git_pass(git_index_add_all(index, NULL, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/file.bar", true);
  	check_status(g_repo, 5, 0, 0, 0, 0, 0, 1, 0);
  
  	git_index_free(index);
  #endif
  }
  
  static int addall_match_prefix(
  	const char *path, const char *matched_pathspec, void *payload)
  {
  	GIT_UNUSED(matched_pathspec);
  	return !git__prefixcmp(path, payload) ? 0 : 1;
  }
  
  static int addall_match_suffix(
  	const char *path, const char *matched_pathspec, void *payload)
  {
  	GIT_UNUSED(matched_pathspec);
  	return !git__suffixcmp(path, payload) ? 0 : 1;
  }
  
  static int addall_cancel_at(
  	const char *path, const char *matched_pathspec, void *payload)
  {
  	GIT_UNUSED(matched_pathspec);
  	return !strcmp(path, payload) ? -123 : 0;
  }
  
  void test_index_addall__callback_filtering(void)
  {
  	git_index *index;
  
  	addall_create_test_repo(false);
  
  	cl_git_pass(git_repository_index(&index, g_repo));
  
  	cl_git_pass(
  		git_index_add_all(index, NULL, 0, addall_match_prefix, "file."));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/file.bar", true);
  	check_status(g_repo, 1, 0, 0, 1, 0, 0, 1, 0);
  
  	cl_git_mkfile(TEST_DIR "/file.zzz", "yet another one");
  	cl_git_mkfile(TEST_DIR "/more.zzz", "yet another one");
  	cl_git_mkfile(TEST_DIR "/other.zzz", "yet another one");
  
  	cl_git_pass(git_index_update_all(index, NULL, NULL, NULL));
  	check_stat_data(index, TEST_DIR "/file.bar", true);
  	check_status(g_repo, 1, 0, 0, 4, 0, 0, 1, 0);
  
  	cl_git_pass(
  		git_index_add_all(index, NULL, 0, addall_match_prefix, "other"));
  	cl_git_pass(git_index_write(index));
  	check_stat_data(index, TEST_DIR "/other.zzz", true);
  	check_status(g_repo, 2, 0, 0, 3, 0, 0, 1, 0);
  
  	cl_git_pass(
  		git_index_add_all(index, NULL, 0, addall_match_suffix, ".zzz"));
  	cl_git_pass(git_index_write(index));
  	check_status(g_repo, 4, 0, 0, 1, 0, 0, 1, 0);
  
  	cl_git_pass(
  		git_index_remove_all(index, NULL, addall_match_suffix, ".zzz"));
  	check_status(g_repo, 1, 0, 0, 4, 0, 0, 1, 0);
  
  	cl_git_fail_with(
  		git_index_add_all(index, NULL, 0, addall_cancel_at, "more.zzz"), -123);
  	check_status(g_repo, 3, 0, 0, 2, 0, 0, 1, 0);
  
  	cl_git_fail_with(
  		git_index_add_all(index, NULL, 0, addall_cancel_at, "other.zzz"), -123);
  	check_status(g_repo, 4, 0, 0, 1, 0, 0, 1, 0);
  
  	cl_git_pass(
  		git_index_add_all(index, NULL, 0, addall_match_suffix, ".zzz"));
  	cl_git_pass(git_index_write(index));
  	check_status(g_repo, 5, 0, 0, 0, 0, 0, 1, 0);
  
  	cl_must_pass(p_unlink(TEST_DIR "/file.zzz"));
  	cl_must_pass(p_unlink(TEST_DIR "/more.zzz"));
  	cl_must_pass(p_unlink(TEST_DIR "/other.zzz"));
  
  	cl_git_fail_with(
  		git_index_update_all(index, NULL, addall_cancel_at, "more.zzz"), -123);
  	/* file.zzz removed from index (so Index Adds 5 -> 4) and
  	 * more.zzz + other.zzz removed (so Worktree Dels 0 -> 2) */
  	check_status(g_repo, 4, 0, 0, 0, 2, 0, 1, 0);
  
  	cl_git_fail_with(
  		git_index_update_all(index, NULL, addall_cancel_at, "other.zzz"), -123);
  	/* more.zzz removed from index (so Index Adds 4 -> 3) and
  	 * Just other.zzz removed (so Worktree Dels 2 -> 1) */
  	check_status(g_repo, 3, 0, 0, 0, 1, 0, 1, 0);
  
  	git_index_free(index);
  }
  
  void test_index_addall__adds_conflicts(void)
  {
  	git_index *index;
  	git_reference *ref;
  	git_annotated_commit *annotated;
  
  	g_repo = cl_git_sandbox_init("merge-resolve");
  	cl_git_pass(git_repository_index(&index, g_repo));
  
  	check_status(g_repo, 0, 0, 0, 0, 0, 0, 0, 0);
  
  	cl_git_pass(git_reference_lookup(&ref, g_repo, "refs/heads/branch"));
  	cl_git_pass(git_annotated_commit_from_ref(&annotated, g_repo, ref));
  
  	cl_git_pass(git_merge(g_repo, (const git_annotated_commit**)&annotated, 1, NULL, NULL));
  	check_status(g_repo, 0, 1, 2, 0, 0, 0, 0, 1);
  
  	cl_git_pass(git_index_add_all(index, NULL, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_status(g_repo, 0, 1, 3, 0, 0, 0, 0, 0);
  
  	git_annotated_commit_free(annotated);
  	git_reference_free(ref);
  	git_index_free(index);
  }
  
  void test_index_addall__removes_deleted_conflicted_files(void)
  {
  	git_index *index;
  	git_reference *ref;
  	git_annotated_commit *annotated;
  
  	g_repo = cl_git_sandbox_init("merge-resolve");
  	cl_git_pass(git_repository_index(&index, g_repo));
  
  	check_status(g_repo, 0, 0, 0, 0, 0, 0, 0, 0);
  
  	cl_git_pass(git_reference_lookup(&ref, g_repo, "refs/heads/branch"));
  	cl_git_pass(git_annotated_commit_from_ref(&annotated, g_repo, ref));
  
  	cl_git_pass(git_merge(g_repo, (const git_annotated_commit**)&annotated, 1, NULL, NULL));
  	check_status(g_repo, 0, 1, 2, 0, 0, 0, 0, 1);
  
  	cl_git_rmfile("merge-resolve/conflicting.txt");
  
  	cl_git_pass(git_index_add_all(index, NULL, 0, NULL, NULL));
  	cl_git_pass(git_index_write(index));
  	check_status(g_repo, 0, 2, 2, 0, 0, 0, 0, 0);
  
  	git_annotated_commit_free(annotated);
  	git_reference_free(ref);
  	git_index_free(index);
  }
  