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