1#include <stdlib.h>
2#include <fnmatch.h>
3#include "cache.h"
4#include "commit.h"
5#include "refs.h"
6
7static const char show_branch_usage[] =
8"git-show-branch [--current] [--all] [--heads] [--tags] [--topo-order] [--more=count | --list | --independent | --merge-base ] [<refs>...]";
9
10static int default_num = 0;
11static int default_alloc = 0;
12static char **default_arg = NULL;
13
14#define UNINTERESTING 01
15
16#define REV_SHIFT 2
17#define MAX_REVS 29 /* should not exceed bits_per_int - REV_SHIFT */
18
19static struct commit *interesting(struct commit_list *list)
20{
21 while (list) {
22 struct commit *commit = list->item;
23 list = list->next;
24 if (commit->object.flags & UNINTERESTING)
25 continue;
26 return commit;
27 }
28 return NULL;
29}
30
31static struct commit *pop_one_commit(struct commit_list **list_p)
32{
33 struct commit *commit;
34 struct commit_list *list;
35 list = *list_p;
36 commit = list->item;
37 *list_p = list->next;
38 free(list);
39 return commit;
40}
41
42struct commit_name {
43 const char *head_name; /* which head's ancestor? */
44 int generation; /* how many parents away from head_name */
45};
46
47/* Name the commit as nth generation ancestor of head_name;
48 * we count only the first-parent relationship for naming purposes.
49 */
50static void name_commit(struct commit *commit, const char *head_name, int nth)
51{
52 struct commit_name *name;
53 if (!commit->object.util)
54 commit->object.util = xmalloc(sizeof(struct commit_name));
55 name = commit->object.util;
56 name->head_name = head_name;
57 name->generation = nth;
58}
59
60/* Parent is the first parent of the commit. We may name it
61 * as (n+1)th generation ancestor of the same head_name as
62 * commit is nth generation ancestor of, if that generation
63 * number is better than the name it already has.
64 */
65static void name_parent(struct commit *commit, struct commit *parent)
66{
67 struct commit_name *commit_name = commit->object.util;
68 struct commit_name *parent_name = parent->object.util;
69 if (!commit_name)
70 return;
71 if (!parent_name ||
72 commit_name->generation + 1 < parent_name->generation)
73 name_commit(parent, commit_name->head_name,
74 commit_name->generation + 1);
75}
76
77static int name_first_parent_chain(struct commit *c)
78{
79 int i = 0;
80 while (c) {
81 struct commit *p;
82 if (!c->object.util)
83 break;
84 if (!c->parents)
85 break;
86 p = c->parents->item;
87 if (!p->object.util) {
88 name_parent(c, p);
89 i++;
90 }
91 c = p;
92 }
93 return i;
94}
95
96static void name_commits(struct commit_list *list,
97 struct commit **rev,
98 char **ref_name,
99 int num_rev)
100{
101 struct commit_list *cl;
102 struct commit *c;
103 int i;
104
105 /* First give names to the given heads */
106 for (cl = list; cl; cl = cl->next) {
107 c = cl->item;
108 if (c->object.util)
109 continue;
110 for (i = 0; i < num_rev; i++) {
111 if (rev[i] == c) {
112 name_commit(c, ref_name[i], 0);
113 break;
114 }
115 }
116 }
117
118 /* Then commits on the first parent ancestry chain */
119 do {
120 i = 0;
121 for (cl = list; cl; cl = cl->next) {
122 i += name_first_parent_chain(cl->item);
123 }
124 } while (i);
125
126 /* Finally, any unnamed commits */
127 do {
128 i = 0;
129 for (cl = list; cl; cl = cl->next) {
130 struct commit_list *parents;
131 struct commit_name *n;
132 int nth;
133 c = cl->item;
134 if (!c->object.util)
135 continue;
136 n = c->object.util;
137 parents = c->parents;
138 nth = 0;
139 while (parents) {
140 struct commit *p = parents->item;
141 char newname[1000], *en;
142 parents = parents->next;
143 nth++;
144 if (p->object.util)
145 continue;
146 en = newname;
147 switch (n->generation) {
148 case 0:
149 en += sprintf(en, "%s", n->head_name);
150 break;
151 case 1:
152 en += sprintf(en, "%s^", n->head_name);
153 break;
154 default:
155 en += sprintf(en, "%s~%d",
156 n->head_name, n->generation);
157 break;
158 }
159 if (nth == 1)
160 en += sprintf(en, "^");
161 else
162 en += sprintf(en, "^%d", nth);
163 name_commit(p, strdup(newname), 0);
164 i++;
165 name_first_parent_chain(p);
166 }
167 }
168 } while (i);
169}
170
171static int mark_seen(struct commit *commit, struct commit_list **seen_p)
172{
173 if (!commit->object.flags) {
174 insert_by_date(commit, seen_p);
175 return 1;
176 }
177 return 0;
178}
179
180static void join_revs(struct commit_list **list_p,
181 struct commit_list **seen_p,
182 int num_rev, int extra)
183{
184 int all_mask = ((1u << (REV_SHIFT + num_rev)) - 1);
185 int all_revs = all_mask & ~((1u << REV_SHIFT) - 1);
186
187 while (*list_p) {
188 struct commit_list *parents;
189 int still_interesting = !!interesting(*list_p);
190 struct commit *commit = pop_one_commit(list_p);
191 int flags = commit->object.flags & all_mask;
192
193 if (!still_interesting && extra <= 0)
194 break;
195
196 mark_seen(commit, seen_p);
197 if ((flags & all_revs) == all_revs)
198 flags |= UNINTERESTING;
199 parents = commit->parents;
200
201 while (parents) {
202 struct commit *p = parents->item;
203 int this_flag = p->object.flags;
204 parents = parents->next;
205 if ((this_flag & flags) == flags)
206 continue;
207 if (!p->object.parsed)
208 parse_commit(p);
209 if (mark_seen(p, seen_p) && !still_interesting)
210 extra--;
211 p->object.flags |= flags;
212 insert_by_date(p, list_p);
213 }
214 }
215
216 /*
217 * Postprocess to complete well-poisoning.
218 *
219 * At this point we have all the commits we have seen in
220 * seen_p list (which happens to be sorted chronologically but
221 * it does not really matter). Mark anything that can be
222 * reached from uninteresting commits not interesting.
223 */
224 for (;;) {
225 int changed = 0;
226 struct commit_list *s;
227 for (s = *seen_p; s; s = s->next) {
228 struct commit *c = s->item;
229 struct commit_list *parents;
230
231 if (((c->object.flags & all_revs) != all_revs) &&
232 !(c->object.flags & UNINTERESTING))
233 continue;
234
235 /* The current commit is either a merge base or
236 * already uninteresting one. Mark its parents
237 * as uninteresting commits _only_ if they are
238 * already parsed. No reason to find new ones
239 * here.
240 */
241 parents = c->parents;
242 while (parents) {
243 struct commit *p = parents->item;
244 parents = parents->next;
245 if (!(p->object.flags & UNINTERESTING)) {
246 p->object.flags |= UNINTERESTING;
247 changed = 1;
248 }
249 }
250 }
251 if (!changed)
252 break;
253 }
254}
255
256static void show_one_commit(struct commit *commit, int no_name)
257{
258 char pretty[256], *cp;
259 struct commit_name *name = commit->object.util;
260 if (commit->object.parsed)
261 pretty_print_commit(CMIT_FMT_ONELINE, commit, ~0,
262 pretty, sizeof(pretty), 0);
263 else
264 strcpy(pretty, "(unavailable)");
265 if (!strncmp(pretty, "[PATCH] ", 8))
266 cp = pretty + 8;
267 else
268 cp = pretty;
269
270 if (!no_name) {
271 if (name && name->head_name) {
272 printf("[%s", name->head_name);
273 if (name->generation) {
274 if (name->generation == 1)
275 printf("^");
276 else
277 printf("~%d", name->generation);
278 }
279 printf("] ");
280 }
281 else
282 printf("[%s] ",
283 find_unique_abbrev(commit->object.sha1, 7));
284 }
285 puts(cp);
286}
287
288static char *ref_name[MAX_REVS + 1];
289static int ref_name_cnt;
290
291static const char *find_digit_prefix(const char *s, int *v)
292{
293 const char *p;
294 int ver;
295 char ch;
296
297 for (p = s, ver = 0;
298 '0' <= (ch = *p) && ch <= '9';
299 p++)
300 ver = ver * 10 + ch - '0';
301 *v = ver;
302 return p;
303}
304
305
306static int version_cmp(const char *a, const char *b)
307{
308 while (1) {
309 int va, vb;
310
311 a = find_digit_prefix(a, &va);
312 b = find_digit_prefix(b, &vb);
313 if (va != vb)
314 return va - vb;
315
316 while (1) {
317 int ca = *a;
318 int cb = *b;
319 if ('0' <= ca && ca <= '9')
320 ca = 0;
321 if ('0' <= cb && cb <= '9')
322 cb = 0;
323 if (ca != cb)
324 return ca - cb;
325 if (!ca)
326 break;
327 a++;
328 b++;
329 }
330 if (!*a && !*b)
331 return 0;
332 }
333}
334
335static int compare_ref_name(const void *a_, const void *b_)
336{
337 const char * const*a = a_, * const*b = b_;
338 return version_cmp(*a, *b);
339}
340
341static void sort_ref_range(int bottom, int top)
342{
343 qsort(ref_name + bottom, top - bottom, sizeof(ref_name[0]),
344 compare_ref_name);
345}
346
347static int append_ref(const char *refname, const unsigned char *sha1)
348{
349 struct commit *commit = lookup_commit_reference_gently(sha1, 1);
350 int i;
351
352 if (!commit)
353 return 0;
354 /* Avoid adding the same thing twice */
355 for (i = 0; i < ref_name_cnt; i++)
356 if (!strcmp(refname, ref_name[i]))
357 return 0;
358
359 if (MAX_REVS <= ref_name_cnt) {
360 fprintf(stderr, "warning: ignoring %s; "
361 "cannot handle more than %d refs\n",
362 refname, MAX_REVS);
363 return 0;
364 }
365 ref_name[ref_name_cnt++] = strdup(refname);
366 ref_name[ref_name_cnt] = NULL;
367 return 0;
368}
369
370static int append_head_ref(const char *refname, const unsigned char *sha1)
371{
372 unsigned char tmp[20];
373 int ofs = 11;
374 if (strncmp(refname, "refs/heads/", ofs))
375 return 0;
376 /* If both heads/foo and tags/foo exists, get_sha1 would
377 * get confused.
378 */
379 if (get_sha1(refname + ofs, tmp) || memcmp(tmp, sha1, 20))
380 ofs = 5;
381 return append_ref(refname + ofs, sha1);
382}
383
384static int append_tag_ref(const char *refname, const unsigned char *sha1)
385{
386 if (strncmp(refname, "refs/tags/", 10))
387 return 0;
388 return append_ref(refname + 5, sha1);
389}
390
391static const char *match_ref_pattern = NULL;
392static int match_ref_slash = 0;
393static int count_slash(const char *s)
394{
395 int cnt = 0;
396 while (*s)
397 if (*s++ == '/')
398 cnt++;
399 return cnt;
400}
401
402static int append_matching_ref(const char *refname, const unsigned char *sha1)
403{
404 /* we want to allow pattern hold/<asterisk> to show all
405 * branches under refs/heads/hold/, and v0.99.9? to show
406 * refs/tags/v0.99.9a and friends.
407 */
408 const char *tail;
409 int slash = count_slash(refname);
410 for (tail = refname; *tail && match_ref_slash < slash; )
411 if (*tail++ == '/')
412 slash--;
413 if (!*tail)
414 return 0;
415 if (fnmatch(match_ref_pattern, tail, 0))
416 return 0;
417 if (!strncmp("refs/heads/", refname, 11))
418 return append_head_ref(refname, sha1);
419 if (!strncmp("refs/tags/", refname, 10))
420 return append_tag_ref(refname, sha1);
421 return append_ref(refname, sha1);
422}
423
424static void snarf_refs(int head, int tag)
425{
426 if (head) {
427 int orig_cnt = ref_name_cnt;
428 for_each_ref(append_head_ref);
429 sort_ref_range(orig_cnt, ref_name_cnt);
430 }
431 if (tag) {
432 int orig_cnt = ref_name_cnt;
433 for_each_ref(append_tag_ref);
434 sort_ref_range(orig_cnt, ref_name_cnt);
435 }
436}
437
438static int rev_is_head(char *head_path, int headlen, char *name,
439 unsigned char *head_sha1, unsigned char *sha1)
440{
441 int namelen;
442 if ((!head_path[0]) ||
443 (head_sha1 && sha1 && memcmp(head_sha1, sha1, 20)))
444 return 0;
445 namelen = strlen(name);
446 if ((headlen < namelen) ||
447 memcmp(head_path + headlen - namelen, name, namelen))
448 return 0;
449 if (headlen == namelen ||
450 head_path[headlen - namelen - 1] == '/')
451 return 1;
452 return 0;
453}
454
455static int show_merge_base(struct commit_list *seen, int num_rev)
456{
457 int all_mask = ((1u << (REV_SHIFT + num_rev)) - 1);
458 int all_revs = all_mask & ~((1u << REV_SHIFT) - 1);
459 int exit_status = 1;
460
461 while (seen) {
462 struct commit *commit = pop_one_commit(&seen);
463 int flags = commit->object.flags & all_mask;
464 if (!(flags & UNINTERESTING) &&
465 ((flags & all_revs) == all_revs)) {
466 puts(sha1_to_hex(commit->object.sha1));
467 exit_status = 0;
468 commit->object.flags |= UNINTERESTING;
469 }
470 }
471 return exit_status;
472}
473
474static int show_independent(struct commit **rev,
475 int num_rev,
476 char **ref_name,
477 unsigned int *rev_mask)
478{
479 int i;
480
481 for (i = 0; i < num_rev; i++) {
482 struct commit *commit = rev[i];
483 unsigned int flag = rev_mask[i];
484
485 if (commit->object.flags == flag)
486 puts(sha1_to_hex(commit->object.sha1));
487 commit->object.flags |= UNINTERESTING;
488 }
489 return 0;
490}
491
492static void append_one_rev(const char *av)
493{
494 unsigned char revkey[20];
495 if (!get_sha1(av, revkey)) {
496 append_ref(av, revkey);
497 return;
498 }
499 if (strchr(av, '*') || strchr(av, '?') || strchr(av, '[')) {
500 /* glob style match */
501 int saved_matches = ref_name_cnt;
502 match_ref_pattern = av;
503 match_ref_slash = count_slash(av);
504 for_each_ref(append_matching_ref);
505 if (saved_matches == ref_name_cnt &&
506 ref_name_cnt < MAX_REVS)
507 error("no matching refs with %s", av);
508 if (saved_matches + 1 < ref_name_cnt)
509 sort_ref_range(saved_matches, ref_name_cnt);
510 return;
511 }
512 die("bad sha1 reference %s", av);
513}
514
515static int git_show_branch_config(const char *var, const char *value)
516{
517 if (!strcmp(var, "showbranch.default")) {
518 if (default_alloc <= default_num + 1) {
519 default_alloc = default_alloc * 3 / 2 + 20;
520 default_arg = xrealloc(default_arg, sizeof *default_arg * default_alloc);
521 }
522 default_arg[default_num++] = strdup(value);
523 default_arg[default_num] = NULL;
524 return 0;
525 }
526
527 return git_default_config(var, value);
528}
529
530int main(int ac, char **av)
531{
532 struct commit *rev[MAX_REVS], *commit;
533 struct commit_list *list = NULL, *seen = NULL;
534 unsigned int rev_mask[MAX_REVS];
535 int num_rev, i, extra = 0;
536 int all_heads = 0, all_tags = 0;
537 int all_mask, all_revs;
538 char head_path[128];
539 const char *head_path_p;
540 int head_path_len;
541 unsigned char head_sha1[20];
542 int merge_base = 0;
543 int independent = 0;
544 int no_name = 0;
545 int sha1_name = 0;
546 int shown_merge_point = 0;
547 int topo_order = 0;
548 int with_current_branch = 0;
549 int head_at = -1;
550
551 git_config(git_show_branch_config);
552 setup_git_directory();
553
554 /* If nothing is specified, try the default first */
555 if (ac == 1 && default_num) {
556 ac = default_num + 1;
557 av = default_arg - 1; /* ick; we would not address av[0] */
558 }
559
560 while (1 < ac && av[1][0] == '-') {
561 char *arg = av[1];
562 if (!strcmp(arg, "--")) {
563 ac--; av++;
564 break;
565 }
566 else if (!strcmp(arg, "--all"))
567 all_heads = all_tags = 1;
568 else if (!strcmp(arg, "--heads"))
569 all_heads = 1;
570 else if (!strcmp(arg, "--tags"))
571 all_tags = 1;
572 else if (!strcmp(arg, "--more"))
573 extra = 1;
574 else if (!strcmp(arg, "--list"))
575 extra = -1;
576 else if (!strcmp(arg, "--no-name"))
577 no_name = 1;
578 else if (!strcmp(arg, "--current"))
579 with_current_branch = 1;
580 else if (!strcmp(arg, "--sha1-name"))
581 sha1_name = 1;
582 else if (!strncmp(arg, "--more=", 7))
583 extra = atoi(arg + 7);
584 else if (!strcmp(arg, "--merge-base"))
585 merge_base = 1;
586 else if (!strcmp(arg, "--independent"))
587 independent = 1;
588 else if (!strcmp(arg, "--topo-order"))
589 topo_order = 1;
590 else
591 usage(show_branch_usage);
592 ac--; av++;
593 }
594 ac--; av++;
595
596 /* Only one of these is allowed */
597 if (1 < independent + merge_base + (extra != 0))
598 usage(show_branch_usage);
599
600 /* If nothing is specified, show all branches by default */
601 if (ac + all_heads + all_tags == 0)
602 all_heads = 1;
603
604 if (all_heads + all_tags)
605 snarf_refs(all_heads, all_tags);
606 while (0 < ac) {
607 append_one_rev(*av);
608 ac--; av++;
609 }
610
611 head_path_p = resolve_ref(git_path("HEAD"), head_sha1, 1);
612 if (head_path_p) {
613 head_path_len = strlen(head_path_p);
614 memcpy(head_path, head_path_p, head_path_len + 1);
615 }
616 else {
617 head_path_len = 0;
618 head_path[0] = 0;
619 }
620
621 if (with_current_branch && head_path_p) {
622 int has_head = 0;
623 for (i = 0; !has_head && i < ref_name_cnt; i++) {
624 /* We are only interested in adding the branch
625 * HEAD points at.
626 */
627 if (rev_is_head(head_path,
628 head_path_len,
629 ref_name[i],
630 head_sha1, NULL))
631 has_head++;
632 }
633 if (!has_head) {
634 int pfxlen = strlen(git_path("refs/heads/"));
635 append_one_rev(head_path + pfxlen);
636 }
637 }
638
639 if (!ref_name_cnt) {
640 fprintf(stderr, "No revs to be shown.\n");
641 exit(0);
642 }
643
644 for (num_rev = 0; ref_name[num_rev]; num_rev++) {
645 unsigned char revkey[20];
646 unsigned int flag = 1u << (num_rev + REV_SHIFT);
647
648 if (MAX_REVS <= num_rev)
649 die("cannot handle more than %d revs.", MAX_REVS);
650 if (get_sha1(ref_name[num_rev], revkey))
651 die("'%s' is not a valid ref.", ref_name[num_rev]);
652 commit = lookup_commit_reference(revkey);
653 if (!commit)
654 die("cannot find commit %s (%s)",
655 ref_name[num_rev], revkey);
656 parse_commit(commit);
657 mark_seen(commit, &seen);
658
659 /* rev#0 uses bit REV_SHIFT, rev#1 uses bit REV_SHIFT+1,
660 * and so on. REV_SHIFT bits from bit 0 are used for
661 * internal bookkeeping.
662 */
663 commit->object.flags |= flag;
664 if (commit->object.flags == flag)
665 insert_by_date(commit, &list);
666 rev[num_rev] = commit;
667 }
668 for (i = 0; i < num_rev; i++)
669 rev_mask[i] = rev[i]->object.flags;
670
671 if (0 <= extra)
672 join_revs(&list, &seen, num_rev, extra);
673
674 if (merge_base)
675 return show_merge_base(seen, num_rev);
676
677 if (independent)
678 return show_independent(rev, num_rev, ref_name, rev_mask);
679
680 /* Show list; --more=-1 means list-only */
681 if (1 < num_rev || extra < 0) {
682 for (i = 0; i < num_rev; i++) {
683 int j;
684 int is_head = rev_is_head(head_path,
685 head_path_len,
686 ref_name[i],
687 head_sha1,
688 rev[i]->object.sha1);
689 if (extra < 0)
690 printf("%c [%s] ",
691 is_head ? '*' : ' ', ref_name[i]);
692 else {
693 for (j = 0; j < i; j++)
694 putchar(' ');
695 printf("%c [%s] ",
696 is_head ? '*' : '!', ref_name[i]);
697 }
698 /* header lines never need name */
699 show_one_commit(rev[i], 1);
700 if (is_head)
701 head_at = i;
702 }
703 if (0 <= extra) {
704 for (i = 0; i < num_rev; i++)
705 putchar('-');
706 putchar('\n');
707 }
708 }
709 if (extra < 0)
710 exit(0);
711
712 /* Sort topologically */
713 if (topo_order)
714 sort_in_topological_order(&seen);
715
716 /* Give names to commits */
717 if (!sha1_name && !no_name)
718 name_commits(seen, rev, ref_name, num_rev);
719
720 all_mask = ((1u << (REV_SHIFT + num_rev)) - 1);
721 all_revs = all_mask & ~((1u << REV_SHIFT) - 1);
722
723 while (seen) {
724 struct commit *commit = pop_one_commit(&seen);
725 int this_flag = commit->object.flags;
726
727 shown_merge_point |= ((this_flag & all_revs) == all_revs);
728
729 if (1 < num_rev) {
730 int is_merge = !!(commit->parents && commit->parents->next);
731 for (i = 0; i < num_rev; i++) {
732 int mark;
733 if (!(this_flag & (1u << (i + REV_SHIFT))))
734 mark = ' ';
735 else if (is_merge)
736 mark = '-';
737 else if (i == head_at)
738 mark = '*';
739 else
740 mark = '+';
741 putchar(mark);
742 }
743 putchar(' ');
744 }
745 show_one_commit(commit, no_name);
746
747 if (shown_merge_point && --extra < 0)
748 break;
749 }
750 return 0;
751}