for-each-ref: introduce new structures for better organisation
[git/debian.git] / builtin / for-each-ref.c
blob822838a91198d71c2a08e29ce9ef8e04579c1f95
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 #include "color.h"
14 /* Quoting styles */
15 #define QUOTE_NONE 0
16 #define QUOTE_SHELL 1
17 #define QUOTE_PERL 2
18 #define QUOTE_PYTHON 4
19 #define QUOTE_TCL 8
21 typedef enum { FIELD_STR, FIELD_ULONG, FIELD_TIME } cmp_type;
23 struct atom_value {
24 const char *s;
25 unsigned long ul; /* used for sorting when not FIELD_STR */
28 struct ref_sort {
29 struct ref_sort *next;
30 int atom; /* index into used_atom array */
31 unsigned reverse : 1;
34 struct ref_array_item {
35 unsigned char objectname[20];
36 int flag;
37 const char *symref;
38 struct atom_value *value;
39 char *refname;
42 struct ref_array {
43 int nr, alloc;
44 struct ref_array_item **items;
47 struct ref_filter {
48 const char **name_patterns;
51 struct ref_filter_cbdata {
52 struct ref_array array;
53 struct ref_filter filter;
56 static struct {
57 const char *name;
58 cmp_type cmp_type;
59 } valid_atom[] = {
60 { "refname" },
61 { "objecttype" },
62 { "objectsize", FIELD_ULONG },
63 { "objectname" },
64 { "tree" },
65 { "parent" },
66 { "numparent", FIELD_ULONG },
67 { "object" },
68 { "type" },
69 { "tag" },
70 { "author" },
71 { "authorname" },
72 { "authoremail" },
73 { "authordate", FIELD_TIME },
74 { "committer" },
75 { "committername" },
76 { "committeremail" },
77 { "committerdate", FIELD_TIME },
78 { "tagger" },
79 { "taggername" },
80 { "taggeremail" },
81 { "taggerdate", FIELD_TIME },
82 { "creator" },
83 { "creatordate", FIELD_TIME },
84 { "subject" },
85 { "body" },
86 { "contents" },
87 { "contents:subject" },
88 { "contents:body" },
89 { "contents:signature" },
90 { "upstream" },
91 { "push" },
92 { "symref" },
93 { "flag" },
94 { "HEAD" },
95 { "color" },
99 * An atom is a valid field atom listed above, possibly prefixed with
100 * a "*" to denote deref_tag().
102 * We parse given format string and sort specifiers, and make a list
103 * of properties that we need to extract out of objects. ref_array_item
104 * structure will hold an array of values extracted that can be
105 * indexed with the "atom number", which is an index into this
106 * array.
108 static const char **used_atom;
109 static cmp_type *used_atom_type;
110 static int used_atom_cnt, need_tagged, need_symref;
111 static int need_color_reset_at_eol;
114 * Used to parse format string and sort specifiers
116 static int parse_atom(const char *atom, const char *ep)
118 const char *sp;
119 int i, at;
121 sp = atom;
122 if (*sp == '*' && sp < ep)
123 sp++; /* deref */
124 if (ep <= sp)
125 die("malformed field name: %.*s", (int)(ep-atom), atom);
127 /* Do we have the atom already used elsewhere? */
128 for (i = 0; i < used_atom_cnt; i++) {
129 int len = strlen(used_atom[i]);
130 if (len == ep - atom && !memcmp(used_atom[i], atom, len))
131 return i;
134 /* Is the atom a valid one? */
135 for (i = 0; i < ARRAY_SIZE(valid_atom); i++) {
136 int len = strlen(valid_atom[i].name);
138 * If the atom name has a colon, strip it and everything after
139 * it off - it specifies the format for this entry, and
140 * shouldn't be used for checking against the valid_atom
141 * table.
143 const char *formatp = strchr(sp, ':');
144 if (!formatp || ep < formatp)
145 formatp = ep;
146 if (len == formatp - sp && !memcmp(valid_atom[i].name, sp, len))
147 break;
150 if (ARRAY_SIZE(valid_atom) <= i)
151 die("unknown field name: %.*s", (int)(ep-atom), atom);
153 /* Add it in, including the deref prefix */
154 at = used_atom_cnt;
155 used_atom_cnt++;
156 REALLOC_ARRAY(used_atom, used_atom_cnt);
157 REALLOC_ARRAY(used_atom_type, used_atom_cnt);
158 used_atom[at] = xmemdupz(atom, ep - atom);
159 used_atom_type[at] = valid_atom[i].cmp_type;
160 if (*atom == '*')
161 need_tagged = 1;
162 if (!strcmp(used_atom[at], "symref"))
163 need_symref = 1;
164 return at;
168 * In a format string, find the next occurrence of %(atom).
170 static const char *find_next(const char *cp)
172 while (*cp) {
173 if (*cp == '%') {
175 * %( is the start of an atom;
176 * %% is a quoted per-cent.
178 if (cp[1] == '(')
179 return cp;
180 else if (cp[1] == '%')
181 cp++; /* skip over two % */
182 /* otherwise this is a singleton, literal % */
184 cp++;
186 return NULL;
190 * Make sure the format string is well formed, and parse out
191 * the used atoms.
193 static int verify_format(const char *format)
195 const char *cp, *sp;
197 need_color_reset_at_eol = 0;
198 for (cp = format; *cp && (sp = find_next(cp)); ) {
199 const char *color, *ep = strchr(sp, ')');
200 int at;
202 if (!ep)
203 return error("malformed format string %s", sp);
204 /* sp points at "%(" and ep points at the closing ")" */
205 at = parse_atom(sp + 2, ep);
206 cp = ep + 1;
208 if (skip_prefix(used_atom[at], "color:", &color))
209 need_color_reset_at_eol = !!strcmp(color, "reset");
211 return 0;
215 * Given an object name, read the object data and size, and return a
216 * "struct object". If the object data we are returning is also borrowed
217 * by the "struct object" representation, set *eaten as well---it is a
218 * signal from parse_object_buffer to us not to free the buffer.
220 static void *get_obj(const unsigned char *sha1, struct object **obj, unsigned long *sz, int *eaten)
222 enum object_type type;
223 void *buf = read_sha1_file(sha1, &type, sz);
225 if (buf)
226 *obj = parse_object_buffer(sha1, type, *sz, buf, eaten);
227 else
228 *obj = NULL;
229 return buf;
232 static int grab_objectname(const char *name, const unsigned char *sha1,
233 struct atom_value *v)
235 if (!strcmp(name, "objectname")) {
236 char *s = xmalloc(41);
237 strcpy(s, sha1_to_hex(sha1));
238 v->s = s;
239 return 1;
241 if (!strcmp(name, "objectname:short")) {
242 v->s = xstrdup(find_unique_abbrev(sha1, DEFAULT_ABBREV));
243 return 1;
245 return 0;
248 /* See grab_values */
249 static void grab_common_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
251 int i;
253 for (i = 0; i < used_atom_cnt; i++) {
254 const char *name = used_atom[i];
255 struct atom_value *v = &val[i];
256 if (!!deref != (*name == '*'))
257 continue;
258 if (deref)
259 name++;
260 if (!strcmp(name, "objecttype"))
261 v->s = typename(obj->type);
262 else if (!strcmp(name, "objectsize")) {
263 char *s = xmalloc(40);
264 sprintf(s, "%lu", sz);
265 v->ul = sz;
266 v->s = s;
268 else if (deref)
269 grab_objectname(name, obj->sha1, v);
273 /* See grab_values */
274 static void grab_tag_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
276 int i;
277 struct tag *tag = (struct tag *) obj;
279 for (i = 0; i < used_atom_cnt; i++) {
280 const char *name = used_atom[i];
281 struct atom_value *v = &val[i];
282 if (!!deref != (*name == '*'))
283 continue;
284 if (deref)
285 name++;
286 if (!strcmp(name, "tag"))
287 v->s = tag->tag;
288 else if (!strcmp(name, "type") && tag->tagged)
289 v->s = typename(tag->tagged->type);
290 else if (!strcmp(name, "object") && tag->tagged) {
291 char *s = xmalloc(41);
292 strcpy(s, sha1_to_hex(tag->tagged->sha1));
293 v->s = s;
298 /* See grab_values */
299 static void grab_commit_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
301 int i;
302 struct commit *commit = (struct commit *) obj;
304 for (i = 0; i < used_atom_cnt; i++) {
305 const char *name = used_atom[i];
306 struct atom_value *v = &val[i];
307 if (!!deref != (*name == '*'))
308 continue;
309 if (deref)
310 name++;
311 if (!strcmp(name, "tree")) {
312 char *s = xmalloc(41);
313 strcpy(s, sha1_to_hex(commit->tree->object.sha1));
314 v->s = s;
316 if (!strcmp(name, "numparent")) {
317 char *s = xmalloc(40);
318 v->ul = commit_list_count(commit->parents);
319 sprintf(s, "%lu", v->ul);
320 v->s = s;
322 else if (!strcmp(name, "parent")) {
323 int num = commit_list_count(commit->parents);
324 int i;
325 struct commit_list *parents;
326 char *s = xmalloc(41 * num + 1);
327 v->s = s;
328 for (i = 0, parents = commit->parents;
329 parents;
330 parents = parents->next, i = i + 41) {
331 struct commit *parent = parents->item;
332 strcpy(s+i, sha1_to_hex(parent->object.sha1));
333 if (parents->next)
334 s[i+40] = ' ';
336 if (!i)
337 *s = '\0';
342 static const char *find_wholine(const char *who, int wholen, const char *buf, unsigned long sz)
344 const char *eol;
345 while (*buf) {
346 if (!strncmp(buf, who, wholen) &&
347 buf[wholen] == ' ')
348 return buf + wholen + 1;
349 eol = strchr(buf, '\n');
350 if (!eol)
351 return "";
352 eol++;
353 if (*eol == '\n')
354 return ""; /* end of header */
355 buf = eol;
357 return "";
360 static const char *copy_line(const char *buf)
362 const char *eol = strchrnul(buf, '\n');
363 return xmemdupz(buf, eol - buf);
366 static const char *copy_name(const char *buf)
368 const char *cp;
369 for (cp = buf; *cp && *cp != '\n'; cp++) {
370 if (!strncmp(cp, " <", 2))
371 return xmemdupz(buf, cp - buf);
373 return "";
376 static const char *copy_email(const char *buf)
378 const char *email = strchr(buf, '<');
379 const char *eoemail;
380 if (!email)
381 return "";
382 eoemail = strchr(email, '>');
383 if (!eoemail)
384 return "";
385 return xmemdupz(email, eoemail + 1 - email);
388 static char *copy_subject(const char *buf, unsigned long len)
390 char *r = xmemdupz(buf, len);
391 int i;
393 for (i = 0; i < len; i++)
394 if (r[i] == '\n')
395 r[i] = ' ';
397 return r;
400 static void grab_date(const char *buf, struct atom_value *v, const char *atomname)
402 const char *eoemail = strstr(buf, "> ");
403 char *zone;
404 unsigned long timestamp;
405 long tz;
406 enum date_mode date_mode = DATE_NORMAL;
407 const char *formatp;
410 * We got here because atomname ends in "date" or "date<something>";
411 * it's not possible that <something> is not ":<format>" because
412 * parse_atom() wouldn't have allowed it, so we can assume that no
413 * ":" means no format is specified, and use the default.
415 formatp = strchr(atomname, ':');
416 if (formatp != NULL) {
417 formatp++;
418 date_mode = parse_date_format(formatp);
421 if (!eoemail)
422 goto bad;
423 timestamp = strtoul(eoemail + 2, &zone, 10);
424 if (timestamp == ULONG_MAX)
425 goto bad;
426 tz = strtol(zone, NULL, 10);
427 if ((tz == LONG_MIN || tz == LONG_MAX) && errno == ERANGE)
428 goto bad;
429 v->s = xstrdup(show_date(timestamp, tz, date_mode));
430 v->ul = timestamp;
431 return;
432 bad:
433 v->s = "";
434 v->ul = 0;
437 /* See grab_values */
438 static void grab_person(const char *who, struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
440 int i;
441 int wholen = strlen(who);
442 const char *wholine = NULL;
444 for (i = 0; i < used_atom_cnt; i++) {
445 const char *name = used_atom[i];
446 struct atom_value *v = &val[i];
447 if (!!deref != (*name == '*'))
448 continue;
449 if (deref)
450 name++;
451 if (strncmp(who, name, wholen))
452 continue;
453 if (name[wholen] != 0 &&
454 strcmp(name + wholen, "name") &&
455 strcmp(name + wholen, "email") &&
456 !starts_with(name + wholen, "date"))
457 continue;
458 if (!wholine)
459 wholine = find_wholine(who, wholen, buf, sz);
460 if (!wholine)
461 return; /* no point looking for it */
462 if (name[wholen] == 0)
463 v->s = copy_line(wholine);
464 else if (!strcmp(name + wholen, "name"))
465 v->s = copy_name(wholine);
466 else if (!strcmp(name + wholen, "email"))
467 v->s = copy_email(wholine);
468 else if (starts_with(name + wholen, "date"))
469 grab_date(wholine, v, name);
473 * For a tag or a commit object, if "creator" or "creatordate" is
474 * requested, do something special.
476 if (strcmp(who, "tagger") && strcmp(who, "committer"))
477 return; /* "author" for commit object is not wanted */
478 if (!wholine)
479 wholine = find_wholine(who, wholen, buf, sz);
480 if (!wholine)
481 return;
482 for (i = 0; i < used_atom_cnt; i++) {
483 const char *name = used_atom[i];
484 struct atom_value *v = &val[i];
485 if (!!deref != (*name == '*'))
486 continue;
487 if (deref)
488 name++;
490 if (starts_with(name, "creatordate"))
491 grab_date(wholine, v, name);
492 else if (!strcmp(name, "creator"))
493 v->s = copy_line(wholine);
497 static void find_subpos(const char *buf, unsigned long sz,
498 const char **sub, unsigned long *sublen,
499 const char **body, unsigned long *bodylen,
500 unsigned long *nonsiglen,
501 const char **sig, unsigned long *siglen)
503 const char *eol;
504 /* skip past header until we hit empty line */
505 while (*buf && *buf != '\n') {
506 eol = strchrnul(buf, '\n');
507 if (*eol)
508 eol++;
509 buf = eol;
511 /* skip any empty lines */
512 while (*buf == '\n')
513 buf++;
515 /* parse signature first; we might not even have a subject line */
516 *sig = buf + parse_signature(buf, strlen(buf));
517 *siglen = strlen(*sig);
519 /* subject is first non-empty line */
520 *sub = buf;
521 /* subject goes to first empty line */
522 while (buf < *sig && *buf && *buf != '\n') {
523 eol = strchrnul(buf, '\n');
524 if (*eol)
525 eol++;
526 buf = eol;
528 *sublen = buf - *sub;
529 /* drop trailing newline, if present */
530 if (*sublen && (*sub)[*sublen - 1] == '\n')
531 *sublen -= 1;
533 /* skip any empty lines */
534 while (*buf == '\n')
535 buf++;
536 *body = buf;
537 *bodylen = strlen(buf);
538 *nonsiglen = *sig - buf;
541 /* See grab_values */
542 static void grab_sub_body_contents(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
544 int i;
545 const char *subpos = NULL, *bodypos = NULL, *sigpos = NULL;
546 unsigned long sublen = 0, bodylen = 0, nonsiglen = 0, siglen = 0;
548 for (i = 0; i < used_atom_cnt; i++) {
549 const char *name = used_atom[i];
550 struct atom_value *v = &val[i];
551 if (!!deref != (*name == '*'))
552 continue;
553 if (deref)
554 name++;
555 if (strcmp(name, "subject") &&
556 strcmp(name, "body") &&
557 strcmp(name, "contents") &&
558 strcmp(name, "contents:subject") &&
559 strcmp(name, "contents:body") &&
560 strcmp(name, "contents:signature"))
561 continue;
562 if (!subpos)
563 find_subpos(buf, sz,
564 &subpos, &sublen,
565 &bodypos, &bodylen, &nonsiglen,
566 &sigpos, &siglen);
568 if (!strcmp(name, "subject"))
569 v->s = copy_subject(subpos, sublen);
570 else if (!strcmp(name, "contents:subject"))
571 v->s = copy_subject(subpos, sublen);
572 else if (!strcmp(name, "body"))
573 v->s = xmemdupz(bodypos, bodylen);
574 else if (!strcmp(name, "contents:body"))
575 v->s = xmemdupz(bodypos, nonsiglen);
576 else if (!strcmp(name, "contents:signature"))
577 v->s = xmemdupz(sigpos, siglen);
578 else if (!strcmp(name, "contents"))
579 v->s = xstrdup(subpos);
584 * We want to have empty print-string for field requests
585 * that do not apply (e.g. "authordate" for a tag object)
587 static void fill_missing_values(struct atom_value *val)
589 int i;
590 for (i = 0; i < used_atom_cnt; i++) {
591 struct atom_value *v = &val[i];
592 if (v->s == NULL)
593 v->s = "";
598 * val is a list of atom_value to hold returned values. Extract
599 * the values for atoms in used_atom array out of (obj, buf, sz).
600 * when deref is false, (obj, buf, sz) is the object that is
601 * pointed at by the ref itself; otherwise it is the object the
602 * ref (which is a tag) refers to.
604 static void grab_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
606 grab_common_values(val, deref, obj, buf, sz);
607 switch (obj->type) {
608 case OBJ_TAG:
609 grab_tag_values(val, deref, obj, buf, sz);
610 grab_sub_body_contents(val, deref, obj, buf, sz);
611 grab_person("tagger", val, deref, obj, buf, sz);
612 break;
613 case OBJ_COMMIT:
614 grab_commit_values(val, deref, obj, buf, sz);
615 grab_sub_body_contents(val, deref, obj, buf, sz);
616 grab_person("author", val, deref, obj, buf, sz);
617 grab_person("committer", val, deref, obj, buf, sz);
618 break;
619 case OBJ_TREE:
620 /* grab_tree_values(val, deref, obj, buf, sz); */
621 break;
622 case OBJ_BLOB:
623 /* grab_blob_values(val, deref, obj, buf, sz); */
624 break;
625 default:
626 die("Eh? Object of type %d?", obj->type);
630 static inline char *copy_advance(char *dst, const char *src)
632 while (*src)
633 *dst++ = *src++;
634 return dst;
638 * Parse the object referred by ref, and grab needed value.
640 static void populate_value(struct ref_array_item *ref)
642 void *buf;
643 struct object *obj;
644 int eaten, i;
645 unsigned long size;
646 const unsigned char *tagged;
648 ref->value = xcalloc(used_atom_cnt, sizeof(struct atom_value));
650 if (need_symref && (ref->flag & REF_ISSYMREF) && !ref->symref) {
651 unsigned char unused1[20];
652 ref->symref = resolve_refdup(ref->refname, RESOLVE_REF_READING,
653 unused1, NULL);
654 if (!ref->symref)
655 ref->symref = "";
658 /* Fill in specials first */
659 for (i = 0; i < used_atom_cnt; i++) {
660 const char *name = used_atom[i];
661 struct atom_value *v = &ref->value[i];
662 int deref = 0;
663 const char *refname;
664 const char *formatp;
665 struct branch *branch = NULL;
667 if (*name == '*') {
668 deref = 1;
669 name++;
672 if (starts_with(name, "refname"))
673 refname = ref->refname;
674 else if (starts_with(name, "symref"))
675 refname = ref->symref ? ref->symref : "";
676 else if (starts_with(name, "upstream")) {
677 const char *branch_name;
678 /* only local branches may have an upstream */
679 if (!skip_prefix(ref->refname, "refs/heads/",
680 &branch_name))
681 continue;
682 branch = branch_get(branch_name);
684 refname = branch_get_upstream(branch, NULL);
685 if (!refname)
686 continue;
687 } else if (starts_with(name, "push")) {
688 const char *branch_name;
689 if (!skip_prefix(ref->refname, "refs/heads/",
690 &branch_name))
691 continue;
692 branch = branch_get(branch_name);
694 refname = branch_get_push(branch, NULL);
695 if (!refname)
696 continue;
697 } else if (starts_with(name, "color:")) {
698 char color[COLOR_MAXLEN] = "";
700 if (color_parse(name + 6, color) < 0)
701 die(_("unable to parse format"));
702 v->s = xstrdup(color);
703 continue;
704 } else if (!strcmp(name, "flag")) {
705 char buf[256], *cp = buf;
706 if (ref->flag & REF_ISSYMREF)
707 cp = copy_advance(cp, ",symref");
708 if (ref->flag & REF_ISPACKED)
709 cp = copy_advance(cp, ",packed");
710 if (cp == buf)
711 v->s = "";
712 else {
713 *cp = '\0';
714 v->s = xstrdup(buf + 1);
716 continue;
717 } else if (!deref && grab_objectname(name, ref->objectname, v)) {
718 continue;
719 } else if (!strcmp(name, "HEAD")) {
720 const char *head;
721 unsigned char sha1[20];
723 head = resolve_ref_unsafe("HEAD", RESOLVE_REF_READING,
724 sha1, NULL);
725 if (!strcmp(ref->refname, head))
726 v->s = "*";
727 else
728 v->s = " ";
729 continue;
730 } else
731 continue;
733 formatp = strchr(name, ':');
734 if (formatp) {
735 int num_ours, num_theirs;
737 formatp++;
738 if (!strcmp(formatp, "short"))
739 refname = shorten_unambiguous_ref(refname,
740 warn_ambiguous_refs);
741 else if (!strcmp(formatp, "track") &&
742 (starts_with(name, "upstream") ||
743 starts_with(name, "push"))) {
744 char buf[40];
746 if (stat_tracking_info(branch, &num_ours,
747 &num_theirs, NULL))
748 continue;
750 if (!num_ours && !num_theirs)
751 v->s = "";
752 else if (!num_ours) {
753 sprintf(buf, "[behind %d]", num_theirs);
754 v->s = xstrdup(buf);
755 } else if (!num_theirs) {
756 sprintf(buf, "[ahead %d]", num_ours);
757 v->s = xstrdup(buf);
758 } else {
759 sprintf(buf, "[ahead %d, behind %d]",
760 num_ours, num_theirs);
761 v->s = xstrdup(buf);
763 continue;
764 } else if (!strcmp(formatp, "trackshort") &&
765 (starts_with(name, "upstream") ||
766 starts_with(name, "push"))) {
767 assert(branch);
769 if (stat_tracking_info(branch, &num_ours,
770 &num_theirs, NULL))
771 continue;
773 if (!num_ours && !num_theirs)
774 v->s = "=";
775 else if (!num_ours)
776 v->s = "<";
777 else if (!num_theirs)
778 v->s = ">";
779 else
780 v->s = "<>";
781 continue;
782 } else
783 die("unknown %.*s format %s",
784 (int)(formatp - name), name, formatp);
787 if (!deref)
788 v->s = refname;
789 else {
790 int len = strlen(refname);
791 char *s = xmalloc(len + 4);
792 sprintf(s, "%s^{}", refname);
793 v->s = s;
797 for (i = 0; i < used_atom_cnt; i++) {
798 struct atom_value *v = &ref->value[i];
799 if (v->s == NULL)
800 goto need_obj;
802 return;
804 need_obj:
805 buf = get_obj(ref->objectname, &obj, &size, &eaten);
806 if (!buf)
807 die("missing object %s for %s",
808 sha1_to_hex(ref->objectname), ref->refname);
809 if (!obj)
810 die("parse_object_buffer failed on %s for %s",
811 sha1_to_hex(ref->objectname), ref->refname);
813 grab_values(ref->value, 0, obj, buf, size);
814 if (!eaten)
815 free(buf);
818 * If there is no atom that wants to know about tagged
819 * object, we are done.
821 if (!need_tagged || (obj->type != OBJ_TAG))
822 return;
825 * If it is a tag object, see if we use a value that derefs
826 * the object, and if we do grab the object it refers to.
828 tagged = ((struct tag *)obj)->tagged->sha1;
831 * NEEDSWORK: This derefs tag only once, which
832 * is good to deal with chains of trust, but
833 * is not consistent with what deref_tag() does
834 * which peels the onion to the core.
836 buf = get_obj(tagged, &obj, &size, &eaten);
837 if (!buf)
838 die("missing object %s for %s",
839 sha1_to_hex(tagged), ref->refname);
840 if (!obj)
841 die("parse_object_buffer failed on %s for %s",
842 sha1_to_hex(tagged), ref->refname);
843 grab_values(ref->value, 1, obj, buf, size);
844 if (!eaten)
845 free(buf);
849 * Given a ref, return the value for the atom. This lazily gets value
850 * out of the object by calling populate value.
852 static void get_value(struct ref_array_item *ref, int atom, struct atom_value **v)
854 if (!ref->value) {
855 populate_value(ref);
856 fill_missing_values(ref->value);
858 *v = &ref->value[atom];
862 * Return 1 if the refname matches one of the patterns, otherwise 0.
863 * A pattern can be path prefix (e.g. a refname "refs/heads/master"
864 * matches a pattern "refs/heads/") or a wildcard (e.g. the same ref
865 * matches "refs/heads/m*",too).
867 static int match_name_as_path(const char **pattern, const char *refname)
869 int namelen = strlen(refname);
870 for (; *pattern; pattern++) {
871 const char *p = *pattern;
872 int plen = strlen(p);
874 if ((plen <= namelen) &&
875 !strncmp(refname, p, plen) &&
876 (refname[plen] == '\0' ||
877 refname[plen] == '/' ||
878 p[plen-1] == '/'))
879 return 1;
880 if (!wildmatch(p, refname, WM_PATHNAME, NULL))
881 return 1;
883 return 0;
886 /* Allocate space for a new ref_array_item and copy the objectname and flag to it */
887 static struct ref_array_item *new_ref_array_item(const char *refname,
888 const unsigned char *objectname,
889 int flag)
891 struct ref_array_item *ref = xcalloc(1, sizeof(struct ref_array_item));
892 ref->refname = xstrdup(refname);
893 hashcpy(ref->objectname, objectname);
894 ref->flag = flag;
896 return ref;
900 * A call-back given to for_each_ref(). Filter refs and keep them for
901 * later object processing.
903 static int grab_single_ref(const char *refname, const struct object_id *oid,
904 int flag, void *cb_data)
906 struct ref_filter_cbdata *ref_cbdata = cb_data;
907 struct ref_filter *filter = &ref_cbdata->filter;
908 struct ref_array_item *ref;
910 if (flag & REF_BAD_NAME) {
911 warning("ignoring ref with broken name %s", refname);
912 return 0;
915 if (*filter->name_patterns && !match_name_as_path(filter->name_patterns, refname))
916 return 0;
919 * We do not open the object yet; sort may only need refname
920 * to do its job and the resulting list may yet to be pruned
921 * by maxcount logic.
923 ref = new_ref_array_item(refname, oid->hash, flag);
925 REALLOC_ARRAY(ref_cbdata->array.items, ref_cbdata->array.nr + 1);
926 ref_cbdata->array.items[ref_cbdata->array.nr++] = ref;
927 return 0;
930 static int cmp_ref_sort(struct ref_sort *s, struct ref_array_item *a, struct ref_array_item *b)
932 struct atom_value *va, *vb;
933 int cmp;
934 cmp_type cmp_type = used_atom_type[s->atom];
936 get_value(a, s->atom, &va);
937 get_value(b, s->atom, &vb);
938 switch (cmp_type) {
939 case FIELD_STR:
940 cmp = strcmp(va->s, vb->s);
941 break;
942 default:
943 if (va->ul < vb->ul)
944 cmp = -1;
945 else if (va->ul == vb->ul)
946 cmp = 0;
947 else
948 cmp = 1;
949 break;
951 return (s->reverse) ? -cmp : cmp;
954 static struct ref_sort *ref_sort;
955 static int compare_refs(const void *a_, const void *b_)
957 struct ref_array_item *a = *((struct ref_array_item **)a_);
958 struct ref_array_item *b = *((struct ref_array_item **)b_);
959 struct ref_sort *s;
961 for (s = ref_sort; s; s = s->next) {
962 int cmp = cmp_ref_sort(s, a, b);
963 if (cmp)
964 return cmp;
966 return 0;
969 static void sort_refs(struct ref_sort *sort, struct ref_array *array)
971 ref_sort = sort;
972 qsort(array->items, array->nr, sizeof(struct ref_array_item *), compare_refs);
975 static void print_value(struct atom_value *v, int quote_style)
977 struct strbuf sb = STRBUF_INIT;
978 switch (quote_style) {
979 case QUOTE_NONE:
980 fputs(v->s, stdout);
981 break;
982 case QUOTE_SHELL:
983 sq_quote_buf(&sb, v->s);
984 break;
985 case QUOTE_PERL:
986 perl_quote_buf(&sb, v->s);
987 break;
988 case QUOTE_PYTHON:
989 python_quote_buf(&sb, v->s);
990 break;
991 case QUOTE_TCL:
992 tcl_quote_buf(&sb, v->s);
993 break;
995 if (quote_style != QUOTE_NONE) {
996 fputs(sb.buf, stdout);
997 strbuf_release(&sb);
1001 static int hex1(char ch)
1003 if ('0' <= ch && ch <= '9')
1004 return ch - '0';
1005 else if ('a' <= ch && ch <= 'f')
1006 return ch - 'a' + 10;
1007 else if ('A' <= ch && ch <= 'F')
1008 return ch - 'A' + 10;
1009 return -1;
1011 static int hex2(const char *cp)
1013 if (cp[0] && cp[1])
1014 return (hex1(cp[0]) << 4) | hex1(cp[1]);
1015 else
1016 return -1;
1019 static void emit(const char *cp, const char *ep)
1021 while (*cp && (!ep || cp < ep)) {
1022 if (*cp == '%') {
1023 if (cp[1] == '%')
1024 cp++;
1025 else {
1026 int ch = hex2(cp + 1);
1027 if (0 <= ch) {
1028 putchar(ch);
1029 cp += 3;
1030 continue;
1034 putchar(*cp);
1035 cp++;
1039 static void show_ref(struct ref_array_item *info, const char *format, int quote_style)
1041 const char *cp, *sp, *ep;
1043 for (cp = format; *cp && (sp = find_next(cp)); cp = ep + 1) {
1044 struct atom_value *atomv;
1046 ep = strchr(sp, ')');
1047 if (cp < sp)
1048 emit(cp, sp);
1049 get_value(info, parse_atom(sp + 2, ep), &atomv);
1050 print_value(atomv, quote_style);
1052 if (*cp) {
1053 sp = cp + strlen(cp);
1054 emit(cp, sp);
1056 if (need_color_reset_at_eol) {
1057 struct atom_value resetv;
1058 char color[COLOR_MAXLEN] = "";
1060 if (color_parse("reset", color) < 0)
1061 die("BUG: couldn't parse 'reset' as a color");
1062 resetv.s = color;
1063 print_value(&resetv, quote_style);
1065 putchar('\n');
1068 static struct ref_sort *default_sort(void)
1070 static const char cstr_name[] = "refname";
1072 struct ref_sort *sort = xcalloc(1, sizeof(*sort));
1074 sort->next = NULL;
1075 sort->atom = parse_atom(cstr_name, cstr_name + strlen(cstr_name));
1076 return sort;
1079 static int opt_parse_sort(const struct option *opt, const char *arg, int unset)
1081 struct ref_sort **sort_tail = opt->value;
1082 struct ref_sort *s;
1083 int len;
1085 if (!arg) /* should --no-sort void the list ? */
1086 return -1;
1088 s = xcalloc(1, sizeof(*s));
1089 s->next = *sort_tail;
1090 *sort_tail = s;
1092 if (*arg == '-') {
1093 s->reverse = 1;
1094 arg++;
1096 len = strlen(arg);
1097 s->atom = parse_atom(arg, arg+len);
1098 return 0;
1101 static char const * const for_each_ref_usage[] = {
1102 N_("git for-each-ref [<options>] [<pattern>]"),
1103 NULL
1106 int cmd_for_each_ref(int argc, const char **argv, const char *prefix)
1108 int i;
1109 const char *format = "%(objectname) %(objecttype)\t%(refname)";
1110 struct ref_sort *sort = NULL, **sort_tail = &sort;
1111 int maxcount = 0, quote_style = 0;
1112 struct ref_filter_cbdata ref_cbdata;
1114 struct option opts[] = {
1115 OPT_BIT('s', "shell", &quote_style,
1116 N_("quote placeholders suitably for shells"), QUOTE_SHELL),
1117 OPT_BIT('p', "perl", &quote_style,
1118 N_("quote placeholders suitably for perl"), QUOTE_PERL),
1119 OPT_BIT(0 , "python", &quote_style,
1120 N_("quote placeholders suitably for python"), QUOTE_PYTHON),
1121 OPT_BIT(0 , "tcl", &quote_style,
1122 N_("quote placeholders suitably for Tcl"), QUOTE_TCL),
1124 OPT_GROUP(""),
1125 OPT_INTEGER( 0 , "count", &maxcount, N_("show only <n> matched refs")),
1126 OPT_STRING( 0 , "format", &format, N_("format"), N_("format to use for the output")),
1127 OPT_CALLBACK(0 , "sort", sort_tail, N_("key"),
1128 N_("field name to sort on"), &opt_parse_sort),
1129 OPT_END(),
1132 parse_options(argc, argv, prefix, opts, for_each_ref_usage, 0);
1133 if (maxcount < 0) {
1134 error("invalid --count argument: `%d'", maxcount);
1135 usage_with_options(for_each_ref_usage, opts);
1137 if (HAS_MULTI_BITS(quote_style)) {
1138 error("more than one quoting style?");
1139 usage_with_options(for_each_ref_usage, opts);
1141 if (verify_format(format))
1142 usage_with_options(for_each_ref_usage, opts);
1144 if (!sort)
1145 sort = default_sort();
1147 /* for warn_ambiguous_refs */
1148 git_config(git_default_config, NULL);
1150 memset(&ref_cbdata, 0, sizeof(ref_cbdata));
1151 ref_cbdata.filter.name_patterns = argv;
1152 for_each_rawref(grab_single_ref, &ref_cbdata);
1154 sort_refs(sort, &ref_cbdata.array);
1156 if (!maxcount || ref_cbdata.array.nr < maxcount)
1157 maxcount = ref_cbdata.array.nr;
1158 for (i = 0; i < maxcount; i++)
1159 show_ref(ref_cbdata.array.items[i], format, quote_style);
1160 return 0;