pretty.con commit grep: recurse in-process using 'struct repository' (f9ee2fc)
   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
  16static char *user_format;
  17static 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;
  25static size_t builtin_formats_len;
  26static size_t commit_formats_len;
  27static size_t commit_formats_alloc;
  28static struct cmt_fmt_map *find_commit_format(const char *sought);
  29
  30int commit_format_is_empty(enum cmit_fmt fmt)
  31{
  32        return fmt == CMIT_FMT_USERFORMAT && !*user_format;
  33}
  34
  35static 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
  44static 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
  90static 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
 111static 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
 146static 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
 154void 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 */
 189static 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 */
 205static int non_ascii(int ch)
 206{
 207        return !isascii(ch) || ch == '\033';
 208}
 209
 210int 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
 222static 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
 244static 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
 253static 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
 264static 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
 285enum rfc2047_type {
 286        RFC2047_SUBJECT,
 287        RFC2047_ADDRESS
 288};
 289
 290static 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
 342static 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
 358static 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
 406const 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
 425void 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
 514static 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
 523const 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
 537static 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
 560static 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
 567static 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
 598const 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
 666static 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
 677static 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
 732skip:
 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
 745struct chunk {
 746        size_t off;
 747        size_t len;
 748};
 749
 750enum flush_type {
 751        no_flush,
 752        flush_right,
 753        flush_left,
 754        flush_left_and_steal,
 755        flush_both
 756};
 757
 758enum trunc_type {
 759        trunc_none,
 760        trunc_left,
 761        trunc_middle,
 762        trunc_right
 763};
 764
 765struct 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
 790static 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
 816static int istitlechar(char c)
 817{
 818        return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') ||
 819                (c >= '0' && c <= '9') || c == '.' || c == '_';
 820}
 821
 822static 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
 850const 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
 874static 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
 884static 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
 899static 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
 912static 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
 928static 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
 945static 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
 978static 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
1050static 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
1284static 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
1386static 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
1435static 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
1451void 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
1464void 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
1509static 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
1565void 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
1630static 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
1644static 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 */
1685static 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
1696static 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
1706void 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
1745void 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
1822void 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}