color.c

   1 #include "color.h"
   2 #include "cache.h"
   3 #include "git-compat-util.h"
   4
   5 #include <stdarg.h>
   6
   7 #define COLOR_RESET "\033[m"
   8
   9 static int parse_color(const char *name, int len)
  10 {
  11         static const char * const color_names[] = {
  12                 "normal", "black", "red", "green", "yellow",
  13                 "blue", "magenta", "cyan", "white"
  14         };
  15         char *end;
  16         int i;
  17         for (i = 0; i < ARRAY_SIZE(color_names); i++) {
  18                 const char *str = color_names[i];
  19                 if (!strncasecmp(name, str, len) && !str[len])
  20                         return i - 1;
  21         }
  22         i = strtol(name, &end, 10);
  23         if (*name && !*end && i >= -1 && i <= 255)
  24                 return i;
  25         return -2;
  26 }
  27
  28 static int parse_attr(const char *name, int len)
  29 {
  30         static const int attr_values[] = { 1, 2, 4, 5, 7 };
  31         static const char * const attr_names[] = {
  32                 "bold", "dim", "ul", "blink", "reverse"
  33         };
  34         int i;
  35         for (i = 0; i < ARRAY_SIZE(attr_names); i++) {
  36                 const char *str = attr_names[i];
  37                 if (!strncasecmp(name, str, len) && !str[len])
  38                         return attr_values[i];
  39         }
  40         return -1;
  41 }
  42
  43 void color_parse(const char *value, const char *var, char *dst)
  44 {
  45         const char *ptr = value;
  46         int attr = -1;
  47         int fg = -2;
  48         int bg = -2;
  49
  50         if (!strcasecmp(value, "reset")) {
  51                 strcpy(dst, "\033[m");
  52                 return;
  53         }
  54
  55         /* [fg [bg]] [attr] */
  56         while (*ptr) {
  57                 const char *word = ptr;
  58                 int val, len = 0;
  59
  60                 while (word[len] && !isspace(word[len]))
  61                         len++;
  62
  63                 ptr = word + len;
  64                 while (*ptr && isspace(*ptr))
  65                         ptr++;
  66
  67                 val = parse_color(word, len);
  68                 if (val >= -1) {
  69                         if (fg == -2) {
  70                                 fg = val;
  71                                 continue;
  72                         }
  73                         if (bg == -2) {
  74                                 bg = val;
  75                                 continue;
  76                         }
  77                         goto bad;
  78                 }
  79                 val = parse_attr(word, len);
  80                 if (val < 0 || attr != -1)
  81                         goto bad;
  82                 attr = val;
  83         }
  84
  85         if (attr >= 0 || fg >= 0 || bg >= 0) {
  86                 int sep = 0;
  87
  88                 *dst++ = '\033';
  89                 *dst++ = '[';
  90                 if (attr >= 0) {
  91                         *dst++ = '0' + attr;
  92                         sep++;
  93                 }
  94                 if (fg >= 0) {
  95                         if (sep++)
  96                                 *dst++ = ';';
  97                         if (fg < 8) {
  98                                 *dst++ = '3';
  99                                 *dst++ = '0' + fg;
 100                         } else {
 101                                 dst += sprintf(dst, "38;5;%d", fg);
 102                         }
 103                 }
 104                 if (bg >= 0) {
 105                         if (sep++)
 106                                 *dst++ = ';';
 107                         if (bg < 8) {
 108                                 *dst++ = '4';
 109                                 *dst++ = '0' + bg;
 110                         } else {
 111                                 dst += sprintf(dst, "48;5;%d", bg);
 112                         }
 113                 }
 114                 *dst++ = 'm';
 115         }
 116         *dst = 0;
 117         return;
 118 bad:
 119         die("bad config value '%s' for variable '%s'", value, var);
 120 }
 121
 122 int git_config_colorbool(const char *var, const char *value)
 123 {
 124         if (!value)
 125                 return 1;
 126         if (!strcasecmp(value, "auto")) {
 127                 if (isatty(1) || (pager_in_use && pager_use_color)) {
 128                         char *term = getenv("TERM");
 129                         if (term && strcmp(term, "dumb"))
 130                                 return 1;
 131                 }
 132                 return 0;
 133         }
 134         if (!strcasecmp(value, "never"))
 135                 return 0;
 136         if (!strcasecmp(value, "always"))
 137                 return 1;
 138         return git_config_bool(var, value);
 139 }
 140
 141 static int color_vprintf(const char *color, const char *fmt,
 142                 va_list args, const char *trail)
 143 {
 144         int r = 0;
 145
 146         if (*color)
 147                 r += printf("%s", color);
 148         r += vprintf(fmt, args);
 149         if (*color)
 150                 r += printf("%s", COLOR_RESET);
 151         if (trail)
 152                 r += printf("%s", trail);
 153         return r;
 154 }
 155
 156
 157
 158 int color_printf(const char *color, const char *fmt, ...)
 159 {
 160         va_list args;
 161         int r;
 162         va_start(args, fmt);
 163         r = color_vprintf(color, fmt, args, NULL);
 164         va_end(args);
 165         return r;
 166 }
 167
 168 int color_printf_ln(const char *color, const char *fmt, ...)
 169 {
 170         va_list args;
 171         int r;
 172         va_start(args, fmt);
 173         r = color_vprintf(color, fmt, args, "\n");
 174         va_end(args);
 175         return r;
 176 }