refs.c: ref_transaction_commit: distinguish name conflicts from other errors
[git.git] / builtin / for-each-ref.c
blobfda0f047125f16f080288c1cda76b0e337784a54
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 refinfo {
35 char *refname;
36 unsigned char objectname[20];
37 int flag;
38 const char *symref;
39 struct atom_value *value;
42 static struct {
43 const char *name;
44 cmp_type cmp_type;
45 } valid_atom[] = {
46 { "refname" },
47 { "objecttype" },
48 { "objectsize", FIELD_ULONG },
49 { "objectname" },
50 { "tree" },
51 { "parent" },
52 { "numparent", FIELD_ULONG },
53 { "object" },
54 { "type" },
55 { "tag" },
56 { "author" },
57 { "authorname" },
58 { "authoremail" },
59 { "authordate", FIELD_TIME },
60 { "committer" },
61 { "committername" },
62 { "committeremail" },
63 { "committerdate", FIELD_TIME },
64 { "tagger" },
65 { "taggername" },
66 { "taggeremail" },
67 { "taggerdate", FIELD_TIME },
68 { "creator" },
69 { "creatordate", FIELD_TIME },
70 { "subject" },
71 { "body" },
72 { "contents" },
73 { "contents:subject" },
74 { "contents:body" },
75 { "contents:signature" },
76 { "upstream" },
77 { "symref" },
78 { "flag" },
79 { "HEAD" },
80 { "color" },
84 * An atom is a valid field atom listed above, possibly prefixed with
85 * a "*" to denote deref_tag().
87 * We parse given format string and sort specifiers, and make a list
88 * of properties that we need to extract out of objects. refinfo
89 * structure will hold an array of values extracted that can be
90 * indexed with the "atom number", which is an index into this
91 * array.
93 static const char **used_atom;
94 static cmp_type *used_atom_type;
95 static int used_atom_cnt, need_tagged, need_symref;
96 static int need_color_reset_at_eol;
99 * Used to parse format string and sort specifiers
101 static int parse_atom(const char *atom, const char *ep)
103 const char *sp;
104 int i, at;
106 sp = atom;
107 if (*sp == '*' && sp < ep)
108 sp++; /* deref */
109 if (ep <= sp)
110 die("malformed field name: %.*s", (int)(ep-atom), atom);
112 /* Do we have the atom already used elsewhere? */
113 for (i = 0; i < used_atom_cnt; i++) {
114 int len = strlen(used_atom[i]);
115 if (len == ep - atom && !memcmp(used_atom[i], atom, len))
116 return i;
119 /* Is the atom a valid one? */
120 for (i = 0; i < ARRAY_SIZE(valid_atom); i++) {
121 int len = strlen(valid_atom[i].name);
123 * If the atom name has a colon, strip it and everything after
124 * it off - it specifies the format for this entry, and
125 * shouldn't be used for checking against the valid_atom
126 * table.
128 const char *formatp = strchr(sp, ':');
129 if (!formatp || ep < formatp)
130 formatp = ep;
131 if (len == formatp - sp && !memcmp(valid_atom[i].name, sp, len))
132 break;
135 if (ARRAY_SIZE(valid_atom) <= i)
136 die("unknown field name: %.*s", (int)(ep-atom), atom);
138 /* Add it in, including the deref prefix */
139 at = used_atom_cnt;
140 used_atom_cnt++;
141 REALLOC_ARRAY(used_atom, used_atom_cnt);
142 REALLOC_ARRAY(used_atom_type, used_atom_cnt);
143 used_atom[at] = xmemdupz(atom, ep - atom);
144 used_atom_type[at] = valid_atom[i].cmp_type;
145 if (*atom == '*')
146 need_tagged = 1;
147 if (!strcmp(used_atom[at], "symref"))
148 need_symref = 1;
149 return at;
153 * In a format string, find the next occurrence of %(atom).
155 static const char *find_next(const char *cp)
157 while (*cp) {
158 if (*cp == '%') {
160 * %( is the start of an atom;
161 * %% is a quoted per-cent.
163 if (cp[1] == '(')
164 return cp;
165 else if (cp[1] == '%')
166 cp++; /* skip over two % */
167 /* otherwise this is a singleton, literal % */
169 cp++;
171 return NULL;
175 * Make sure the format string is well formed, and parse out
176 * the used atoms.
178 static int verify_format(const char *format)
180 const char *cp, *sp;
181 static const char color_reset[] = "color:reset";
183 need_color_reset_at_eol = 0;
184 for (cp = format; *cp && (sp = find_next(cp)); ) {
185 const char *ep = strchr(sp, ')');
186 int at;
188 if (!ep)
189 return error("malformed format string %s", sp);
190 /* sp points at "%(" and ep points at the closing ")" */
191 at = parse_atom(sp + 2, ep);
192 cp = ep + 1;
194 if (starts_with(used_atom[at], "color:"))
195 need_color_reset_at_eol = !!strcmp(used_atom[at], color_reset);
197 return 0;
201 * Given an object name, read the object data and size, and return a
202 * "struct object". If the object data we are returning is also borrowed
203 * by the "struct object" representation, set *eaten as well---it is a
204 * signal from parse_object_buffer to us not to free the buffer.
206 static void *get_obj(const unsigned char *sha1, struct object **obj, unsigned long *sz, int *eaten)
208 enum object_type type;
209 void *buf = read_sha1_file(sha1, &type, sz);
211 if (buf)
212 *obj = parse_object_buffer(sha1, type, *sz, buf, eaten);
213 else
214 *obj = NULL;
215 return buf;
218 static int grab_objectname(const char *name, const unsigned char *sha1,
219 struct atom_value *v)
221 if (!strcmp(name, "objectname")) {
222 char *s = xmalloc(41);
223 strcpy(s, sha1_to_hex(sha1));
224 v->s = s;
225 return 1;
227 if (!strcmp(name, "objectname:short")) {
228 v->s = xstrdup(find_unique_abbrev(sha1, DEFAULT_ABBREV));
229 return 1;
231 return 0;
234 /* See grab_values */
235 static void grab_common_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
237 int i;
239 for (i = 0; i < used_atom_cnt; i++) {
240 const char *name = used_atom[i];
241 struct atom_value *v = &val[i];
242 if (!!deref != (*name == '*'))
243 continue;
244 if (deref)
245 name++;
246 if (!strcmp(name, "objecttype"))
247 v->s = typename(obj->type);
248 else if (!strcmp(name, "objectsize")) {
249 char *s = xmalloc(40);
250 sprintf(s, "%lu", sz);
251 v->ul = sz;
252 v->s = s;
254 else if (deref)
255 grab_objectname(name, obj->sha1, v);
259 /* See grab_values */
260 static void grab_tag_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
262 int i;
263 struct tag *tag = (struct tag *) obj;
265 for (i = 0; i < used_atom_cnt; i++) {
266 const char *name = used_atom[i];
267 struct atom_value *v = &val[i];
268 if (!!deref != (*name == '*'))
269 continue;
270 if (deref)
271 name++;
272 if (!strcmp(name, "tag"))
273 v->s = tag->tag;
274 else if (!strcmp(name, "type") && tag->tagged)
275 v->s = typename(tag->tagged->type);
276 else if (!strcmp(name, "object") && tag->tagged) {
277 char *s = xmalloc(41);
278 strcpy(s, sha1_to_hex(tag->tagged->sha1));
279 v->s = s;
284 /* See grab_values */
285 static void grab_commit_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
287 int i;
288 struct commit *commit = (struct commit *) obj;
290 for (i = 0; i < used_atom_cnt; i++) {
291 const char *name = used_atom[i];
292 struct atom_value *v = &val[i];
293 if (!!deref != (*name == '*'))
294 continue;
295 if (deref)
296 name++;
297 if (!strcmp(name, "tree")) {
298 char *s = xmalloc(41);
299 strcpy(s, sha1_to_hex(commit->tree->object.sha1));
300 v->s = s;
302 if (!strcmp(name, "numparent")) {
303 char *s = xmalloc(40);
304 v->ul = commit_list_count(commit->parents);
305 sprintf(s, "%lu", v->ul);
306 v->s = s;
308 else if (!strcmp(name, "parent")) {
309 int num = commit_list_count(commit->parents);
310 int i;
311 struct commit_list *parents;
312 char *s = xmalloc(41 * num + 1);
313 v->s = s;
314 for (i = 0, parents = commit->parents;
315 parents;
316 parents = parents->next, i = i + 41) {
317 struct commit *parent = parents->item;
318 strcpy(s+i, sha1_to_hex(parent->object.sha1));
319 if (parents->next)
320 s[i+40] = ' ';
322 if (!i)
323 *s = '\0';
328 static const char *find_wholine(const char *who, int wholen, const char *buf, unsigned long sz)
330 const char *eol;
331 while (*buf) {
332 if (!strncmp(buf, who, wholen) &&
333 buf[wholen] == ' ')
334 return buf + wholen + 1;
335 eol = strchr(buf, '\n');
336 if (!eol)
337 return "";
338 eol++;
339 if (*eol == '\n')
340 return ""; /* end of header */
341 buf = eol;
343 return "";
346 static const char *copy_line(const char *buf)
348 const char *eol = strchrnul(buf, '\n');
349 return xmemdupz(buf, eol - buf);
352 static const char *copy_name(const char *buf)
354 const char *cp;
355 for (cp = buf; *cp && *cp != '\n'; cp++) {
356 if (!strncmp(cp, " <", 2))
357 return xmemdupz(buf, cp - buf);
359 return "";
362 static const char *copy_email(const char *buf)
364 const char *email = strchr(buf, '<');
365 const char *eoemail;
366 if (!email)
367 return "";
368 eoemail = strchr(email, '>');
369 if (!eoemail)
370 return "";
371 return xmemdupz(email, eoemail + 1 - email);
374 static char *copy_subject(const char *buf, unsigned long len)
376 char *r = xmemdupz(buf, len);
377 int i;
379 for (i = 0; i < len; i++)
380 if (r[i] == '\n')
381 r[i] = ' ';
383 return r;
386 static void grab_date(const char *buf, struct atom_value *v, const char *atomname)
388 const char *eoemail = strstr(buf, "> ");
389 char *zone;
390 unsigned long timestamp;
391 long tz;
392 enum date_mode date_mode = DATE_NORMAL;
393 const char *formatp;
396 * We got here because atomname ends in "date" or "date<something>";
397 * it's not possible that <something> is not ":<format>" because
398 * parse_atom() wouldn't have allowed it, so we can assume that no
399 * ":" means no format is specified, and use the default.
401 formatp = strchr(atomname, ':');
402 if (formatp != NULL) {
403 formatp++;
404 date_mode = parse_date_format(formatp);
407 if (!eoemail)
408 goto bad;
409 timestamp = strtoul(eoemail + 2, &zone, 10);
410 if (timestamp == ULONG_MAX)
411 goto bad;
412 tz = strtol(zone, NULL, 10);
413 if ((tz == LONG_MIN || tz == LONG_MAX) && errno == ERANGE)
414 goto bad;
415 v->s = xstrdup(show_date(timestamp, tz, date_mode));
416 v->ul = timestamp;
417 return;
418 bad:
419 v->s = "";
420 v->ul = 0;
423 /* See grab_values */
424 static void grab_person(const char *who, struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
426 int i;
427 int wholen = strlen(who);
428 const char *wholine = NULL;
430 for (i = 0; i < used_atom_cnt; i++) {
431 const char *name = used_atom[i];
432 struct atom_value *v = &val[i];
433 if (!!deref != (*name == '*'))
434 continue;
435 if (deref)
436 name++;
437 if (strncmp(who, name, wholen))
438 continue;
439 if (name[wholen] != 0 &&
440 strcmp(name + wholen, "name") &&
441 strcmp(name + wholen, "email") &&
442 !starts_with(name + wholen, "date"))
443 continue;
444 if (!wholine)
445 wholine = find_wholine(who, wholen, buf, sz);
446 if (!wholine)
447 return; /* no point looking for it */
448 if (name[wholen] == 0)
449 v->s = copy_line(wholine);
450 else if (!strcmp(name + wholen, "name"))
451 v->s = copy_name(wholine);
452 else if (!strcmp(name + wholen, "email"))
453 v->s = copy_email(wholine);
454 else if (starts_with(name + wholen, "date"))
455 grab_date(wholine, v, name);
459 * For a tag or a commit object, if "creator" or "creatordate" is
460 * requested, do something special.
462 if (strcmp(who, "tagger") && strcmp(who, "committer"))
463 return; /* "author" for commit object is not wanted */
464 if (!wholine)
465 wholine = find_wholine(who, wholen, buf, sz);
466 if (!wholine)
467 return;
468 for (i = 0; i < used_atom_cnt; i++) {
469 const char *name = used_atom[i];
470 struct atom_value *v = &val[i];
471 if (!!deref != (*name == '*'))
472 continue;
473 if (deref)
474 name++;
476 if (starts_with(name, "creatordate"))
477 grab_date(wholine, v, name);
478 else if (!strcmp(name, "creator"))
479 v->s = copy_line(wholine);
483 static void find_subpos(const char *buf, unsigned long sz,
484 const char **sub, unsigned long *sublen,
485 const char **body, unsigned long *bodylen,
486 unsigned long *nonsiglen,
487 const char **sig, unsigned long *siglen)
489 const char *eol;
490 /* skip past header until we hit empty line */
491 while (*buf && *buf != '\n') {
492 eol = strchrnul(buf, '\n');
493 if (*eol)
494 eol++;
495 buf = eol;
497 /* skip any empty lines */
498 while (*buf == '\n')
499 buf++;
501 /* parse signature first; we might not even have a subject line */
502 *sig = buf + parse_signature(buf, strlen(buf));
503 *siglen = strlen(*sig);
505 /* subject is first non-empty line */
506 *sub = buf;
507 /* subject goes to first empty line */
508 while (buf < *sig && *buf && *buf != '\n') {
509 eol = strchrnul(buf, '\n');
510 if (*eol)
511 eol++;
512 buf = eol;
514 *sublen = buf - *sub;
515 /* drop trailing newline, if present */
516 if (*sublen && (*sub)[*sublen - 1] == '\n')
517 *sublen -= 1;
519 /* skip any empty lines */
520 while (*buf == '\n')
521 buf++;
522 *body = buf;
523 *bodylen = strlen(buf);
524 *nonsiglen = *sig - buf;
527 /* See grab_values */
528 static void grab_sub_body_contents(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
530 int i;
531 const char *subpos = NULL, *bodypos = NULL, *sigpos = NULL;
532 unsigned long sublen = 0, bodylen = 0, nonsiglen = 0, siglen = 0;
534 for (i = 0; i < used_atom_cnt; i++) {
535 const char *name = used_atom[i];
536 struct atom_value *v = &val[i];
537 if (!!deref != (*name == '*'))
538 continue;
539 if (deref)
540 name++;
541 if (strcmp(name, "subject") &&
542 strcmp(name, "body") &&
543 strcmp(name, "contents") &&
544 strcmp(name, "contents:subject") &&
545 strcmp(name, "contents:body") &&
546 strcmp(name, "contents:signature"))
547 continue;
548 if (!subpos)
549 find_subpos(buf, sz,
550 &subpos, &sublen,
551 &bodypos, &bodylen, &nonsiglen,
552 &sigpos, &siglen);
554 if (!strcmp(name, "subject"))
555 v->s = copy_subject(subpos, sublen);
556 else if (!strcmp(name, "contents:subject"))
557 v->s = copy_subject(subpos, sublen);
558 else if (!strcmp(name, "body"))
559 v->s = xmemdupz(bodypos, bodylen);
560 else if (!strcmp(name, "contents:body"))
561 v->s = xmemdupz(bodypos, nonsiglen);
562 else if (!strcmp(name, "contents:signature"))
563 v->s = xmemdupz(sigpos, siglen);
564 else if (!strcmp(name, "contents"))
565 v->s = xstrdup(subpos);
570 * We want to have empty print-string for field requests
571 * that do not apply (e.g. "authordate" for a tag object)
573 static void fill_missing_values(struct atom_value *val)
575 int i;
576 for (i = 0; i < used_atom_cnt; i++) {
577 struct atom_value *v = &val[i];
578 if (v->s == NULL)
579 v->s = "";
584 * val is a list of atom_value to hold returned values. Extract
585 * the values for atoms in used_atom array out of (obj, buf, sz).
586 * when deref is false, (obj, buf, sz) is the object that is
587 * pointed at by the ref itself; otherwise it is the object the
588 * ref (which is a tag) refers to.
590 static void grab_values(struct atom_value *val, int deref, struct object *obj, void *buf, unsigned long sz)
592 grab_common_values(val, deref, obj, buf, sz);
593 switch (obj->type) {
594 case OBJ_TAG:
595 grab_tag_values(val, deref, obj, buf, sz);
596 grab_sub_body_contents(val, deref, obj, buf, sz);
597 grab_person("tagger", val, deref, obj, buf, sz);
598 break;
599 case OBJ_COMMIT:
600 grab_commit_values(val, deref, obj, buf, sz);
601 grab_sub_body_contents(val, deref, obj, buf, sz);
602 grab_person("author", val, deref, obj, buf, sz);
603 grab_person("committer", val, deref, obj, buf, sz);
604 break;
605 case OBJ_TREE:
606 /* grab_tree_values(val, deref, obj, buf, sz); */
607 break;
608 case OBJ_BLOB:
609 /* grab_blob_values(val, deref, obj, buf, sz); */
610 break;
611 default:
612 die("Eh? Object of type %d?", obj->type);
616 static inline char *copy_advance(char *dst, const char *src)
618 while (*src)
619 *dst++ = *src++;
620 return dst;
624 * Parse the object referred by ref, and grab needed value.
626 static void populate_value(struct refinfo *ref)
628 void *buf;
629 struct object *obj;
630 int eaten, i;
631 unsigned long size;
632 const unsigned char *tagged;
634 ref->value = xcalloc(used_atom_cnt, sizeof(struct atom_value));
636 if (need_symref && (ref->flag & REF_ISSYMREF) && !ref->symref) {
637 unsigned char unused1[20];
638 ref->symref = resolve_refdup(ref->refname, unused1, 1, NULL);
639 if (!ref->symref)
640 ref->symref = "";
643 /* Fill in specials first */
644 for (i = 0; i < used_atom_cnt; i++) {
645 const char *name = used_atom[i];
646 struct atom_value *v = &ref->value[i];
647 int deref = 0;
648 const char *refname;
649 const char *formatp;
650 struct branch *branch = NULL;
652 if (*name == '*') {
653 deref = 1;
654 name++;
657 if (starts_with(name, "refname"))
658 refname = ref->refname;
659 else if (starts_with(name, "symref"))
660 refname = ref->symref ? ref->symref : "";
661 else if (starts_with(name, "upstream")) {
662 /* only local branches may have an upstream */
663 if (!starts_with(ref->refname, "refs/heads/"))
664 continue;
665 branch = branch_get(ref->refname + 11);
667 if (!branch || !branch->merge || !branch->merge[0] ||
668 !branch->merge[0]->dst)
669 continue;
670 refname = branch->merge[0]->dst;
671 } else if (starts_with(name, "color:")) {
672 char color[COLOR_MAXLEN] = "";
674 color_parse(name + 6, "--format", color);
675 v->s = xstrdup(color);
676 continue;
677 } else if (!strcmp(name, "flag")) {
678 char buf[256], *cp = buf;
679 if (ref->flag & REF_ISSYMREF)
680 cp = copy_advance(cp, ",symref");
681 if (ref->flag & REF_ISPACKED)
682 cp = copy_advance(cp, ",packed");
683 if (cp == buf)
684 v->s = "";
685 else {
686 *cp = '\0';
687 v->s = xstrdup(buf + 1);
689 continue;
690 } else if (!deref && grab_objectname(name, ref->objectname, v)) {
691 continue;
692 } else if (!strcmp(name, "HEAD")) {
693 const char *head;
694 unsigned char sha1[20];
696 head = resolve_ref_unsafe("HEAD", sha1, 1, NULL);
697 if (!strcmp(ref->refname, head))
698 v->s = "*";
699 else
700 v->s = " ";
701 continue;
702 } else
703 continue;
705 formatp = strchr(name, ':');
706 if (formatp) {
707 int num_ours, num_theirs;
709 formatp++;
710 if (!strcmp(formatp, "short"))
711 refname = shorten_unambiguous_ref(refname,
712 warn_ambiguous_refs);
713 else if (!strcmp(formatp, "track") &&
714 starts_with(name, "upstream")) {
715 char buf[40];
717 stat_tracking_info(branch, &num_ours, &num_theirs);
718 if (!num_ours && !num_theirs)
719 v->s = "";
720 else if (!num_ours) {
721 sprintf(buf, "[behind %d]", num_theirs);
722 v->s = xstrdup(buf);
723 } else if (!num_theirs) {
724 sprintf(buf, "[ahead %d]", num_ours);
725 v->s = xstrdup(buf);
726 } else {
727 sprintf(buf, "[ahead %d, behind %d]",
728 num_ours, num_theirs);
729 v->s = xstrdup(buf);
731 continue;
732 } else if (!strcmp(formatp, "trackshort") &&
733 starts_with(name, "upstream")) {
734 assert(branch);
735 stat_tracking_info(branch, &num_ours, &num_theirs);
736 if (!num_ours && !num_theirs)
737 v->s = "=";
738 else if (!num_ours)
739 v->s = "<";
740 else if (!num_theirs)
741 v->s = ">";
742 else
743 v->s = "<>";
744 continue;
745 } else
746 die("unknown %.*s format %s",
747 (int)(formatp - name), name, formatp);
750 if (!deref)
751 v->s = refname;
752 else {
753 int len = strlen(refname);
754 char *s = xmalloc(len + 4);
755 sprintf(s, "%s^{}", refname);
756 v->s = s;
760 for (i = 0; i < used_atom_cnt; i++) {
761 struct atom_value *v = &ref->value[i];
762 if (v->s == NULL)
763 goto need_obj;
765 return;
767 need_obj:
768 buf = get_obj(ref->objectname, &obj, &size, &eaten);
769 if (!buf)
770 die("missing object %s for %s",
771 sha1_to_hex(ref->objectname), ref->refname);
772 if (!obj)
773 die("parse_object_buffer failed on %s for %s",
774 sha1_to_hex(ref->objectname), ref->refname);
776 grab_values(ref->value, 0, obj, buf, size);
777 if (!eaten)
778 free(buf);
781 * If there is no atom that wants to know about tagged
782 * object, we are done.
784 if (!need_tagged || (obj->type != OBJ_TAG))
785 return;
788 * If it is a tag object, see if we use a value that derefs
789 * the object, and if we do grab the object it refers to.
791 tagged = ((struct tag *)obj)->tagged->sha1;
794 * NEEDSWORK: This derefs tag only once, which
795 * is good to deal with chains of trust, but
796 * is not consistent with what deref_tag() does
797 * which peels the onion to the core.
799 buf = get_obj(tagged, &obj, &size, &eaten);
800 if (!buf)
801 die("missing object %s for %s",
802 sha1_to_hex(tagged), ref->refname);
803 if (!obj)
804 die("parse_object_buffer failed on %s for %s",
805 sha1_to_hex(tagged), ref->refname);
806 grab_values(ref->value, 1, obj, buf, size);
807 if (!eaten)
808 free(buf);
812 * Given a ref, return the value for the atom. This lazily gets value
813 * out of the object by calling populate value.
815 static void get_value(struct refinfo *ref, int atom, struct atom_value **v)
817 if (!ref->value) {
818 populate_value(ref);
819 fill_missing_values(ref->value);
821 *v = &ref->value[atom];
824 struct grab_ref_cbdata {
825 struct refinfo **grab_array;
826 const char **grab_pattern;
827 int grab_cnt;
831 * A call-back given to for_each_ref(). Filter refs and keep them for
832 * later object processing.
834 static int grab_single_ref(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
836 struct grab_ref_cbdata *cb = cb_data;
837 struct refinfo *ref;
838 int cnt;
840 if (*cb->grab_pattern) {
841 const char **pattern;
842 int namelen = strlen(refname);
843 for (pattern = cb->grab_pattern; *pattern; pattern++) {
844 const char *p = *pattern;
845 int plen = strlen(p);
847 if ((plen <= namelen) &&
848 !strncmp(refname, p, plen) &&
849 (refname[plen] == '\0' ||
850 refname[plen] == '/' ||
851 p[plen-1] == '/'))
852 break;
853 if (!wildmatch(p, refname, WM_PATHNAME, NULL))
854 break;
856 if (!*pattern)
857 return 0;
861 * We do not open the object yet; sort may only need refname
862 * to do its job and the resulting list may yet to be pruned
863 * by maxcount logic.
865 ref = xcalloc(1, sizeof(*ref));
866 ref->refname = xstrdup(refname);
867 hashcpy(ref->objectname, sha1);
868 ref->flag = flag;
870 cnt = cb->grab_cnt;
871 REALLOC_ARRAY(cb->grab_array, cnt + 1);
872 cb->grab_array[cnt++] = ref;
873 cb->grab_cnt = cnt;
874 return 0;
877 static int cmp_ref_sort(struct ref_sort *s, struct refinfo *a, struct refinfo *b)
879 struct atom_value *va, *vb;
880 int cmp;
881 cmp_type cmp_type = used_atom_type[s->atom];
883 get_value(a, s->atom, &va);
884 get_value(b, s->atom, &vb);
885 switch (cmp_type) {
886 case FIELD_STR:
887 cmp = strcmp(va->s, vb->s);
888 break;
889 default:
890 if (va->ul < vb->ul)
891 cmp = -1;
892 else if (va->ul == vb->ul)
893 cmp = 0;
894 else
895 cmp = 1;
896 break;
898 return (s->reverse) ? -cmp : cmp;
901 static struct ref_sort *ref_sort;
902 static int compare_refs(const void *a_, const void *b_)
904 struct refinfo *a = *((struct refinfo **)a_);
905 struct refinfo *b = *((struct refinfo **)b_);
906 struct ref_sort *s;
908 for (s = ref_sort; s; s = s->next) {
909 int cmp = cmp_ref_sort(s, a, b);
910 if (cmp)
911 return cmp;
913 return 0;
916 static void sort_refs(struct ref_sort *sort, struct refinfo **refs, int num_refs)
918 ref_sort = sort;
919 qsort(refs, num_refs, sizeof(struct refinfo *), compare_refs);
922 static void print_value(struct atom_value *v, int quote_style)
924 struct strbuf sb = STRBUF_INIT;
925 switch (quote_style) {
926 case QUOTE_NONE:
927 fputs(v->s, stdout);
928 break;
929 case QUOTE_SHELL:
930 sq_quote_buf(&sb, v->s);
931 break;
932 case QUOTE_PERL:
933 perl_quote_buf(&sb, v->s);
934 break;
935 case QUOTE_PYTHON:
936 python_quote_buf(&sb, v->s);
937 break;
938 case QUOTE_TCL:
939 tcl_quote_buf(&sb, v->s);
940 break;
942 if (quote_style != QUOTE_NONE) {
943 fputs(sb.buf, stdout);
944 strbuf_release(&sb);
948 static int hex1(char ch)
950 if ('0' <= ch && ch <= '9')
951 return ch - '0';
952 else if ('a' <= ch && ch <= 'f')
953 return ch - 'a' + 10;
954 else if ('A' <= ch && ch <= 'F')
955 return ch - 'A' + 10;
956 return -1;
958 static int hex2(const char *cp)
960 if (cp[0] && cp[1])
961 return (hex1(cp[0]) << 4) | hex1(cp[1]);
962 else
963 return -1;
966 static void emit(const char *cp, const char *ep)
968 while (*cp && (!ep || cp < ep)) {
969 if (*cp == '%') {
970 if (cp[1] == '%')
971 cp++;
972 else {
973 int ch = hex2(cp + 1);
974 if (0 <= ch) {
975 putchar(ch);
976 cp += 3;
977 continue;
981 putchar(*cp);
982 cp++;
986 static void show_ref(struct refinfo *info, const char *format, int quote_style)
988 const char *cp, *sp, *ep;
990 for (cp = format; *cp && (sp = find_next(cp)); cp = ep + 1) {
991 struct atom_value *atomv;
993 ep = strchr(sp, ')');
994 if (cp < sp)
995 emit(cp, sp);
996 get_value(info, parse_atom(sp + 2, ep), &atomv);
997 print_value(atomv, quote_style);
999 if (*cp) {
1000 sp = cp + strlen(cp);
1001 emit(cp, sp);
1003 if (need_color_reset_at_eol) {
1004 struct atom_value resetv;
1005 char color[COLOR_MAXLEN] = "";
1007 color_parse("reset", "--format", color);
1008 resetv.s = color;
1009 print_value(&resetv, quote_style);
1011 putchar('\n');
1014 static struct ref_sort *default_sort(void)
1016 static const char cstr_name[] = "refname";
1018 struct ref_sort *sort = xcalloc(1, sizeof(*sort));
1020 sort->next = NULL;
1021 sort->atom = parse_atom(cstr_name, cstr_name + strlen(cstr_name));
1022 return sort;
1025 static int opt_parse_sort(const struct option *opt, const char *arg, int unset)
1027 struct ref_sort **sort_tail = opt->value;
1028 struct ref_sort *s;
1029 int len;
1031 if (!arg) /* should --no-sort void the list ? */
1032 return -1;
1034 s = xcalloc(1, sizeof(*s));
1035 s->next = *sort_tail;
1036 *sort_tail = s;
1038 if (*arg == '-') {
1039 s->reverse = 1;
1040 arg++;
1042 len = strlen(arg);
1043 s->atom = parse_atom(arg, arg+len);
1044 return 0;
1047 static char const * const for_each_ref_usage[] = {
1048 N_("git for-each-ref [options] [<pattern>]"),
1049 NULL
1052 int cmd_for_each_ref(int argc, const char **argv, const char *prefix)
1054 int i, num_refs;
1055 const char *format = "%(objectname) %(objecttype)\t%(refname)";
1056 struct ref_sort *sort = NULL, **sort_tail = &sort;
1057 int maxcount = 0, quote_style = 0;
1058 struct refinfo **refs;
1059 struct grab_ref_cbdata cbdata;
1061 struct option opts[] = {
1062 OPT_BIT('s', "shell", &quote_style,
1063 N_("quote placeholders suitably for shells"), QUOTE_SHELL),
1064 OPT_BIT('p', "perl", &quote_style,
1065 N_("quote placeholders suitably for perl"), QUOTE_PERL),
1066 OPT_BIT(0 , "python", &quote_style,
1067 N_("quote placeholders suitably for python"), QUOTE_PYTHON),
1068 OPT_BIT(0 , "tcl", &quote_style,
1069 N_("quote placeholders suitably for tcl"), QUOTE_TCL),
1071 OPT_GROUP(""),
1072 OPT_INTEGER( 0 , "count", &maxcount, N_("show only <n> matched refs")),
1073 OPT_STRING( 0 , "format", &format, N_("format"), N_("format to use for the output")),
1074 OPT_CALLBACK(0 , "sort", sort_tail, N_("key"),
1075 N_("field name to sort on"), &opt_parse_sort),
1076 OPT_END(),
1079 parse_options(argc, argv, prefix, opts, for_each_ref_usage, 0);
1080 if (maxcount < 0) {
1081 error("invalid --count argument: `%d'", maxcount);
1082 usage_with_options(for_each_ref_usage, opts);
1084 if (HAS_MULTI_BITS(quote_style)) {
1085 error("more than one quoting style?");
1086 usage_with_options(for_each_ref_usage, opts);
1088 if (verify_format(format))
1089 usage_with_options(for_each_ref_usage, opts);
1091 if (!sort)
1092 sort = default_sort();
1094 /* for warn_ambiguous_refs */
1095 git_config(git_default_config, NULL);
1097 memset(&cbdata, 0, sizeof(cbdata));
1098 cbdata.grab_pattern = argv;
1099 for_each_rawref(grab_single_ref, &cbdata);
1100 refs = cbdata.grab_array;
1101 num_refs = cbdata.grab_cnt;
1103 sort_refs(sort, refs, num_refs);
1105 if (!maxcount || num_refs < maxcount)
1106 maxcount = num_refs;
1107 for (i = 0; i < maxcount; i++)
1108 show_ref(refs[i], format, quote_style);
1109 return 0;