From: Slavica Djukic <slawica92@xxxxxxxxxxx> Implement add --interactive's status command in add-interactive.c and use it in builtin add--helper.c. This is the first interactive add command implemented in C of those anticipated by the previous commit, which introduced the add--helper built-in. Implement additional helper functions dealing with struct choice and colors. Use list_modified to get the data, and then pass it to list_and_choose to show it in appropriate format. We're a bit lax with the command-line parsing, as the command is intended to be called only by one internal user: the add--interactive script. Mentored-by: Johannes Schindelin <Johannes.Schindelin@xxxxxx> Original-patch-by: Daniel Ferreira <bnmvco@xxxxxxxxx> Signed-off-by: Slavica Djukic <slawica92@xxxxxxxxxxx> --- add-interactive.c | 119 ++++++++++++++++++++++++++++++++++++++++++ add-interactive.h | 8 +++ builtin/add--helper.c | 32 ++++++++++++ 3 files changed, 159 insertions(+) create mode 100644 add-interactive.h diff --git a/add-interactive.c b/add-interactive.c index 9a475a5d48..6bf8a90d9d 100644 --- a/add-interactive.c +++ b/add-interactive.c @@ -1,3 +1,4 @@ +#include "add-interactive.h" #include "cache.h" #include "commit.h" #include "color.h" @@ -7,6 +8,8 @@ #define HEADER_INDENT " " +#define HEADER_MAXLEN 30 + enum collection_phase { WORKTREE, INDEX @@ -29,6 +32,11 @@ struct collection_status { struct hashmap file_map; }; +struct command { + char *name; + void (*command_fn)(void); +}; + struct list_and_choose_options { int column_n; unsigned singleton:1; @@ -84,6 +92,42 @@ static const char *get_color(enum color_add_i ix) return ""; } +static int parse_color_slot(const char *slot) +{ + if (!strcasecmp(slot, "prompt")) + return COLOR_PROMPT; + if (!strcasecmp(slot, "header")) + return COLOR_HEADER; + if (!strcasecmp(slot, "help")) + return COLOR_HELP; + if (!strcasecmp(slot, "error")) + return COLOR_ERROR; + + return -1; +} + +int add_i_config(const char *var, + const char *value, void *cbdata) +{ + const char *name; + + if (!strcmp(var, "color.interactive")) { + use_color = git_config_colorbool(var, value); + return 0; + } + + if (skip_prefix(var, "color.interactive.", &name)) { + int slot = parse_color_slot(name); + if (slot < 0) + return 0; + if (!value) + return config_error_nonbool(var); + return color_parse(value, colors[slot]); + } + + return git_default_config(var, value, cbdata); +} + static int hash_cmp(const void *unused_cmp_data, const void *entry, const void *entry_or_key, const void *keydata) { @@ -313,3 +357,78 @@ static struct choices *list_and_choose(struct choices *data, return NULL; } } + +static struct choice *make_choice(const char *name ) +{ + struct choice *choice; + FLEXPTR_ALLOC_STR(choice, name, name); + return choice; +} + +static struct choice *add_choice(struct choices *choices, const char type, + struct file_stat *file, struct command *command) +{ + struct choice *choice; + switch (type) { + case 'f': + choice = make_choice(file->name); + choice->u.file.index.added = file->index.added; + choice->u.file.index.deleted = file->index.deleted; + choice->u.file.worktree.added = file->worktree.added; + choice->u.file.worktree.deleted = file->worktree.deleted; + break; + case 'c': + choice = make_choice(command->name); + choice->u.command_fn = command->command_fn; + break; + } + choice->type = type; + + ALLOC_GROW(choices->choices, choices->nr + 1, choices->alloc); + choices->choices[choices->nr++] = choice; + + return choice; +} + +static void free_choices(struct choices *choices) +{ + int i; + + for (i = 0; i < choices->nr; i++) + free(choices->choices[i]); + free(choices->choices); + choices->choices = NULL; + choices->nr = choices->alloc = 0; +} + +void add_i_status(void) +{ + struct collection_status *s; + struct list_and_choose_options opts = { 0 }; + struct hashmap *map; + struct hashmap_iter iter; + struct choices choices = CHOICES_INIT; + struct file_stat *entry; + const char *modified_fmt = _("%12s %12s %s"); + const char type = 'f'; + + opts.header = xmalloc(sizeof(char) * (HEADER_MAXLEN + 1)); + snprintf(opts.header, HEADER_MAXLEN + 1, modified_fmt, + _("staged"), _("unstaged"), _("path")); + + s = list_modified(the_repository, NULL); + if (s == NULL) + return; + + map = &s->file_map; + hashmap_iter_init(map, &iter); + while ((entry = hashmap_iter_next(&iter))) { + add_choice(&choices, type, entry, NULL); + } + + list_and_choose(&choices, &opts); + + hashmap_free(&s->file_map, 1); + free(s); + free_choices(&choices); +} diff --git a/add-interactive.h b/add-interactive.h new file mode 100644 index 0000000000..8ef3d2e82b --- /dev/null +++ b/add-interactive.h @@ -0,0 +1,8 @@ +#ifndef ADD_INTERACTIVE_H +#define ADD_INTERACTIVE_H + +int add_i_config(const char *var, const char *value, void *cbdata); + +void add_i_status(void); + +#endif diff --git a/builtin/add--helper.c b/builtin/add--helper.c index 6a97f0e191..464d2245f3 100644 --- a/builtin/add--helper.c +++ b/builtin/add--helper.c @@ -1,6 +1,38 @@ +#include "add-interactive.h" #include "builtin.h" +#include "config.h" +#include "revision.h" + +static const char * const builtin_add_helper_usage[] = { + N_("git add-interactive--helper <command>"), + NULL +}; + +enum cmd_mode { + DEFAULT = 0, + STATUS +}; int cmd_add__helper(int argc, const char **argv, const char *prefix) { + enum cmd_mode mode = DEFAULT; + + struct option options[] = { + OPT_CMDMODE(0, "status", &mode, + N_("print status information with diffstat"), STATUS), + OPT_END() + }; + + git_config(add_i_config, NULL); + argc = parse_options(argc, argv, NULL, options, + builtin_add_helper_usage, + PARSE_OPT_KEEP_ARGV0); + + if (mode == STATUS) + add_i_status(); + else + usage_with_options(builtin_add_helper_usage, + options); + return 0; } -- gitgitgadget