Merge branch 'master' into next
[git] / builtin-commit.c
1 /*
2  * Builtin "git commit"
3  *
4  * Copyright (c) 2007 Kristian Høgsberg <krh@redhat.com>
5  * Based on git-commit.sh by Junio C Hamano and Linus Torvalds
6  */
7
8 #include "cache.h"
9 #include "cache-tree.h"
10 #include "color.h"
11 #include "dir.h"
12 #include "builtin.h"
13 #include "diff.h"
14 #include "diffcore.h"
15 #include "commit.h"
16 #include "revision.h"
17 #include "wt-status.h"
18 #include "run-command.h"
19 #include "refs.h"
20 #include "log-tree.h"
21 #include "strbuf.h"
22 #include "utf8.h"
23 #include "parse-options.h"
24 #include "string-list.h"
25 #include "rerere.h"
26 #include "unpack-trees.h"
27 #include "quote.h"
28
29 static const char * const builtin_commit_usage[] = {
30         "git commit [options] [--] <filepattern>...",
31         NULL
32 };
33
34 static const char * const builtin_status_usage[] = {
35         "git status [options] [--] <filepattern>...",
36         NULL
37 };
38
39 static unsigned char head_sha1[20], merge_head_sha1[20];
40 static char *use_message_buffer;
41 static const char commit_editmsg[] = "COMMIT_EDITMSG";
42 static struct lock_file index_lock; /* real index */
43 static struct lock_file false_lock; /* used only for partial commits */
44 static enum {
45         COMMIT_AS_IS = 1,
46         COMMIT_NORMAL,
47         COMMIT_PARTIAL,
48 } commit_style;
49
50 static const char *logfile, *force_author;
51 static const char *template_file;
52 static char *edit_message, *use_message;
53 static char *author_name, *author_email, *author_date;
54 static int all, edit_flag, also, interactive, only, amend, signoff;
55 static int quiet, verbose, no_verify, allow_empty, dry_run;
56 static char *untracked_files_arg;
57 /*
58  * The default commit message cleanup mode will remove the lines
59  * beginning with # (shell comments) and leading and trailing
60  * whitespaces (empty lines or containing only whitespaces)
61  * if editor is used, and only the whitespaces if the message
62  * is specified explicitly.
63  */
64 static enum {
65         CLEANUP_SPACE,
66         CLEANUP_NONE,
67         CLEANUP_ALL,
68 } cleanup_mode;
69 static char *cleanup_arg;
70
71 static int use_editor = 1, initial_commit, in_merge;
72 static const char *only_include_assumed;
73 static struct strbuf message;
74
75 static int opt_parse_m(const struct option *opt, const char *arg, int unset)
76 {
77         struct strbuf *buf = opt->value;
78         if (unset)
79                 strbuf_setlen(buf, 0);
80         else {
81                 strbuf_addstr(buf, arg);
82                 strbuf_addstr(buf, "\n\n");
83         }
84         return 0;
85 }
86
87 static struct option builtin_commit_options[] = {
88         OPT__QUIET(&quiet),
89         OPT__VERBOSE(&verbose),
90         OPT_GROUP("Commit message options"),
91
92         OPT_FILENAME('F', "file", &logfile, "read log from file"),
93         OPT_STRING(0, "author", &force_author, "AUTHOR", "override author for commit"),
94         OPT_CALLBACK('m', "message", &message, "MESSAGE", "specify commit message", opt_parse_m),
95         OPT_STRING('c', "reedit-message", &edit_message, "COMMIT", "reuse and edit message from specified commit "),
96         OPT_STRING('C', "reuse-message", &use_message, "COMMIT", "reuse message from specified commit"),
97         OPT_BOOLEAN('s', "signoff", &signoff, "add Signed-off-by:"),
98         OPT_FILENAME('t', "template", &template_file, "use specified template file"),
99         OPT_BOOLEAN('e', "edit", &edit_flag, "force edit of commit"),
100
101         OPT_GROUP("Commit contents options"),
102         OPT_BOOLEAN('a', "all", &all, "commit all changed files"),
103         OPT_BOOLEAN('i', "include", &also, "add specified files to index for commit"),
104         OPT_BOOLEAN(0, "interactive", &interactive, "interactively add files"),
105         OPT_BOOLEAN('o', "only", &only, "commit only specified files"),
106         OPT_BOOLEAN('n', "no-verify", &no_verify, "bypass pre-commit hook"),
107         OPT_BOOLEAN(0, "dry-run", &dry_run, "show what would be committed"),
108         OPT_BOOLEAN(0, "amend", &amend, "amend previous commit"),
109         { OPTION_STRING, 'u', "untracked-files", &untracked_files_arg, "mode", "show untracked files, optional modes: all, normal, no. (Default: all)", PARSE_OPT_OPTARG, NULL, (intptr_t)"all" },
110         OPT_BOOLEAN(0, "allow-empty", &allow_empty, "ok to record an empty change"),
111         OPT_STRING(0, "cleanup", &cleanup_arg, "default", "how to strip spaces and #comments from message"),
112
113         OPT_END()
114 };
115
116 static void rollback_index_files(void)
117 {
118         switch (commit_style) {
119         case COMMIT_AS_IS:
120                 break; /* nothing to do */
121         case COMMIT_NORMAL:
122                 rollback_lock_file(&index_lock);
123                 break;
124         case COMMIT_PARTIAL:
125                 rollback_lock_file(&index_lock);
126                 rollback_lock_file(&false_lock);
127                 break;
128         }
129 }
130
131 static int commit_index_files(void)
132 {
133         int err = 0;
134
135         switch (commit_style) {
136         case COMMIT_AS_IS:
137                 break; /* nothing to do */
138         case COMMIT_NORMAL:
139                 err = commit_lock_file(&index_lock);
140                 break;
141         case COMMIT_PARTIAL:
142                 err = commit_lock_file(&index_lock);
143                 rollback_lock_file(&false_lock);
144                 break;
145         }
146
147         return err;
148 }
149
150 /*
151  * Take a union of paths in the index and the named tree (typically, "HEAD"),
152  * and return the paths that match the given pattern in list.
153  */
154 static int list_paths(struct string_list *list, const char *with_tree,
155                       const char *prefix, const char **pattern)
156 {
157         int i;
158         char *m;
159
160         for (i = 0; pattern[i]; i++)
161                 ;
162         m = xcalloc(1, i);
163
164         if (with_tree)
165                 overlay_tree_on_cache(with_tree, prefix);
166
167         for (i = 0; i < active_nr; i++) {
168                 struct cache_entry *ce = active_cache[i];
169                 if (ce->ce_flags & CE_UPDATE)
170                         continue;
171                 if (!match_pathspec(pattern, ce->name, ce_namelen(ce), 0, m))
172                         continue;
173                 string_list_insert(ce->name, list);
174         }
175
176         return report_path_error(m, pattern, prefix ? strlen(prefix) : 0);
177 }
178
179 static void add_remove_files(struct string_list *list)
180 {
181         int i;
182         for (i = 0; i < list->nr; i++) {
183                 struct stat st;
184                 struct string_list_item *p = &(list->items[i]);
185
186                 if (!lstat(p->string, &st)) {
187                         if (add_to_cache(p->string, &st, 0))
188                                 die("updating files failed");
189                 } else
190                         remove_file_from_cache(p->string);
191         }
192 }
193
194 static void create_base_index(void)
195 {
196         struct tree *tree;
197         struct unpack_trees_options opts;
198         struct tree_desc t;
199
200         if (initial_commit) {
201                 discard_cache();
202                 return;
203         }
204
205         memset(&opts, 0, sizeof(opts));
206         opts.head_idx = 1;
207         opts.index_only = 1;
208         opts.merge = 1;
209         opts.src_index = &the_index;
210         opts.dst_index = &the_index;
211
212         opts.fn = oneway_merge;
213         tree = parse_tree_indirect(head_sha1);
214         if (!tree)
215                 die("failed to unpack HEAD tree object");
216         parse_tree(tree);
217         init_tree_desc(&t, tree->buffer, tree->size);
218         if (unpack_trees(1, &t, &opts))
219                 exit(128); /* We've already reported the error, finish dying */
220 }
221
222 static char *prepare_index(int argc, const char **argv, const char *prefix, int is_status)
223 {
224         int fd;
225         struct string_list partial;
226         const char **pathspec = NULL;
227         int refresh_flags = REFRESH_QUIET;
228
229         if (is_status)
230                 refresh_flags |= REFRESH_UNMERGED;
231         if (interactive) {
232                 if (interactive_add(argc, argv, prefix) != 0)
233                         die("interactive add failed");
234                 if (read_cache_preload(NULL) < 0)
235                         die("index file corrupt");
236                 commit_style = COMMIT_AS_IS;
237                 return get_index_file();
238         }
239
240         if (*argv)
241                 pathspec = get_pathspec(prefix, argv);
242
243         if (read_cache_preload(pathspec) < 0)
244                 die("index file corrupt");
245
246         /*
247          * Non partial, non as-is commit.
248          *
249          * (1) get the real index;
250          * (2) update the_index as necessary;
251          * (3) write the_index out to the real index (still locked);
252          * (4) return the name of the locked index file.
253          *
254          * The caller should run hooks on the locked real index, and
255          * (A) if all goes well, commit the real index;
256          * (B) on failure, rollback the real index.
257          */
258         if (all || (also && pathspec && *pathspec)) {
259                 int fd = hold_locked_index(&index_lock, 1);
260                 add_files_to_cache(also ? prefix : NULL, pathspec, 0);
261                 refresh_cache(refresh_flags);
262                 if (write_cache(fd, active_cache, active_nr) ||
263                     close_lock_file(&index_lock))
264                         die("unable to write new_index file");
265                 commit_style = COMMIT_NORMAL;
266                 return index_lock.filename;
267         }
268
269         /*
270          * As-is commit.
271          *
272          * (1) return the name of the real index file.
273          *
274          * The caller should run hooks on the real index, and run
275          * hooks on the real index, and create commit from the_index.
276          * We still need to refresh the index here.
277          */
278         if (!pathspec || !*pathspec) {
279                 fd = hold_locked_index(&index_lock, 1);
280                 refresh_cache(refresh_flags);
281                 if (write_cache(fd, active_cache, active_nr) ||
282                     commit_locked_index(&index_lock))
283                         die("unable to write new_index file");
284                 commit_style = COMMIT_AS_IS;
285                 return get_index_file();
286         }
287
288         /*
289          * A partial commit.
290          *
291          * (0) find the set of affected paths;
292          * (1) get lock on the real index file;
293          * (2) update the_index with the given paths;
294          * (3) write the_index out to the real index (still locked);
295          * (4) get lock on the false index file;
296          * (5) reset the_index from HEAD;
297          * (6) update the_index the same way as (2);
298          * (7) write the_index out to the false index file;
299          * (8) return the name of the false index file (still locked);
300          *
301          * The caller should run hooks on the locked false index, and
302          * create commit from it.  Then
303          * (A) if all goes well, commit the real index;
304          * (B) on failure, rollback the real index;
305          * In either case, rollback the false index.
306          */
307         commit_style = COMMIT_PARTIAL;
308
309         if (file_exists(git_path("MERGE_HEAD")))
310                 die("cannot do a partial commit during a merge.");
311
312         memset(&partial, 0, sizeof(partial));
313         partial.strdup_strings = 1;
314         if (list_paths(&partial, initial_commit ? NULL : "HEAD", prefix, pathspec))
315                 exit(1);
316
317         discard_cache();
318         if (read_cache() < 0)
319                 die("cannot read the index");
320
321         fd = hold_locked_index(&index_lock, 1);
322         add_remove_files(&partial);
323         refresh_cache(REFRESH_QUIET);
324         if (write_cache(fd, active_cache, active_nr) ||
325             close_lock_file(&index_lock))
326                 die("unable to write new_index file");
327
328         fd = hold_lock_file_for_update(&false_lock,
329                                        git_path("next-index-%"PRIuMAX,
330                                                 (uintmax_t) getpid()),
331                                        LOCK_DIE_ON_ERROR);
332
333         create_base_index();
334         add_remove_files(&partial);
335         refresh_cache(REFRESH_QUIET);
336
337         if (write_cache(fd, active_cache, active_nr) ||
338             close_lock_file(&false_lock))
339                 die("unable to write temporary index file");
340
341         discard_cache();
342         read_cache_from(false_lock.filename);
343
344         return false_lock.filename;
345 }
346
347 static int run_status(FILE *fp, const char *index_file, const char *prefix, int nowarn,
348                       struct wt_status *s)
349 {
350         unsigned char sha1[20];
351
352         if (s->relative_paths)
353                 s->prefix = prefix;
354
355         if (amend) {
356                 s->amend = 1;
357                 s->reference = "HEAD^1";
358         }
359         s->verbose = verbose;
360         s->index_file = index_file;
361         s->fp = fp;
362         s->nowarn = nowarn;
363         s->is_initial = get_sha1(s->reference, sha1) ? 1 : 0;
364
365         wt_status_collect(s);
366         wt_status_print(s);
367
368         return s->commitable;
369 }
370
371 static int is_a_merge(const unsigned char *sha1)
372 {
373         struct commit *commit = lookup_commit(sha1);
374         if (!commit || parse_commit(commit))
375                 die("could not parse HEAD commit");
376         return !!(commit->parents && commit->parents->next);
377 }
378
379 static const char sign_off_header[] = "Signed-off-by: ";
380
381 static void determine_author_info(void)
382 {
383         char *name, *email, *date;
384
385         name = getenv("GIT_AUTHOR_NAME");
386         email = getenv("GIT_AUTHOR_EMAIL");
387         date = getenv("GIT_AUTHOR_DATE");
388
389         if (use_message) {
390                 const char *a, *lb, *rb, *eol;
391
392                 a = strstr(use_message_buffer, "\nauthor ");
393                 if (!a)
394                         die("invalid commit: %s", use_message);
395
396                 lb = strstr(a + 8, " <");
397                 rb = strstr(a + 8, "> ");
398                 eol = strchr(a + 8, '\n');
399                 if (!lb || !rb || !eol)
400                         die("invalid commit: %s", use_message);
401
402                 name = xstrndup(a + 8, lb - (a + 8));
403                 email = xstrndup(lb + 2, rb - (lb + 2));
404                 date = xstrndup(rb + 2, eol - (rb + 2));
405         }
406
407         if (force_author) {
408                 const char *lb = strstr(force_author, " <");
409                 const char *rb = strchr(force_author, '>');
410
411                 if (!lb || !rb)
412                         die("malformed --author parameter");
413                 name = xstrndup(force_author, lb - force_author);
414                 email = xstrndup(lb + 2, rb - (lb + 2));
415         }
416
417         author_name = name;
418         author_email = email;
419         author_date = date;
420 }
421
422 static int prepare_to_commit(const char *index_file, const char *prefix,
423                              struct wt_status *s)
424 {
425         struct stat statbuf;
426         int commitable, saved_color_setting;
427         struct strbuf sb = STRBUF_INIT;
428         char *buffer;
429         FILE *fp;
430         const char *hook_arg1 = NULL;
431         const char *hook_arg2 = NULL;
432         int ident_shown = 0;
433
434         if (!no_verify && run_hook(index_file, "pre-commit", NULL))
435                 return 0;
436
437         if (message.len) {
438                 strbuf_addbuf(&sb, &message);
439                 hook_arg1 = "message";
440         } else if (logfile && !strcmp(logfile, "-")) {
441                 if (isatty(0))
442                         fprintf(stderr, "(reading log message from standard input)\n");
443                 if (strbuf_read(&sb, 0, 0) < 0)
444                         die_errno("could not read log from standard input");
445                 hook_arg1 = "message";
446         } else if (logfile) {
447                 if (strbuf_read_file(&sb, logfile, 0) < 0)
448                         die_errno("could not read log file '%s'",
449                                   logfile);
450                 hook_arg1 = "message";
451         } else if (use_message) {
452                 buffer = strstr(use_message_buffer, "\n\n");
453                 if (!buffer || buffer[2] == '\0')
454                         die("commit has empty message");
455                 strbuf_add(&sb, buffer + 2, strlen(buffer + 2));
456                 hook_arg1 = "commit";
457                 hook_arg2 = use_message;
458         } else if (!stat(git_path("MERGE_MSG"), &statbuf)) {
459                 if (strbuf_read_file(&sb, git_path("MERGE_MSG"), 0) < 0)
460                         die_errno("could not read MERGE_MSG");
461                 hook_arg1 = "merge";
462         } else if (!stat(git_path("SQUASH_MSG"), &statbuf)) {
463                 if (strbuf_read_file(&sb, git_path("SQUASH_MSG"), 0) < 0)
464                         die_errno("could not read SQUASH_MSG");
465                 hook_arg1 = "squash";
466         } else if (template_file && !stat(template_file, &statbuf)) {
467                 if (strbuf_read_file(&sb, template_file, 0) < 0)
468                         die_errno("could not read '%s'", template_file);
469                 hook_arg1 = "template";
470         }
471
472         /*
473          * This final case does not modify the template message,
474          * it just sets the argument to the prepare-commit-msg hook.
475          */
476         else if (in_merge)
477                 hook_arg1 = "merge";
478
479         fp = fopen(git_path(commit_editmsg), "w");
480         if (fp == NULL)
481                 die_errno("could not open '%s'", git_path(commit_editmsg));
482
483         if (cleanup_mode != CLEANUP_NONE)
484                 stripspace(&sb, 0);
485
486         if (signoff) {
487                 struct strbuf sob = STRBUF_INIT;
488                 int i;
489
490                 strbuf_addstr(&sob, sign_off_header);
491                 strbuf_addstr(&sob, fmt_name(getenv("GIT_COMMITTER_NAME"),
492                                              getenv("GIT_COMMITTER_EMAIL")));
493                 strbuf_addch(&sob, '\n');
494                 for (i = sb.len - 1; i > 0 && sb.buf[i - 1] != '\n'; i--)
495                         ; /* do nothing */
496                 if (prefixcmp(sb.buf + i, sob.buf)) {
497                         if (prefixcmp(sb.buf + i, sign_off_header))
498                                 strbuf_addch(&sb, '\n');
499                         strbuf_addbuf(&sb, &sob);
500                 }
501                 strbuf_release(&sob);
502         }
503
504         if (fwrite(sb.buf, 1, sb.len, fp) < sb.len)
505                 die_errno("could not write commit template");
506
507         strbuf_release(&sb);
508
509         determine_author_info();
510
511         /* This checks if committer ident is explicitly given */
512         git_committer_info(0);
513         if (use_editor) {
514                 char *author_ident;
515                 const char *committer_ident;
516
517                 if (in_merge)
518                         fprintf(fp,
519                                 "#\n"
520                                 "# It looks like you may be committing a MERGE.\n"
521                                 "# If this is not correct, please remove the file\n"
522                                 "#      %s\n"
523                                 "# and try again.\n"
524                                 "#\n",
525                                 git_path("MERGE_HEAD"));
526
527                 fprintf(fp,
528                         "\n"
529                         "# Please enter the commit message for your changes.");
530                 if (cleanup_mode == CLEANUP_ALL)
531                         fprintf(fp,
532                                 " Lines starting\n"
533                                 "# with '#' will be ignored, and an empty"
534                                 " message aborts the commit.\n");
535                 else /* CLEANUP_SPACE, that is. */
536                         fprintf(fp,
537                                 " Lines starting\n"
538                                 "# with '#' will be kept; you may remove them"
539                                 " yourself if you want to.\n"
540                                 "# An empty message aborts the commit.\n");
541                 if (only_include_assumed)
542                         fprintf(fp, "# %s\n", only_include_assumed);
543
544                 author_ident = xstrdup(fmt_name(author_name, author_email));
545                 committer_ident = fmt_name(getenv("GIT_COMMITTER_NAME"),
546                                            getenv("GIT_COMMITTER_EMAIL"));
547                 if (strcmp(author_ident, committer_ident))
548                         fprintf(fp,
549                                 "%s"
550                                 "# Author:    %s\n",
551                                 ident_shown++ ? "" : "#\n",
552                                 author_ident);
553                 free(author_ident);
554
555                 if (!user_ident_explicitly_given)
556                         fprintf(fp,
557                                 "%s"
558                                 "# Committer: %s\n",
559                                 ident_shown++ ? "" : "#\n",
560                                 committer_ident);
561
562                 if (ident_shown)
563                         fprintf(fp, "#\n");
564
565                 saved_color_setting = s->use_color;
566                 s->use_color = 0;
567                 commitable = run_status(fp, index_file, prefix, 1, s);
568                 s->use_color = saved_color_setting;
569         } else {
570                 unsigned char sha1[20];
571                 const char *parent = "HEAD";
572
573                 if (!active_nr && read_cache() < 0)
574                         die("Cannot read index");
575
576                 if (amend)
577                         parent = "HEAD^1";
578
579                 if (get_sha1(parent, sha1))
580                         commitable = !!active_nr;
581                 else
582                         commitable = index_differs_from(parent, 0);
583         }
584
585         fclose(fp);
586
587         if (!commitable && !in_merge && !allow_empty &&
588             !(amend && is_a_merge(head_sha1))) {
589                 run_status(stdout, index_file, prefix, 0, s);
590                 return 0;
591         }
592
593         /*
594          * Re-read the index as pre-commit hook could have updated it,
595          * and write it out as a tree.  We must do this before we invoke
596          * the editor and after we invoke run_status above.
597          */
598         discard_cache();
599         read_cache_from(index_file);
600         if (!active_cache_tree)
601                 active_cache_tree = cache_tree();
602         if (cache_tree_update(active_cache_tree,
603                               active_cache, active_nr, 0, 0) < 0) {
604                 error("Error building trees");
605                 return 0;
606         }
607
608         if (run_hook(index_file, "prepare-commit-msg",
609                      git_path(commit_editmsg), hook_arg1, hook_arg2, NULL))
610                 return 0;
611
612         if (use_editor) {
613                 char index[PATH_MAX];
614                 const char *env[2] = { index, NULL };
615                 snprintf(index, sizeof(index), "GIT_INDEX_FILE=%s", index_file);
616                 if (launch_editor(git_path(commit_editmsg), NULL, env)) {
617                         fprintf(stderr,
618                         "Please supply the message using either -m or -F option.\n");
619                         exit(1);
620                 }
621         }
622
623         if (!no_verify &&
624             run_hook(index_file, "commit-msg", git_path(commit_editmsg), NULL)) {
625                 return 0;
626         }
627
628         return 1;
629 }
630
631 /*
632  * Find out if the message in the strbuf contains only whitespace and
633  * Signed-off-by lines.
634  */
635 static int message_is_empty(struct strbuf *sb)
636 {
637         struct strbuf tmpl = STRBUF_INIT;
638         const char *nl;
639         int eol, i, start = 0;
640
641         if (cleanup_mode == CLEANUP_NONE && sb->len)
642                 return 0;
643
644         /* See if the template is just a prefix of the message. */
645         if (template_file && strbuf_read_file(&tmpl, template_file, 0) > 0) {
646                 stripspace(&tmpl, cleanup_mode == CLEANUP_ALL);
647                 if (start + tmpl.len <= sb->len &&
648                     memcmp(tmpl.buf, sb->buf + start, tmpl.len) == 0)
649                         start += tmpl.len;
650         }
651         strbuf_release(&tmpl);
652
653         /* Check if the rest is just whitespace and Signed-of-by's. */
654         for (i = start; i < sb->len; i++) {
655                 nl = memchr(sb->buf + i, '\n', sb->len - i);
656                 if (nl)
657                         eol = nl - sb->buf;
658                 else
659                         eol = sb->len;
660
661                 if (strlen(sign_off_header) <= eol - i &&
662                     !prefixcmp(sb->buf + i, sign_off_header)) {
663                         i = eol;
664                         continue;
665                 }
666                 while (i < eol)
667                         if (!isspace(sb->buf[i++]))
668                                 return 0;
669         }
670
671         return 1;
672 }
673
674 static const char *find_author_by_nickname(const char *name)
675 {
676         struct rev_info revs;
677         struct commit *commit;
678         struct strbuf buf = STRBUF_INIT;
679         const char *av[20];
680         int ac = 0;
681
682         init_revisions(&revs, NULL);
683         strbuf_addf(&buf, "--author=%s", name);
684         av[++ac] = "--all";
685         av[++ac] = "-i";
686         av[++ac] = buf.buf;
687         av[++ac] = NULL;
688         setup_revisions(ac, av, &revs, NULL);
689         prepare_revision_walk(&revs);
690         commit = get_revision(&revs);
691         if (commit) {
692                 strbuf_release(&buf);
693                 format_commit_message(commit, "%an <%ae>", &buf, DATE_NORMAL);
694                 return strbuf_detach(&buf, NULL);
695         }
696         die("No existing author found with '%s'", name);
697 }
698
699
700 static void handle_untracked_files_arg(struct wt_status *s)
701 {
702         if (!untracked_files_arg)
703                 ; /* default already initialized */
704         else if (!strcmp(untracked_files_arg, "no"))
705                 s->show_untracked_files = SHOW_NO_UNTRACKED_FILES;
706         else if (!strcmp(untracked_files_arg, "normal"))
707                 s->show_untracked_files = SHOW_NORMAL_UNTRACKED_FILES;
708         else if (!strcmp(untracked_files_arg, "all"))
709                 s->show_untracked_files = SHOW_ALL_UNTRACKED_FILES;
710         else
711                 die("Invalid untracked files mode '%s'", untracked_files_arg);
712 }
713
714 static int parse_and_validate_options(int argc, const char *argv[],
715                                       const char * const usage[],
716                                       const char *prefix,
717                                       struct wt_status *s)
718 {
719         int f = 0;
720
721         argc = parse_options(argc, argv, prefix, builtin_commit_options, usage,
722                              0);
723
724         if (force_author && !strchr(force_author, '>'))
725                 force_author = find_author_by_nickname(force_author);
726
727         if (logfile || message.len || use_message)
728                 use_editor = 0;
729         if (edit_flag)
730                 use_editor = 1;
731         if (!use_editor)
732                 setenv("GIT_EDITOR", ":", 1);
733
734         if (get_sha1("HEAD", head_sha1))
735                 initial_commit = 1;
736
737         if (!get_sha1("MERGE_HEAD", merge_head_sha1))
738                 in_merge = 1;
739
740         /* Sanity check options */
741         if (amend && initial_commit)
742                 die("You have nothing to amend.");
743         if (amend && in_merge)
744                 die("You are in the middle of a merge -- cannot amend.");
745
746         if (use_message)
747                 f++;
748         if (edit_message)
749                 f++;
750         if (logfile)
751                 f++;
752         if (f > 1)
753                 die("Only one of -c/-C/-F can be used.");
754         if (message.len && f > 0)
755                 die("Option -m cannot be combined with -c/-C/-F.");
756         if (edit_message)
757                 use_message = edit_message;
758         if (amend && !use_message)
759                 use_message = "HEAD";
760         if (use_message) {
761                 unsigned char sha1[20];
762                 static char utf8[] = "UTF-8";
763                 const char *out_enc;
764                 char *enc, *end;
765                 struct commit *commit;
766
767                 if (get_sha1(use_message, sha1))
768                         die("could not lookup commit %s", use_message);
769                 commit = lookup_commit_reference(sha1);
770                 if (!commit || parse_commit(commit))
771                         die("could not parse commit %s", use_message);
772
773                 enc = strstr(commit->buffer, "\nencoding");
774                 if (enc) {
775                         end = strchr(enc + 10, '\n');
776                         enc = xstrndup(enc + 10, end - (enc + 10));
777                 } else {
778                         enc = utf8;
779                 }
780                 out_enc = git_commit_encoding ? git_commit_encoding : utf8;
781
782                 if (strcmp(out_enc, enc))
783                         use_message_buffer =
784                                 reencode_string(commit->buffer, out_enc, enc);
785
786                 /*
787                  * If we failed to reencode the buffer, just copy it
788                  * byte for byte so the user can try to fix it up.
789                  * This also handles the case where input and output
790                  * encodings are identical.
791                  */
792                 if (use_message_buffer == NULL)
793                         use_message_buffer = xstrdup(commit->buffer);
794                 if (enc != utf8)
795                         free(enc);
796         }
797
798         if (!!also + !!only + !!all + !!interactive > 1)
799                 die("Only one of --include/--only/--all/--interactive can be used.");
800         if (argc == 0 && (also || (only && !amend)))
801                 die("No paths with --include/--only does not make sense.");
802         if (argc == 0 && only && amend)
803                 only_include_assumed = "Clever... amending the last one with dirty index.";
804         if (argc > 0 && !also && !only)
805                 only_include_assumed = "Explicit paths specified without -i nor -o; assuming --only paths...";
806         if (!cleanup_arg || !strcmp(cleanup_arg, "default"))
807                 cleanup_mode = use_editor ? CLEANUP_ALL : CLEANUP_SPACE;
808         else if (!strcmp(cleanup_arg, "verbatim"))
809                 cleanup_mode = CLEANUP_NONE;
810         else if (!strcmp(cleanup_arg, "whitespace"))
811                 cleanup_mode = CLEANUP_SPACE;
812         else if (!strcmp(cleanup_arg, "strip"))
813                 cleanup_mode = CLEANUP_ALL;
814         else
815                 die("Invalid cleanup mode %s", cleanup_arg);
816
817         handle_untracked_files_arg(s);
818
819         if (all && argc > 0)
820                 die("Paths with -a does not make sense.");
821         else if (interactive && argc > 0)
822                 die("Paths with --interactive does not make sense.");
823
824         return argc;
825 }
826
827 static int dry_run_commit(int argc, const char **argv, const char *prefix,
828                           struct wt_status *s)
829 {
830         int commitable;
831         const char *index_file;
832
833         index_file = prepare_index(argc, argv, prefix, 1);
834         commitable = run_status(stdout, index_file, prefix, 0, s);
835         rollback_index_files();
836
837         return commitable ? 0 : 1;
838 }
839
840 static int parse_status_slot(const char *var, int offset)
841 {
842         if (!strcasecmp(var+offset, "header"))
843                 return WT_STATUS_HEADER;
844         if (!strcasecmp(var+offset, "updated")
845                 || !strcasecmp(var+offset, "added"))
846                 return WT_STATUS_UPDATED;
847         if (!strcasecmp(var+offset, "changed"))
848                 return WT_STATUS_CHANGED;
849         if (!strcasecmp(var+offset, "untracked"))
850                 return WT_STATUS_UNTRACKED;
851         if (!strcasecmp(var+offset, "nobranch"))
852                 return WT_STATUS_NOBRANCH;
853         if (!strcasecmp(var+offset, "unmerged"))
854                 return WT_STATUS_UNMERGED;
855         die("bad config variable '%s'", var);
856 }
857
858 static int git_status_config(const char *k, const char *v, void *cb)
859 {
860         struct wt_status *s = cb;
861
862         if (!strcmp(k, "status.submodulesummary")) {
863                 int is_bool;
864                 s->submodule_summary = git_config_bool_or_int(k, v, &is_bool);
865                 if (is_bool && s->submodule_summary)
866                         s->submodule_summary = -1;
867                 return 0;
868         }
869         if (!strcmp(k, "status.color") || !strcmp(k, "color.status")) {
870                 s->use_color = git_config_colorbool(k, v, -1);
871                 return 0;
872         }
873         if (!prefixcmp(k, "status.color.") || !prefixcmp(k, "color.status.")) {
874                 int slot = parse_status_slot(k, 13);
875                 if (!v)
876                         return config_error_nonbool(k);
877                 color_parse(v, k, s->color_palette[slot]);
878                 return 0;
879         }
880         if (!strcmp(k, "status.relativepaths")) {
881                 s->relative_paths = git_config_bool(k, v);
882                 return 0;
883         }
884         if (!strcmp(k, "status.showuntrackedfiles")) {
885                 if (!v)
886                         return config_error_nonbool(k);
887                 else if (!strcmp(v, "no"))
888                         s->show_untracked_files = SHOW_NO_UNTRACKED_FILES;
889                 else if (!strcmp(v, "normal"))
890                         s->show_untracked_files = SHOW_NORMAL_UNTRACKED_FILES;
891                 else if (!strcmp(v, "all"))
892                         s->show_untracked_files = SHOW_ALL_UNTRACKED_FILES;
893                 else
894                         return error("Invalid untracked files mode '%s'", v);
895                 return 0;
896         }
897         return git_diff_ui_config(k, v, NULL);
898 }
899
900 #define quote_path quote_path_relative
901
902 static void short_unmerged(int null_termination, struct string_list_item *it,
903                            struct wt_status *s)
904 {
905         struct wt_status_change_data *d = it->util;
906         const char *how = "??";
907
908         switch (d->stagemask) {
909         case 1: how = "DD"; break; /* both deleted */
910         case 2: how = "AU"; break; /* added by us */
911         case 3: how = "UD"; break; /* deleted by them */
912         case 4: how = "UA"; break; /* added by them */
913         case 5: how = "DU"; break; /* deleted by us */
914         case 6: how = "AA"; break; /* both added */
915         case 7: how = "UU"; break; /* both modified */
916         }
917         printf("%s ", how);
918         if (null_termination) {
919                 fprintf(stdout, "%s%c", it->string, 0);
920         } else {
921                 struct strbuf onebuf = STRBUF_INIT;
922                 const char *one;
923                 one = quote_path(it->string, -1, &onebuf, s->prefix);
924                 printf("%s\n", one);
925                 strbuf_release(&onebuf);
926         }
927 }
928
929 static void short_status(int null_termination, struct string_list_item *it,
930                          struct wt_status *s)
931 {
932         struct wt_status_change_data *d = it->util;
933
934         printf("%c%c ",
935                !d->index_status ? ' ' : d->index_status,
936                !d->worktree_status ? ' ' : d->worktree_status);
937         if (null_termination) {
938                 fprintf(stdout, "%s%c", it->string, 0);
939                 if (d->head_path)
940                         fprintf(stdout, "%s%c", d->head_path, 0);
941         } else {
942                 struct strbuf onebuf = STRBUF_INIT;
943                 const char *one;
944                 if (d->head_path) {
945                         one = quote_path(d->head_path, -1, &onebuf, s->prefix);
946                         printf("%s -> ", one);
947                         strbuf_release(&onebuf);
948                 }
949                 one = quote_path(it->string, -1, &onebuf, s->prefix);
950                 printf("%s\n", one);
951                 strbuf_release(&onebuf);
952         }
953 }
954
955 static void short_untracked(int null_termination, struct string_list_item *it,
956                             struct wt_status *s)
957 {
958         if (null_termination) {
959                 fprintf(stdout, "?? %s%c", it->string, 0);
960         } else {
961                 struct strbuf onebuf = STRBUF_INIT;
962                 const char *one;
963                 one = quote_path(it->string, -1, &onebuf, s->prefix);
964                 printf("?? %s\n", one);
965                 strbuf_release(&onebuf);
966         }
967 }
968
969 int cmd_status(int argc, const char **argv, const char *prefix)
970 {
971         struct wt_status s;
972         static int null_termination, shortstatus;
973         int i;
974         unsigned char sha1[20];
975         static struct option builtin_status_options[] = {
976                 OPT__VERBOSE(&verbose),
977                 OPT_BOOLEAN('s', "short", &shortstatus,
978                             "show status concicely"),
979                 OPT_BOOLEAN('z', "null", &null_termination,
980                             "terminate entries with NUL"),
981                 { OPTION_STRING, 'u', "untracked-files", &untracked_files_arg,
982                   "mode",
983                   "show untracked files, optional modes: all, normal, no. (Default: all)",
984                   PARSE_OPT_OPTARG, NULL, (intptr_t)"all" },
985                 OPT_END(),
986         };
987
988         if (null_termination)
989                 shortstatus = 1;
990
991         wt_status_prepare(&s);
992         git_config(git_status_config, &s);
993         argc = parse_options(argc, argv, prefix,
994                              builtin_status_options,
995                              builtin_status_usage, 0);
996         handle_untracked_files_arg(&s);
997
998         if (*argv)
999                 s.pathspec = get_pathspec(prefix, argv);
1000
1001         read_cache();
1002         refresh_cache(REFRESH_QUIET|REFRESH_UNMERGED);
1003         s.is_initial = get_sha1(s.reference, sha1) ? 1 : 0;
1004         wt_status_collect(&s);
1005
1006         if (shortstatus) {
1007                 for (i = 0; i < s.change.nr; i++) {
1008                         struct wt_status_change_data *d;
1009                         struct string_list_item *it;
1010
1011                         it = &(s.change.items[i]);
1012                         d = it->util;
1013                         if (d->stagemask)
1014                                 short_unmerged(null_termination, it, &s);
1015                         else
1016                                 short_status(null_termination, it, &s);
1017                 }
1018                 for (i = 0; i < s.untracked.nr; i++) {
1019                         struct string_list_item *it;
1020
1021                         it = &(s.untracked.items[i]);
1022                         short_untracked(null_termination, it, &s);
1023                 }
1024         } else {
1025                 s.verbose = verbose;
1026                 if (s.relative_paths)
1027                         s.prefix = prefix;
1028                 if (s.use_color == -1)
1029                         s.use_color = git_use_color_default;
1030                 if (diff_use_color_default == -1)
1031                         diff_use_color_default = git_use_color_default;
1032                 wt_status_print(&s);
1033         }
1034         return 0;
1035 }
1036
1037 static void print_summary(const char *prefix, const unsigned char *sha1)
1038 {
1039         struct rev_info rev;
1040         struct commit *commit;
1041         static const char *format = "format:%h] %s";
1042         unsigned char junk_sha1[20];
1043         const char *head = resolve_ref("HEAD", junk_sha1, 0, NULL);
1044
1045         commit = lookup_commit(sha1);
1046         if (!commit)
1047                 die("couldn't look up newly created commit");
1048         if (!commit || parse_commit(commit))
1049                 die("could not parse newly created commit");
1050
1051         init_revisions(&rev, prefix);
1052         setup_revisions(0, NULL, &rev, NULL);
1053
1054         rev.abbrev = 0;
1055         rev.diff = 1;
1056         rev.diffopt.output_format =
1057                 DIFF_FORMAT_SHORTSTAT | DIFF_FORMAT_SUMMARY;
1058
1059         rev.verbose_header = 1;
1060         rev.show_root_diff = 1;
1061         get_commit_format(format, &rev);
1062         rev.always_show_header = 0;
1063         rev.diffopt.detect_rename = 1;
1064         rev.diffopt.rename_limit = 100;
1065         rev.diffopt.break_opt = 0;
1066         diff_setup_done(&rev.diffopt);
1067
1068         printf("[%s%s ",
1069                 !prefixcmp(head, "refs/heads/") ?
1070                         head + 11 :
1071                         !strcmp(head, "HEAD") ?
1072                                 "detached HEAD" :
1073                                 head,
1074                 initial_commit ? " (root-commit)" : "");
1075
1076         if (!log_tree_commit(&rev, commit)) {
1077                 struct strbuf buf = STRBUF_INIT;
1078                 format_commit_message(commit, format + 7, &buf, DATE_NORMAL);
1079                 printf("%s\n", buf.buf);
1080                 strbuf_release(&buf);
1081         }
1082 }
1083
1084 static int git_commit_config(const char *k, const char *v, void *cb)
1085 {
1086         struct wt_status *s = cb;
1087
1088         if (!strcmp(k, "commit.template"))
1089                 return git_config_string(&template_file, k, v);
1090
1091         return git_status_config(k, v, s);
1092 }
1093
1094 int cmd_commit(int argc, const char **argv, const char *prefix)
1095 {
1096         struct strbuf sb = STRBUF_INIT;
1097         const char *index_file, *reflog_msg;
1098         char *nl, *p;
1099         unsigned char commit_sha1[20];
1100         struct ref_lock *ref_lock;
1101         struct commit_list *parents = NULL, **pptr = &parents;
1102         struct stat statbuf;
1103         int allow_fast_forward = 1;
1104         struct wt_status s;
1105
1106         wt_status_prepare(&s);
1107         git_config(git_commit_config, &s);
1108
1109         if (s.use_color == -1)
1110                 s.use_color = git_use_color_default;
1111
1112         argc = parse_and_validate_options(argc, argv, builtin_commit_usage,
1113                                           prefix, &s);
1114         if (dry_run) {
1115                 if (diff_use_color_default == -1)
1116                         diff_use_color_default = git_use_color_default;
1117                 return dry_run_commit(argc, argv, prefix, &s);
1118         }
1119         index_file = prepare_index(argc, argv, prefix, 0);
1120
1121         /* Set up everything for writing the commit object.  This includes
1122            running hooks, writing the trees, and interacting with the user.  */
1123         if (!prepare_to_commit(index_file, prefix, &s)) {
1124                 rollback_index_files();
1125                 return 1;
1126         }
1127
1128         /* Determine parents */
1129         if (initial_commit) {
1130                 reflog_msg = "commit (initial)";
1131         } else if (amend) {
1132                 struct commit_list *c;
1133                 struct commit *commit;
1134
1135                 reflog_msg = "commit (amend)";
1136                 commit = lookup_commit(head_sha1);
1137                 if (!commit || parse_commit(commit))
1138                         die("could not parse HEAD commit");
1139
1140                 for (c = commit->parents; c; c = c->next)
1141                         pptr = &commit_list_insert(c->item, pptr)->next;
1142         } else if (in_merge) {
1143                 struct strbuf m = STRBUF_INIT;
1144                 FILE *fp;
1145
1146                 reflog_msg = "commit (merge)";
1147                 pptr = &commit_list_insert(lookup_commit(head_sha1), pptr)->next;
1148                 fp = fopen(git_path("MERGE_HEAD"), "r");
1149                 if (fp == NULL)
1150                         die_errno("could not open '%s' for reading",
1151                                   git_path("MERGE_HEAD"));
1152                 while (strbuf_getline(&m, fp, '\n') != EOF) {
1153                         unsigned char sha1[20];
1154                         if (get_sha1_hex(m.buf, sha1) < 0)
1155                                 die("Corrupt MERGE_HEAD file (%s)", m.buf);
1156                         pptr = &commit_list_insert(lookup_commit(sha1), pptr)->next;
1157                 }
1158                 fclose(fp);
1159                 strbuf_release(&m);
1160                 if (!stat(git_path("MERGE_MODE"), &statbuf)) {
1161                         if (strbuf_read_file(&sb, git_path("MERGE_MODE"), 0) < 0)
1162                                 die_errno("could not read MERGE_MODE");
1163                         if (!strcmp(sb.buf, "no-ff"))
1164                                 allow_fast_forward = 0;
1165                 }
1166                 if (allow_fast_forward)
1167                         parents = reduce_heads(parents);
1168         } else {
1169                 reflog_msg = "commit";
1170                 pptr = &commit_list_insert(lookup_commit(head_sha1), pptr)->next;
1171         }
1172
1173         /* Finally, get the commit message */
1174         strbuf_reset(&sb);
1175         if (strbuf_read_file(&sb, git_path(commit_editmsg), 0) < 0) {
1176                 int saved_errno = errno;
1177                 rollback_index_files();
1178                 die("could not read commit message: %s", strerror(saved_errno));
1179         }
1180
1181         /* Truncate the message just before the diff, if any. */
1182         if (verbose) {
1183                 p = strstr(sb.buf, "\ndiff --git ");
1184                 if (p != NULL)
1185                         strbuf_setlen(&sb, p - sb.buf + 1);
1186         }
1187
1188         if (cleanup_mode != CLEANUP_NONE)
1189                 stripspace(&sb, cleanup_mode == CLEANUP_ALL);
1190         if (message_is_empty(&sb)) {
1191                 rollback_index_files();
1192                 fprintf(stderr, "Aborting commit due to empty commit message.\n");
1193                 exit(1);
1194         }
1195
1196         if (commit_tree(sb.buf, active_cache_tree->sha1, parents, commit_sha1,
1197                         fmt_ident(author_name, author_email, author_date,
1198                                 IDENT_ERROR_ON_NO_NAME))) {
1199                 rollback_index_files();
1200                 die("failed to write commit object");
1201         }
1202
1203         ref_lock = lock_any_ref_for_update("HEAD",
1204                                            initial_commit ? NULL : head_sha1,
1205                                            0);
1206
1207         nl = strchr(sb.buf, '\n');
1208         if (nl)
1209                 strbuf_setlen(&sb, nl + 1 - sb.buf);
1210         else
1211                 strbuf_addch(&sb, '\n');
1212         strbuf_insert(&sb, 0, reflog_msg, strlen(reflog_msg));
1213         strbuf_insert(&sb, strlen(reflog_msg), ": ", 2);
1214
1215         if (!ref_lock) {
1216                 rollback_index_files();
1217                 die("cannot lock HEAD ref");
1218         }
1219         if (write_ref_sha1(ref_lock, commit_sha1, sb.buf) < 0) {
1220                 rollback_index_files();
1221                 die("cannot update HEAD ref");
1222         }
1223
1224         unlink(git_path("MERGE_HEAD"));
1225         unlink(git_path("MERGE_MSG"));
1226         unlink(git_path("MERGE_MODE"));
1227         unlink(git_path("SQUASH_MSG"));
1228
1229         if (commit_index_files())
1230                 die ("Repository has been updated, but unable to write\n"
1231                      "new_index file. Check that disk is not full or quota is\n"
1232                      "not exceeded, and then \"git reset HEAD\" to recover.");
1233
1234         rerere();
1235         run_hook(get_index_file(), "post-commit", NULL);
1236         if (!quiet)
1237                 print_summary(prefix, commit_sha1);
1238
1239         return 0;
1240 }