Merge branch 'kh/fetch-optparse'
[git] / builtin-shortlog.c
1 #include "builtin.h"
2 #include "cache.h"
3 #include "commit.h"
4 #include "diff.h"
5 #include "path-list.h"
6 #include "revision.h"
7 #include "utf8.h"
8 #include "mailmap.h"
9
10 static const char shortlog_usage[] =
11 "git-shortlog [-n] [-s] [<commit-id>... ]";
12
13 static char *common_repo_prefix;
14
15 static int compare_by_number(const void *a1, const void *a2)
16 {
17         const struct path_list_item *i1 = a1, *i2 = a2;
18         const struct path_list *l1 = i1->util, *l2 = i2->util;
19
20         if (l1->nr < l2->nr)
21                 return 1;
22         else if (l1->nr == l2->nr)
23                 return 0;
24         else
25                 return -1;
26 }
27
28 static struct path_list mailmap = {NULL, 0, 0, 0};
29
30 static void insert_author_oneline(struct path_list *list,
31                 const char *author, int authorlen,
32                 const char *oneline, int onelinelen)
33 {
34         const char *dot3 = common_repo_prefix;
35         char *buffer, *p;
36         struct path_list_item *item;
37         struct path_list *onelines;
38
39         while (authorlen > 0 && isspace(author[authorlen - 1]))
40                 authorlen--;
41
42         buffer = xmemdupz(author, authorlen);
43         item = path_list_insert(buffer, list);
44         if (item->util == NULL)
45                 item->util = xcalloc(1, sizeof(struct path_list));
46         else
47                 free(buffer);
48
49         if (!prefixcmp(oneline, "[PATCH")) {
50                 char *eob = strchr(oneline, ']');
51
52                 if (eob) {
53                         while (isspace(eob[1]) && eob[1] != '\n')
54                                 eob++;
55                         if (eob - oneline < onelinelen) {
56                                 onelinelen -= eob - oneline;
57                                 oneline = eob;
58                         }
59                 }
60         }
61
62         while (onelinelen > 0 && isspace(oneline[0])) {
63                 oneline++;
64                 onelinelen--;
65         }
66         while (onelinelen > 0 && isspace(oneline[onelinelen - 1]))
67                 onelinelen--;
68         buffer = xmemdupz(oneline, onelinelen);
69
70         if (dot3) {
71                 int dot3len = strlen(dot3);
72                 if (dot3len > 5) {
73                         while ((p = strstr(buffer, dot3)) != NULL) {
74                                 int taillen = strlen(p) - dot3len;
75                                 memcpy(p, "/.../", 5);
76                                 memmove(p + 5, p + dot3len, taillen + 1);
77                         }
78                 }
79         }
80
81         onelines = item->util;
82         if (onelines->nr >= onelines->alloc) {
83                 onelines->alloc = alloc_nr(onelines->nr);
84                 onelines->items = xrealloc(onelines->items,
85                                 onelines->alloc
86                                 * sizeof(struct path_list_item));
87         }
88
89         onelines->items[onelines->nr].util = NULL;
90         onelines->items[onelines->nr++].path = buffer;
91 }
92
93 static void read_from_stdin(struct path_list *list)
94 {
95         char buffer[1024];
96
97         while (fgets(buffer, sizeof(buffer), stdin) != NULL) {
98                 char *bob;
99                 if ((buffer[0] == 'A' || buffer[0] == 'a') &&
100                                 !prefixcmp(buffer + 1, "uthor: ") &&
101                                 (bob = strchr(buffer + 7, '<')) != NULL) {
102                         char buffer2[1024], offset = 0;
103
104                         if (map_email(&mailmap, bob + 1, buffer, sizeof(buffer)))
105                                 bob = buffer + strlen(buffer);
106                         else {
107                                 offset = 8;
108                                 while (buffer + offset < bob &&
109                                        isspace(bob[-1]))
110                                         bob--;
111                         }
112
113                         while (fgets(buffer2, sizeof(buffer2), stdin) &&
114                                         buffer2[0] != '\n')
115                                 ; /* chomp input */
116                         if (fgets(buffer2, sizeof(buffer2), stdin)) {
117                                 int l2 = strlen(buffer2);
118                                 int i;
119                                 for (i = 0; i < l2; i++)
120                                         if (!isspace(buffer2[i]))
121                                                 break;
122                                 insert_author_oneline(list,
123                                                 buffer + offset,
124                                                 bob - buffer - offset,
125                                                 buffer2 + i, l2 - i);
126                         }
127                 }
128         }
129 }
130
131 static void get_from_rev(struct rev_info *rev, struct path_list *list)
132 {
133         char scratch[1024];
134         struct commit *commit;
135
136         prepare_revision_walk(rev);
137         while ((commit = get_revision(rev)) != NULL) {
138                 const char *author = NULL, *oneline, *buffer;
139                 int authorlen = authorlen, onelinelen;
140
141                 /* get author and oneline */
142                 for (buffer = commit->buffer; buffer && *buffer != '\0' &&
143                                 *buffer != '\n'; ) {
144                         const char *eol = strchr(buffer, '\n');
145
146                         if (eol == NULL)
147                                 eol = buffer + strlen(buffer);
148                         else
149                                 eol++;
150
151                         if (!prefixcmp(buffer, "author ")) {
152                                 char *bracket = strchr(buffer, '<');
153
154                                 if (bracket == NULL || bracket > eol)
155                                         die("Invalid commit buffer: %s",
156                                             sha1_to_hex(commit->object.sha1));
157
158                                 if (map_email(&mailmap, bracket + 1, scratch,
159                                                         sizeof(scratch))) {
160                                         author = scratch;
161                                         authorlen = strlen(scratch);
162                                 } else {
163                                         if (bracket[-1] == ' ')
164                                                 bracket--;
165
166                                         author = buffer + 7;
167                                         authorlen = bracket - buffer - 7;
168                                 }
169                         }
170                         buffer = eol;
171                 }
172
173                 if (author == NULL)
174                         die ("Missing author: %s",
175                                         sha1_to_hex(commit->object.sha1));
176
177                 if (buffer == NULL || *buffer == '\0') {
178                         oneline = "<none>";
179                         onelinelen = sizeof(oneline) + 1;
180                 } else {
181                         char *eol;
182
183                         oneline = buffer + 1;
184                         eol = strchr(oneline, '\n');
185                         if (eol == NULL)
186                                 onelinelen = strlen(oneline);
187                         else
188                                 onelinelen = eol - oneline;
189                 }
190
191                 insert_author_oneline(list,
192                                 author, authorlen, oneline, onelinelen);
193         }
194
195 }
196
197 static int parse_uint(char const **arg, int comma)
198 {
199         unsigned long ul;
200         int ret;
201         char *endp;
202
203         ul = strtoul(*arg, &endp, 10);
204         if (endp != *arg && *endp && *endp != comma)
205                 return -1;
206         ret = (int) ul;
207         if (ret != ul)
208                 return -1;
209         *arg = endp;
210         if (**arg)
211                 (*arg)++;
212         return ret;
213 }
214
215 static const char wrap_arg_usage[] = "-w[<width>[,<indent1>[,<indent2>]]]";
216 #define DEFAULT_WRAPLEN 76
217 #define DEFAULT_INDENT1 6
218 #define DEFAULT_INDENT2 9
219
220 static void parse_wrap_args(const char *arg, int *in1, int *in2, int *wrap)
221 {
222         arg += 2; /* skip -w */
223
224         *wrap = parse_uint(&arg, ',');
225         if (*wrap < 0)
226                 die(wrap_arg_usage);
227         *in1 = parse_uint(&arg, ',');
228         if (*in1 < 0)
229                 die(wrap_arg_usage);
230         *in2 = parse_uint(&arg, '\0');
231         if (*in2 < 0)
232                 die(wrap_arg_usage);
233
234         if (!*wrap)
235                 *wrap = DEFAULT_WRAPLEN;
236         if (!*in1)
237                 *in1 = DEFAULT_INDENT1;
238         if (!*in2)
239                 *in2 = DEFAULT_INDENT2;
240         if (*wrap &&
241             ((*in1 && *wrap <= *in1) ||
242              (*in2 && *wrap <= *in2)))
243                 die(wrap_arg_usage);
244 }
245
246 int cmd_shortlog(int argc, const char **argv, const char *prefix)
247 {
248         struct rev_info rev;
249         struct path_list list = { NULL, 0, 0, 1 };
250         int i, j, sort_by_number = 0, summary = 0;
251         int wrap_lines = 0;
252         int wrap = DEFAULT_WRAPLEN;
253         int in1 = DEFAULT_INDENT1;
254         int in2 = DEFAULT_INDENT2;
255
256         /* since -n is a shadowed rev argument, parse our args first */
257         while (argc > 1) {
258                 if (!strcmp(argv[1], "-n") || !strcmp(argv[1], "--numbered"))
259                         sort_by_number = 1;
260                 else if (!strcmp(argv[1], "-s") ||
261                                 !strcmp(argv[1], "--summary"))
262                         summary = 1;
263                 else if (!prefixcmp(argv[1], "-w")) {
264                         wrap_lines = 1;
265                         parse_wrap_args(argv[1], &in1, &in2, &wrap);
266                 }
267                 else if (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help"))
268                         usage(shortlog_usage);
269                 else
270                         break;
271                 argv++;
272                 argc--;
273         }
274         init_revisions(&rev, prefix);
275         argc = setup_revisions(argc, argv, &rev, NULL);
276         if (argc > 1)
277                 die ("unrecognized argument: %s", argv[1]);
278
279         read_mailmap(&mailmap, ".mailmap", &common_repo_prefix);
280
281         if (rev.pending.nr == 0) {
282                 if (isatty(0))
283                         fprintf(stderr, "(reading log to summarize from standard input)\n");
284                 read_from_stdin(&list);
285         }
286         else
287                 get_from_rev(&rev, &list);
288
289         if (sort_by_number)
290                 qsort(list.items, list.nr, sizeof(struct path_list_item),
291                         compare_by_number);
292
293         for (i = 0; i < list.nr; i++) {
294                 struct path_list *onelines = list.items[i].util;
295
296                 if (summary) {
297                         printf("%s: %d\n", list.items[i].path, onelines->nr);
298                 } else {
299                         printf("%s (%d):\n", list.items[i].path, onelines->nr);
300                         for (j = onelines->nr - 1; j >= 0; j--) {
301                                 const char *msg = onelines->items[j].path;
302
303                                 if (wrap_lines) {
304                                         int col = print_wrapped_text(msg, in1, in2, wrap);
305                                         if (col != wrap)
306                                                 putchar('\n');
307                                 }
308                                 else
309                                         printf("      %s\n", msg);
310                         }
311                         putchar('\n');
312                 }
313
314                 onelines->strdup_paths = 1;
315                 path_list_clear(onelines, 1);
316                 free(onelines);
317                 list.items[i].util = NULL;
318         }
319
320         list.strdup_paths = 1;
321         path_list_clear(&list, 1);
322         mailmap.strdup_paths = 1;
323         path_list_clear(&mailmap, 1);
324
325         return 0;
326 }