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