builtin-shortlog.con commit Reduce the number of connects when fetching (ba22785)
   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
  10static const char shortlog_usage[] =
  11"git-shortlog [-n] [-s] [-e] [<commit-id>... ]";
  12
  13static char *common_repo_prefix;
  14static int email;
  15
  16static 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
  29static struct path_list mailmap = {NULL, 0, 0, 0};
  30
  31static 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
 117static 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
 135static 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
 165static 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
 183static 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
 188static 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
 214int 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}