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