1#include "cache.h"
2#include "wt-status.h"
3#include "color.h"
4#include "object.h"
5#include "dir.h"
6#include "commit.h"
7#include "diff.h"
8#include "revision.h"
9#include "diffcore.h"
10#include "quote.h"
11
12int wt_status_relative_paths = 1;
13int wt_status_use_color = -1;
14static char wt_status_colors[][COLOR_MAXLEN] = {
15 "", /* WT_STATUS_HEADER: normal */
16 "\033[32m", /* WT_STATUS_UPDATED: green */
17 "\033[31m", /* WT_STATUS_CHANGED: red */
18 "\033[31m", /* WT_STATUS_UNTRACKED: red */
19};
20
21static const char use_add_msg[] =
22"use \"git add <file>...\" to update what will be committed";
23static const char use_add_rm_msg[] =
24"use \"git add/rm <file>...\" to update what will be committed";
25static const char use_add_to_include_msg[] =
26"use \"git add <file>...\" to include in what will be committed";
27
28static int parse_status_slot(const char *var, int offset)
29{
30 if (!strcasecmp(var+offset, "header"))
31 return WT_STATUS_HEADER;
32 if (!strcasecmp(var+offset, "updated")
33 || !strcasecmp(var+offset, "added"))
34 return WT_STATUS_UPDATED;
35 if (!strcasecmp(var+offset, "changed"))
36 return WT_STATUS_CHANGED;
37 if (!strcasecmp(var+offset, "untracked"))
38 return WT_STATUS_UNTRACKED;
39 die("bad config variable '%s'", var);
40}
41
42static const char* color(int slot)
43{
44 return wt_status_use_color > 0 ? wt_status_colors[slot] : "";
45}
46
47void wt_status_prepare(struct wt_status *s)
48{
49 unsigned char sha1[20];
50 const char *head;
51
52 memset(s, 0, sizeof(*s));
53 head = resolve_ref("HEAD", sha1, 0, NULL);
54 s->branch = head ? xstrdup(head) : NULL;
55 s->reference = "HEAD";
56 s->fp = stdout;
57 s->index_file = get_index_file();
58}
59
60static void wt_status_print_cached_header(struct wt_status *s)
61{
62 const char *c = color(WT_STATUS_HEADER);
63 color_fprintf_ln(s->fp, c, "# Changes to be committed:");
64 if (!s->is_initial) {
65 color_fprintf_ln(s->fp, c, "# (use \"git reset %s <file>...\" to unstage)", s->reference);
66 } else {
67 color_fprintf_ln(s->fp, c, "# (use \"git rm --cached <file>...\" to unstage)");
68 }
69 color_fprintf_ln(s->fp, c, "#");
70}
71
72static void wt_status_print_header(struct wt_status *s,
73 const char *main, const char *sub)
74{
75 const char *c = color(WT_STATUS_HEADER);
76 color_fprintf_ln(s->fp, c, "# %s:", main);
77 color_fprintf_ln(s->fp, c, "# (%s)", sub);
78 color_fprintf_ln(s->fp, c, "#");
79}
80
81static void wt_status_print_trailer(struct wt_status *s)
82{
83 color_fprintf_ln(s->fp, color(WT_STATUS_HEADER), "#");
84}
85
86#define quote_path quote_path_relative
87
88static void wt_status_print_filepair(struct wt_status *s,
89 int t, struct diff_filepair *p)
90{
91 const char *c = color(t);
92 const char *one, *two;
93 struct strbuf onebuf, twobuf;
94
95 strbuf_init(&onebuf, 0);
96 strbuf_init(&twobuf, 0);
97 one = quote_path(p->one->path, -1, &onebuf, s->prefix);
98 two = quote_path(p->two->path, -1, &twobuf, s->prefix);
99
100 color_fprintf(s->fp, color(WT_STATUS_HEADER), "#\t");
101 switch (p->status) {
102 case DIFF_STATUS_ADDED:
103 color_fprintf(s->fp, c, "new file: %s", one);
104 break;
105 case DIFF_STATUS_COPIED:
106 color_fprintf(s->fp, c, "copied: %s -> %s", one, two);
107 break;
108 case DIFF_STATUS_DELETED:
109 color_fprintf(s->fp, c, "deleted: %s", one);
110 break;
111 case DIFF_STATUS_MODIFIED:
112 color_fprintf(s->fp, c, "modified: %s", one);
113 break;
114 case DIFF_STATUS_RENAMED:
115 color_fprintf(s->fp, c, "renamed: %s -> %s", one, two);
116 break;
117 case DIFF_STATUS_TYPE_CHANGED:
118 color_fprintf(s->fp, c, "typechange: %s", one);
119 break;
120 case DIFF_STATUS_UNKNOWN:
121 color_fprintf(s->fp, c, "unknown: %s", one);
122 break;
123 case DIFF_STATUS_UNMERGED:
124 color_fprintf(s->fp, c, "unmerged: %s", one);
125 break;
126 default:
127 die("bug: unhandled diff status %c", p->status);
128 }
129 fprintf(s->fp, "\n");
130 strbuf_release(&onebuf);
131 strbuf_release(&twobuf);
132}
133
134static void wt_status_print_updated_cb(struct diff_queue_struct *q,
135 struct diff_options *options,
136 void *data)
137{
138 struct wt_status *s = data;
139 int shown_header = 0;
140 int i;
141 for (i = 0; i < q->nr; i++) {
142 if (q->queue[i]->status == 'U')
143 continue;
144 if (!shown_header) {
145 wt_status_print_cached_header(s);
146 s->commitable = 1;
147 shown_header = 1;
148 }
149 wt_status_print_filepair(s, WT_STATUS_UPDATED, q->queue[i]);
150 }
151 if (shown_header)
152 wt_status_print_trailer(s);
153}
154
155static void wt_status_print_changed_cb(struct diff_queue_struct *q,
156 struct diff_options *options,
157 void *data)
158{
159 struct wt_status *s = data;
160 int i;
161 if (q->nr) {
162 const char *msg = use_add_msg;
163 s->workdir_dirty = 1;
164 for (i = 0; i < q->nr; i++)
165 if (q->queue[i]->status == DIFF_STATUS_DELETED) {
166 msg = use_add_rm_msg;
167 break;
168 }
169 wt_status_print_header(s, "Changed but not updated", msg);
170 }
171 for (i = 0; i < q->nr; i++)
172 wt_status_print_filepair(s, WT_STATUS_CHANGED, q->queue[i]);
173 if (q->nr)
174 wt_status_print_trailer(s);
175}
176
177static void wt_status_print_initial(struct wt_status *s)
178{
179 int i;
180 struct strbuf buf;
181
182 strbuf_init(&buf, 0);
183 if (active_nr) {
184 s->commitable = 1;
185 wt_status_print_cached_header(s);
186 }
187 for (i = 0; i < active_nr; i++) {
188 color_fprintf(s->fp, color(WT_STATUS_HEADER), "#\t");
189 color_fprintf_ln(s->fp, color(WT_STATUS_UPDATED), "new file: %s",
190 quote_path(active_cache[i]->name, -1,
191 &buf, s->prefix));
192 }
193 if (active_nr)
194 wt_status_print_trailer(s);
195 strbuf_release(&buf);
196}
197
198static void wt_status_print_updated(struct wt_status *s)
199{
200 struct rev_info rev;
201 init_revisions(&rev, NULL);
202 setup_revisions(0, NULL, &rev, s->reference);
203 rev.diffopt.output_format |= DIFF_FORMAT_CALLBACK;
204 rev.diffopt.format_callback = wt_status_print_updated_cb;
205 rev.diffopt.format_callback_data = s;
206 rev.diffopt.detect_rename = 1;
207 rev.diffopt.rename_limit = 100;
208 rev.diffopt.break_opt = 0;
209 run_diff_index(&rev, 1);
210}
211
212static void wt_status_print_changed(struct wt_status *s)
213{
214 struct rev_info rev;
215 init_revisions(&rev, "");
216 setup_revisions(0, NULL, &rev, NULL);
217 rev.diffopt.output_format |= DIFF_FORMAT_CALLBACK;
218 rev.diffopt.format_callback = wt_status_print_changed_cb;
219 rev.diffopt.format_callback_data = s;
220 run_diff_files(&rev, 0);
221}
222
223static void wt_status_print_untracked(struct wt_status *s)
224{
225 struct dir_struct dir;
226 int i;
227 int shown_header = 0;
228 struct strbuf buf;
229
230 strbuf_init(&buf, 0);
231 memset(&dir, 0, sizeof(dir));
232
233 if (!s->untracked) {
234 dir.show_other_directories = 1;
235 dir.hide_empty_directories = 1;
236 }
237 setup_standard_excludes(&dir);
238
239 read_directory(&dir, ".", "", 0, NULL);
240 for(i = 0; i < dir.nr; i++) {
241 /* check for matching entry, which is unmerged; lifted from
242 * builtin-ls-files:show_other_files */
243 struct dir_entry *ent = dir.entries[i];
244 int pos = cache_name_pos(ent->name, ent->len);
245 struct cache_entry *ce;
246 if (0 <= pos)
247 die("bug in wt_status_print_untracked");
248 pos = -pos - 1;
249 if (pos < active_nr) {
250 ce = active_cache[pos];
251 if (ce_namelen(ce) == ent->len &&
252 !memcmp(ce->name, ent->name, ent->len))
253 continue;
254 }
255 if (!shown_header) {
256 s->workdir_untracked = 1;
257 wt_status_print_header(s, "Untracked files",
258 use_add_to_include_msg);
259 shown_header = 1;
260 }
261 color_fprintf(s->fp, color(WT_STATUS_HEADER), "#\t");
262 color_fprintf_ln(s->fp, color(WT_STATUS_UNTRACKED), "%s",
263 quote_path(ent->name, ent->len,
264 &buf, s->prefix));
265 }
266 strbuf_release(&buf);
267}
268
269static void wt_status_print_verbose(struct wt_status *s)
270{
271 struct rev_info rev;
272 int saved_stdout;
273
274 fflush(s->fp);
275
276 /* Sigh, the entire diff machinery is hardcoded to output to
277 * stdout. Do the dup-dance...*/
278 saved_stdout = dup(STDOUT_FILENO);
279 if (saved_stdout < 0 ||dup2(fileno(s->fp), STDOUT_FILENO) < 0)
280 die("couldn't redirect stdout\n");
281
282 init_revisions(&rev, NULL);
283 setup_revisions(0, NULL, &rev, s->reference);
284 rev.diffopt.output_format |= DIFF_FORMAT_PATCH;
285 rev.diffopt.detect_rename = 1;
286 run_diff_index(&rev, 1);
287
288 fflush(stdout);
289
290 if (dup2(saved_stdout, STDOUT_FILENO) < 0)
291 die("couldn't restore stdout\n");
292 close(saved_stdout);
293}
294
295void wt_status_print(struct wt_status *s)
296{
297 unsigned char sha1[20];
298 s->is_initial = get_sha1(s->reference, sha1) ? 1 : 0;
299
300 if (s->branch) {
301 const char *on_what = "On branch ";
302 const char *branch_name = s->branch;
303 if (!prefixcmp(branch_name, "refs/heads/"))
304 branch_name += 11;
305 else if (!strcmp(branch_name, "HEAD")) {
306 branch_name = "";
307 on_what = "Not currently on any branch.";
308 }
309 color_fprintf_ln(s->fp, color(WT_STATUS_HEADER),
310 "# %s%s", on_what, branch_name);
311 }
312
313 if (s->is_initial) {
314 color_fprintf_ln(s->fp, color(WT_STATUS_HEADER), "#");
315 color_fprintf_ln(s->fp, color(WT_STATUS_HEADER), "# Initial commit");
316 color_fprintf_ln(s->fp, color(WT_STATUS_HEADER), "#");
317 wt_status_print_initial(s);
318 }
319 else {
320 wt_status_print_updated(s);
321 }
322
323 wt_status_print_changed(s);
324 wt_status_print_untracked(s);
325
326 if (s->verbose && !s->is_initial)
327 wt_status_print_verbose(s);
328 if (!s->commitable) {
329 if (s->amend)
330 fprintf(s->fp, "# No changes\n");
331 else if (s->nowarn)
332 ; /* nothing */
333 else if (s->workdir_dirty)
334 printf("no changes added to commit (use \"git add\" and/or \"git commit -a\")\n");
335 else if (s->workdir_untracked)
336 printf("nothing added to commit but untracked files present (use \"git add\" to track)\n");
337 else if (s->is_initial)
338 printf("nothing to commit (create/copy files and use \"git add\" to track)\n");
339 else
340 printf("nothing to commit (working directory clean)\n");
341 }
342}
343
344int git_status_config(const char *k, const char *v)
345{
346 if (!strcmp(k, "status.color") || !strcmp(k, "color.status")) {
347 wt_status_use_color = git_config_colorbool(k, v, -1);
348 return 0;
349 }
350 if (!prefixcmp(k, "status.color.") || !prefixcmp(k, "color.status.")) {
351 int slot = parse_status_slot(k, 13);
352 if (!v)
353 return config_error_nonbool(k);
354 color_parse(v, k, wt_status_colors[slot]);
355 return 0;
356 }
357 if (!strcmp(k, "status.relativepaths")) {
358 wt_status_relative_paths = git_config_bool(k, v);
359 return 0;
360 }
361 return git_color_default_config(k, v);
362}