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