mailmap: handle mailmap blobs without trailing newlines
[git] / mailmap.c
1 #include "cache.h"
2 #include "string-list.h"
3 #include "mailmap.h"
4
5 #define DEBUG_MAILMAP 0
6 #if DEBUG_MAILMAP
7 #define debug_mm(...) fprintf(stderr, __VA_ARGS__)
8 #else
9 static inline void debug_mm(const char *format, ...) {}
10 #endif
11
12 const char *git_mailmap_file;
13 const char *git_mailmap_blob;
14
15 struct mailmap_info {
16         char *name;
17         char *email;
18 };
19
20 struct mailmap_entry {
21         /* name and email for the simple mail-only case */
22         char *name;
23         char *email;
24
25         /* name and email for the complex mail and name matching case */
26         struct string_list namemap;
27 };
28
29 static void free_mailmap_info(void *p, const char *s)
30 {
31         struct mailmap_info *mi = (struct mailmap_info *)p;
32         debug_mm("mailmap: -- complex: '%s' -> '%s' <%s>\n", s, mi->name, mi->email);
33         free(mi->name);
34         free(mi->email);
35 }
36
37 static void free_mailmap_entry(void *p, const char *s)
38 {
39         struct mailmap_entry *me = (struct mailmap_entry *)p;
40         debug_mm("mailmap: removing entries for <%s>, with %d sub-entries\n", s, me->namemap.nr);
41         debug_mm("mailmap: - simple: '%s' <%s>\n", me->name, me->email);
42         free(me->name);
43         free(me->email);
44
45         me->namemap.strdup_strings = 1;
46         string_list_clear_func(&me->namemap, free_mailmap_info);
47 }
48
49 static void add_mapping(struct string_list *map,
50                         char *new_name, char *new_email, char *old_name, char *old_email)
51 {
52         struct mailmap_entry *me;
53         int index;
54         char *p;
55
56         if (old_email)
57                 for (p = old_email; *p; p++)
58                         *p = tolower(*p);
59         if (new_email)
60                 for (p = new_email; *p; p++)
61                         *p = tolower(*p);
62
63         if (old_email == NULL) {
64                 old_email = new_email;
65                 new_email = NULL;
66         }
67
68         if ((index = string_list_find_insert_index(map, old_email, 1)) < 0) {
69                 /* mailmap entry exists, invert index value */
70                 index = -1 - index;
71         } else {
72                 /* create mailmap entry */
73                 struct string_list_item *item = string_list_insert_at_index(map, index, old_email);
74                 item->util = xcalloc(1, sizeof(struct mailmap_entry));
75                 ((struct mailmap_entry *)item->util)->namemap.strdup_strings = 1;
76         }
77         me = (struct mailmap_entry *)map->items[index].util;
78
79         if (old_name == NULL) {
80                 debug_mm("mailmap: adding (simple) entry for %s at index %d\n", old_email, index);
81                 /* Replace current name and new email for simple entry */
82                 if (new_name) {
83                         free(me->name);
84                         me->name = xstrdup(new_name);
85                 }
86                 if (new_email) {
87                         free(me->email);
88                         me->email = xstrdup(new_email);
89                 }
90         } else {
91                 struct mailmap_info *mi = xcalloc(1, sizeof(struct mailmap_info));
92                 debug_mm("mailmap: adding (complex) entry for %s at index %d\n", old_email, index);
93                 if (new_name)
94                         mi->name = xstrdup(new_name);
95                 if (new_email)
96                         mi->email = xstrdup(new_email);
97                 string_list_insert(&me->namemap, old_name)->util = mi;
98         }
99
100         debug_mm("mailmap:  '%s' <%s> -> '%s' <%s>\n",
101                  old_name, old_email, new_name, new_email);
102 }
103
104 static char *parse_name_and_email(char *buffer, char **name,
105                 char **email, int allow_empty_email)
106 {
107         char *left, *right, *nstart, *nend;
108         *name = *email = NULL;
109
110         if ((left = strchr(buffer, '<')) == NULL)
111                 return NULL;
112         if ((right = strchr(left+1, '>')) == NULL)
113                 return NULL;
114         if (!allow_empty_email && (left+1 == right))
115                 return NULL;
116
117         /* remove whitespace from beginning and end of name */
118         nstart = buffer;
119         while (isspace(*nstart) && nstart < left)
120                 ++nstart;
121         nend = left-1;
122         while (nend > nstart && isspace(*nend))
123                 --nend;
124
125         *name = (nstart < nend ? nstart : NULL);
126         *email = left+1;
127         *(nend+1) = '\0';
128         *right++ = '\0';
129
130         return (*right == '\0' ? NULL : right);
131 }
132
133 static void read_mailmap_line(struct string_list *map, char *buffer,
134                               char **repo_abbrev)
135 {
136         char *name1 = NULL, *email1 = NULL, *name2 = NULL, *email2 = NULL;
137         if (buffer[0] == '#') {
138                 static const char abbrev[] = "# repo-abbrev:";
139                 int abblen = sizeof(abbrev) - 1;
140                 int len = strlen(buffer);
141
142                 if (!repo_abbrev)
143                         return;
144
145                 if (len && buffer[len - 1] == '\n')
146                         buffer[--len] = 0;
147                 if (!strncmp(buffer, abbrev, abblen)) {
148                         char *cp;
149
150                         if (repo_abbrev)
151                                 free(*repo_abbrev);
152                         *repo_abbrev = xmalloc(len);
153
154                         for (cp = buffer + abblen; isspace(*cp); cp++)
155                                 ; /* nothing */
156                         strcpy(*repo_abbrev, cp);
157                 }
158                 return;
159         }
160         if ((name2 = parse_name_and_email(buffer, &name1, &email1, 0)) != NULL)
161                 parse_name_and_email(name2, &name2, &email2, 1);
162
163         if (email1)
164                 add_mapping(map, name1, email1, name2, email2);
165 }
166
167 static int read_mailmap_file(struct string_list *map, const char *filename,
168                              char **repo_abbrev)
169 {
170         char buffer[1024];
171         FILE *f;
172
173         if (!filename)
174                 return 0;
175
176         f = fopen(filename, "r");
177         if (!f) {
178                 if (errno == ENOENT)
179                         return 0;
180                 return error("unable to open mailmap at %s: %s",
181                              filename, strerror(errno));
182         }
183
184         while (fgets(buffer, sizeof(buffer), f) != NULL)
185                 read_mailmap_line(map, buffer, repo_abbrev);
186         fclose(f);
187         return 0;
188 }
189
190 static void read_mailmap_string(struct string_list *map, char *buf,
191                                 char **repo_abbrev)
192 {
193         while (*buf) {
194                 char *end = strchrnul(buf, '\n');
195
196                 if (*end)
197                         *end++ = '\0';
198
199                 read_mailmap_line(map, buf, repo_abbrev);
200                 buf = end;
201         }
202 }
203
204 static int read_mailmap_blob(struct string_list *map,
205                              const char *name,
206                              char **repo_abbrev)
207 {
208         unsigned char sha1[20];
209         char *buf;
210         unsigned long size;
211         enum object_type type;
212
213         if (!name)
214                 return 0;
215         if (get_sha1(name, sha1) < 0)
216                 return 0;
217
218         buf = read_sha1_file(sha1, &type, &size);
219         if (!buf)
220                 return error("unable to read mailmap object at %s", name);
221         if (type != OBJ_BLOB)
222                 return error("mailmap is not a blob: %s", name);
223
224         read_mailmap_string(map, buf, repo_abbrev);
225
226         free(buf);
227         return 0;
228 }
229
230 int read_mailmap(struct string_list *map, char **repo_abbrev)
231 {
232         int err = 0;
233
234         map->strdup_strings = 1;
235         map->cmp = strcasecmp;
236
237         if (!git_mailmap_blob && is_bare_repository())
238                 git_mailmap_blob = "HEAD:.mailmap";
239
240         err |= read_mailmap_file(map, ".mailmap", repo_abbrev);
241         err |= read_mailmap_blob(map, git_mailmap_blob, repo_abbrev);
242         err |= read_mailmap_file(map, git_mailmap_file, repo_abbrev);
243         return err;
244 }
245
246 void clear_mailmap(struct string_list *map)
247 {
248         debug_mm("mailmap: clearing %d entries...\n", map->nr);
249         map->strdup_strings = 1;
250         string_list_clear_func(map, free_mailmap_entry);
251         debug_mm("mailmap: cleared\n");
252 }
253
254 /*
255  * Look for an entry in map that match string[0:len]; string[len]
256  * does not have to be NUL (but it could be).
257  */
258 static struct string_list_item *lookup_prefix(struct string_list *map,
259                                               const char *string, size_t len)
260 {
261         int i = string_list_find_insert_index(map, string, 1);
262         if (i < 0) {
263                 /* exact match */
264                 i = -1 - i;
265                 if (!string[len])
266                         return &map->items[i];
267                 /*
268                  * that map entry matches exactly to the string, including
269                  * the cruft at the end beyond "len".  That is not a match
270                  * with string[0:len] that we are looking for.
271                  */
272         } else if (!string[len]) {
273                 /*
274                  * asked with the whole string, and got nothing.  No
275                  * matching entry can exist in the map.
276                  */
277                 return NULL;
278         }
279
280         /*
281          * i is at the exact match to an overlong key, or location the
282          * overlong key would be inserted, which must come after the
283          * real location of the key if one exists.
284          */
285         while (0 <= --i && i < map->nr) {
286                 int cmp = strncasecmp(map->items[i].string, string, len);
287                 if (cmp < 0)
288                         /*
289                          * "i" points at a key definitely below the prefix;
290                          * the map does not have string[0:len] in it.
291                          */
292                         break;
293                 else if (!cmp && !map->items[i].string[len])
294                         /* found it */
295                         return &map->items[i];
296                 /*
297                  * otherwise, the string at "i" may be string[0:len]
298                  * followed by a string that sorts later than string[len:];
299                  * keep trying.
300                  */
301         }
302         return NULL;
303 }
304
305 int map_user(struct string_list *map,
306                          const char **email, size_t *emaillen,
307                          const char **name, size_t *namelen)
308 {
309         struct string_list_item *item;
310         struct mailmap_entry *me;
311
312         debug_mm("map_user: map '%.*s' <%.*s>\n",
313                  *name, *namelen, *emaillen, *email);
314
315         item = lookup_prefix(map, *email, *emaillen);
316         if (item != NULL) {
317                 me = (struct mailmap_entry *)item->util;
318                 if (me->namemap.nr) {
319                         /* The item has multiple items, so we'll look up on name too */
320                         /* If the name is not found, we choose the simple entry      */
321                         struct string_list_item *subitem;
322                         subitem = lookup_prefix(&me->namemap, *name, *namelen);
323                         if (subitem)
324                                 item = subitem;
325                 }
326         }
327         if (item != NULL) {
328                 struct mailmap_info *mi = (struct mailmap_info *)item->util;
329                 if (mi->name == NULL && mi->email == NULL) {
330                         debug_mm("map_user:  -- (no simple mapping)\n");
331                         return 0;
332                 }
333                 if (mi->email) {
334                                 *email = mi->email;
335                                 *emaillen = strlen(*email);
336                 }
337                 if (mi->name) {
338                                 *name = mi->name;
339                                 *namelen = strlen(*name);
340                 }
341                 debug_mm("map_user:  to '%.*s' <.*%s>\n", *namelen, *name,
342                                  *emaillen, *email);
343                 return 1;
344         }
345         debug_mm("map_user:  --\n");
346         return 0;
347 }