parse-opt: add PARSE_OPT_KEEP_ARGV0 parser option.
[gitweb.git] / builtin-shortlog.c
index 3fe754677d3f7ab11419a04dd828c70b5958ed87..e6a2865019cceadfcbfc8575a2bd1f97b7159dcb 100644 (file)
@@ -6,11 +6,10 @@
 #include "revision.h"
 #include "utf8.h"
 #include "mailmap.h"
+#include "shortlog.h"
 
 static const char shortlog_usage[] =
-"git-shortlog [-n] [-s] [<commit-id>... ]";
-
-static char *common_repo_prefix;
+"git-shortlog [-n] [-s] [-e] [-w] [<commit-id>... ]";
 
 static int compare_by_number(const void *a1, const void *a2)
 {
@@ -25,47 +24,69 @@ static int compare_by_number(const void *a1, const void *a2)
                return -1;
 }
 
-static struct path_list mailmap = {NULL, 0, 0, 0};
-
-static void insert_author_oneline(struct path_list *list,
-               const char *author, int authorlen,
-               const char *oneline, int onelinelen)
+static void insert_one_record(struct shortlog *log,
+                             const char *author,
+                             const char *oneline)
 {
-       const char *dot3 = common_repo_prefix;
+       const char *dot3 = log->common_repo_prefix;
        char *buffer, *p;
        struct path_list_item *item;
        struct path_list *onelines;
+       char namebuf[1024];
+       size_t len;
+       const char *eol;
+       const char *boemail, *eoemail;
+
+       boemail = strchr(author, '<');
+       if (!boemail)
+               return;
+       eoemail = strchr(boemail, '>');
+       if (!eoemail)
+               return;
+       if (!map_email(&log->mailmap, boemail+1, namebuf, sizeof(namebuf))) {
+               while (author < boemail && isspace(*author))
+                       author++;
+               for (len = 0;
+                    len < sizeof(namebuf) - 1 && author + len < boemail;
+                    len++)
+                       namebuf[len] = author[len];
+               while (0 < len && isspace(namebuf[len-1]))
+                       len--;
+               namebuf[len] = '\0';
+       }
+       else
+               len = strlen(namebuf);
 
-       while (authorlen > 0 && isspace(author[authorlen - 1]))
-               authorlen--;
+       if (log->email) {
+               size_t room = sizeof(namebuf) - len - 1;
+               int maillen = eoemail - boemail + 1;
+               snprintf(namebuf + len, room, " %.*s", maillen, boemail);
+       }
 
-       buffer = xmemdupz(author, authorlen);
-       item = path_list_insert(buffer, list);
+       buffer = xstrdup(namebuf);
+       item = path_list_insert(buffer, &log->list);
        if (item->util == NULL)
                item->util = xcalloc(1, sizeof(struct path_list));
        else
                free(buffer);
 
+       /* Skip any leading whitespace, including any blank lines. */
+       while (*oneline && isspace(*oneline))
+               oneline++;
+       eol = strchr(oneline, '\n');
+       if (!eol)
+               eol = oneline + strlen(oneline);
        if (!prefixcmp(oneline, "[PATCH")) {
                char *eob = strchr(oneline, ']');
-
-               if (eob) {
-                       while (isspace(eob[1]) && eob[1] != '\n')
-                               eob++;
-                       if (eob - oneline < onelinelen) {
-                               onelinelen -= eob - oneline;
-                               oneline = eob;
-                       }
-               }
+               if (eob && (!eol || eob < eol))
+                       oneline = eob + 1;
        }
-
-       while (onelinelen > 0 && isspace(oneline[0])) {
+       while (*oneline && isspace(*oneline) && *oneline != '\n')
                oneline++;
-               onelinelen--;
-       }
-       while (onelinelen > 0 && isspace(oneline[onelinelen - 1]))
-               onelinelen--;
-       buffer = xmemdupz(oneline, onelinelen);
+       len = eol - oneline;
+       while (len && isspace(oneline[len-1]))
+               len--;
+       buffer = xmemdupz(oneline, len);
 
        if (dot3) {
                int dot3len = strlen(dot3);
@@ -90,108 +111,57 @@ static void insert_author_oneline(struct path_list *list,
        onelines->items[onelines->nr++].path = buffer;
 }
 
-static void read_from_stdin(struct path_list *list)
+static void read_from_stdin(struct shortlog *log)
 {
-       char buffer[1024];
-
-       while (fgets(buffer, sizeof(buffer), stdin) != NULL) {
-               char *bob;
-               if ((buffer[0] == 'A' || buffer[0] == 'a') &&
-                               !prefixcmp(buffer + 1, "uthor: ") &&
-                               (bob = strchr(buffer + 7, '<')) != NULL) {
-                       char buffer2[1024], offset = 0;
-
-                       if (map_email(&mailmap, bob + 1, buffer, sizeof(buffer)))
-                               bob = buffer + strlen(buffer);
-                       else {
-                               offset = 8;
-                               while (buffer + offset < bob &&
-                                      isspace(bob[-1]))
-                                       bob--;
-                       }
-
-                       while (fgets(buffer2, sizeof(buffer2), stdin) &&
-                                       buffer2[0] != '\n')
-                               ; /* chomp input */
-                       if (fgets(buffer2, sizeof(buffer2), stdin)) {
-                               int l2 = strlen(buffer2);
-                               int i;
-                               for (i = 0; i < l2; i++)
-                                       if (!isspace(buffer2[i]))
-                                               break;
-                               insert_author_oneline(list,
-                                               buffer + offset,
-                                               bob - buffer - offset,
-                                               buffer2 + i, l2 - i);
-                       }
-               }
+       char author[1024], oneline[1024];
+
+       while (fgets(author, sizeof(author), stdin) != NULL) {
+               if (!(author[0] == 'A' || author[0] == 'a') ||
+                   prefixcmp(author + 1, "uthor: "))
+                       continue;
+               while (fgets(oneline, sizeof(oneline), stdin) &&
+                      oneline[0] != '\n')
+                       ; /* discard headers */
+               while (fgets(oneline, sizeof(oneline), stdin) &&
+                      oneline[0] == '\n')
+                       ; /* discard blanks */
+               insert_one_record(log, author + 8, oneline);
        }
 }
 
-static void get_from_rev(struct rev_info *rev, struct path_list *list)
+void shortlog_add_commit(struct shortlog *log, struct commit *commit)
 {
-       char scratch[1024];
-       struct commit *commit;
+       const char *author = NULL, *buffer;
 
-       prepare_revision_walk(rev);
-       while ((commit = get_revision(rev)) != NULL) {
-               const char *author = NULL, *oneline, *buffer;
-               int authorlen = authorlen, onelinelen;
-
-               /* get author and oneline */
-               for (buffer = commit->buffer; buffer && *buffer != '\0' &&
-                               *buffer != '\n'; ) {
-                       const char *eol = strchr(buffer, '\n');
-
-                       if (eol == NULL)
-                               eol = buffer + strlen(buffer);
-                       else
-                               eol++;
-
-                       if (!prefixcmp(buffer, "author ")) {
-                               char *bracket = strchr(buffer, '<');
-
-                               if (bracket == NULL || bracket > eol)
-                                       die("Invalid commit buffer: %s",
-                                           sha1_to_hex(commit->object.sha1));
-
-                               if (map_email(&mailmap, bracket + 1, scratch,
-                                                       sizeof(scratch))) {
-                                       author = scratch;
-                                       authorlen = strlen(scratch);
-                               } else {
-                                       if (bracket[-1] == ' ')
-                                               bracket--;
-
-                                       author = buffer + 7;
-                                       authorlen = bracket - buffer - 7;
-                               }
-                       }
-                       buffer = eol;
-               }
+       buffer = commit->buffer;
+       while (*buffer && *buffer != '\n') {
+               const char *eol = strchr(buffer, '\n');
 
-               if (author == NULL)
-                       die ("Missing author: %s",
-                                       sha1_to_hex(commit->object.sha1));
-
-               if (buffer == NULL || *buffer == '\0') {
-                       oneline = "<none>";
-                       onelinelen = sizeof(oneline) + 1;
-               } else {
-                       char *eol;
-
-                       oneline = buffer + 1;
-                       eol = strchr(oneline, '\n');
-                       if (eol == NULL)
-                               onelinelen = strlen(oneline);
-                       else
-                               onelinelen = eol - oneline;
-               }
+               if (eol == NULL)
+                       eol = buffer + strlen(buffer);
+               else
+                       eol++;
 
-               insert_author_oneline(list,
-                               author, authorlen, oneline, onelinelen);
+               if (!prefixcmp(buffer, "author "))
+                       author = buffer + 7;
+               buffer = eol;
        }
+       if (!author)
+               die("Missing author: %s",
+                   sha1_to_hex(commit->object.sha1));
+       if (*buffer)
+               buffer++;
+       insert_one_record(log, author, !*buffer ? "<none>" : buffer);
+}
+
+static void get_from_rev(struct rev_info *rev, struct shortlog *log)
+{
+       struct commit *commit;
 
+       if (prepare_revision_walk(rev))
+               die("revision walk setup failed");
+       while ((commit = get_revision(rev)) != NULL)
+               shortlog_add_commit(log, commit);
 }
 
 static int parse_uint(char const **arg, int comma)
@@ -243,26 +213,40 @@ static void parse_wrap_args(const char *arg, int *in1, int *in2, int *wrap)
                die(wrap_arg_usage);
 }
 
+void shortlog_init(struct shortlog *log)
+{
+       memset(log, 0, sizeof(*log));
+
+       read_mailmap(&log->mailmap, ".mailmap", &log->common_repo_prefix);
+
+       log->list.strdup_paths = 1;
+       log->wrap = DEFAULT_WRAPLEN;
+       log->in1 = DEFAULT_INDENT1;
+       log->in2 = DEFAULT_INDENT2;
+}
+
 int cmd_shortlog(int argc, const char **argv, const char *prefix)
 {
+       struct shortlog log;
        struct rev_info rev;
-       struct path_list list = { NULL, 0, 0, 1 };
-       int i, j, sort_by_number = 0, summary = 0;
-       int wrap_lines = 0;
-       int wrap = DEFAULT_WRAPLEN;
-       int in1 = DEFAULT_INDENT1;
-       int in2 = DEFAULT_INDENT2;
+       int nongit;
+
+       prefix = setup_git_directory_gently(&nongit);
+       shortlog_init(&log);
 
        /* since -n is a shadowed rev argument, parse our args first */
        while (argc > 1) {
                if (!strcmp(argv[1], "-n") || !strcmp(argv[1], "--numbered"))
-                       sort_by_number = 1;
+                       log.sort_by_number = 1;
                else if (!strcmp(argv[1], "-s") ||
                                !strcmp(argv[1], "--summary"))
-                       summary = 1;
+                       log.summary = 1;
+               else if (!strcmp(argv[1], "-e") ||
+                        !strcmp(argv[1], "--email"))
+                       log.email = 1;
                else if (!prefixcmp(argv[1], "-w")) {
-                       wrap_lines = 1;
-                       parse_wrap_args(argv[1], &in1, &in2, &wrap);
+                       log.wrap_lines = 1;
+                       parse_wrap_args(argv[1], &log.in1, &log.in2, &log.wrap);
                }
                else if (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help"))
                        usage(shortlog_usage);
@@ -276,33 +260,38 @@ int cmd_shortlog(int argc, const char **argv, const char *prefix)
        if (argc > 1)
                die ("unrecognized argument: %s", argv[1]);
 
-       read_mailmap(&mailmap, ".mailmap", &common_repo_prefix);
-
+       /* assume HEAD if from a tty */
+       if (!nongit && !rev.pending.nr && isatty(0))
+               add_head_to_pending(&rev);
        if (rev.pending.nr == 0) {
-               if (isatty(0))
-                       fprintf(stderr, "(reading log to summarize from standard input)\n");
-               read_from_stdin(&list);
+               read_from_stdin(&log);
        }
        else
-               get_from_rev(&rev, &list);
+               get_from_rev(&rev, &log);
 
-       if (sort_by_number)
-               qsort(list.items, list.nr, sizeof(struct path_list_item),
-                       compare_by_number);
+       shortlog_output(&log);
+       return 0;
+}
 
-       for (i = 0; i < list.nr; i++) {
-               struct path_list *onelines = list.items[i].util;
+void shortlog_output(struct shortlog *log)
+{
+       int i, j;
+       if (log->sort_by_number)
+               qsort(log->list.items, log->list.nr, sizeof(struct path_list_item),
+                       compare_by_number);
+       for (i = 0; i < log->list.nr; i++) {
+               struct path_list *onelines = log->list.items[i].util;
 
-               if (summary) {
-                       printf("%s: %d\n", list.items[i].path, onelines->nr);
+               if (log->summary) {
+                       printf("%6d\t%s\n", onelines->nr, log->list.items[i].path);
                } else {
-                       printf("%s (%d):\n", list.items[i].path, onelines->nr);
+                       printf("%s (%d):\n", log->list.items[i].path, onelines->nr);
                        for (j = onelines->nr - 1; j >= 0; j--) {
                                const char *msg = onelines->items[j].path;
 
-                               if (wrap_lines) {
-                                       int col = print_wrapped_text(msg, in1, in2, wrap);
-                                       if (col != wrap)
+                               if (log->wrap_lines) {
+                                       int col = print_wrapped_text(msg, log->in1, log->in2, log->wrap);
+                                       if (col != log->wrap)
                                                putchar('\n');
                                }
                                else
@@ -314,13 +303,11 @@ int cmd_shortlog(int argc, const char **argv, const char *prefix)
                onelines->strdup_paths = 1;
                path_list_clear(onelines, 1);
                free(onelines);
-               list.items[i].util = NULL;
+               log->list.items[i].util = NULL;
        }
 
-       list.strdup_paths = 1;
-       path_list_clear(&list, 1);
-       mailmap.strdup_paths = 1;
-       path_list_clear(&mailmap, 1);
-
-       return 0;
+       log->list.strdup_paths = 1;
+       path_list_clear(&log->list, 1);
+       log->mailmap.strdup_paths = 1;
+       path_list_clear(&log->mailmap, 1);
 }