Merge branch 'jk/add-p-commentchar-fix'
[git] / pretty.c
1 #include "cache.h"
2 #include "config.h"
3 #include "commit.h"
4 #include "utf8.h"
5 #include "diff.h"
6 #include "revision.h"
7 #include "string-list.h"
8 #include "mailmap.h"
9 #include "log-tree.h"
10 #include "notes.h"
11 #include "color.h"
12 #include "reflog-walk.h"
13 #include "gpg-interface.h"
14 #include "trailer.h"
15
16 static char *user_format;
17 static struct cmt_fmt_map {
18         const char *name;
19         enum cmit_fmt format;
20         int is_tformat;
21         int expand_tabs_in_log;
22         int is_alias;
23         const char *user_format;
24 } *commit_formats;
25 static size_t builtin_formats_len;
26 static size_t commit_formats_len;
27 static size_t commit_formats_alloc;
28 static struct cmt_fmt_map *find_commit_format(const char *sought);
29
30 int commit_format_is_empty(enum cmit_fmt fmt)
31 {
32         return fmt == CMIT_FMT_USERFORMAT && !*user_format;
33 }
34
35 static void save_user_format(struct rev_info *rev, const char *cp, int is_tformat)
36 {
37         free(user_format);
38         user_format = xstrdup(cp);
39         if (is_tformat)
40                 rev->use_terminator = 1;
41         rev->commit_format = CMIT_FMT_USERFORMAT;
42 }
43
44 static int git_pretty_formats_config(const char *var, const char *value, void *cb)
45 {
46         struct cmt_fmt_map *commit_format = NULL;
47         const char *name;
48         const char *fmt;
49         int i;
50
51         if (!skip_prefix(var, "pretty.", &name))
52                 return 0;
53
54         for (i = 0; i < builtin_formats_len; i++) {
55                 if (!strcmp(commit_formats[i].name, name))
56                         return 0;
57         }
58
59         for (i = builtin_formats_len; i < commit_formats_len; i++) {
60                 if (!strcmp(commit_formats[i].name, name)) {
61                         commit_format = &commit_formats[i];
62                         break;
63                 }
64         }
65
66         if (!commit_format) {
67                 ALLOC_GROW(commit_formats, commit_formats_len+1,
68                            commit_formats_alloc);
69                 commit_format = &commit_formats[commit_formats_len];
70                 memset(commit_format, 0, sizeof(*commit_format));
71                 commit_formats_len++;
72         }
73
74         commit_format->name = xstrdup(name);
75         commit_format->format = CMIT_FMT_USERFORMAT;
76         if (git_config_string(&fmt, var, value))
77                 return -1;
78
79         if (skip_prefix(fmt, "format:", &fmt))
80                 commit_format->is_tformat = 0;
81         else if (skip_prefix(fmt, "tformat:", &fmt) || strchr(fmt, '%'))
82                 commit_format->is_tformat = 1;
83         else
84                 commit_format->is_alias = 1;
85         commit_format->user_format = fmt;
86
87         return 0;
88 }
89
90 static void setup_commit_formats(void)
91 {
92         struct cmt_fmt_map builtin_formats[] = {
93                 { "raw",        CMIT_FMT_RAW,           0,      0 },
94                 { "medium",     CMIT_FMT_MEDIUM,        0,      8 },
95                 { "short",      CMIT_FMT_SHORT,         0,      0 },
96                 { "email",      CMIT_FMT_EMAIL,         0,      0 },
97                 { "mboxrd",     CMIT_FMT_MBOXRD,        0,      0 },
98                 { "fuller",     CMIT_FMT_FULLER,        0,      8 },
99                 { "full",       CMIT_FMT_FULL,          0,      8 },
100                 { "oneline",    CMIT_FMT_ONELINE,       1,      0 }
101         };
102         commit_formats_len = ARRAY_SIZE(builtin_formats);
103         builtin_formats_len = commit_formats_len;
104         ALLOC_GROW(commit_formats, commit_formats_len, commit_formats_alloc);
105         memcpy(commit_formats, builtin_formats,
106                sizeof(*builtin_formats)*ARRAY_SIZE(builtin_formats));
107
108         git_config(git_pretty_formats_config, NULL);
109 }
110
111 static struct cmt_fmt_map *find_commit_format_recursive(const char *sought,
112                                                         const char *original,
113                                                         int num_redirections)
114 {
115         struct cmt_fmt_map *found = NULL;
116         size_t found_match_len = 0;
117         int i;
118
119         if (num_redirections >= commit_formats_len)
120                 die("invalid --pretty format: "
121                     "'%s' references an alias which points to itself",
122                     original);
123
124         for (i = 0; i < commit_formats_len; i++) {
125                 size_t match_len;
126
127                 if (!starts_with(commit_formats[i].name, sought))
128                         continue;
129
130                 match_len = strlen(commit_formats[i].name);
131                 if (found == NULL || found_match_len > match_len) {
132                         found = &commit_formats[i];
133                         found_match_len = match_len;
134                 }
135         }
136
137         if (found && found->is_alias) {
138                 found = find_commit_format_recursive(found->user_format,
139                                                      original,
140                                                      num_redirections+1);
141         }
142
143         return found;
144 }
145
146 static struct cmt_fmt_map *find_commit_format(const char *sought)
147 {
148         if (!commit_formats)
149                 setup_commit_formats();
150
151         return find_commit_format_recursive(sought, sought, 0);
152 }
153
154 void get_commit_format(const char *arg, struct rev_info *rev)
155 {
156         struct cmt_fmt_map *commit_format;
157
158         rev->use_terminator = 0;
159         if (!arg) {
160                 rev->commit_format = CMIT_FMT_DEFAULT;
161                 return;
162         }
163         if (skip_prefix(arg, "format:", &arg)) {
164                 save_user_format(rev, arg, 0);
165                 return;
166         }
167
168         if (!*arg || skip_prefix(arg, "tformat:", &arg) || strchr(arg, '%')) {
169                 save_user_format(rev, arg, 1);
170                 return;
171         }
172
173         commit_format = find_commit_format(arg);
174         if (!commit_format)
175                 die("invalid --pretty format: %s", arg);
176
177         rev->commit_format = commit_format->format;
178         rev->use_terminator = commit_format->is_tformat;
179         rev->expand_tabs_in_log_default = commit_format->expand_tabs_in_log;
180         if (commit_format->format == CMIT_FMT_USERFORMAT) {
181                 save_user_format(rev, commit_format->user_format,
182                                  commit_format->is_tformat);
183         }
184 }
185
186 /*
187  * Generic support for pretty-printing the header
188  */
189 static int get_one_line(const char *msg)
190 {
191         int ret = 0;
192
193         for (;;) {
194                 char c = *msg++;
195                 if (!c)
196                         break;
197                 ret++;
198                 if (c == '\n')
199                         break;
200         }
201         return ret;
202 }
203
204 /* High bit set, or ISO-2022-INT */
205 static int non_ascii(int ch)
206 {
207         return !isascii(ch) || ch == '\033';
208 }
209
210 int has_non_ascii(const char *s)
211 {
212         int ch;
213         if (!s)
214                 return 0;
215         while ((ch = *s++) != '\0') {
216                 if (non_ascii(ch))
217                         return 1;
218         }
219         return 0;
220 }
221
222 static int is_rfc822_special(char ch)
223 {
224         switch (ch) {
225         case '(':
226         case ')':
227         case '<':
228         case '>':
229         case '[':
230         case ']':
231         case ':':
232         case ';':
233         case '@':
234         case ',':
235         case '.':
236         case '"':
237         case '\\':
238                 return 1;
239         default:
240                 return 0;
241         }
242 }
243
244 static int needs_rfc822_quoting(const char *s, int len)
245 {
246         int i;
247         for (i = 0; i < len; i++)
248                 if (is_rfc822_special(s[i]))
249                         return 1;
250         return 0;
251 }
252
253 static int last_line_length(struct strbuf *sb)
254 {
255         int i;
256
257         /* How many bytes are already used on the last line? */
258         for (i = sb->len - 1; i >= 0; i--)
259                 if (sb->buf[i] == '\n')
260                         break;
261         return sb->len - (i + 1);
262 }
263
264 static void add_rfc822_quoted(struct strbuf *out, const char *s, int len)
265 {
266         int i;
267
268         /* just a guess, we may have to also backslash-quote */
269         strbuf_grow(out, len + 2);
270
271         strbuf_addch(out, '"');
272         for (i = 0; i < len; i++) {
273                 switch (s[i]) {
274                 case '"':
275                 case '\\':
276                         strbuf_addch(out, '\\');
277                         /* fall through */
278                 default:
279                         strbuf_addch(out, s[i]);
280                 }
281         }
282         strbuf_addch(out, '"');
283 }
284
285 enum rfc2047_type {
286         RFC2047_SUBJECT,
287         RFC2047_ADDRESS
288 };
289
290 static int is_rfc2047_special(char ch, enum rfc2047_type type)
291 {
292         /*
293          * rfc2047, section 4.2:
294          *
295          *    8-bit values which correspond to printable ASCII characters other
296          *    than "=", "?", and "_" (underscore), MAY be represented as those
297          *    characters.  (But see section 5 for restrictions.)  In
298          *    particular, SPACE and TAB MUST NOT be represented as themselves
299          *    within encoded words.
300          */
301
302         /*
303          * rule out non-ASCII characters and non-printable characters (the
304          * non-ASCII check should be redundant as isprint() is not localized
305          * and only knows about ASCII, but be defensive about that)
306          */
307         if (non_ascii(ch) || !isprint(ch))
308                 return 1;
309
310         /*
311          * rule out special printable characters (' ' should be the only
312          * whitespace character considered printable, but be defensive and use
313          * isspace())
314          */
315         if (isspace(ch) || ch == '=' || ch == '?' || ch == '_')
316                 return 1;
317
318         /*
319          * rfc2047, section 5.3:
320          *
321          *    As a replacement for a 'word' entity within a 'phrase', for example,
322          *    one that precedes an address in a From, To, or Cc header.  The ABNF
323          *    definition for 'phrase' from RFC 822 thus becomes:
324          *
325          *    phrase = 1*( encoded-word / word )
326          *
327          *    In this case the set of characters that may be used in a "Q"-encoded
328          *    'encoded-word' is restricted to: <upper and lower case ASCII
329          *    letters, decimal digits, "!", "*", "+", "-", "/", "=", and "_"
330          *    (underscore, ASCII 95.)>.  An 'encoded-word' that appears within a
331          *    'phrase' MUST be separated from any adjacent 'word', 'text' or
332          *    'special' by 'linear-white-space'.
333          */
334
335         if (type != RFC2047_ADDRESS)
336                 return 0;
337
338         /* '=' and '_' are special cases and have been checked above */
339         return !(isalnum(ch) || ch == '!' || ch == '*' || ch == '+' || ch == '-' || ch == '/');
340 }
341
342 static int needs_rfc2047_encoding(const char *line, int len,
343                                   enum rfc2047_type type)
344 {
345         int i;
346
347         for (i = 0; i < len; i++) {
348                 int ch = line[i];
349                 if (non_ascii(ch) || ch == '\n')
350                         return 1;
351                 if ((i + 1 < len) && (ch == '=' && line[i+1] == '?'))
352                         return 1;
353         }
354
355         return 0;
356 }
357
358 static void add_rfc2047(struct strbuf *sb, const char *line, size_t len,
359                        const char *encoding, enum rfc2047_type type)
360 {
361         static const int max_encoded_length = 76; /* per rfc2047 */
362         int i;
363         int line_len = last_line_length(sb);
364
365         strbuf_grow(sb, len * 3 + strlen(encoding) + 100);
366         strbuf_addf(sb, "=?%s?q?", encoding);
367         line_len += strlen(encoding) + 5; /* 5 for =??q? */
368
369         while (len) {
370                 /*
371                  * RFC 2047, section 5 (3):
372                  *
373                  * Each 'encoded-word' MUST represent an integral number of
374                  * characters.  A multi-octet character may not be split across
375                  * adjacent 'encoded- word's.
376                  */
377                 const unsigned char *p = (const unsigned char *)line;
378                 int chrlen = mbs_chrlen(&line, &len, encoding);
379                 int is_special = (chrlen > 1) || is_rfc2047_special(*p, type);
380
381                 /* "=%02X" * chrlen, or the byte itself */
382                 const char *encoded_fmt = is_special ? "=%02X"    : "%c";
383                 int         encoded_len = is_special ? 3 * chrlen : 1;
384
385                 /*
386                  * According to RFC 2047, we could encode the special character
387                  * ' ' (space) with '_' (underscore) for readability. But many
388                  * programs do not understand this and just leave the
389                  * underscore in place. Thus, we do nothing special here, which
390                  * causes ' ' to be encoded as '=20', avoiding this problem.
391                  */
392
393                 if (line_len + encoded_len + 2 > max_encoded_length) {
394                         /* It won't fit with trailing "?=" --- break the line */
395                         strbuf_addf(sb, "?=\n =?%s?q?", encoding);
396                         line_len = strlen(encoding) + 5 + 1; /* =??q? plus SP */
397                 }
398
399                 for (i = 0; i < chrlen; i++)
400                         strbuf_addf(sb, encoded_fmt, p[i]);
401                 line_len += encoded_len;
402         }
403         strbuf_addstr(sb, "?=");
404 }
405
406 const char *show_ident_date(const struct ident_split *ident,
407                             const struct date_mode *mode)
408 {
409         timestamp_t date = 0;
410         long tz = 0;
411
412         if (ident->date_begin && ident->date_end)
413                 date = parse_timestamp(ident->date_begin, NULL, 10);
414         if (date_overflows(date))
415                 date = 0;
416         else {
417                 if (ident->tz_begin && ident->tz_end)
418                         tz = strtol(ident->tz_begin, NULL, 10);
419                 if (tz >= INT_MAX || tz <= INT_MIN)
420                         tz = 0;
421         }
422         return show_date(date, tz, mode);
423 }
424
425 void pp_user_info(struct pretty_print_context *pp,
426                   const char *what, struct strbuf *sb,
427                   const char *line, const char *encoding)
428 {
429         struct ident_split ident;
430         char *line_end;
431         const char *mailbuf, *namebuf;
432         size_t namelen, maillen;
433         int max_length = 78; /* per rfc2822 */
434
435         if (pp->fmt == CMIT_FMT_ONELINE)
436                 return;
437
438         line_end = strchrnul(line, '\n');
439         if (split_ident_line(&ident, line, line_end - line))
440                 return;
441
442         mailbuf = ident.mail_begin;
443         maillen = ident.mail_end - ident.mail_begin;
444         namebuf = ident.name_begin;
445         namelen = ident.name_end - ident.name_begin;
446
447         if (pp->mailmap)
448                 map_user(pp->mailmap, &mailbuf, &maillen, &namebuf, &namelen);
449
450         if (cmit_fmt_is_mail(pp->fmt)) {
451                 if (pp->from_ident && ident_cmp(pp->from_ident, &ident)) {
452                         struct strbuf buf = STRBUF_INIT;
453
454                         strbuf_addstr(&buf, "From: ");
455                         strbuf_add(&buf, namebuf, namelen);
456                         strbuf_addstr(&buf, " <");
457                         strbuf_add(&buf, mailbuf, maillen);
458                         strbuf_addstr(&buf, ">\n");
459                         string_list_append(&pp->in_body_headers,
460                                            strbuf_detach(&buf, NULL));
461
462                         mailbuf = pp->from_ident->mail_begin;
463                         maillen = pp->from_ident->mail_end - mailbuf;
464                         namebuf = pp->from_ident->name_begin;
465                         namelen = pp->from_ident->name_end - namebuf;
466                 }
467
468                 strbuf_addstr(sb, "From: ");
469                 if (needs_rfc2047_encoding(namebuf, namelen, RFC2047_ADDRESS)) {
470                         add_rfc2047(sb, namebuf, namelen,
471                                     encoding, RFC2047_ADDRESS);
472                         max_length = 76; /* per rfc2047 */
473                 } else if (needs_rfc822_quoting(namebuf, namelen)) {
474                         struct strbuf quoted = STRBUF_INIT;
475                         add_rfc822_quoted(&quoted, namebuf, namelen);
476                         strbuf_add_wrapped_bytes(sb, quoted.buf, quoted.len,
477                                                         -6, 1, max_length);
478                         strbuf_release(&quoted);
479                 } else {
480                         strbuf_add_wrapped_bytes(sb, namebuf, namelen,
481                                                  -6, 1, max_length);
482                 }
483
484                 if (max_length <
485                     last_line_length(sb) + strlen(" <") + maillen + strlen(">"))
486                         strbuf_addch(sb, '\n');
487                 strbuf_addf(sb, " <%.*s>\n", (int)maillen, mailbuf);
488         } else {
489                 strbuf_addf(sb, "%s: %.*s%.*s <%.*s>\n", what,
490                             (pp->fmt == CMIT_FMT_FULLER) ? 4 : 0, "    ",
491                             (int)namelen, namebuf, (int)maillen, mailbuf);
492         }
493
494         switch (pp->fmt) {
495         case CMIT_FMT_MEDIUM:
496                 strbuf_addf(sb, "Date:   %s\n",
497                             show_ident_date(&ident, &pp->date_mode));
498                 break;
499         case CMIT_FMT_EMAIL:
500         case CMIT_FMT_MBOXRD:
501                 strbuf_addf(sb, "Date: %s\n",
502                             show_ident_date(&ident, DATE_MODE(RFC2822)));
503                 break;
504         case CMIT_FMT_FULLER:
505                 strbuf_addf(sb, "%sDate: %s\n", what,
506                             show_ident_date(&ident, &pp->date_mode));
507                 break;
508         default:
509                 /* notin' */
510                 break;
511         }
512 }
513
514 static int is_blank_line(const char *line, int *len_p)
515 {
516         int len = *len_p;
517         while (len && isspace(line[len - 1]))
518                 len--;
519         *len_p = len;
520         return !len;
521 }
522
523 const char *skip_blank_lines(const char *msg)
524 {
525         for (;;) {
526                 int linelen = get_one_line(msg);
527                 int ll = linelen;
528                 if (!linelen)
529                         break;
530                 if (!is_blank_line(msg, &ll))
531                         break;
532                 msg += linelen;
533         }
534         return msg;
535 }
536
537 static void add_merge_info(const struct pretty_print_context *pp,
538                            struct strbuf *sb, const struct commit *commit)
539 {
540         struct commit_list *parent = commit->parents;
541
542         if ((pp->fmt == CMIT_FMT_ONELINE) || (cmit_fmt_is_mail(pp->fmt)) ||
543             !parent || !parent->next)
544                 return;
545
546         strbuf_addstr(sb, "Merge:");
547
548         while (parent) {
549                 struct object_id *oidp = &parent->item->object.oid;
550                 strbuf_addch(sb, ' ');
551                 if (pp->abbrev)
552                         strbuf_add_unique_abbrev(sb, oidp->hash, pp->abbrev);
553                 else
554                         strbuf_addstr(sb, oid_to_hex(oidp));
555                 parent = parent->next;
556         }
557         strbuf_addch(sb, '\n');
558 }
559
560 static char *get_header(const char *msg, const char *key)
561 {
562         size_t len;
563         const char *v = find_commit_header(msg, key, &len);
564         return v ? xmemdupz(v, len) : NULL;
565 }
566
567 static char *replace_encoding_header(char *buf, const char *encoding)
568 {
569         struct strbuf tmp = STRBUF_INIT;
570         size_t start, len;
571         char *cp = buf;
572
573         /* guess if there is an encoding header before a \n\n */
574         while (!starts_with(cp, "encoding ")) {
575                 cp = strchr(cp, '\n');
576                 if (!cp || *++cp == '\n')
577                         return buf;
578         }
579         start = cp - buf;
580         cp = strchr(cp, '\n');
581         if (!cp)
582                 return buf; /* should not happen but be defensive */
583         len = cp + 1 - (buf + start);
584
585         strbuf_attach(&tmp, buf, strlen(buf), strlen(buf) + 1);
586         if (is_encoding_utf8(encoding)) {
587                 /* we have re-coded to UTF-8; drop the header */
588                 strbuf_remove(&tmp, start, len);
589         } else {
590                 /* just replaces XXXX in 'encoding XXXX\n' */
591                 strbuf_splice(&tmp, start + strlen("encoding "),
592                                           len - strlen("encoding \n"),
593                                           encoding, strlen(encoding));
594         }
595         return strbuf_detach(&tmp, NULL);
596 }
597
598 const char *logmsg_reencode(const struct commit *commit,
599                             char **commit_encoding,
600                             const char *output_encoding)
601 {
602         static const char *utf8 = "UTF-8";
603         const char *use_encoding;
604         char *encoding;
605         const char *msg = get_commit_buffer(commit, NULL);
606         char *out;
607
608         if (!output_encoding || !*output_encoding) {
609                 if (commit_encoding)
610                         *commit_encoding = get_header(msg, "encoding");
611                 return msg;
612         }
613         encoding = get_header(msg, "encoding");
614         if (commit_encoding)
615                 *commit_encoding = encoding;
616         use_encoding = encoding ? encoding : utf8;
617         if (same_encoding(use_encoding, output_encoding)) {
618                 /*
619                  * No encoding work to be done. If we have no encoding header
620                  * at all, then there's nothing to do, and we can return the
621                  * message verbatim (whether newly allocated or not).
622                  */
623                 if (!encoding)
624                         return msg;
625
626                 /*
627                  * Otherwise, we still want to munge the encoding header in the
628                  * result, which will be done by modifying the buffer. If we
629                  * are using a fresh copy, we can reuse it. But if we are using
630                  * the cached copy from get_commit_buffer, we need to duplicate it
631                  * to avoid munging the cached copy.
632                  */
633                 if (msg == get_cached_commit_buffer(commit, NULL))
634                         out = xstrdup(msg);
635                 else
636                         out = (char *)msg;
637         }
638         else {
639                 /*
640                  * There's actual encoding work to do. Do the reencoding, which
641                  * still leaves the header to be replaced in the next step. At
642                  * this point, we are done with msg. If we allocated a fresh
643                  * copy, we can free it.
644                  */
645                 out = reencode_string(msg, output_encoding, use_encoding);
646                 if (out)
647                         unuse_commit_buffer(commit, msg);
648         }
649
650         /*
651          * This replacement actually consumes the buffer we hand it, so we do
652          * not have to worry about freeing the old "out" here.
653          */
654         if (out)
655                 out = replace_encoding_header(out, output_encoding);
656
657         if (!commit_encoding)
658                 free(encoding);
659         /*
660          * If the re-encoding failed, out might be NULL here; in that
661          * case we just return the commit message verbatim.
662          */
663         return out ? out : msg;
664 }
665
666 static int mailmap_name(const char **email, size_t *email_len,
667                         const char **name, size_t *name_len)
668 {
669         static struct string_list *mail_map;
670         if (!mail_map) {
671                 mail_map = xcalloc(1, sizeof(*mail_map));
672                 read_mailmap(mail_map, NULL);
673         }
674         return mail_map->nr && map_user(mail_map, email, email_len, name, name_len);
675 }
676
677 static size_t format_person_part(struct strbuf *sb, char part,
678                                  const char *msg, int len,
679                                  const struct date_mode *dmode)
680 {
681         /* currently all placeholders have same length */
682         const int placeholder_len = 2;
683         struct ident_split s;
684         const char *name, *mail;
685         size_t maillen, namelen;
686
687         if (split_ident_line(&s, msg, len) < 0)
688                 goto skip;
689
690         name = s.name_begin;
691         namelen = s.name_end - s.name_begin;
692         mail = s.mail_begin;
693         maillen = s.mail_end - s.mail_begin;
694
695         if (part == 'N' || part == 'E') /* mailmap lookup */
696                 mailmap_name(&mail, &maillen, &name, &namelen);
697         if (part == 'n' || part == 'N') {       /* name */
698                 strbuf_add(sb, name, namelen);
699                 return placeholder_len;
700         }
701         if (part == 'e' || part == 'E') {       /* email */
702                 strbuf_add(sb, mail, maillen);
703                 return placeholder_len;
704         }
705
706         if (!s.date_begin)
707                 goto skip;
708
709         if (part == 't') {      /* date, UNIX timestamp */
710                 strbuf_add(sb, s.date_begin, s.date_end - s.date_begin);
711                 return placeholder_len;
712         }
713
714         switch (part) {
715         case 'd':       /* date */
716                 strbuf_addstr(sb, show_ident_date(&s, dmode));
717                 return placeholder_len;
718         case 'D':       /* date, RFC2822 style */
719                 strbuf_addstr(sb, show_ident_date(&s, DATE_MODE(RFC2822)));
720                 return placeholder_len;
721         case 'r':       /* date, relative */
722                 strbuf_addstr(sb, show_ident_date(&s, DATE_MODE(RELATIVE)));
723                 return placeholder_len;
724         case 'i':       /* date, ISO 8601-like */
725                 strbuf_addstr(sb, show_ident_date(&s, DATE_MODE(ISO8601)));
726                 return placeholder_len;
727         case 'I':       /* date, ISO 8601 strict */
728                 strbuf_addstr(sb, show_ident_date(&s, DATE_MODE(ISO8601_STRICT)));
729                 return placeholder_len;
730         }
731
732 skip:
733         /*
734          * reading from either a bogus commit, or a reflog entry with
735          * %gn, %ge, etc.; 'sb' cannot be updated, but we still need
736          * to compute a valid return value.
737          */
738         if (part == 'n' || part == 'e' || part == 't' || part == 'd'
739             || part == 'D' || part == 'r' || part == 'i')
740                 return placeholder_len;
741
742         return 0; /* unknown placeholder */
743 }
744
745 struct chunk {
746         size_t off;
747         size_t len;
748 };
749
750 enum flush_type {
751         no_flush,
752         flush_right,
753         flush_left,
754         flush_left_and_steal,
755         flush_both
756 };
757
758 enum trunc_type {
759         trunc_none,
760         trunc_left,
761         trunc_middle,
762         trunc_right
763 };
764
765 struct format_commit_context {
766         const struct commit *commit;
767         const struct pretty_print_context *pretty_ctx;
768         unsigned commit_header_parsed:1;
769         unsigned commit_message_parsed:1;
770         struct signature_check signature_check;
771         enum flush_type flush_type;
772         enum trunc_type truncate;
773         const char *message;
774         char *commit_encoding;
775         size_t width, indent1, indent2;
776         int auto_color;
777         int padding;
778
779         /* These offsets are relative to the start of the commit message. */
780         struct chunk author;
781         struct chunk committer;
782         size_t message_off;
783         size_t subject_off;
784         size_t body_off;
785
786         /* The following ones are relative to the result struct strbuf. */
787         size_t wrap_start;
788 };
789
790 static void parse_commit_header(struct format_commit_context *context)
791 {
792         const char *msg = context->message;
793         int i;
794
795         for (i = 0; msg[i]; i++) {
796                 const char *name;
797                 int eol;
798                 for (eol = i; msg[eol] && msg[eol] != '\n'; eol++)
799                         ; /* do nothing */
800
801                 if (i == eol) {
802                         break;
803                 } else if (skip_prefix(msg + i, "author ", &name)) {
804                         context->author.off = name - msg;
805                         context->author.len = msg + eol - name;
806                 } else if (skip_prefix(msg + i, "committer ", &name)) {
807                         context->committer.off = name - msg;
808                         context->committer.len = msg + eol - name;
809                 }
810                 i = eol;
811         }
812         context->message_off = i;
813         context->commit_header_parsed = 1;
814 }
815
816 static int istitlechar(char c)
817 {
818         return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') ||
819                 (c >= '0' && c <= '9') || c == '.' || c == '_';
820 }
821
822 static void format_sanitized_subject(struct strbuf *sb, const char *msg)
823 {
824         size_t trimlen;
825         size_t start_len = sb->len;
826         int space = 2;
827
828         for (; *msg && *msg != '\n'; msg++) {
829                 if (istitlechar(*msg)) {
830                         if (space == 1)
831                                 strbuf_addch(sb, '-');
832                         space = 0;
833                         strbuf_addch(sb, *msg);
834                         if (*msg == '.')
835                                 while (*(msg+1) == '.')
836                                         msg++;
837                 } else
838                         space |= 1;
839         }
840
841         /* trim any trailing '.' or '-' characters */
842         trimlen = 0;
843         while (sb->len - trimlen > start_len &&
844                 (sb->buf[sb->len - 1 - trimlen] == '.'
845                 || sb->buf[sb->len - 1 - trimlen] == '-'))
846                 trimlen++;
847         strbuf_remove(sb, sb->len - trimlen, trimlen);
848 }
849
850 const char *format_subject(struct strbuf *sb, const char *msg,
851                            const char *line_separator)
852 {
853         int first = 1;
854
855         for (;;) {
856                 const char *line = msg;
857                 int linelen = get_one_line(line);
858
859                 msg += linelen;
860                 if (!linelen || is_blank_line(line, &linelen))
861                         break;
862
863                 if (!sb)
864                         continue;
865                 strbuf_grow(sb, linelen + 2);
866                 if (!first)
867                         strbuf_addstr(sb, line_separator);
868                 strbuf_add(sb, line, linelen);
869                 first = 0;
870         }
871         return msg;
872 }
873
874 static void format_trailers(struct strbuf *sb, const char *msg)
875 {
876         struct trailer_info info;
877
878         trailer_info_get(&info, msg);
879         strbuf_add(sb, info.trailer_start,
880                    info.trailer_end - info.trailer_start);
881         trailer_info_release(&info);
882 }
883
884 static void parse_commit_message(struct format_commit_context *c)
885 {
886         const char *msg = c->message + c->message_off;
887         const char *start = c->message;
888
889         msg = skip_blank_lines(msg);
890         c->subject_off = msg - start;
891
892         msg = format_subject(NULL, msg, NULL);
893         msg = skip_blank_lines(msg);
894         c->body_off = msg - start;
895
896         c->commit_message_parsed = 1;
897 }
898
899 static void strbuf_wrap(struct strbuf *sb, size_t pos,
900                         size_t width, size_t indent1, size_t indent2)
901 {
902         struct strbuf tmp = STRBUF_INIT;
903
904         if (pos)
905                 strbuf_add(&tmp, sb->buf, pos);
906         strbuf_add_wrapped_text(&tmp, sb->buf + pos,
907                                 (int) indent1, (int) indent2, (int) width);
908         strbuf_swap(&tmp, sb);
909         strbuf_release(&tmp);
910 }
911
912 static void rewrap_message_tail(struct strbuf *sb,
913                                 struct format_commit_context *c,
914                                 size_t new_width, size_t new_indent1,
915                                 size_t new_indent2)
916 {
917         if (c->width == new_width && c->indent1 == new_indent1 &&
918             c->indent2 == new_indent2)
919                 return;
920         if (c->wrap_start < sb->len)
921                 strbuf_wrap(sb, c->wrap_start, c->width, c->indent1, c->indent2);
922         c->wrap_start = sb->len;
923         c->width = new_width;
924         c->indent1 = new_indent1;
925         c->indent2 = new_indent2;
926 }
927
928 static int format_reflog_person(struct strbuf *sb,
929                                 char part,
930                                 struct reflog_walk_info *log,
931                                 const struct date_mode *dmode)
932 {
933         const char *ident;
934
935         if (!log)
936                 return 2;
937
938         ident = get_reflog_ident(log);
939         if (!ident)
940                 return 2;
941
942         return format_person_part(sb, part, ident, strlen(ident), dmode);
943 }
944
945 static size_t parse_color(struct strbuf *sb, /* in UTF-8 */
946                           const char *placeholder,
947                           struct format_commit_context *c)
948 {
949         const char *rest = placeholder;
950
951         if (placeholder[1] == '(') {
952                 const char *begin = placeholder + 2;
953                 const char *end = strchr(begin, ')');
954                 char color[COLOR_MAXLEN];
955
956                 if (!end)
957                         return 0;
958                 if (skip_prefix(begin, "auto,", &begin)) {
959                         if (!want_color(c->pretty_ctx->color))
960                                 return end - placeholder + 1;
961                 }
962                 if (color_parse_mem(begin, end - begin, color) < 0)
963                         die(_("unable to parse --pretty format"));
964                 strbuf_addstr(sb, color);
965                 return end - placeholder + 1;
966         }
967         if (skip_prefix(placeholder + 1, "red", &rest))
968                 strbuf_addstr(sb, GIT_COLOR_RED);
969         else if (skip_prefix(placeholder + 1, "green", &rest))
970                 strbuf_addstr(sb, GIT_COLOR_GREEN);
971         else if (skip_prefix(placeholder + 1, "blue", &rest))
972                 strbuf_addstr(sb, GIT_COLOR_BLUE);
973         else if (skip_prefix(placeholder + 1, "reset", &rest))
974                 strbuf_addstr(sb, GIT_COLOR_RESET);
975         return rest - placeholder;
976 }
977
978 static size_t parse_padding_placeholder(struct strbuf *sb,
979                                         const char *placeholder,
980                                         struct format_commit_context *c)
981 {
982         const char *ch = placeholder;
983         enum flush_type flush_type;
984         int to_column = 0;
985
986         switch (*ch++) {
987         case '<':
988                 flush_type = flush_right;
989                 break;
990         case '>':
991                 if (*ch == '<') {
992                         flush_type = flush_both;
993                         ch++;
994                 } else if (*ch == '>') {
995                         flush_type = flush_left_and_steal;
996                         ch++;
997                 } else
998                         flush_type = flush_left;
999                 break;
1000         default:
1001                 return 0;
1002         }
1003
1004         /* the next value means "wide enough to that column" */
1005         if (*ch == '|') {
1006                 to_column = 1;
1007                 ch++;
1008         }
1009
1010         if (*ch == '(') {
1011                 const char *start = ch + 1;
1012                 const char *end = start + strcspn(start, ",)");
1013                 char *next;
1014                 int width;
1015                 if (!end || end == start)
1016                         return 0;
1017                 width = strtol(start, &next, 10);
1018                 if (next == start || width == 0)
1019                         return 0;
1020                 if (width < 0) {
1021                         if (to_column)
1022                                 width += term_columns();
1023                         if (width < 0)
1024                                 return 0;
1025                 }
1026                 c->padding = to_column ? -width : width;
1027                 c->flush_type = flush_type;
1028
1029                 if (*end == ',') {
1030                         start = end + 1;
1031                         end = strchr(start, ')');
1032                         if (!end || end == start)
1033                                 return 0;
1034                         if (starts_with(start, "trunc)"))
1035                                 c->truncate = trunc_right;
1036                         else if (starts_with(start, "ltrunc)"))
1037                                 c->truncate = trunc_left;
1038                         else if (starts_with(start, "mtrunc)"))
1039                                 c->truncate = trunc_middle;
1040                         else
1041                                 return 0;
1042                 } else
1043                         c->truncate = trunc_none;
1044
1045                 return end - placeholder + 1;
1046         }
1047         return 0;
1048 }
1049
1050 static size_t format_commit_one(struct strbuf *sb, /* in UTF-8 */
1051                                 const char *placeholder,
1052                                 void *context)
1053 {
1054         struct format_commit_context *c = context;
1055         const struct commit *commit = c->commit;
1056         const char *msg = c->message;
1057         struct commit_list *p;
1058         int ch;
1059
1060         /* these are independent of the commit */
1061         switch (placeholder[0]) {
1062         case 'C':
1063                 if (starts_with(placeholder + 1, "(auto)")) {
1064                         c->auto_color = want_color(c->pretty_ctx->color);
1065                         if (c->auto_color && sb->len)
1066                                 strbuf_addstr(sb, GIT_COLOR_RESET);
1067                         return 7; /* consumed 7 bytes, "C(auto)" */
1068                 } else {
1069                         int ret = parse_color(sb, placeholder, c);
1070                         if (ret)
1071                                 c->auto_color = 0;
1072                         /*
1073                          * Otherwise, we decided to treat %C<unknown>
1074                          * as a literal string, and the previous
1075                          * %C(auto) is still valid.
1076                          */
1077                         return ret;
1078                 }
1079         case 'n':               /* newline */
1080                 strbuf_addch(sb, '\n');
1081                 return 1;
1082         case 'x':
1083                 /* %x00 == NUL, %x0a == LF, etc. */
1084                 ch = hex2chr(placeholder + 1);
1085                 if (ch < 0)
1086                         return 0;
1087                 strbuf_addch(sb, ch);
1088                 return 3;
1089         case 'w':
1090                 if (placeholder[1] == '(') {
1091                         unsigned long width = 0, indent1 = 0, indent2 = 0;
1092                         char *next;
1093                         const char *start = placeholder + 2;
1094                         const char *end = strchr(start, ')');
1095                         if (!end)
1096                                 return 0;
1097                         if (end > start) {
1098                                 width = strtoul(start, &next, 10);
1099                                 if (*next == ',') {
1100                                         indent1 = strtoul(next + 1, &next, 10);
1101                                         if (*next == ',') {
1102                                                 indent2 = strtoul(next + 1,
1103                                                                  &next, 10);
1104                                         }
1105                                 }
1106                                 if (*next != ')')
1107                                         return 0;
1108                         }
1109                         rewrap_message_tail(sb, c, width, indent1, indent2);
1110                         return end - placeholder + 1;
1111                 } else
1112                         return 0;
1113
1114         case '<':
1115         case '>':
1116                 return parse_padding_placeholder(sb, placeholder, c);
1117         }
1118
1119         /* these depend on the commit */
1120         if (!commit->object.parsed)
1121                 parse_object(&commit->object.oid);
1122
1123         switch (placeholder[0]) {
1124         case 'H':               /* commit hash */
1125                 strbuf_addstr(sb, diff_get_color(c->auto_color, DIFF_COMMIT));
1126                 strbuf_addstr(sb, oid_to_hex(&commit->object.oid));
1127                 strbuf_addstr(sb, diff_get_color(c->auto_color, DIFF_RESET));
1128                 return 1;
1129         case 'h':               /* abbreviated commit hash */
1130                 strbuf_addstr(sb, diff_get_color(c->auto_color, DIFF_COMMIT));
1131                 strbuf_add_unique_abbrev(sb, commit->object.oid.hash,
1132                                          c->pretty_ctx->abbrev);
1133                 strbuf_addstr(sb, diff_get_color(c->auto_color, DIFF_RESET));
1134                 return 1;
1135         case 'T':               /* tree hash */
1136                 strbuf_addstr(sb, oid_to_hex(&commit->tree->object.oid));
1137                 return 1;
1138         case 't':               /* abbreviated tree hash */
1139                 strbuf_add_unique_abbrev(sb, commit->tree->object.oid.hash,
1140                                          c->pretty_ctx->abbrev);
1141                 return 1;
1142         case 'P':               /* parent hashes */
1143                 for (p = commit->parents; p; p = p->next) {
1144                         if (p != commit->parents)
1145                                 strbuf_addch(sb, ' ');
1146                         strbuf_addstr(sb, oid_to_hex(&p->item->object.oid));
1147                 }
1148                 return 1;
1149         case 'p':               /* abbreviated parent hashes */
1150                 for (p = commit->parents; p; p = p->next) {
1151                         if (p != commit->parents)
1152                                 strbuf_addch(sb, ' ');
1153                         strbuf_add_unique_abbrev(sb, p->item->object.oid.hash,
1154                                                  c->pretty_ctx->abbrev);
1155                 }
1156                 return 1;
1157         case 'm':               /* left/right/bottom */
1158                 strbuf_addstr(sb, get_revision_mark(NULL, commit));
1159                 return 1;
1160         case 'd':
1161                 load_ref_decorations(DECORATE_SHORT_REFS);
1162                 format_decorations(sb, commit, c->auto_color);
1163                 return 1;
1164         case 'D':
1165                 load_ref_decorations(DECORATE_SHORT_REFS);
1166                 format_decorations_extended(sb, commit, c->auto_color, "", ", ", "");
1167                 return 1;
1168         case 'g':               /* reflog info */
1169                 switch(placeholder[1]) {
1170                 case 'd':       /* reflog selector */
1171                 case 'D':
1172                         if (c->pretty_ctx->reflog_info)
1173                                 get_reflog_selector(sb,
1174                                                     c->pretty_ctx->reflog_info,
1175                                                     &c->pretty_ctx->date_mode,
1176                                                     c->pretty_ctx->date_mode_explicit,
1177                                                     (placeholder[1] == 'd'));
1178                         return 2;
1179                 case 's':       /* reflog message */
1180                         if (c->pretty_ctx->reflog_info)
1181                                 get_reflog_message(sb, c->pretty_ctx->reflog_info);
1182                         return 2;
1183                 case 'n':
1184                 case 'N':
1185                 case 'e':
1186                 case 'E':
1187                         return format_reflog_person(sb,
1188                                                     placeholder[1],
1189                                                     c->pretty_ctx->reflog_info,
1190                                                     &c->pretty_ctx->date_mode);
1191                 }
1192                 return 0;       /* unknown %g placeholder */
1193         case 'N':
1194                 if (c->pretty_ctx->notes_message) {
1195                         strbuf_addstr(sb, c->pretty_ctx->notes_message);
1196                         return 1;
1197                 }
1198                 return 0;
1199         }
1200
1201         if (placeholder[0] == 'G') {
1202                 if (!c->signature_check.result)
1203                         check_commit_signature(c->commit, &(c->signature_check));
1204                 switch (placeholder[1]) {
1205                 case 'G':
1206                         if (c->signature_check.gpg_output)
1207                                 strbuf_addstr(sb, c->signature_check.gpg_output);
1208                         break;
1209                 case '?':
1210                         switch (c->signature_check.result) {
1211                         case 'G':
1212                         case 'B':
1213                         case 'E':
1214                         case 'U':
1215                         case 'N':
1216                         case 'X':
1217                         case 'Y':
1218                         case 'R':
1219                                 strbuf_addch(sb, c->signature_check.result);
1220                         }
1221                         break;
1222                 case 'S':
1223                         if (c->signature_check.signer)
1224                                 strbuf_addstr(sb, c->signature_check.signer);
1225                         break;
1226                 case 'K':
1227                         if (c->signature_check.key)
1228                                 strbuf_addstr(sb, c->signature_check.key);
1229                         break;
1230                 default:
1231                         return 0;
1232                 }
1233                 return 2;
1234         }
1235
1236
1237         /* For the rest we have to parse the commit header. */
1238         if (!c->commit_header_parsed)
1239                 parse_commit_header(c);
1240
1241         switch (placeholder[0]) {
1242         case 'a':       /* author ... */
1243                 return format_person_part(sb, placeholder[1],
1244                                    msg + c->author.off, c->author.len,
1245                                    &c->pretty_ctx->date_mode);
1246         case 'c':       /* committer ... */
1247                 return format_person_part(sb, placeholder[1],
1248                                    msg + c->committer.off, c->committer.len,
1249                                    &c->pretty_ctx->date_mode);
1250         case 'e':       /* encoding */
1251                 if (c->commit_encoding)
1252                         strbuf_addstr(sb, c->commit_encoding);
1253                 return 1;
1254         case 'B':       /* raw body */
1255                 /* message_off is always left at the initial newline */
1256                 strbuf_addstr(sb, msg + c->message_off + 1);
1257                 return 1;
1258         }
1259
1260         /* Now we need to parse the commit message. */
1261         if (!c->commit_message_parsed)
1262                 parse_commit_message(c);
1263
1264         switch (placeholder[0]) {
1265         case 's':       /* subject */
1266                 format_subject(sb, msg + c->subject_off, " ");
1267                 return 1;
1268         case 'f':       /* sanitized subject */
1269                 format_sanitized_subject(sb, msg + c->subject_off);
1270                 return 1;
1271         case 'b':       /* body */
1272                 strbuf_addstr(sb, msg + c->body_off);
1273                 return 1;
1274         }
1275
1276         if (starts_with(placeholder, "(trailers)")) {
1277                 format_trailers(sb, msg + c->subject_off);
1278                 return strlen("(trailers)");
1279         }
1280
1281         return 0;       /* unknown placeholder */
1282 }
1283
1284 static size_t format_and_pad_commit(struct strbuf *sb, /* in UTF-8 */
1285                                     const char *placeholder,
1286                                     struct format_commit_context *c)
1287 {
1288         struct strbuf local_sb = STRBUF_INIT;
1289         int total_consumed = 0, len, padding = c->padding;
1290         if (padding < 0) {
1291                 const char *start = strrchr(sb->buf, '\n');
1292                 int occupied;
1293                 if (!start)
1294                         start = sb->buf;
1295                 occupied = utf8_strnwidth(start, -1, 1);
1296                 occupied += c->pretty_ctx->graph_width;
1297                 padding = (-padding) - occupied;
1298         }
1299         while (1) {
1300                 int modifier = *placeholder == 'C';
1301                 int consumed = format_commit_one(&local_sb, placeholder, c);
1302                 total_consumed += consumed;
1303
1304                 if (!modifier)
1305                         break;
1306
1307                 placeholder += consumed;
1308                 if (*placeholder != '%')
1309                         break;
1310                 placeholder++;
1311                 total_consumed++;
1312         }
1313         len = utf8_strnwidth(local_sb.buf, -1, 1);
1314
1315         if (c->flush_type == flush_left_and_steal) {
1316                 const char *ch = sb->buf + sb->len - 1;
1317                 while (len > padding && ch > sb->buf) {
1318                         const char *p;
1319                         if (*ch == ' ') {
1320                                 ch--;
1321                                 padding++;
1322                                 continue;
1323                         }
1324                         /* check for trailing ansi sequences */
1325                         if (*ch != 'm')
1326                                 break;
1327                         p = ch - 1;
1328                         while (ch - p < 10 && *p != '\033')
1329                                 p--;
1330                         if (*p != '\033' ||
1331                             ch + 1 - p != display_mode_esc_sequence_len(p))
1332                                 break;
1333                         /*
1334                          * got a good ansi sequence, put it back to
1335                          * local_sb as we're cutting sb
1336                          */
1337                         strbuf_insert(&local_sb, 0, p, ch + 1 - p);
1338                         ch = p - 1;
1339                 }
1340                 strbuf_setlen(sb, ch + 1 - sb->buf);
1341                 c->flush_type = flush_left;
1342         }
1343
1344         if (len > padding) {
1345                 switch (c->truncate) {
1346                 case trunc_left:
1347                         strbuf_utf8_replace(&local_sb,
1348                                             0, len - (padding - 2),
1349                                             "..");
1350                         break;
1351                 case trunc_middle:
1352                         strbuf_utf8_replace(&local_sb,
1353                                             padding / 2 - 1,
1354                                             len - (padding - 2),
1355                                             "..");
1356                         break;
1357                 case trunc_right:
1358                         strbuf_utf8_replace(&local_sb,
1359                                             padding - 2, len - (padding - 2),
1360                                             "..");
1361                         break;
1362                 case trunc_none:
1363                         break;
1364                 }
1365                 strbuf_addbuf(sb, &local_sb);
1366         } else {
1367                 int sb_len = sb->len, offset = 0;
1368                 if (c->flush_type == flush_left)
1369                         offset = padding - len;
1370                 else if (c->flush_type == flush_both)
1371                         offset = (padding - len) / 2;
1372                 /*
1373                  * we calculate padding in columns, now
1374                  * convert it back to chars
1375                  */
1376                 padding = padding - len + local_sb.len;
1377                 strbuf_addchars(sb, ' ', padding);
1378                 memcpy(sb->buf + sb_len + offset, local_sb.buf,
1379                        local_sb.len);
1380         }
1381         strbuf_release(&local_sb);
1382         c->flush_type = no_flush;
1383         return total_consumed;
1384 }
1385
1386 static size_t format_commit_item(struct strbuf *sb, /* in UTF-8 */
1387                                  const char *placeholder,
1388                                  void *context)
1389 {
1390         int consumed;
1391         size_t orig_len;
1392         enum {
1393                 NO_MAGIC,
1394                 ADD_LF_BEFORE_NON_EMPTY,
1395                 DEL_LF_BEFORE_EMPTY,
1396                 ADD_SP_BEFORE_NON_EMPTY
1397         } magic = NO_MAGIC;
1398
1399         switch (placeholder[0]) {
1400         case '-':
1401                 magic = DEL_LF_BEFORE_EMPTY;
1402                 break;
1403         case '+':
1404                 magic = ADD_LF_BEFORE_NON_EMPTY;
1405                 break;
1406         case ' ':
1407                 magic = ADD_SP_BEFORE_NON_EMPTY;
1408                 break;
1409         default:
1410                 break;
1411         }
1412         if (magic != NO_MAGIC)
1413                 placeholder++;
1414
1415         orig_len = sb->len;
1416         if (((struct format_commit_context *)context)->flush_type != no_flush)
1417                 consumed = format_and_pad_commit(sb, placeholder, context);
1418         else
1419                 consumed = format_commit_one(sb, placeholder, context);
1420         if (magic == NO_MAGIC)
1421                 return consumed;
1422
1423         if ((orig_len == sb->len) && magic == DEL_LF_BEFORE_EMPTY) {
1424                 while (sb->len && sb->buf[sb->len - 1] == '\n')
1425                         strbuf_setlen(sb, sb->len - 1);
1426         } else if (orig_len != sb->len) {
1427                 if (magic == ADD_LF_BEFORE_NON_EMPTY)
1428                         strbuf_insert(sb, orig_len, "\n", 1);
1429                 else if (magic == ADD_SP_BEFORE_NON_EMPTY)
1430                         strbuf_insert(sb, orig_len, " ", 1);
1431         }
1432         return consumed + 1;
1433 }
1434
1435 static size_t userformat_want_item(struct strbuf *sb, const char *placeholder,
1436                                    void *context)
1437 {
1438         struct userformat_want *w = context;
1439
1440         if (*placeholder == '+' || *placeholder == '-' || *placeholder == ' ')
1441                 placeholder++;
1442
1443         switch (*placeholder) {
1444         case 'N':
1445                 w->notes = 1;
1446                 break;
1447         }
1448         return 0;
1449 }
1450
1451 void userformat_find_requirements(const char *fmt, struct userformat_want *w)
1452 {
1453         struct strbuf dummy = STRBUF_INIT;
1454
1455         if (!fmt) {
1456                 if (!user_format)
1457                         return;
1458                 fmt = user_format;
1459         }
1460         strbuf_expand(&dummy, fmt, userformat_want_item, w);
1461         strbuf_release(&dummy);
1462 }
1463
1464 void format_commit_message(const struct commit *commit,
1465                            const char *format, struct strbuf *sb,
1466                            const struct pretty_print_context *pretty_ctx)
1467 {
1468         struct format_commit_context context;
1469         const char *output_enc = pretty_ctx->output_encoding;
1470         const char *utf8 = "UTF-8";
1471
1472         memset(&context, 0, sizeof(context));
1473         context.commit = commit;
1474         context.pretty_ctx = pretty_ctx;
1475         context.wrap_start = sb->len;
1476         /*
1477          * convert a commit message to UTF-8 first
1478          * as far as 'format_commit_item' assumes it in UTF-8
1479          */
1480         context.message = logmsg_reencode(commit,
1481                                           &context.commit_encoding,
1482                                           utf8);
1483
1484         strbuf_expand(sb, format, format_commit_item, &context);
1485         rewrap_message_tail(sb, &context, 0, 0, 0);
1486
1487         /* then convert a commit message to an actual output encoding */
1488         if (output_enc) {
1489                 if (same_encoding(utf8, output_enc))
1490                         output_enc = NULL;
1491         } else {
1492                 if (context.commit_encoding &&
1493                     !same_encoding(context.commit_encoding, utf8))
1494                         output_enc = context.commit_encoding;
1495         }
1496
1497         if (output_enc) {
1498                 int outsz;
1499                 char *out = reencode_string_len(sb->buf, sb->len,
1500                                                 output_enc, utf8, &outsz);
1501                 if (out)
1502                         strbuf_attach(sb, out, outsz, outsz + 1);
1503         }
1504
1505         free(context.commit_encoding);
1506         unuse_commit_buffer(commit, context.message);
1507 }
1508
1509 static void pp_header(struct pretty_print_context *pp,
1510                       const char *encoding,
1511                       const struct commit *commit,
1512                       const char **msg_p,
1513                       struct strbuf *sb)
1514 {
1515         int parents_shown = 0;
1516
1517         for (;;) {
1518                 const char *name, *line = *msg_p;
1519                 int linelen = get_one_line(*msg_p);
1520
1521                 if (!linelen)
1522                         return;
1523                 *msg_p += linelen;
1524
1525                 if (linelen == 1)
1526                         /* End of header */
1527                         return;
1528
1529                 if (pp->fmt == CMIT_FMT_RAW) {
1530                         strbuf_add(sb, line, linelen);
1531                         continue;
1532                 }
1533
1534                 if (starts_with(line, "parent ")) {
1535                         if (linelen != 48)
1536                                 die("bad parent line in commit");
1537                         continue;
1538                 }
1539
1540                 if (!parents_shown) {
1541                         unsigned num = commit_list_count(commit->parents);
1542                         /* with enough slop */
1543                         strbuf_grow(sb, num * 50 + 20);
1544                         add_merge_info(pp, sb, commit);
1545                         parents_shown = 1;
1546                 }
1547
1548                 /*
1549                  * MEDIUM == DEFAULT shows only author with dates.
1550                  * FULL shows both authors but not dates.
1551                  * FULLER shows both authors and dates.
1552                  */
1553                 if (skip_prefix(line, "author ", &name)) {
1554                         strbuf_grow(sb, linelen + 80);
1555                         pp_user_info(pp, "Author", sb, name, encoding);
1556                 }
1557                 if (skip_prefix(line, "committer ", &name) &&
1558                     (pp->fmt == CMIT_FMT_FULL || pp->fmt == CMIT_FMT_FULLER)) {
1559                         strbuf_grow(sb, linelen + 80);
1560                         pp_user_info(pp, "Commit", sb, name, encoding);
1561                 }
1562         }
1563 }
1564
1565 void pp_title_line(struct pretty_print_context *pp,
1566                    const char **msg_p,
1567                    struct strbuf *sb,
1568                    const char *encoding,
1569                    int need_8bit_cte)
1570 {
1571         static const int max_length = 78; /* per rfc2047 */
1572         struct strbuf title;
1573
1574         strbuf_init(&title, 80);
1575         *msg_p = format_subject(&title, *msg_p,
1576                                 pp->preserve_subject ? "\n" : " ");
1577
1578         strbuf_grow(sb, title.len + 1024);
1579         if (pp->print_email_subject) {
1580                 if (pp->rev)
1581                         fmt_output_email_subject(sb, pp->rev);
1582                 if (needs_rfc2047_encoding(title.buf, title.len, RFC2047_SUBJECT))
1583                         add_rfc2047(sb, title.buf, title.len,
1584                                                 encoding, RFC2047_SUBJECT);
1585                 else
1586                         strbuf_add_wrapped_bytes(sb, title.buf, title.len,
1587                                          -last_line_length(sb), 1, max_length);
1588         } else {
1589                 strbuf_addbuf(sb, &title);
1590         }
1591         strbuf_addch(sb, '\n');
1592
1593         if (need_8bit_cte == 0) {
1594                 int i;
1595                 for (i = 0; i < pp->in_body_headers.nr; i++) {
1596                         if (has_non_ascii(pp->in_body_headers.items[i].string)) {
1597                                 need_8bit_cte = 1;
1598                                 break;
1599                         }
1600                 }
1601         }
1602
1603         if (need_8bit_cte > 0) {
1604                 const char *header_fmt =
1605                         "MIME-Version: 1.0\n"
1606                         "Content-Type: text/plain; charset=%s\n"
1607                         "Content-Transfer-Encoding: 8bit\n";
1608                 strbuf_addf(sb, header_fmt, encoding);
1609         }
1610         if (pp->after_subject) {
1611                 strbuf_addstr(sb, pp->after_subject);
1612         }
1613         if (cmit_fmt_is_mail(pp->fmt)) {
1614                 strbuf_addch(sb, '\n');
1615         }
1616
1617         if (pp->in_body_headers.nr) {
1618                 int i;
1619                 for (i = 0; i < pp->in_body_headers.nr; i++) {
1620                         strbuf_addstr(sb, pp->in_body_headers.items[i].string);
1621                         free(pp->in_body_headers.items[i].string);
1622                 }
1623                 string_list_clear(&pp->in_body_headers, 0);
1624                 strbuf_addch(sb, '\n');
1625         }
1626
1627         strbuf_release(&title);
1628 }
1629
1630 static int pp_utf8_width(const char *start, const char *end)
1631 {
1632         int width = 0;
1633         size_t remain = end - start;
1634
1635         while (remain) {
1636                 int n = utf8_width(&start, &remain);
1637                 if (n < 0 || !start)
1638                         return -1;
1639                 width += n;
1640         }
1641         return width;
1642 }
1643
1644 static void strbuf_add_tabexpand(struct strbuf *sb, int tabwidth,
1645                                  const char *line, int linelen)
1646 {
1647         const char *tab;
1648
1649         while ((tab = memchr(line, '\t', linelen)) != NULL) {
1650                 int width = pp_utf8_width(line, tab);
1651
1652                 /*
1653                  * If it wasn't well-formed utf8, or it
1654                  * had characters with badly defined
1655                  * width (control characters etc), just
1656                  * give up on trying to align things.
1657                  */
1658                 if (width < 0)
1659                         break;
1660
1661                 /* Output the data .. */
1662                 strbuf_add(sb, line, tab - line);
1663
1664                 /* .. and the de-tabified tab */
1665                 strbuf_addchars(sb, ' ', tabwidth - (width % tabwidth));
1666
1667                 /* Skip over the printed part .. */
1668                 linelen -= tab + 1 - line;
1669                 line = tab + 1;
1670         }
1671
1672         /*
1673          * Print out everything after the last tab without
1674          * worrying about width - there's nothing more to
1675          * align.
1676          */
1677         strbuf_add(sb, line, linelen);
1678 }
1679
1680 /*
1681  * pp_handle_indent() prints out the intendation, and
1682  * the whole line (without the final newline), after
1683  * de-tabifying.
1684  */
1685 static void pp_handle_indent(struct pretty_print_context *pp,
1686                              struct strbuf *sb, int indent,
1687                              const char *line, int linelen)
1688 {
1689         strbuf_addchars(sb, ' ', indent);
1690         if (pp->expand_tabs_in_log)
1691                 strbuf_add_tabexpand(sb, pp->expand_tabs_in_log, line, linelen);
1692         else
1693                 strbuf_add(sb, line, linelen);
1694 }
1695
1696 static int is_mboxrd_from(const char *line, int len)
1697 {
1698         /*
1699          * a line matching /^From $/ here would only have len == 4
1700          * at this point because is_empty_line would've trimmed all
1701          * trailing space
1702          */
1703         return len > 4 && starts_with(line + strspn(line, ">"), "From ");
1704 }
1705
1706 void pp_remainder(struct pretty_print_context *pp,
1707                   const char **msg_p,
1708                   struct strbuf *sb,
1709                   int indent)
1710 {
1711         int first = 1;
1712         for (;;) {
1713                 const char *line = *msg_p;
1714                 int linelen = get_one_line(line);
1715                 *msg_p += linelen;
1716
1717                 if (!linelen)
1718                         break;
1719
1720                 if (is_blank_line(line, &linelen)) {
1721                         if (first)
1722                                 continue;
1723                         if (pp->fmt == CMIT_FMT_SHORT)
1724                                 break;
1725                 }
1726                 first = 0;
1727
1728                 strbuf_grow(sb, linelen + indent + 20);
1729                 if (indent)
1730                         pp_handle_indent(pp, sb, indent, line, linelen);
1731                 else if (pp->expand_tabs_in_log)
1732                         strbuf_add_tabexpand(sb, pp->expand_tabs_in_log,
1733                                              line, linelen);
1734                 else {
1735                         if (pp->fmt == CMIT_FMT_MBOXRD &&
1736                                         is_mboxrd_from(line, linelen))
1737                                 strbuf_addch(sb, '>');
1738
1739                         strbuf_add(sb, line, linelen);
1740                 }
1741                 strbuf_addch(sb, '\n');
1742         }
1743 }
1744
1745 void pretty_print_commit(struct pretty_print_context *pp,
1746                          const struct commit *commit,
1747                          struct strbuf *sb)
1748 {
1749         unsigned long beginning_of_body;
1750         int indent = 4;
1751         const char *msg;
1752         const char *reencoded;
1753         const char *encoding;
1754         int need_8bit_cte = pp->need_8bit_cte;
1755
1756         if (pp->fmt == CMIT_FMT_USERFORMAT) {
1757                 format_commit_message(commit, user_format, sb, pp);
1758                 return;
1759         }
1760
1761         encoding = get_log_output_encoding();
1762         msg = reencoded = logmsg_reencode(commit, NULL, encoding);
1763
1764         if (pp->fmt == CMIT_FMT_ONELINE || cmit_fmt_is_mail(pp->fmt))
1765                 indent = 0;
1766
1767         /*
1768          * We need to check and emit Content-type: to mark it
1769          * as 8-bit if we haven't done so.
1770          */
1771         if (cmit_fmt_is_mail(pp->fmt) && need_8bit_cte == 0) {
1772                 int i, ch, in_body;
1773
1774                 for (in_body = i = 0; (ch = msg[i]); i++) {
1775                         if (!in_body) {
1776                                 /* author could be non 7-bit ASCII but
1777                                  * the log may be so; skip over the
1778                                  * header part first.
1779                                  */
1780                                 if (ch == '\n' && msg[i+1] == '\n')
1781                                         in_body = 1;
1782                         }
1783                         else if (non_ascii(ch)) {
1784                                 need_8bit_cte = 1;
1785                                 break;
1786                         }
1787                 }
1788         }
1789
1790         pp_header(pp, encoding, commit, &msg, sb);
1791         if (pp->fmt != CMIT_FMT_ONELINE && !pp->print_email_subject) {
1792                 strbuf_addch(sb, '\n');
1793         }
1794
1795         /* Skip excess blank lines at the beginning of body, if any... */
1796         msg = skip_blank_lines(msg);
1797
1798         /* These formats treat the title line specially. */
1799         if (pp->fmt == CMIT_FMT_ONELINE || cmit_fmt_is_mail(pp->fmt))
1800                 pp_title_line(pp, &msg, sb, encoding, need_8bit_cte);
1801
1802         beginning_of_body = sb->len;
1803         if (pp->fmt != CMIT_FMT_ONELINE)
1804                 pp_remainder(pp, &msg, sb, indent);
1805         strbuf_rtrim(sb);
1806
1807         /* Make sure there is an EOLN for the non-oneline case */
1808         if (pp->fmt != CMIT_FMT_ONELINE)
1809                 strbuf_addch(sb, '\n');
1810
1811         /*
1812          * The caller may append additional body text in e-mail
1813          * format.  Make sure we did not strip the blank line
1814          * between the header and the body.
1815          */
1816         if (cmit_fmt_is_mail(pp->fmt) && sb->len <= beginning_of_body)
1817                 strbuf_addch(sb, '\n');
1818
1819         unuse_commit_buffer(commit, reencoded);
1820 }
1821
1822 void pp_commit_easy(enum cmit_fmt fmt, const struct commit *commit,
1823                     struct strbuf *sb)
1824 {
1825         struct pretty_print_context pp = {0};
1826         pp.fmt = fmt;
1827         pretty_print_commit(&pp, commit, sb);
1828 }