built-in add -i: re-implement add-untracked in C

This is yet another command, ported to C. It builds nicely on the
support functions introduced for other commands, with the notable
difference that only names are displayed for untracked files, no
file type or diff summary.

Signed-off-by: Johannes Schindelin <[email protected]>

Author: dscho

add-interactive.c

@@ -7,6 +7,8 @@
 #include "refs.h"
 #include "prefix-map.h"
 #include "lockfile.h"
+#include "pathspec.h"
+#include "dir.h"
 
 struct add_i_state {
 	struct repository *r;
@@ -455,6 +457,7 @@ static int is_valid_prefix(const char *prefix, size_t prefix_len)
 struct print_file_item_data {
 	const char *modified_fmt, *color, *reset;
 	struct strbuf buf, name, index, worktree;
+	unsigned only_names:1;
 };
 
 static void print_file_item(int i, int selected, struct prefix_item *item,
@@ -478,6 +481,12 @@ static void print_file_item(int i, int selected, struct prefix_item *item,
 		highlighted = d->name.buf;
 	}
 
+	if (d->only_names) {
+		printf("%c%2d: %s", selected ? '*' : ' ', i + 1,
+		       highlighted ? highlighted : item->name);
+		return;
+	}
+
 	populate_wi_changes(&d->worktree, &c->worktree, _("nothing"));
 	populate_wi_changes(&d->index, &c->index, _("unchanged"));
 	strbuf_addf(&d->buf, d->modified_fmt,
@@ -672,6 +681,92 @@ static int run_revert(struct add_i_state *s, const struct pathspec *ps,
 	return res;
 }
 
+static int get_untracked_files(struct repository *r, struct file_list *list,
+			       const struct pathspec *ps)
+{
+	struct dir_struct dir = { 0 };
+	size_t i;
+
+	if (repo_read_index(r) < 0)
+		return error(_("could not read index"));
+
+	setup_standard_excludes(&dir);
+	add_exclude_list(&dir, EXC_CMDL, "--exclude option");
+	fill_directory(&dir, r->index, ps);
+
+	for (i = 0; i < dir.nr; i++) {
+		struct dir_entry *ent = dir.entries[i];
+
+		if (index_name_is_other(r->index, ent->name, ent->len)) {
+			struct file_item *item;
+
+			FLEXPTR_ALLOC_MEM(item, item.name, ent->name, ent->len);
+
+			ALLOC_GROW(list->file, list->nr + 1, list->alloc);
+			list->file[list->nr++] = item;
+		}
+	}
+
+	return 0;
+}
+
+static int run_add_untracked(struct add_i_state *s, const struct pathspec *ps,
+		      struct file_list *files,
+		      struct list_and_choose_options *opts)
+{
+	struct print_file_item_data *d = opts->list_opts.print_item_data;
+	int res = 0, fd, *selected = NULL;
+	size_t count, i;
+
+	struct lock_file index_lock;
+
+	reset_file_list(files);
+	if (get_untracked_files(s->r, files, ps) < 0)
+		return -1;
+
+	if (!files->nr) {
+		printf(_("No untracked files.\n"));
+		goto finish_add_untracked;
+	}
+
+	opts->prompt = N_("Add untracked");
+	CALLOC_ARRAY(selected, files->nr);
+
+	d->only_names = 1;
+	count = list_and_choose((struct prefix_item **)files->file,
+				selected, files->nr, s, opts);
+	d->only_names = 0;
+	if (count <= 0)
+		goto finish_add_untracked;
+
+	fd = repo_hold_locked_index(s->r, &index_lock, LOCK_REPORT_ON_ERROR);
+	if (fd < 0) {
+		res = -1;
+		goto finish_add_untracked;
+	}
+
+	for (i = 0; i < files->nr; i++) {
+		const char *name = files->file[i]->item.name;
+		if (selected[i] &&
+		    add_file_to_index(s->r->index, name, 0) < 0) {
+			res = error(_("could not stage '%s'"), name);
+			break;
+		}
+	}
+
+	if (!res && write_locked_index(s->r->index, &index_lock, COMMIT_LOCK) < 0)
+		res = error(_("could not write index"));
+
+	if (!res)
+		printf(Q_("added %d path\n",
+			  "added %d paths\n", count), (int)count);
+
+finish_add_untracked:
+	putchar('\n');
+	free(selected);
+	return res;
+}
+
 static int run_help(struct add_i_state *s, const struct pathspec *ps,
 		    struct file_list *files,
 		    struct list_and_choose_options *opts)
@@ -764,9 +859,10 @@ int run_add_i(struct repository *r, const struct pathspec *ps)
 		status = { { "status" }, run_status },
 		update = { { "update" }, run_update },
 		revert = { { "revert" }, run_revert },
+		add_untracked = { { "add untracked" }, run_add_untracked },
 		help = { { "help" }, run_help };
 	struct command_item *commands[] = {
-		&status, &update, &revert,
+		&status, &update, &revert, &add_untracked,
 		&help
 	};