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