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