Andrew's git - gitweb.git/blob - 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 <string.h>
   8
   9static const char shortlog_usage[] =
  10"git-shortlog [-n] [-s] [<commit-id>... ]";
  11
  12static int compare_by_number(const void *a1, const void *a2)
  13{
  14        const struct path_list_item *i1 = a1, *i2 = a2;
  15        const struct path_list *l1 = i1->util, *l2 = i2->util;
  16
  17        if (l1->nr < l2->nr)
  18                return 1;
  19        else if (l1->nr == l2->nr)
  20                return 0;
  21        else
  22                return -1;
  23}
  24
  25static struct path_list mailmap = {NULL, 0, 0, 0};
  26
  27static int read_mailmap(const char *filename)
  28{
  29        char buffer[1024];
  30        FILE *f = fopen(filename, "r");
  31
  32        if (f == NULL)
  33                return 1;
  34        while (fgets(buffer, sizeof(buffer), f) != NULL) {
  35                char *end_of_name, *left_bracket, *right_bracket;
  36                char *name, *email;
  37                int i;
  38                if (buffer[0] == '#')
  39                        continue;
  40                if ((left_bracket = strchr(buffer, '<')) == NULL)
  41                        continue;
  42                if ((right_bracket = strchr(left_bracket + 1, '>')) == NULL)
  43                        continue;
  44                if (right_bracket == left_bracket + 1)
  45                        continue;
  46                for (end_of_name = left_bracket; end_of_name != buffer
  47                                && isspace(end_of_name[-1]); end_of_name--)
  48                        /* keep on looking */
  49                if (end_of_name == buffer)
  50                        continue;
  51                name = xmalloc(end_of_name - buffer + 1);
  52                strlcpy(name, buffer, end_of_name - buffer + 1);
  53                email = xmalloc(right_bracket - left_bracket);
  54                for (i = 0; i < right_bracket - left_bracket - 1; i++)
  55                        email[i] = tolower(left_bracket[i + 1]);
  56                email[right_bracket - left_bracket - 1] = '\0';
  57                path_list_insert(email, &mailmap)->util = name;
  58        }
  59        fclose(f);
  60        return 0;
  61}
  62
  63static int map_email(char *email, char *name, int maxlen)
  64{
  65        char *p;
  66        struct path_list_item *item;
  67
  68        /* autocomplete common developers */
  69        p = strchr(email, '>');
  70        if (!p)
  71                return 0;
  72
  73        *p = '\0';
  74        /* downcase the email address */
  75        for (p = email; *p; p++)
  76                *p = tolower(*p);
  77        item = path_list_lookup(email, &mailmap);
  78        if (item != NULL) {
  79                const char *realname = (const char *)item->util;
  80                strncpy(name, realname, maxlen);
  81                return 1;
  82        }
  83        return 0;
  84}
  85
  86static void insert_author_oneline(struct path_list *list,
  87                const char *author, int authorlen,
  88                const char *oneline, int onelinelen)
  89{
  90        const char *dot3 = "/pub/scm/linux/kernel/git/";
  91        char *buffer, *p;
  92        struct path_list_item *item;
  93        struct path_list *onelines;
  94
  95        while (authorlen > 0 && isspace(author[authorlen - 1]))
  96                authorlen--;
  97
  98        buffer = xmalloc(authorlen + 1);
  99        memcpy(buffer, author, authorlen);
 100        buffer[authorlen] = '\0';
 101
 102        item = path_list_insert(buffer, list);
 103        if (item->util == NULL)
 104                item->util = xcalloc(1, sizeof(struct path_list));
 105        else
 106                free(buffer);
 107
 108        if (!strncmp(oneline, "[PATCH", 6)) {
 109                char *eob = strchr(oneline, ']');
 110
 111                if (eob) {
 112                        while (isspace(eob[1]) && eob[1] != '\n')
 113                                eob++;
 114                        if (eob - oneline < onelinelen) {
 115                                onelinelen -= eob - oneline;
 116                                oneline = eob;
 117                        }
 118                }
 119        }
 120
 121        while (onelinelen > 0 && isspace(oneline[0])) {
 122                oneline++;
 123                onelinelen--;
 124        }
 125
 126        while (onelinelen > 0 && isspace(oneline[onelinelen - 1]))
 127                onelinelen--;
 128
 129        buffer = xmalloc(onelinelen + 1);
 130        memcpy(buffer, oneline, onelinelen);
 131        buffer[onelinelen] = '\0';
 132
 133        while ((p = strstr(buffer, dot3)) != NULL) {
 134                memcpy(p, "...", 3);
 135                strcpy(p + 2, p + sizeof(dot3) - 1);
 136        }
 137
 138
 139        onelines = item->util;
 140        if (onelines->nr >= onelines->alloc) {
 141                onelines->alloc = alloc_nr(onelines->nr);
 142                onelines->items = xrealloc(onelines->items,
 143                                onelines->alloc
 144                                * sizeof(struct path_list_item));
 145        }
 146
 147        onelines->items[onelines->nr].util = NULL;
 148        onelines->items[onelines->nr++].path = buffer;
 149}
 150
 151static void read_from_stdin(struct path_list *list)
 152{
 153        char buffer[1024];
 154
 155        while (fgets(buffer, sizeof(buffer), stdin) != NULL) {
 156                char *bob;
 157                if ((buffer[0] == 'A' || buffer[0] == 'a') &&
 158                                !strncmp(buffer + 1, "uthor: ", 7) &&
 159                                (bob = strchr(buffer + 7, '<')) != NULL) {
 160                        char buffer2[1024], offset = 0;
 161
 162                        if (map_email(bob + 1, buffer, sizeof(buffer)))
 163                                bob = buffer + strlen(buffer);
 164                        else {
 165                                offset = 8;
 166                                while (isspace(bob[-1]))
 167                                        bob--;
 168                        }
 169
 170                        while (fgets(buffer2, sizeof(buffer2), stdin) &&
 171                                        buffer2[0] != '\n')
 172                                ; /* chomp input */
 173                        if (fgets(buffer2, sizeof(buffer2), stdin))
 174                                insert_author_oneline(list,
 175                                                buffer + offset,
 176                                                bob - buffer - offset,
 177                                                buffer2, strlen(buffer2));
 178                }
 179        }
 180}
 181
 182static void get_from_rev(struct rev_info *rev, struct path_list *list)
 183{
 184        char scratch[1024];
 185        struct commit *commit;
 186
 187        prepare_revision_walk(rev);
 188        while ((commit = get_revision(rev)) != NULL) {
 189                char *author = NULL, *oneline, *buffer;
 190                int authorlen = authorlen, onelinelen;
 191
 192                /* get author and oneline */
 193                for (buffer = commit->buffer; buffer && *buffer != '\0' &&
 194                                *buffer != '\n'; ) {
 195                        char *eol = strchr(buffer, '\n');
 196
 197                        if (eol == NULL)
 198                                eol = buffer + strlen(buffer);
 199                        else
 200                                eol++;
 201
 202                        if (!strncmp(buffer, "author ", 7)) {
 203                                char *bracket = strchr(buffer, '<');
 204
 205                                if (bracket == NULL || bracket > eol)
 206                                        die("Invalid commit buffer: %s",
 207                                            sha1_to_hex(commit->object.sha1));
 208
 209                                if (map_email(bracket + 1, scratch,
 210                                                        sizeof(scratch))) {
 211                                        author = scratch;
 212                                        authorlen = strlen(scratch);
 213                                } else {
 214                                        while (bracket[-1] == ' ')
 215                                                bracket--;
 216
 217                                        author = buffer + 7;
 218                                        authorlen = bracket - buffer - 7;
 219                                }
 220                        }
 221                        buffer = eol;
 222                }
 223
 224                if (author == NULL)
 225                        die ("Missing author: %s",
 226                                        sha1_to_hex(commit->object.sha1));
 227
 228                if (buffer == NULL || *buffer == '\0') {
 229                        oneline = "<none>";
 230                        onelinelen = sizeof(oneline) + 1;
 231                } else {
 232                        char *eol;
 233
 234                        oneline = buffer + 1;
 235                        eol = strchr(oneline, '\n');
 236                        if (eol == NULL)
 237                                onelinelen = strlen(oneline);
 238                        else
 239                                onelinelen = eol - oneline;
 240                }
 241
 242                insert_author_oneline(list,
 243                                author, authorlen, oneline, onelinelen);
 244        }
 245
 246}
 247
 248int cmd_shortlog(int argc, const char **argv, const char *prefix)
 249{
 250        struct rev_info rev;
 251        struct path_list list = { NULL, 0, 0, 1 };
 252        int i, j, sort_by_number = 0, summary = 0;
 253
 254        /* since -n is a shadowed rev argument, parse our args first */
 255        while (argc > 1) {
 256                if (!strcmp(argv[1], "-n") || !strcmp(argv[1], "--numbered"))
 257                        sort_by_number = 1;
 258                else if (!strcmp(argv[1], "-s") ||
 259                                !strcmp(argv[1], "--summary"))
 260                        summary = 1;
 261                else if (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help"))
 262                        usage(shortlog_usage);
 263                else
 264                        break;
 265                argv++;
 266                argc--;
 267        }
 268        init_revisions(&rev, prefix);
 269        argc = setup_revisions(argc, argv, &rev, NULL);
 270        if (argc > 1)
 271                die ("unrecognized argument: %s", argv[1]);
 272
 273        if (!access(".mailmap", R_OK))
 274                read_mailmap(".mailmap");
 275
 276        if (rev.pending.nr == 1)
 277                die ("Need a range!");
 278        else if (rev.pending.nr == 0)
 279                read_from_stdin(&list);
 280        else
 281                get_from_rev(&rev, &list);
 282
 283        if (sort_by_number)
 284                qsort(list.items, list.nr, sizeof(struct path_list_item),
 285                        compare_by_number);
 286
 287        for (i = 0; i < list.nr; i++) {
 288                struct path_list *onelines = list.items[i].util;
 289
 290                if (summary) {
 291                        printf("%s: %d\n", list.items[i].path, onelines->nr);
 292                } else {
 293                        printf("%s (%d):\n", list.items[i].path, onelines->nr);
 294                        for (j = onelines->nr - 1; j >= 0; j--)
 295                                printf("      %s\n", onelines->items[j].path);
 296                        printf("\n");
 297                }
 298
 299                onelines->strdup_paths = 1;
 300                path_list_clear(onelines, 1);
 301                free(onelines);
 302                list.items[i].util = NULL;
 303        }
 304
 305        list.strdup_paths = 1;
 306        path_list_clear(&list, 1);
 307        mailmap.strdup_paths = 1;
 308        path_list_clear(&mailmap, 1);
 309
 310        return 0;
 311}
 312