]>
Commit | Line | Data |
---|---|---|
7c92fe0e JK |
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 | } |