[PATCH v3 05/11] built-in add -i: color the header in the `status` command

[Johannes Schindelin via GitGitGadget]

From: Johannes Schindelin <johannes.schinde...@gmx.de>

For simplicity, we only implemented the `status` command without colors.
This patch starts adding color, matching what the Perl script
`git-add--interactive.perl` does.

Original-Patch-By: Daniel Ferreira <bnm...@gmail.com>
Signed-off-by: Slavica Djukic <slawic...@hotmail.com>
Signed-off-by: Johannes Schindelin <johannes.schinde...@gmx.de>
---
 add-interactive.c | 60 ++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 54 insertions(+), 6 deletions(-)

diff --git a/add-interactive.c b/add-interactive.c
index 2dbf29dee2..6c2fca12c1 100644
--- a/add-interactive.c
+++ b/add-interactive.c
@@ -1,9 +1,51 @@
 #include "cache.h"
 #include "add-interactive.h"
+#include "color.h"
+#include "config.h"
 #include "diffcore.h"
 #include "revision.h"
 #include "refs.h"
 
+struct add_i_state {
+       struct repository *r;
+       int use_color;
+       char header_color[COLOR_MAXLEN];
+};
+
+static void init_color(struct repository *r, struct add_i_state *s,
+                      const char *slot_name, char *dst,
+                      const char *default_color)
+{
+       char *key = xstrfmt("color.interactive.%s", slot_name);
+       const char *value;
+
+       if (!s->use_color)
+               dst[0] = '\0';
+       else if (repo_config_get_value(r, key, &value) ||
+                color_parse(value, dst))
+               strlcpy(dst, default_color, COLOR_MAXLEN);
+
+       free(key);
+}
+
+static int init_add_i_state(struct repository *r, struct add_i_state *s)
+{
+       const char *value;
+
+       s->r = r;
+
+       if (repo_config_get_value(r, "color.interactive", &value))
+               s->use_color = -1;
+       else
+               s->use_color =
+                       git_config_colorbool("color.interactive", value);
+       s->use_color = want_color(s->use_color);
+
+       init_color(r, s, "header", s->header_color, GIT_COLOR_BOLD);
+
+       return 0;
+}
+
 struct item {
        const char *name;
 };
@@ -14,7 +56,8 @@ struct list_options {
        void *print_item_data;
 };
 
-static void list(struct item **list, size_t nr, struct list_options *opts)
+static void list(struct item **list, size_t nr,
+                struct add_i_state *s, struct list_options *opts)
 {
        int i;
 
@@ -22,7 +65,8 @@ static void list(struct item **list, size_t nr, struct 
list_options *opts)
                return;
 
        if (opts->header)
-               printf("%s\n", opts->header);
+               color_fprintf_ln(stdout, s->header_color,
+                                "%s", opts->header);
 
        for (i = 0; i < nr; i++) {
                opts->print_item(i, list[i], opts->print_item_data);
@@ -226,16 +270,16 @@ static void print_file_item(int i, struct item *item,
        printf(" %2d: %s", i + 1, d->buf.buf);
 }
 
-static int run_status(struct repository *r, const struct pathspec *ps,
+static int run_status(struct add_i_state *s, const struct pathspec *ps,
                      struct file_list *files, struct list_options *opts)
 {
        reset_file_list(files);
 
-       if (get_modified_files(r, files, ps) < 0)
+       if (get_modified_files(s->r, files, ps) < 0)
                return -1;
 
        if (files->nr)
-               list((struct item **)files->file, files->nr, opts);
+               list((struct item **)files->file, files->nr, s, opts);
        putchar('\n');
 
        return 0;
@@ -243,6 +287,7 @@ static int run_status(struct repository *r, const struct 
pathspec *ps,
 
 int run_add_i(struct repository *r, const struct pathspec *ps)
 {
+       struct add_i_state s = { NULL };
        struct print_file_item_data print_file_item_data = {
                "%12s %12s %s", STRBUF_INIT, STRBUF_INIT, STRBUF_INIT
        };
@@ -253,13 +298,16 @@ int run_add_i(struct repository *r, const struct pathspec 
*ps)
        struct file_list files = { NULL };
        int res = 0;
 
+       if (init_add_i_state(r, &s))
+               return error("could not parse `add -i` config");
+
        strbuf_addstr(&header, "      ");
        strbuf_addf(&header, print_file_item_data.modified_fmt,
                    _("staged"), _("unstaged"), _("path"));
        opts.header = header.buf;
 
        repo_refresh_and_write_index(r, REFRESH_QUIET, 1);
-       if (run_status(r, ps, &files, &opts) < 0)
+       if (run_status(&s, ps, &files, &opts) < 0)
                res = -1;
 
        release_file_list(&files);
-- 
gitgitgadget