Felipe Contreras <felipe.contre...@gmail.com> writes:

> Otherwise when using 'git branch -vv' it's hard to see them among so
> much output.
>
> Signed-off-by: Felipe Contreras <felipe.contre...@gmail.com>
> ---
>
> This time with strbuf.
>
>  Documentation/config.txt |  3 ++-
>  builtin/branch.c         | 38 ++++++++++++++++++++++++++++++--------
>  2 files changed, 32 insertions(+), 9 deletions(-)
>
> diff --git a/Documentation/config.txt b/Documentation/config.txt
> index bc750d5..302533f 100644
> --- a/Documentation/config.txt
> +++ b/Documentation/config.txt
> @@ -794,7 +794,8 @@ color.branch::
>  color.branch.<slot>::
>       Use customized color for branch coloration. `<slot>` is one of
>       `current` (the current branch), `local` (a local branch),
> -     `remote` (a remote-tracking branch in refs/remotes/), `plain` (other
> +     `remote` (a remote-tracking branch in refs/remotes/),
> +     `upstream` (upstream tracking branch), `plain` (other
>       refs).
>  +
>  The value for these configuration variables is a list of colors (at most
> diff --git a/builtin/branch.c b/builtin/branch.c
> index 00d17d2..26e9322 100644
> --- a/builtin/branch.c
> +++ b/builtin/branch.c
> @@ -40,13 +40,15 @@ static char branch_colors[][COLOR_MAXLEN] = {
>       GIT_COLOR_RED,          /* REMOTE */
>       GIT_COLOR_NORMAL,       /* LOCAL */
>       GIT_COLOR_GREEN,        /* CURRENT */
> +     GIT_COLOR_BLUE,         /* UPSTREAM */
>  };
>  enum color_branch {
>       BRANCH_COLOR_RESET = 0,
>       BRANCH_COLOR_PLAIN = 1,
>       BRANCH_COLOR_REMOTE = 2,
>       BRANCH_COLOR_LOCAL = 3,
> -     BRANCH_COLOR_CURRENT = 4
> +     BRANCH_COLOR_CURRENT = 4,
> +     BRANCH_COLOR_UPSTREAM = 5,
>  };

We had to fix these "trailing comma in enums" to help other peoples'
compilers a few times.  Yes, they happily take the trailing comma at
the end of array values above.  Don't complain to me ;-)

> @@ -417,36 +421,54 @@ static void fill_tracking_info(struct strbuf *stat, 
> const char *branch_name,
>       int ours, theirs;
>       char *ref = NULL;
>       struct branch *branch = branch_get(branch_name);
> +     struct strbuf fancy;
> +
> +     strbuf_init(&fancy, 0);
>  
>       if (!stat_tracking_info(branch, &ours, &theirs)) {
>               if (branch && branch->merge && branch->merge[0]->dst &&
> -                 show_upstream_ref)
> -                     strbuf_addf(stat, "[%s] ",
> -                         shorten_unambiguous_ref(branch->merge[0]->dst, 0));
> +                 show_upstream_ref) {
> +                     ref = shorten_unambiguous_ref(branch->merge[0]->dst, 0);
> +                     if (want_color(branch_use_color))
> +                             strbuf_addf(stat, "[%s%s%s] ",
> +                                             
> branch_get_color(BRANCH_COLOR_UPSTREAM),
> +                                             ref, 
> branch_get_color(BRANCH_COLOR_RESET));
> +                     else
> +                             strbuf_addf(stat, "[%s] ", ref);
> +             }
>               return;
>       }
>  
> -     if (show_upstream_ref)
> +     if (show_upstream_ref) {
>               ref = shorten_unambiguous_ref(branch->merge[0]->dst, 0);
> +             if (want_color(branch_use_color))
> +                     strbuf_addf(&fancy, "%s%s%s",
> +                                     branch_get_color(BRANCH_COLOR_UPSTREAM),
> +                                     ref, 
> branch_get_color(BRANCH_COLOR_RESET));
> +             else
> +                     strbuf_addstr(&fancy, ref);
> +     }
> +
>       if (!ours) {
>               if (ref)
> -                     strbuf_addf(stat, _("[%s: behind %d]"), ref, theirs);
> +                     strbuf_addf(stat, _("[%s: behind %d]"), fancy.buf, 
> theirs);
>               else
>                       strbuf_addf(stat, _("[behind %d]"), theirs);
>  
>       } else if (!theirs) {
>               if (ref)
> -                     strbuf_addf(stat, _("[%s: ahead %d]"), ref, ours);
> +                     strbuf_addf(stat, _("[%s: ahead %d]"), fancy.buf, ours);
>               else
>                       strbuf_addf(stat, _("[ahead %d]"), ours);
>       } else {
>               if (ref)
>                       strbuf_addf(stat, _("[%s: ahead %d, behind %d]"),
> -                                 ref, ours, theirs);
> +                                 fancy.buf, ours, theirs);
>               else
>                       strbuf_addf(stat, _("[ahead %d, behind %d]"),
>                                   ours, theirs);
>       }
> +     strbuf_release(&fancy);
>       strbuf_addch(stat, ' ');
>       free(ref);
>  }

Fairly straight-forward and looks good.
--
To unsubscribe from this list: send the line "unsubscribe git" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to