ref-filter.con commit for-each-ref: add '--points-at' option (d325406)
   1#include "builtin.h"
   2#include "cache.h"
   3#include "parse-options.h"
   4#include "refs.h"
   5#include "wildmatch.h"
   6#include "commit.h"
   7#include "remote.h"
   8#include "color.h"
   9#include "tag.h"
  10#include "quote.h"
  11#include "ref-filter.h"
  12
  13typedef enum { FIELD_STR, FIELD_ULONG, FIELD_TIME } cmp_type;
  14
  15static struct {
  16        const char *name;
  17        cmp_type cmp_type;
  18} valid_atom[] = {
  19        { "refname" },
  20        { "objecttype" },
  21        { "objectsize", FIELD_ULONG },
  22        { "objectname" },
  23        { "tree" },
  24        { "parent" },
  25        { "numparent", FIELD_ULONG },
  26        { "object" },
  27        { "type" },
  28        { "tag" },
  29        { "author" },
  30        { "authorname" },
  31        { "authoremail" },
  32        { "authordate", FIELD_TIME },
  33        { "committer" },
  34        { "committername" },
  35        { "committeremail" },
  36        { "committerdate", FIELD_TIME },
  37        { "tagger" },
  38        { "taggername" },
  39        { "taggeremail" },
  40        { "taggerdate", FIELD_TIME },
  41        { "creator" },
  42        { "creatordate", FIELD_TIME },
  43        { "subject" },
  44        { "body" },
  45        { "contents" },
  46        { "contents:subject" },
  47        { "contents:body" },
  48        { "contents:signature" },
  49        { "upstream" },
  50        { "push" },
  51        { "symref" },
  52        { "flag" },
  53        { "HEAD" },
  54        { "color" },
  55};
  56
  57/*
  58 * An atom is a valid field atom listed above, possibly prefixed with
  59 * a "*" to denote deref_tag().
  60 *
  61 * We parse given format string and sort specifiers, and make a list
  62 * of properties that we need to extract out of objects.  ref_array_item
  63 * structure will hold an array of values extracted that can be
  64 * indexed with the "atom number", which is an index into this
  65 * array.
  66 */
  67static const char **used_atom;
  68static cmp_type *used_atom_type;
  69static int used_atom_cnt, need_tagged, need_symref;
  70static int need_color_reset_at_eol;
  71
  72/*
  73 * Used to parse format string and sort specifiers
  74 */
  75int parse_ref_filter_atom(const char *atom, const char *ep)
  76{
  77        const char *sp;
  78        int i, at;
  79
  80        sp = atom;
  81        if (*sp == '*' && sp < ep)
  82                sp++; /* deref */
  83        if (ep <= sp)
  84                die("malformed field name: %.*s", (int)(ep-atom), atom);
  85
  86        /* Do we have the atom already used elsewhere? */
  87        for (i = 0; i < used_atom_cnt; i++) {
  88                int len = strlen(used_atom[i]);
  89                if (len == ep - atom && !memcmp(used_atom[i], atom, len))
  90                        return i;
  91        }
  92
  93        /* Is the atom a valid one? */
  94        for (i = 0; i < ARRAY_SIZE(valid_atom); i++) {
  95                int len = strlen(valid_atom[i].name);
  96                /*
  97                 * If the atom name has a colon, strip it and everything after
  98                 * it off - it specifies the format for this entry, and
  99                 * shouldn't be used for checking against the valid_atom
 100                 * table.
 101                 */
 102                const char *formatp = strchr(sp, ':');
 103                if (!formatp || ep < formatp)
 104                        formatp = ep;
 105                if (len == formatp - sp && !memcmp(valid_atom[i].name, sp, len))
 106                        break;
 107        }
 108
 109        if (ARRAY_SIZE(valid_atom) <= i)
 110                die("unknown field name: %.*s", (int)(ep-atom), atom);
 111
 112        /* Add it in, including the deref prefix */
 113        at = used_atom_cnt;
 114        used_atom_cnt++;
 115        REALLOC_ARRAY(used_atom, used_atom_cnt);
 116        REALLOC_ARRAY(used_atom_type, used_atom_cnt);
 117        used_atom[at] = xmemdupz(atom, ep - atom);
 118        used_atom_type[at] = valid_atom[i].cmp_type;
 119        if (*atom == '*')
 120                need_tagged = 1;
 121        if (!strcmp(used_atom[at], "symref"))
 122                need_symref = 1;
 123        return at;
 124}
 125
 126/*
 127 * In a format string, find the next occurrence of %(atom).
 128 */
 129static const char *find_next(const char *cp)
 130{
 131        while (*cp) {
 132                if (*cp == '%') {
 133                        /*
 134                         * %( is the start of an atom;
 135                         * %% is a quoted per-cent.
 136                         */
 137                        if (cp[1] == '(')
 138                                return cp;
 139                        else if (cp[1] == '%')
 140                                cp++; /* skip over two % */
 141                        /* otherwise this is a singleton, literal % */
 142                }
 143                cp++;
 144        }
 145        return NULL;
 146}
 147
 148/*
 149 * Make sure the format string is well formed, and parse out
 150 * the used atoms.
 151 */
 152int verify_ref_format(const char *format)
 153{
 154        const char *cp, *sp;
 155
 156        need_color_reset_at_eol = 0;
 157        for (cp = format; *cp && (sp = find_next(cp)); ) {
 158                const char *color, *ep = strchr(sp, ')');
 159                int at;
 160
 161                if (!ep)
 162                        return error("malformed format string %s", sp);
 163                /* sp points at "%(" and ep points at the closing ")" */
 164                at = parse_ref_filter_atom(sp + 2, ep);
 165                cp = ep + 1;
 166
 167                if (skip_prefix(used_atom[at], "color:", &color))
 168                        need_color_reset_at_eol = !!strcmp(color, "reset");
 169        }
 170        return 0;
 171}
 172
 173/*
 174 * Given an object name, read the object data and size, and return a
 175 * "struct object".  If the object data we are returning is also borrowed
 176 * by the "struct object" representation, set *eaten as well---it is a
 177 * signal from parse_object_buffer to us not to free the buffer.
 178 */
 179static void *get_obj(const unsigned char *sha1, struct object **obj, unsigned long *sz, int *eaten)
 180{
 181        enum object_type type;
 182        void *buf = read_sha1_file(sha1, &type, sz);
 183
 184        if (buf)
 185                *obj = parse_object_buffer(sha1, type, *sz, buf, eaten);
 186        else
 187                *obj = NULL;
 188        return buf;
 189}
 190
 191static int grab_objectname(const char *name, const unsigned char *sha1,
 192                            struct atom_value *v)
 193{
 194        if (!strcmp(name, "objectname")) {
 195                char *s = xmalloc(41);
 196                strcpy(s, sha1_to_hex(sha1));
 197                v->s = s;
 198                return 1;
 199        }
 200        if (!strcmp(name, "objectname:short")) {
 201                v->s = xstrdup(find_unique_abbrev(sha1, DEFAULT_ABBREV));
 202                return 1;
 203        }
 204        return 0;
 205}
 206
 207/* See grab_values */
 208static void grab_common_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
 209{
 210        int i;
 211
 212        for (i = 0; i < used_atom_cnt; i++) {
 213                const char *name = used_atom[i];
 214                struct atom_value *v = &val[i];
 215                if (!!deref != (*name == '*'))
 216                        continue;
 217                if (deref)
 218                        name++;
 219                if (!strcmp(name, "objecttype"))
 220                        v->s = typename(obj->type);
 221                else if (!strcmp(name, "objectsize")) {
 222                        char *s = xmalloc(40);
 223                        sprintf(s, "%lu", sz);
 224                        v->ul = sz;
 225                        v->s = s;
 226                }
 227                else if (deref)
 228                        grab_objectname(name, obj->sha1, v);
 229        }
 230}
 231
 232/* See grab_values */
 233static void grab_tag_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
 234{
 235        int i;
 236        struct tag *tag = (struct tag *) obj;
 237
 238        for (i = 0; i < used_atom_cnt; i++) {
 239                const char *name = used_atom[i];
 240                struct atom_value *v = &val[i];
 241                if (!!deref != (*name == '*'))
 242                        continue;
 243                if (deref)
 244                        name++;
 245                if (!strcmp(name, "tag"))
 246                        v->s = tag->tag;
 247                else if (!strcmp(name, "type") && tag->tagged)
 248                        v->s = typename(tag->tagged->type);
 249                else if (!strcmp(name, "object") && tag->tagged) {
 250                        char *s = xmalloc(41);
 251                        strcpy(s, sha1_to_hex(tag->tagged->sha1));
 252                        v->s = s;
 253                }
 254        }
 255}
 256
 257/* See grab_values */
 258static void grab_commit_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
 259{
 260        int i;
 261        struct commit *commit = (struct commit *) obj;
 262
 263        for (i = 0; i < used_atom_cnt; i++) {
 264                const char *name = used_atom[i];
 265                struct atom_value *v = &val[i];
 266                if (!!deref != (*name == '*'))
 267                        continue;
 268                if (deref)
 269                        name++;
 270                if (!strcmp(name, "tree")) {
 271                        char *s = xmalloc(41);
 272                        strcpy(s, sha1_to_hex(commit->tree->object.sha1));
 273                        v->s = s;
 274                }
 275                if (!strcmp(name, "numparent")) {
 276                        char *s = xmalloc(40);
 277                        v->ul = commit_list_count(commit->parents);
 278                        sprintf(s, "%lu", v->ul);
 279                        v->s = s;
 280                }
 281                else if (!strcmp(name, "parent")) {
 282                        int num = commit_list_count(commit->parents);
 283                        int i;
 284                        struct commit_list *parents;
 285                        char *s = xmalloc(41 * num + 1);
 286                        v->s = s;
 287                        for (i = 0, parents = commit->parents;
 288                             parents;
 289                             parents = parents->next, i = i + 41) {
 290                                struct commit *parent = parents->item;
 291                                strcpy(s+i, sha1_to_hex(parent->object.sha1));
 292                                if (parents->next)
 293                                        s[i+40] = ' ';
 294                        }
 295                        if (!i)
 296                                *s = '\0';
 297                }
 298        }
 299}
 300
 301static const char *find_wholine(const char *who, int wholen, const char *buf, unsigned long sz)
 302{
 303        const char *eol;
 304        while (*buf) {
 305                if (!strncmp(buf, who, wholen) &&
 306                    buf[wholen] == ' ')
 307                        return buf + wholen + 1;
 308                eol = strchr(buf, '\n');
 309                if (!eol)
 310                        return "";
 311                eol++;
 312                if (*eol == '\n')
 313                        return ""; /* end of header */
 314                buf = eol;
 315        }
 316        return "";
 317}
 318
 319static const char *copy_line(const char *buf)
 320{
 321        const char *eol = strchrnul(buf, '\n');
 322        return xmemdupz(buf, eol - buf);
 323}
 324
 325static const char *copy_name(const char *buf)
 326{
 327        const char *cp;
 328        for (cp = buf; *cp && *cp != '\n'; cp++) {
 329                if (!strncmp(cp, " <", 2))
 330                        return xmemdupz(buf, cp - buf);
 331        }
 332        return "";
 333}
 334
 335static const char *copy_email(const char *buf)
 336{
 337        const char *email = strchr(buf, '<');
 338        const char *eoemail;
 339        if (!email)
 340                return "";
 341        eoemail = strchr(email, '>');
 342        if (!eoemail)
 343                return "";
 344        return xmemdupz(email, eoemail + 1 - email);
 345}
 346
 347static char *copy_subject(const char *buf, unsigned long len)
 348{
 349        char *r = xmemdupz(buf, len);
 350        int i;
 351
 352        for (i = 0; i < len; i++)
 353                if (r[i] == '\n')
 354                        r[i] = ' ';
 355
 356        return r;
 357}
 358
 359static void grab_date(const char *buf, struct atom_value *v, const char *atomname)
 360{
 361        const char *eoemail = strstr(buf, "> ");
 362        char *zone;
 363        unsigned long timestamp;
 364        long tz;
 365        enum date_mode date_mode = DATE_NORMAL;
 366        const char *formatp;
 367
 368        /*
 369         * We got here because atomname ends in "date" or "date<something>";
 370         * it's not possible that <something> is not ":<format>" because
 371         * parse_ref_filter_atom() wouldn't have allowed it, so we can assume that no
 372         * ":" means no format is specified, and use the default.
 373         */
 374        formatp = strchr(atomname, ':');
 375        if (formatp != NULL) {
 376                formatp++;
 377                date_mode = parse_date_format(formatp);
 378        }
 379
 380        if (!eoemail)
 381                goto bad;
 382        timestamp = strtoul(eoemail + 2, &zone, 10);
 383        if (timestamp == ULONG_MAX)
 384                goto bad;
 385        tz = strtol(zone, NULL, 10);
 386        if ((tz == LONG_MIN || tz == LONG_MAX) && errno == ERANGE)
 387                goto bad;
 388        v->s = xstrdup(show_date(timestamp, tz, date_mode));
 389        v->ul = timestamp;
 390        return;
 391 bad:
 392        v->s = "";
 393        v->ul = 0;
 394}
 395
 396/* See grab_values */
 397static void grab_person(const char *who, struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
 398{
 399        int i;
 400        int wholen = strlen(who);
 401        const char *wholine = NULL;
 402
 403        for (i = 0; i < used_atom_cnt; i++) {
 404                const char *name = used_atom[i];
 405                struct atom_value *v = &val[i];
 406                if (!!deref != (*name == '*'))
 407                        continue;
 408                if (deref)
 409                        name++;
 410                if (strncmp(who, name, wholen))
 411                        continue;
 412                if (name[wholen] != 0 &&
 413                    strcmp(name + wholen, "name") &&
 414                    strcmp(name + wholen, "email") &&
 415                    !starts_with(name + wholen, "date"))
 416                        continue;
 417                if (!wholine)
 418                        wholine = find_wholine(who, wholen, buf, sz);
 419                if (!wholine)
 420                        return; /* no point looking for it */
 421                if (name[wholen] == 0)
 422                        v->s = copy_line(wholine);
 423                else if (!strcmp(name + wholen, "name"))
 424                        v->s = copy_name(wholine);
 425                else if (!strcmp(name + wholen, "email"))
 426                        v->s = copy_email(wholine);
 427                else if (starts_with(name + wholen, "date"))
 428                        grab_date(wholine, v, name);
 429        }
 430
 431        /*
 432         * For a tag or a commit object, if "creator" or "creatordate" is
 433         * requested, do something special.
 434         */
 435        if (strcmp(who, "tagger") && strcmp(who, "committer"))
 436                return; /* "author" for commit object is not wanted */
 437        if (!wholine)
 438                wholine = find_wholine(who, wholen, buf, sz);
 439        if (!wholine)
 440                return;
 441        for (i = 0; i < used_atom_cnt; i++) {
 442                const char *name = used_atom[i];
 443                struct atom_value *v = &val[i];
 444                if (!!deref != (*name == '*'))
 445                        continue;
 446                if (deref)
 447                        name++;
 448
 449                if (starts_with(name, "creatordate"))
 450                        grab_date(wholine, v, name);
 451                else if (!strcmp(name, "creator"))
 452                        v->s = copy_line(wholine);
 453        }
 454}
 455
 456static void find_subpos(const char *buf, unsigned long sz,
 457                        const char **sub, unsigned long *sublen,
 458                        const char **body, unsigned long *bodylen,
 459                        unsigned long *nonsiglen,
 460                        const char **sig, unsigned long *siglen)
 461{
 462        const char *eol;
 463        /* skip past header until we hit empty line */
 464        while (*buf && *buf != '\n') {
 465                eol = strchrnul(buf, '\n');
 466                if (*eol)
 467                        eol++;
 468                buf = eol;
 469        }
 470        /* skip any empty lines */
 471        while (*buf == '\n')
 472                buf++;
 473
 474        /* parse signature first; we might not even have a subject line */
 475        *sig = buf + parse_signature(buf, strlen(buf));
 476        *siglen = strlen(*sig);
 477
 478        /* subject is first non-empty line */
 479        *sub = buf;
 480        /* subject goes to first empty line */
 481        while (buf < *sig && *buf && *buf != '\n') {
 482                eol = strchrnul(buf, '\n');
 483                if (*eol)
 484                        eol++;
 485                buf = eol;
 486        }
 487        *sublen = buf - *sub;
 488        /* drop trailing newline, if present */
 489        if (*sublen && (*sub)[*sublen - 1] == '\n')
 490                *sublen -= 1;
 491
 492        /* skip any empty lines */
 493        while (*buf == '\n')
 494                buf++;
 495        *body = buf;
 496        *bodylen = strlen(buf);
 497        *nonsiglen = *sig - buf;
 498}
 499
 500/* See grab_values */
 501static void grab_sub_body_contents(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
 502{
 503        int i;
 504        const char *subpos = NULL, *bodypos = NULL, *sigpos = NULL;
 505        unsigned long sublen = 0, bodylen = 0, nonsiglen = 0, siglen = 0;
 506
 507        for (i = 0; i < used_atom_cnt; i++) {
 508                const char *name = used_atom[i];
 509                struct atom_value *v = &val[i];
 510                if (!!deref != (*name == '*'))
 511                        continue;
 512                if (deref)
 513                        name++;
 514                if (strcmp(name, "subject") &&
 515                    strcmp(name, "body") &&
 516                    strcmp(name, "contents") &&
 517                    strcmp(name, "contents:subject") &&
 518                    strcmp(name, "contents:body") &&
 519                    strcmp(name, "contents:signature"))
 520                        continue;
 521                if (!subpos)
 522                        find_subpos(buf, sz,
 523                                    &subpos, &sublen,
 524                                    &bodypos, &bodylen, &nonsiglen,
 525                                    &sigpos, &siglen);
 526
 527                if (!strcmp(name, "subject"))
 528                        v->s = copy_subject(subpos, sublen);
 529                else if (!strcmp(name, "contents:subject"))
 530                        v->s = copy_subject(subpos, sublen);
 531                else if (!strcmp(name, "body"))
 532                        v->s = xmemdupz(bodypos, bodylen);
 533                else if (!strcmp(name, "contents:body"))
 534                        v->s = xmemdupz(bodypos, nonsiglen);
 535                else if (!strcmp(name, "contents:signature"))
 536                        v->s = xmemdupz(sigpos, siglen);
 537                else if (!strcmp(name, "contents"))
 538                        v->s = xstrdup(subpos);
 539        }
 540}
 541
 542/*
 543 * We want to have empty print-string for field requests
 544 * that do not apply (e.g. "authordate" for a tag object)
 545 */
 546static void fill_missing_values(struct atom_value *val)
 547{
 548        int i;
 549        for (i = 0; i < used_atom_cnt; i++) {
 550                struct atom_value *v = &val[i];
 551                if (v->s == NULL)
 552                        v->s = "";
 553        }
 554}
 555
 556/*
 557 * val is a list of atom_value to hold returned values.  Extract
 558 * the values for atoms in used_atom array out of (obj, buf, sz).
 559 * when deref is false, (obj, buf, sz) is the object that is
 560 * pointed at by the ref itself; otherwise it is the object the
 561 * ref (which is a tag) refers to.
 562 */
 563static void grab_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
 564{
 565        grab_common_values(val, deref, obj, buf, sz);
 566        switch (obj->type) {
 567        case OBJ_TAG:
 568                grab_tag_values(val, deref, obj, buf, sz);
 569                grab_sub_body_contents(val, deref, obj, buf, sz);
 570                grab_person("tagger", val, deref, obj, buf, sz);
 571                break;
 572        case OBJ_COMMIT:
 573                grab_commit_values(val, deref, obj, buf, sz);
 574                grab_sub_body_contents(val, deref, obj, buf, sz);
 575                grab_person("author", val, deref, obj, buf, sz);
 576                grab_person("committer", val, deref, obj, buf, sz);
 577                break;
 578        case OBJ_TREE:
 579                /* grab_tree_values(val, deref, obj, buf, sz); */
 580                break;
 581        case OBJ_BLOB:
 582                /* grab_blob_values(val, deref, obj, buf, sz); */
 583                break;
 584        default:
 585                die("Eh?  Object of type %d?", obj->type);
 586        }
 587}
 588
 589static inline char *copy_advance(char *dst, const char *src)
 590{
 591        while (*src)
 592                *dst++ = *src++;
 593        return dst;
 594}
 595
 596/*
 597 * Parse the object referred by ref, and grab needed value.
 598 */
 599static void populate_value(struct ref_array_item *ref)
 600{
 601        void *buf;
 602        struct object *obj;
 603        int eaten, i;
 604        unsigned long size;
 605        const unsigned char *tagged;
 606
 607        ref->value = xcalloc(used_atom_cnt, sizeof(struct atom_value));
 608
 609        if (need_symref && (ref->flag & REF_ISSYMREF) && !ref->symref) {
 610                unsigned char unused1[20];
 611                ref->symref = resolve_refdup(ref->refname, RESOLVE_REF_READING,
 612                                             unused1, NULL);
 613                if (!ref->symref)
 614                        ref->symref = "";
 615        }
 616
 617        /* Fill in specials first */
 618        for (i = 0; i < used_atom_cnt; i++) {
 619                const char *name = used_atom[i];
 620                struct atom_value *v = &ref->value[i];
 621                int deref = 0;
 622                const char *refname;
 623                const char *formatp;
 624                struct branch *branch = NULL;
 625
 626                if (*name == '*') {
 627                        deref = 1;
 628                        name++;
 629                }
 630
 631                if (starts_with(name, "refname"))
 632                        refname = ref->refname;
 633                else if (starts_with(name, "symref"))
 634                        refname = ref->symref ? ref->symref : "";
 635                else if (starts_with(name, "upstream")) {
 636                        const char *branch_name;
 637                        /* only local branches may have an upstream */
 638                        if (!skip_prefix(ref->refname, "refs/heads/",
 639                                         &branch_name))
 640                                continue;
 641                        branch = branch_get(branch_name);
 642
 643                        refname = branch_get_upstream(branch, NULL);
 644                        if (!refname)
 645                                continue;
 646                } else if (starts_with(name, "push")) {
 647                        const char *branch_name;
 648                        if (!skip_prefix(ref->refname, "refs/heads/",
 649                                         &branch_name))
 650                                continue;
 651                        branch = branch_get(branch_name);
 652
 653                        refname = branch_get_push(branch, NULL);
 654                        if (!refname)
 655                                continue;
 656                } else if (starts_with(name, "color:")) {
 657                        char color[COLOR_MAXLEN] = "";
 658
 659                        if (color_parse(name + 6, color) < 0)
 660                                die(_("unable to parse format"));
 661                        v->s = xstrdup(color);
 662                        continue;
 663                } else if (!strcmp(name, "flag")) {
 664                        char buf[256], *cp = buf;
 665                        if (ref->flag & REF_ISSYMREF)
 666                                cp = copy_advance(cp, ",symref");
 667                        if (ref->flag & REF_ISPACKED)
 668                                cp = copy_advance(cp, ",packed");
 669                        if (cp == buf)
 670                                v->s = "";
 671                        else {
 672                                *cp = '\0';
 673                                v->s = xstrdup(buf + 1);
 674                        }
 675                        continue;
 676                } else if (!deref && grab_objectname(name, ref->objectname, v)) {
 677                        continue;
 678                } else if (!strcmp(name, "HEAD")) {
 679                        const char *head;
 680                        unsigned char sha1[20];
 681
 682                        head = resolve_ref_unsafe("HEAD", RESOLVE_REF_READING,
 683                                                  sha1, NULL);
 684                        if (!strcmp(ref->refname, head))
 685                                v->s = "*";
 686                        else
 687                                v->s = " ";
 688                        continue;
 689                } else
 690                        continue;
 691
 692                formatp = strchr(name, ':');
 693                if (formatp) {
 694                        int num_ours, num_theirs;
 695
 696                        formatp++;
 697                        if (!strcmp(formatp, "short"))
 698                                refname = shorten_unambiguous_ref(refname,
 699                                                      warn_ambiguous_refs);
 700                        else if (!strcmp(formatp, "track") &&
 701                                 (starts_with(name, "upstream") ||
 702                                  starts_with(name, "push"))) {
 703                                char buf[40];
 704
 705                                if (stat_tracking_info(branch, &num_ours,
 706                                                       &num_theirs, NULL))
 707                                        continue;
 708
 709                                if (!num_ours && !num_theirs)
 710                                        v->s = "";
 711                                else if (!num_ours) {
 712                                        sprintf(buf, "[behind %d]", num_theirs);
 713                                        v->s = xstrdup(buf);
 714                                } else if (!num_theirs) {
 715                                        sprintf(buf, "[ahead %d]", num_ours);
 716                                        v->s = xstrdup(buf);
 717                                } else {
 718                                        sprintf(buf, "[ahead %d, behind %d]",
 719                                                num_ours, num_theirs);
 720                                        v->s = xstrdup(buf);
 721                                }
 722                                continue;
 723                        } else if (!strcmp(formatp, "trackshort") &&
 724                                   (starts_with(name, "upstream") ||
 725                                    starts_with(name, "push"))) {
 726                                assert(branch);
 727
 728                                if (stat_tracking_info(branch, &num_ours,
 729                                                        &num_theirs, NULL))
 730                                        continue;
 731
 732                                if (!num_ours && !num_theirs)
 733                                        v->s = "=";
 734                                else if (!num_ours)
 735                                        v->s = "<";
 736                                else if (!num_theirs)
 737                                        v->s = ">";
 738                                else
 739                                        v->s = "<>";
 740                                continue;
 741                        } else
 742                                die("unknown %.*s format %s",
 743                                    (int)(formatp - name), name, formatp);
 744                }
 745
 746                if (!deref)
 747                        v->s = refname;
 748                else {
 749                        int len = strlen(refname);
 750                        char *s = xmalloc(len + 4);
 751                        sprintf(s, "%s^{}", refname);
 752                        v->s = s;
 753                }
 754        }
 755
 756        for (i = 0; i < used_atom_cnt; i++) {
 757                struct atom_value *v = &ref->value[i];
 758                if (v->s == NULL)
 759                        goto need_obj;
 760        }
 761        return;
 762
 763 need_obj:
 764        buf = get_obj(ref->objectname, &obj, &size, &eaten);
 765        if (!buf)
 766                die("missing object %s for %s",
 767                    sha1_to_hex(ref->objectname), ref->refname);
 768        if (!obj)
 769                die("parse_object_buffer failed on %s for %s",
 770                    sha1_to_hex(ref->objectname), ref->refname);
 771
 772        grab_values(ref->value, 0, obj, buf, size);
 773        if (!eaten)
 774                free(buf);
 775
 776        /*
 777         * If there is no atom that wants to know about tagged
 778         * object, we are done.
 779         */
 780        if (!need_tagged || (obj->type != OBJ_TAG))
 781                return;
 782
 783        /*
 784         * If it is a tag object, see if we use a value that derefs
 785         * the object, and if we do grab the object it refers to.
 786         */
 787        tagged = ((struct tag *)obj)->tagged->sha1;
 788
 789        /*
 790         * NEEDSWORK: This derefs tag only once, which
 791         * is good to deal with chains of trust, but
 792         * is not consistent with what deref_tag() does
 793         * which peels the onion to the core.
 794         */
 795        buf = get_obj(tagged, &obj, &size, &eaten);
 796        if (!buf)
 797                die("missing object %s for %s",
 798                    sha1_to_hex(tagged), ref->refname);
 799        if (!obj)
 800                die("parse_object_buffer failed on %s for %s",
 801                    sha1_to_hex(tagged), ref->refname);
 802        grab_values(ref->value, 1, obj, buf, size);
 803        if (!eaten)
 804                free(buf);
 805}
 806
 807/*
 808 * Given a ref, return the value for the atom.  This lazily gets value
 809 * out of the object by calling populate value.
 810 */
 811static void get_ref_atom_value(struct ref_array_item *ref, int atom, struct atom_value **v)
 812{
 813        if (!ref->value) {
 814                populate_value(ref);
 815                fill_missing_values(ref->value);
 816        }
 817        *v = &ref->value[atom];
 818}
 819
 820/*
 821 * Return 1 if the refname matches one of the patterns, otherwise 0.
 822 * A pattern can be path prefix (e.g. a refname "refs/heads/master"
 823 * matches a pattern "refs/heads/") or a wildcard (e.g. the same ref
 824 * matches "refs/heads/m*",too).
 825 */
 826static int match_name_as_path(const char **pattern, const char *refname)
 827{
 828        int namelen = strlen(refname);
 829        for (; *pattern; pattern++) {
 830                const char *p = *pattern;
 831                int plen = strlen(p);
 832
 833                if ((plen <= namelen) &&
 834                    !strncmp(refname, p, plen) &&
 835                    (refname[plen] == '\0' ||
 836                     refname[plen] == '/' ||
 837                     p[plen-1] == '/'))
 838                        return 1;
 839                if (!wildmatch(p, refname, WM_PATHNAME, NULL))
 840                        return 1;
 841        }
 842        return 0;
 843}
 844
 845/*
 846 * Given a ref (sha1, refname), check if the ref belongs to the array
 847 * of sha1s. If the given ref is a tag, check if the given tag points
 848 * at one of the sha1s in the given sha1 array.
 849 * the given sha1_array.
 850 * NEEDSWORK:
 851 * 1. Only a single level of inderection is obtained, we might want to
 852 * change this to account for multiple levels (e.g. annotated tags
 853 * pointing to annotated tags pointing to a commit.)
 854 * 2. As the refs are cached we might know what refname peels to without
 855 * the need to parse the object via parse_object(). peel_ref() might be a
 856 * more efficient alternative to obtain the pointee.
 857 */
 858static const unsigned char *match_points_at(struct sha1_array *points_at,
 859                                            const unsigned char *sha1,
 860                                            const char *refname)
 861{
 862        const unsigned char *tagged_sha1 = NULL;
 863        struct object *obj;
 864
 865        if (sha1_array_lookup(points_at, sha1) >= 0)
 866                return sha1;
 867        obj = parse_object(sha1);
 868        if (!obj)
 869                die(_("malformed object at '%s'"), refname);
 870        if (obj->type == OBJ_TAG)
 871                tagged_sha1 = ((struct tag *)obj)->tagged->sha1;
 872        if (tagged_sha1 && sha1_array_lookup(points_at, tagged_sha1) >= 0)
 873                return tagged_sha1;
 874        return NULL;
 875}
 876
 877/* Allocate space for a new ref_array_item and copy the objectname and flag to it */
 878static struct ref_array_item *new_ref_array_item(const char *refname,
 879                                                 const unsigned char *objectname,
 880                                                 int flag)
 881{
 882        size_t len = strlen(refname);
 883        struct ref_array_item *ref = xcalloc(1, sizeof(struct ref_array_item) + len + 1);
 884        memcpy(ref->refname, refname, len);
 885        ref->refname[len] = '\0';
 886        hashcpy(ref->objectname, objectname);
 887        ref->flag = flag;
 888
 889        return ref;
 890}
 891
 892/*
 893 * A call-back given to for_each_ref().  Filter refs and keep them for
 894 * later object processing.
 895 */
 896static int ref_filter_handler(const char *refname, const struct object_id *oid, int flag, void *cb_data)
 897{
 898        struct ref_filter_cbdata *ref_cbdata = cb_data;
 899        struct ref_filter *filter = ref_cbdata->filter;
 900        struct ref_array_item *ref;
 901
 902        if (flag & REF_BAD_NAME) {
 903                warning("ignoring ref with broken name %s", refname);
 904                return 0;
 905        }
 906
 907        if (*filter->name_patterns && !match_name_as_path(filter->name_patterns, refname))
 908                return 0;
 909
 910        if (filter->points_at.nr && !match_points_at(&filter->points_at, oid->hash, refname))
 911                return 0;
 912
 913        /*
 914         * We do not open the object yet; sort may only need refname
 915         * to do its job and the resulting list may yet to be pruned
 916         * by maxcount logic.
 917         */
 918        ref = new_ref_array_item(refname, oid->hash, flag);
 919
 920        REALLOC_ARRAY(ref_cbdata->array->items, ref_cbdata->array->nr + 1);
 921        ref_cbdata->array->items[ref_cbdata->array->nr++] = ref;
 922        return 0;
 923}
 924
 925/*  Free memory allocated for a ref_array_item */
 926static void free_array_item(struct ref_array_item *item)
 927{
 928        free((char *)item->symref);
 929        free(item);
 930}
 931
 932/* Free all memory allocated for ref_array */
 933void ref_array_clear(struct ref_array *array)
 934{
 935        int i;
 936
 937        for (i = 0; i < array->nr; i++)
 938                free_array_item(array->items[i]);
 939        free(array->items);
 940        array->items = NULL;
 941        array->nr = array->alloc = 0;
 942}
 943
 944/*
 945 * API for filtering a set of refs. Based on the type of refs the user
 946 * has requested, we iterate through those refs and apply filters
 947 * as per the given ref_filter structure and finally store the
 948 * filtered refs in the ref_array structure.
 949 */
 950int filter_refs(struct ref_array *array, struct ref_filter *filter, unsigned int type)
 951{
 952        struct ref_filter_cbdata ref_cbdata;
 953
 954        ref_cbdata.array = array;
 955        ref_cbdata.filter = filter;
 956
 957        if (type & (FILTER_REFS_ALL | FILTER_REFS_INCLUDE_BROKEN))
 958                return for_each_rawref(ref_filter_handler, &ref_cbdata);
 959        else if (type & FILTER_REFS_ALL)
 960                return for_each_ref(ref_filter_handler, &ref_cbdata);
 961        else
 962                die("filter_refs: invalid type");
 963        return 0;
 964}
 965
 966static int cmp_ref_sorting(struct ref_sorting *s, struct ref_array_item *a, struct ref_array_item *b)
 967{
 968        struct atom_value *va, *vb;
 969        int cmp;
 970        cmp_type cmp_type = used_atom_type[s->atom];
 971
 972        get_ref_atom_value(a, s->atom, &va);
 973        get_ref_atom_value(b, s->atom, &vb);
 974        switch (cmp_type) {
 975        case FIELD_STR:
 976                cmp = strcmp(va->s, vb->s);
 977                break;
 978        default:
 979                if (va->ul < vb->ul)
 980                        cmp = -1;
 981                else if (va->ul == vb->ul)
 982                        cmp = 0;
 983                else
 984                        cmp = 1;
 985                break;
 986        }
 987        return (s->reverse) ? -cmp : cmp;
 988}
 989
 990static struct ref_sorting *ref_sorting;
 991static int compare_refs(const void *a_, const void *b_)
 992{
 993        struct ref_array_item *a = *((struct ref_array_item **)a_);
 994        struct ref_array_item *b = *((struct ref_array_item **)b_);
 995        struct ref_sorting *s;
 996
 997        for (s = ref_sorting; s; s = s->next) {
 998                int cmp = cmp_ref_sorting(s, a, b);
 999                if (cmp)
1000                        return cmp;
1001        }
1002        return 0;
1003}
1004
1005void ref_array_sort(struct ref_sorting *sorting, struct ref_array *array)
1006{
1007        ref_sorting = sorting;
1008        qsort(array->items, array->nr, sizeof(struct ref_array_item *), compare_refs);
1009}
1010
1011static void print_value(struct atom_value *v, int quote_style)
1012{
1013        struct strbuf sb = STRBUF_INIT;
1014        switch (quote_style) {
1015        case QUOTE_NONE:
1016                fputs(v->s, stdout);
1017                break;
1018        case QUOTE_SHELL:
1019                sq_quote_buf(&sb, v->s);
1020                break;
1021        case QUOTE_PERL:
1022                perl_quote_buf(&sb, v->s);
1023                break;
1024        case QUOTE_PYTHON:
1025                python_quote_buf(&sb, v->s);
1026                break;
1027        case QUOTE_TCL:
1028                tcl_quote_buf(&sb, v->s);
1029                break;
1030        }
1031        if (quote_style != QUOTE_NONE) {
1032                fputs(sb.buf, stdout);
1033                strbuf_release(&sb);
1034        }
1035}
1036
1037static int hex1(char ch)
1038{
1039        if ('0' <= ch && ch <= '9')
1040                return ch - '0';
1041        else if ('a' <= ch && ch <= 'f')
1042                return ch - 'a' + 10;
1043        else if ('A' <= ch && ch <= 'F')
1044                return ch - 'A' + 10;
1045        return -1;
1046}
1047static int hex2(const char *cp)
1048{
1049        if (cp[0] && cp[1])
1050                return (hex1(cp[0]) << 4) | hex1(cp[1]);
1051        else
1052                return -1;
1053}
1054
1055static void emit(const char *cp, const char *ep)
1056{
1057        while (*cp && (!ep || cp < ep)) {
1058                if (*cp == '%') {
1059                        if (cp[1] == '%')
1060                                cp++;
1061                        else {
1062                                int ch = hex2(cp + 1);
1063                                if (0 <= ch) {
1064                                        putchar(ch);
1065                                        cp += 3;
1066                                        continue;
1067                                }
1068                        }
1069                }
1070                putchar(*cp);
1071                cp++;
1072        }
1073}
1074
1075void show_ref_array_item(struct ref_array_item *info, const char *format, int quote_style)
1076{
1077        const char *cp, *sp, *ep;
1078
1079        for (cp = format; *cp && (sp = find_next(cp)); cp = ep + 1) {
1080                struct atom_value *atomv;
1081
1082                ep = strchr(sp, ')');
1083                if (cp < sp)
1084                        emit(cp, sp);
1085                get_ref_atom_value(info, parse_ref_filter_atom(sp + 2, ep), &atomv);
1086                print_value(atomv, quote_style);
1087        }
1088        if (*cp) {
1089                sp = cp + strlen(cp);
1090                emit(cp, sp);
1091        }
1092        if (need_color_reset_at_eol) {
1093                struct atom_value resetv;
1094                char color[COLOR_MAXLEN] = "";
1095
1096                if (color_parse("reset", color) < 0)
1097                        die("BUG: couldn't parse 'reset' as a color");
1098                resetv.s = color;
1099                print_value(&resetv, quote_style);
1100        }
1101        putchar('\n');
1102}
1103
1104/*  If no sorting option is given, use refname to sort as default */
1105struct ref_sorting *ref_default_sorting(void)
1106{
1107        static const char cstr_name[] = "refname";
1108
1109        struct ref_sorting *sorting = xcalloc(1, sizeof(*sorting));
1110
1111        sorting->next = NULL;
1112        sorting->atom = parse_ref_filter_atom(cstr_name, cstr_name + strlen(cstr_name));
1113        return sorting;
1114}
1115
1116int parse_opt_ref_sorting(const struct option *opt, const char *arg, int unset)
1117{
1118        struct ref_sorting **sorting_tail = opt->value;
1119        struct ref_sorting *s;
1120        int len;
1121
1122        if (!arg) /* should --no-sort void the list ? */
1123                return -1;
1124
1125        s = xcalloc(1, sizeof(*s));
1126        s->next = *sorting_tail;
1127        *sorting_tail = s;
1128
1129        if (*arg == '-') {
1130                s->reverse = 1;
1131                arg++;
1132        }
1133        len = strlen(arg);
1134        s->atom = parse_ref_filter_atom(arg, arg+len);
1135        return 0;
1136}