builtin/fmt-merge-msg.c

   1 #include "builtin.h"
   2 #include "cache.h"
   3 #include "commit.h"
   4 #include "diff.h"
   5 #include "revision.h"
   6 #include "tag.h"
   7 #include "string-list.h"
   8
   9 static const char * const fmt_merge_msg_usage[] = {
  10         "git fmt-merge-msg [--log|--no-log] [--file <file>]",
  11         NULL
  12 };
  13
  14 static int merge_summary;
  15
  16 static int fmt_merge_msg_config(const char *key, const char *value, void *cb)
  17 {
  18         static int found_merge_log = 0;
  19         if (!strcmp("merge.log", key)) {
  20                 found_merge_log = 1;
  21                 merge_summary = git_config_bool(key, value);
  22         }
  23         if (!found_merge_log && !strcmp("merge.summary", key))
  24                 merge_summary = git_config_bool(key, value);
  25         return 0;
  26 }
  27
  28 struct src_data {
  29         struct string_list branch, tag, r_branch, generic;
  30         int head_status;
  31 };
  32
  33 void init_src_data(struct src_data *data)
  34 {
  35         data->branch.strdup_strings = 1;
  36         data->tag.strdup_strings = 1;
  37         data->r_branch.strdup_strings = 1;
  38         data->generic.strdup_strings = 1;
  39 }
  40
  41 static struct string_list srcs = { NULL, 0, 0, 1 };
  42 static struct string_list origins = { NULL, 0, 0, 1 };
  43
  44 static int handle_line(char *line)
  45 {
  46         int i, len = strlen(line);
  47         unsigned char *sha1;
  48         char *src, *origin;
  49         struct src_data *src_data;
  50         struct string_list_item *item;
  51         int pulling_head = 0;
  52
  53         if (len < 43 || line[40] != '\t')
  54                 return 1;
  55
  56         if (!prefixcmp(line + 41, "not-for-merge"))
  57                 return 0;
  58
  59         if (line[41] != '\t')
  60                 return 2;
  61
  62         line[40] = 0;
  63         sha1 = xmalloc(20);
  64         i = get_sha1(line, sha1);
  65         line[40] = '\t';
  66         if (i)
  67                 return 3;
  68
  69         if (line[len - 1] == '\n')
  70                 line[len - 1] = 0;
  71         line += 42;
  72
  73         src = strstr(line, " of ");
  74         if (src) {
  75                 *src = 0;
  76                 src += 4;
  77                 pulling_head = 0;
  78         } else {
  79                 src = line;
  80                 pulling_head = 1;
  81         }
  82
  83         item = unsorted_string_list_lookup(&srcs, src);
  84         if (!item) {
  85                 item = string_list_append(src, &srcs);
  86                 item->util = xcalloc(1, sizeof(struct src_data));
  87                 init_src_data(item->util);
  88         }
  89         src_data = item->util;
  90
  91         if (pulling_head) {
  92                 origin = src;
  93                 src_data->head_status |= 1;
  94         } else if (!prefixcmp(line, "branch ")) {
  95                 origin = line + 7;
  96                 string_list_append(origin, &src_data->branch);
  97                 src_data->head_status |= 2;
  98         } else if (!prefixcmp(line, "tag ")) {
  99                 origin = line;
 100                 string_list_append(origin + 4, &src_data->tag);
 101                 src_data->head_status |= 2;
 102         } else if (!prefixcmp(line, "remote branch ")) {
 103                 origin = line + 14;
 104                 string_list_append(origin, &src_data->r_branch);
 105                 src_data->head_status |= 2;
 106         } else {
 107                 origin = src;
 108                 string_list_append(line, &src_data->generic);
 109                 src_data->head_status |= 2;
 110         }
 111
 112         if (!strcmp(".", src) || !strcmp(src, origin)) {
 113                 int len = strlen(origin);
 114                 if (origin[0] == '\'' && origin[len - 1] == '\'')
 115                         origin = xmemdupz(origin + 1, len - 2);
 116         } else {
 117                 char *new_origin = xmalloc(strlen(origin) + strlen(src) + 5);
 118                 sprintf(new_origin, "%s of %s", origin, src);
 119                 origin = new_origin;
 120         }
 121         string_list_append(origin, &origins)->util = sha1;
 122         return 0;
 123 }
 124
 125 static void print_joined(const char *singular, const char *plural,
 126                 struct string_list *list, struct strbuf *out)
 127 {
 128         if (list->nr == 0)
 129                 return;
 130         if (list->nr == 1) {
 131                 strbuf_addf(out, "%s%s", singular, list->items[0].string);
 132         } else {
 133                 int i;
 134                 strbuf_addstr(out, plural);
 135                 for (i = 0; i < list->nr - 1; i++)
 136                         strbuf_addf(out, "%s%s", i > 0 ? ", " : "",
 137                                     list->items[i].string);
 138                 strbuf_addf(out, " and %s", list->items[list->nr - 1].string);
 139         }
 140 }
 141
 142 static void shortlog(const char *name, unsigned char *sha1,
 143                 struct commit *head, struct rev_info *rev, int limit,
 144                 struct strbuf *out)
 145 {
 146         int i, count = 0;
 147         struct commit *commit;
 148         struct object *branch;
 149         struct string_list subjects = { NULL, 0, 0, 1 };
 150         int flags = UNINTERESTING | TREESAME | SEEN | SHOWN | ADDED;
 151         struct strbuf sb = STRBUF_INIT;
 152
 153         branch = deref_tag(parse_object(sha1), sha1_to_hex(sha1), 40);
 154         if (!branch || branch->type != OBJ_COMMIT)
 155                 return;
 156
 157         setup_revisions(0, NULL, rev, NULL);
 158         rev->ignore_merges = 1;
 159         add_pending_object(rev, branch, name);
 160         add_pending_object(rev, &head->object, "^HEAD");
 161         head->object.flags |= UNINTERESTING;
 162         if (prepare_revision_walk(rev))
 163                 die("revision walk setup failed");
 164         while ((commit = get_revision(rev)) != NULL) {
 165                 struct pretty_print_context ctx = {0};
 166
 167                 /* ignore merges */
 168                 if (commit->parents && commit->parents->next)
 169                         continue;
 170
 171                 count++;
 172                 if (subjects.nr > limit)
 173                         continue;
 174
 175                 format_commit_message(commit, "%s", &sb, &ctx);
 176                 strbuf_ltrim(&sb);
 177
 178                 if (!sb.len)
 179                         string_list_append(sha1_to_hex(commit->object.sha1),
 180                                            &subjects);
 181                 else
 182                         string_list_append(strbuf_detach(&sb, NULL), &subjects);
 183         }
 184
 185         if (count > limit)
 186                 strbuf_addf(out, "\n* %s: (%d commits)\n", name, count);
 187         else
 188                 strbuf_addf(out, "\n* %s:\n", name);
 189
 190         for (i = 0; i < subjects.nr; i++)
 191                 if (i >= limit)
 192                         strbuf_addf(out, "  ...\n");
 193                 else
 194                         strbuf_addf(out, "  %s\n", subjects.items[i].string);
 195
 196         clear_commit_marks((struct commit *)branch, flags);
 197         clear_commit_marks(head, flags);
 198         free_commit_list(rev->commits);
 199         rev->commits = NULL;
 200         rev->pending.nr = 0;
 201
 202         string_list_clear(&subjects, 0);
 203 }
 204
 205 int fmt_merge_msg(int merge_summary, struct strbuf *in, struct strbuf *out) {
 206         int limit = 20, i = 0, pos = 0;
 207         char *sep = "";
 208         unsigned char head_sha1[20];
 209         const char *current_branch;
 210
 211         /* get current branch */
 212         current_branch = resolve_ref("HEAD", head_sha1, 1, NULL);
 213         if (!current_branch)
 214                 die("No current branch");
 215         if (!prefixcmp(current_branch, "refs/heads/"))
 216                 current_branch += 11;
 217
 218         /* get a line */
 219         while (pos < in->len) {
 220                 int len;
 221                 char *newline, *p = in->buf + pos;
 222
 223                 newline = strchr(p, '\n');
 224                 len = newline ? newline - p : strlen(p);
 225                 pos += len + !!newline;
 226                 i++;
 227                 p[len] = 0;
 228                 if (handle_line(p))
 229                         die ("Error in line %d: %.*s", i, len, p);
 230         }
 231
 232         if (!srcs.nr)
 233                 return 0;
 234
 235         strbuf_addstr(out, "Merge ");
 236         for (i = 0; i < srcs.nr; i++) {
 237                 struct src_data *src_data = srcs.items[i].util;
 238                 const char *subsep = "";
 239
 240                 strbuf_addstr(out, sep);
 241                 sep = "; ";
 242
 243                 if (src_data->head_status == 1) {
 244                         strbuf_addstr(out, srcs.items[i].string);
 245                         continue;
 246                 }
 247                 if (src_data->head_status == 3) {
 248                         subsep = ", ";
 249                         strbuf_addstr(out, "HEAD");
 250                 }
 251                 if (src_data->branch.nr) {
 252                         strbuf_addstr(out, subsep);
 253                         subsep = ", ";
 254                         print_joined("branch ", "branches ", &src_data->branch,
 255                                         out);
 256                 }
 257                 if (src_data->r_branch.nr) {
 258                         strbuf_addstr(out, subsep);
 259                         subsep = ", ";
 260                         print_joined("remote branch ", "remote branches ",
 261                                         &src_data->r_branch, out);
 262                 }
 263                 if (src_data->tag.nr) {
 264                         strbuf_addstr(out, subsep);
 265                         subsep = ", ";
 266                         print_joined("tag ", "tags ", &src_data->tag, out);
 267                 }
 268                 if (src_data->generic.nr) {
 269                         strbuf_addstr(out, subsep);
 270                         print_joined("commit ", "commits ", &src_data->generic,
 271                                         out);
 272                 }
 273                 if (strcmp(".", srcs.items[i].string))
 274                         strbuf_addf(out, " of %s", srcs.items[i].string);
 275         }
 276
 277         if (!strcmp("master", current_branch))
 278                 strbuf_addch(out, '\n');
 279         else
 280                 strbuf_addf(out, " into %s\n", current_branch);
 281
 282         if (merge_summary) {
 283                 struct commit *head;
 284                 struct rev_info rev;
 285
 286                 head = lookup_commit(head_sha1);
 287                 init_revisions(&rev, NULL);
 288                 rev.commit_format = CMIT_FMT_ONELINE;
 289                 rev.ignore_merges = 1;
 290                 rev.limited = 1;
 291
 292                 for (i = 0; i < origins.nr; i++)
 293                         shortlog(origins.items[i].string, origins.items[i].util,
 294                                         head, &rev, limit, out);
 295         }
 296         return 0;
 297 }
 298
 299 int cmd_fmt_merge_msg(int argc, const char **argv, const char *prefix)
 300 {
 301         const char *inpath = NULL;
 302         struct option options[] = {
 303                 OPT_BOOLEAN(0, "log",     &merge_summary, "populate log with the shortlog"),
 304                 { OPTION_BOOLEAN, 0, "summary", &merge_summary, NULL,
 305                   "alias for --log (deprecated)",
 306                   PARSE_OPT_NOARG | PARSE_OPT_HIDDEN },
 307                 OPT_FILENAME('F', "file", &inpath, "file to read from"),
 308                 OPT_END()
 309         };
 310
 311         FILE *in = stdin;
 312         struct strbuf input = STRBUF_INIT, output = STRBUF_INIT;
 313         int ret;
 314
 315         git_config(fmt_merge_msg_config, NULL);
 316         argc = parse_options(argc, argv, prefix, options, fmt_merge_msg_usage,
 317                              0);
 318         if (argc > 0)
 319                 usage_with_options(fmt_merge_msg_usage, options);
 320
 321         if (inpath && strcmp(inpath, "-")) {
 322                 in = fopen(inpath, "r");
 323                 if (!in)
 324                         die_errno("cannot open '%s'", inpath);
 325         }
 326
 327         if (strbuf_read(&input, fileno(in), 0) < 0)
 328                 die_errno("could not read input file");
 329         ret = fmt_merge_msg(merge_summary, &input, &output);
 330         if (ret)
 331                 return ret;
 332         write_in_full(STDOUT_FILENO, output.buf, output.len);
 333         return 0;
 334 }