Merge branch 'jc/diff-algo-cleanup' into maint
[git] / notes-merge.c
1 #include "cache.h"
2 #include "commit.h"
3 #include "refs.h"
4 #include "diff.h"
5 #include "diffcore.h"
6 #include "xdiff-interface.h"
7 #include "ll-merge.h"
8 #include "dir.h"
9 #include "notes.h"
10 #include "notes-merge.h"
11 #include "strbuf.h"
12
13 struct notes_merge_pair {
14         unsigned char obj[20], base[20], local[20], remote[20];
15 };
16
17 void init_notes_merge_options(struct notes_merge_options *o)
18 {
19         memset(o, 0, sizeof(struct notes_merge_options));
20         strbuf_init(&(o->commit_msg), 0);
21         o->verbosity = NOTES_MERGE_VERBOSITY_DEFAULT;
22 }
23
24 static int path_to_sha1(const char *path, unsigned char *sha1)
25 {
26         char hex_sha1[40];
27         int i = 0;
28         while (*path && i < 40) {
29                 if (*path != '/')
30                         hex_sha1[i++] = *path;
31                 path++;
32         }
33         if (*path || i != 40)
34                 return -1;
35         return get_sha1_hex(hex_sha1, sha1);
36 }
37
38 static int verify_notes_filepair(struct diff_filepair *p, unsigned char *sha1)
39 {
40         switch (p->status) {
41         case DIFF_STATUS_MODIFIED:
42                 assert(p->one->mode == p->two->mode);
43                 assert(!is_null_sha1(p->one->sha1));
44                 assert(!is_null_sha1(p->two->sha1));
45                 break;
46         case DIFF_STATUS_ADDED:
47                 assert(is_null_sha1(p->one->sha1));
48                 break;
49         case DIFF_STATUS_DELETED:
50                 assert(is_null_sha1(p->two->sha1));
51                 break;
52         default:
53                 return -1;
54         }
55         assert(!strcmp(p->one->path, p->two->path));
56         return path_to_sha1(p->one->path, sha1);
57 }
58
59 static struct notes_merge_pair *find_notes_merge_pair_pos(
60                 struct notes_merge_pair *list, int len, unsigned char *obj,
61                 int insert_new, int *occupied)
62 {
63         /*
64          * Both diff_tree_remote() and diff_tree_local() tend to process
65          * merge_pairs in ascending order. Therefore, cache last returned
66          * index, and search sequentially from there until the appropriate
67          * position is found.
68          *
69          * Since inserts only happen from diff_tree_remote() (which mainly
70          * _appends_), we don't care that inserting into the middle of the
71          * list is expensive (using memmove()).
72          */
73         static int last_index;
74         int i = last_index < len ? last_index : len - 1;
75         int prev_cmp = 0, cmp = -1;
76         while (i >= 0 && i < len) {
77                 cmp = hashcmp(obj, list[i].obj);
78                 if (!cmp) /* obj belongs @ i */
79                         break;
80                 else if (cmp < 0 && prev_cmp <= 0) /* obj belongs < i */
81                         i--;
82                 else if (cmp < 0) /* obj belongs between i-1 and i */
83                         break;
84                 else if (cmp > 0 && prev_cmp >= 0) /* obj belongs > i */
85                         i++;
86                 else /* if (cmp > 0) */ { /* obj belongs between i and i+1 */
87                         i++;
88                         break;
89                 }
90                 prev_cmp = cmp;
91         }
92         if (i < 0)
93                 i = 0;
94         /* obj belongs at, or immediately preceding, index i (0 <= i <= len) */
95
96         if (!cmp)
97                 *occupied = 1;
98         else {
99                 *occupied = 0;
100                 if (insert_new && i < len) {
101                         memmove(list + i + 1, list + i,
102                                 (len - i) * sizeof(struct notes_merge_pair));
103                         memset(list + i, 0, sizeof(struct notes_merge_pair));
104                 }
105         }
106         last_index = i;
107         return list + i;
108 }
109
110 static unsigned char uninitialized[20] =
111         "\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff" \
112         "\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff";
113
114 static struct notes_merge_pair *diff_tree_remote(struct notes_merge_options *o,
115                                                  const unsigned char *base,
116                                                  const unsigned char *remote,
117                                                  int *num_changes)
118 {
119         struct diff_options opt;
120         struct notes_merge_pair *changes;
121         int i, len = 0;
122
123         trace_printf("\tdiff_tree_remote(base = %.7s, remote = %.7s)\n",
124                sha1_to_hex(base), sha1_to_hex(remote));
125
126         diff_setup(&opt);
127         DIFF_OPT_SET(&opt, RECURSIVE);
128         opt.output_format = DIFF_FORMAT_NO_OUTPUT;
129         if (diff_setup_done(&opt) < 0)
130                 die("diff_setup_done failed");
131         diff_tree_sha1(base, remote, "", &opt);
132         diffcore_std(&opt);
133
134         changes = xcalloc(diff_queued_diff.nr, sizeof(struct notes_merge_pair));
135
136         for (i = 0; i < diff_queued_diff.nr; i++) {
137                 struct diff_filepair *p = diff_queued_diff.queue[i];
138                 struct notes_merge_pair *mp;
139                 int occupied;
140                 unsigned char obj[20];
141
142                 if (verify_notes_filepair(p, obj)) {
143                         trace_printf("\t\tCannot merge entry '%s' (%c): "
144                                "%.7s -> %.7s. Skipping!\n", p->one->path,
145                                p->status, sha1_to_hex(p->one->sha1),
146                                sha1_to_hex(p->two->sha1));
147                         continue;
148                 }
149                 mp = find_notes_merge_pair_pos(changes, len, obj, 1, &occupied);
150                 if (occupied) {
151                         /* We've found an addition/deletion pair */
152                         assert(!hashcmp(mp->obj, obj));
153                         if (is_null_sha1(p->one->sha1)) { /* addition */
154                                 assert(is_null_sha1(mp->remote));
155                                 hashcpy(mp->remote, p->two->sha1);
156                         } else if (is_null_sha1(p->two->sha1)) { /* deletion */
157                                 assert(is_null_sha1(mp->base));
158                                 hashcpy(mp->base, p->one->sha1);
159                         } else
160                                 assert(!"Invalid existing change recorded");
161                 } else {
162                         hashcpy(mp->obj, obj);
163                         hashcpy(mp->base, p->one->sha1);
164                         hashcpy(mp->local, uninitialized);
165                         hashcpy(mp->remote, p->two->sha1);
166                         len++;
167                 }
168                 trace_printf("\t\tStored remote change for %s: %.7s -> %.7s\n",
169                        sha1_to_hex(mp->obj), sha1_to_hex(mp->base),
170                        sha1_to_hex(mp->remote));
171         }
172         diff_flush(&opt);
173         diff_tree_release_paths(&opt);
174
175         *num_changes = len;
176         return changes;
177 }
178
179 static void diff_tree_local(struct notes_merge_options *o,
180                             struct notes_merge_pair *changes, int len,
181                             const unsigned char *base,
182                             const unsigned char *local)
183 {
184         struct diff_options opt;
185         int i;
186
187         trace_printf("\tdiff_tree_local(len = %i, base = %.7s, local = %.7s)\n",
188                len, sha1_to_hex(base), sha1_to_hex(local));
189
190         diff_setup(&opt);
191         DIFF_OPT_SET(&opt, RECURSIVE);
192         opt.output_format = DIFF_FORMAT_NO_OUTPUT;
193         if (diff_setup_done(&opt) < 0)
194                 die("diff_setup_done failed");
195         diff_tree_sha1(base, local, "", &opt);
196         diffcore_std(&opt);
197
198         for (i = 0; i < diff_queued_diff.nr; i++) {
199                 struct diff_filepair *p = diff_queued_diff.queue[i];
200                 struct notes_merge_pair *mp;
201                 int match;
202                 unsigned char obj[20];
203
204                 if (verify_notes_filepair(p, obj)) {
205                         trace_printf("\t\tCannot merge entry '%s' (%c): "
206                                "%.7s -> %.7s. Skipping!\n", p->one->path,
207                                p->status, sha1_to_hex(p->one->sha1),
208                                sha1_to_hex(p->two->sha1));
209                         continue;
210                 }
211                 mp = find_notes_merge_pair_pos(changes, len, obj, 0, &match);
212                 if (!match) {
213                         trace_printf("\t\tIgnoring local-only change for %s: "
214                                "%.7s -> %.7s\n", sha1_to_hex(obj),
215                                sha1_to_hex(p->one->sha1),
216                                sha1_to_hex(p->two->sha1));
217                         continue;
218                 }
219
220                 assert(!hashcmp(mp->obj, obj));
221                 if (is_null_sha1(p->two->sha1)) { /* deletion */
222                         /*
223                          * Either this is a true deletion (1), or it is part
224                          * of an A/D pair (2), or D/A pair (3):
225                          *
226                          * (1) mp->local is uninitialized; set it to null_sha1
227                          * (2) mp->local is not uninitialized; don't touch it
228                          * (3) mp->local is uninitialized; set it to null_sha1
229                          *     (will be overwritten by following addition)
230                          */
231                         if (!hashcmp(mp->local, uninitialized))
232                                 hashclr(mp->local);
233                 } else if (is_null_sha1(p->one->sha1)) { /* addition */
234                         /*
235                          * Either this is a true addition (1), or it is part
236                          * of an A/D pair (2), or D/A pair (3):
237                          *
238                          * (1) mp->local is uninitialized; set to p->two->sha1
239                          * (2) mp->local is uninitialized; set to p->two->sha1
240                          * (3) mp->local is null_sha1;     set to p->two->sha1
241                          */
242                         assert(is_null_sha1(mp->local) ||
243                                !hashcmp(mp->local, uninitialized));
244                         hashcpy(mp->local, p->two->sha1);
245                 } else { /* modification */
246                         /*
247                          * This is a true modification. p->one->sha1 shall
248                          * match mp->base, and mp->local shall be uninitialized.
249                          * Set mp->local to p->two->sha1.
250                          */
251                         assert(!hashcmp(p->one->sha1, mp->base));
252                         assert(!hashcmp(mp->local, uninitialized));
253                         hashcpy(mp->local, p->two->sha1);
254                 }
255                 trace_printf("\t\tStored local change for %s: %.7s -> %.7s\n",
256                        sha1_to_hex(mp->obj), sha1_to_hex(mp->base),
257                        sha1_to_hex(mp->local));
258         }
259         diff_flush(&opt);
260         diff_tree_release_paths(&opt);
261 }
262
263 static void check_notes_merge_worktree(struct notes_merge_options *o)
264 {
265         if (!o->has_worktree) {
266                 /*
267                  * Must establish NOTES_MERGE_WORKTREE.
268                  * Abort if NOTES_MERGE_WORKTREE already exists
269                  */
270                 if (file_exists(git_path(NOTES_MERGE_WORKTREE)) &&
271                     !is_empty_dir(git_path(NOTES_MERGE_WORKTREE))) {
272                         if (advice_resolve_conflict)
273                                 die("You have not concluded your previous "
274                                     "notes merge (%s exists).\nPlease, use "
275                                     "'git notes merge --commit' or 'git notes "
276                                     "merge --abort' to commit/abort the "
277                                     "previous merge before you start a new "
278                                     "notes merge.", git_path("NOTES_MERGE_*"));
279                         else
280                                 die("You have not concluded your notes merge "
281                                     "(%s exists).", git_path("NOTES_MERGE_*"));
282                 }
283
284                 if (safe_create_leading_directories(git_path(
285                                 NOTES_MERGE_WORKTREE "/.test")))
286                         die_errno("unable to create directory %s",
287                                   git_path(NOTES_MERGE_WORKTREE));
288                 o->has_worktree = 1;
289         } else if (!file_exists(git_path(NOTES_MERGE_WORKTREE)))
290                 /* NOTES_MERGE_WORKTREE should already be established */
291                 die("missing '%s'. This should not happen",
292                     git_path(NOTES_MERGE_WORKTREE));
293 }
294
295 static void write_buf_to_worktree(const unsigned char *obj,
296                                   const char *buf, unsigned long size)
297 {
298         int fd;
299         char *path = git_path(NOTES_MERGE_WORKTREE "/%s", sha1_to_hex(obj));
300         if (safe_create_leading_directories(path))
301                 die_errno("unable to create directory for '%s'", path);
302         if (file_exists(path))
303                 die("found existing file at '%s'", path);
304
305         fd = open(path, O_WRONLY | O_TRUNC | O_CREAT, 0666);
306         if (fd < 0)
307                 die_errno("failed to open '%s'", path);
308
309         while (size > 0) {
310                 long ret = write_in_full(fd, buf, size);
311                 if (ret < 0) {
312                         /* Ignore epipe */
313                         if (errno == EPIPE)
314                                 break;
315                         die_errno("notes-merge");
316                 } else if (!ret) {
317                         die("notes-merge: disk full?");
318                 }
319                 size -= ret;
320                 buf += ret;
321         }
322
323         close(fd);
324 }
325
326 static void write_note_to_worktree(const unsigned char *obj,
327                                    const unsigned char *note)
328 {
329         enum object_type type;
330         unsigned long size;
331         void *buf = read_sha1_file(note, &type, &size);
332
333         if (!buf)
334                 die("cannot read note %s for object %s",
335                     sha1_to_hex(note), sha1_to_hex(obj));
336         if (type != OBJ_BLOB)
337                 die("blob expected in note %s for object %s",
338                     sha1_to_hex(note), sha1_to_hex(obj));
339         write_buf_to_worktree(obj, buf, size);
340         free(buf);
341 }
342
343 static int ll_merge_in_worktree(struct notes_merge_options *o,
344                                 struct notes_merge_pair *p)
345 {
346         mmbuffer_t result_buf;
347         mmfile_t base, local, remote;
348         int status;
349
350         read_mmblob(&base, p->base);
351         read_mmblob(&local, p->local);
352         read_mmblob(&remote, p->remote);
353
354         status = ll_merge(&result_buf, sha1_to_hex(p->obj), &base, NULL,
355                           &local, o->local_ref, &remote, o->remote_ref, NULL);
356
357         free(base.ptr);
358         free(local.ptr);
359         free(remote.ptr);
360
361         if ((status < 0) || !result_buf.ptr)
362                 die("Failed to execute internal merge");
363
364         write_buf_to_worktree(p->obj, result_buf.ptr, result_buf.size);
365         free(result_buf.ptr);
366
367         return status;
368 }
369
370 static int merge_one_change_manual(struct notes_merge_options *o,
371                                    struct notes_merge_pair *p,
372                                    struct notes_tree *t)
373 {
374         const char *lref = o->local_ref ? o->local_ref : "local version";
375         const char *rref = o->remote_ref ? o->remote_ref : "remote version";
376
377         trace_printf("\t\t\tmerge_one_change_manual(obj = %.7s, base = %.7s, "
378                "local = %.7s, remote = %.7s)\n",
379                sha1_to_hex(p->obj), sha1_to_hex(p->base),
380                sha1_to_hex(p->local), sha1_to_hex(p->remote));
381
382         /* add "Conflicts:" section to commit message first time through */
383         if (!o->has_worktree)
384                 strbuf_addstr(&(o->commit_msg), "\n\nConflicts:\n");
385
386         strbuf_addf(&(o->commit_msg), "\t%s\n", sha1_to_hex(p->obj));
387
388         if (o->verbosity >= 2)
389                 printf("Auto-merging notes for %s\n", sha1_to_hex(p->obj));
390         check_notes_merge_worktree(o);
391         if (is_null_sha1(p->local)) {
392                 /* D/F conflict, checkout p->remote */
393                 assert(!is_null_sha1(p->remote));
394                 if (o->verbosity >= 1)
395                         printf("CONFLICT (delete/modify): Notes for object %s "
396                                 "deleted in %s and modified in %s. Version from %s "
397                                 "left in tree.\n",
398                                 sha1_to_hex(p->obj), lref, rref, rref);
399                 write_note_to_worktree(p->obj, p->remote);
400         } else if (is_null_sha1(p->remote)) {
401                 /* D/F conflict, checkout p->local */
402                 assert(!is_null_sha1(p->local));
403                 if (o->verbosity >= 1)
404                         printf("CONFLICT (delete/modify): Notes for object %s "
405                                 "deleted in %s and modified in %s. Version from %s "
406                                 "left in tree.\n",
407                                 sha1_to_hex(p->obj), rref, lref, lref);
408                 write_note_to_worktree(p->obj, p->local);
409         } else {
410                 /* "regular" conflict, checkout result of ll_merge() */
411                 const char *reason = "content";
412                 if (is_null_sha1(p->base))
413                         reason = "add/add";
414                 assert(!is_null_sha1(p->local));
415                 assert(!is_null_sha1(p->remote));
416                 if (o->verbosity >= 1)
417                         printf("CONFLICT (%s): Merge conflict in notes for "
418                                 "object %s\n", reason, sha1_to_hex(p->obj));
419                 ll_merge_in_worktree(o, p);
420         }
421
422         trace_printf("\t\t\tremoving from partial merge result\n");
423         remove_note(t, p->obj);
424
425         return 1;
426 }
427
428 static int merge_one_change(struct notes_merge_options *o,
429                             struct notes_merge_pair *p, struct notes_tree *t)
430 {
431         /*
432          * Return 0 if change is successfully resolved (stored in notes_tree).
433          * Return 1 is change results in a conflict (NOT stored in notes_tree,
434          * but instead written to NOTES_MERGE_WORKTREE with conflict markers).
435          */
436         switch (o->strategy) {
437         case NOTES_MERGE_RESOLVE_MANUAL:
438                 return merge_one_change_manual(o, p, t);
439         case NOTES_MERGE_RESOLVE_OURS:
440                 if (o->verbosity >= 2)
441                         printf("Using local notes for %s\n",
442                                                 sha1_to_hex(p->obj));
443                 /* nothing to do */
444                 return 0;
445         case NOTES_MERGE_RESOLVE_THEIRS:
446                 if (o->verbosity >= 2)
447                         printf("Using remote notes for %s\n",
448                                                 sha1_to_hex(p->obj));
449                 if (add_note(t, p->obj, p->remote, combine_notes_overwrite))
450                         die("BUG: combine_notes_overwrite failed");
451                 return 0;
452         case NOTES_MERGE_RESOLVE_UNION:
453                 if (o->verbosity >= 2)
454                         printf("Concatenating local and remote notes for %s\n",
455                                                         sha1_to_hex(p->obj));
456                 if (add_note(t, p->obj, p->remote, combine_notes_concatenate))
457                         die("failed to concatenate notes "
458                             "(combine_notes_concatenate)");
459                 return 0;
460         case NOTES_MERGE_RESOLVE_CAT_SORT_UNIQ:
461                 if (o->verbosity >= 2)
462                         printf("Concatenating unique lines in local and remote "
463                                 "notes for %s\n", sha1_to_hex(p->obj));
464                 if (add_note(t, p->obj, p->remote, combine_notes_cat_sort_uniq))
465                         die("failed to concatenate notes "
466                             "(combine_notes_cat_sort_uniq)");
467                 return 0;
468         }
469         die("Unknown strategy (%i).", o->strategy);
470 }
471
472 static int merge_changes(struct notes_merge_options *o,
473                          struct notes_merge_pair *changes, int *num_changes,
474                          struct notes_tree *t)
475 {
476         int i, conflicts = 0;
477
478         trace_printf("\tmerge_changes(num_changes = %i)\n", *num_changes);
479         for (i = 0; i < *num_changes; i++) {
480                 struct notes_merge_pair *p = changes + i;
481                 trace_printf("\t\t%.7s: %.7s -> %.7s/%.7s\n",
482                        sha1_to_hex(p->obj), sha1_to_hex(p->base),
483                        sha1_to_hex(p->local), sha1_to_hex(p->remote));
484
485                 if (!hashcmp(p->base, p->remote)) {
486                         /* no remote change; nothing to do */
487                         trace_printf("\t\t\tskipping (no remote change)\n");
488                 } else if (!hashcmp(p->local, p->remote)) {
489                         /* same change in local and remote; nothing to do */
490                         trace_printf("\t\t\tskipping (local == remote)\n");
491                 } else if (!hashcmp(p->local, uninitialized) ||
492                            !hashcmp(p->local, p->base)) {
493                         /* no local change; adopt remote change */
494                         trace_printf("\t\t\tno local change, adopted remote\n");
495                         if (add_note(t, p->obj, p->remote,
496                                      combine_notes_overwrite))
497                                 die("BUG: combine_notes_overwrite failed");
498                 } else {
499                         /* need file-level merge between local and remote */
500                         trace_printf("\t\t\tneed content-level merge\n");
501                         conflicts += merge_one_change(o, p, t);
502                 }
503         }
504
505         return conflicts;
506 }
507
508 static int merge_from_diffs(struct notes_merge_options *o,
509                             const unsigned char *base,
510                             const unsigned char *local,
511                             const unsigned char *remote, struct notes_tree *t)
512 {
513         struct notes_merge_pair *changes;
514         int num_changes, conflicts;
515
516         trace_printf("\tmerge_from_diffs(base = %.7s, local = %.7s, "
517                "remote = %.7s)\n", sha1_to_hex(base), sha1_to_hex(local),
518                sha1_to_hex(remote));
519
520         changes = diff_tree_remote(o, base, remote, &num_changes);
521         diff_tree_local(o, changes, num_changes, base, local);
522
523         conflicts = merge_changes(o, changes, &num_changes, t);
524         free(changes);
525
526         if (o->verbosity >= 4)
527                 printf("Merge result: %i unmerged notes and a %s notes tree\n",
528                         conflicts, t->dirty ? "dirty" : "clean");
529
530         return conflicts ? -1 : 1;
531 }
532
533 void create_notes_commit(struct notes_tree *t, struct commit_list *parents,
534                          const struct strbuf *msg, unsigned char *result_sha1)
535 {
536         unsigned char tree_sha1[20];
537
538         assert(t->initialized);
539
540         if (write_notes_tree(t, tree_sha1))
541                 die("Failed to write notes tree to database");
542
543         if (!parents) {
544                 /* Deduce parent commit from t->ref */
545                 unsigned char parent_sha1[20];
546                 if (!read_ref(t->ref, parent_sha1)) {
547                         struct commit *parent = lookup_commit(parent_sha1);
548                         if (!parent || parse_commit(parent))
549                                 die("Failed to find/parse commit %s", t->ref);
550                         commit_list_insert(parent, &parents);
551                 }
552                 /* else: t->ref points to nothing, assume root/orphan commit */
553         }
554
555         if (commit_tree(msg, tree_sha1, parents, result_sha1, NULL, NULL))
556                 die("Failed to commit notes tree to database");
557 }
558
559 int notes_merge(struct notes_merge_options *o,
560                 struct notes_tree *local_tree,
561                 unsigned char *result_sha1)
562 {
563         unsigned char local_sha1[20], remote_sha1[20];
564         struct commit *local, *remote;
565         struct commit_list *bases = NULL;
566         const unsigned char *base_sha1, *base_tree_sha1;
567         int result = 0;
568
569         assert(o->local_ref && o->remote_ref);
570         assert(!strcmp(o->local_ref, local_tree->ref));
571         hashclr(result_sha1);
572
573         trace_printf("notes_merge(o->local_ref = %s, o->remote_ref = %s)\n",
574                o->local_ref, o->remote_ref);
575
576         /* Dereference o->local_ref into local_sha1 */
577         if (read_ref_full(o->local_ref, local_sha1, 0, NULL))
578                 die("Failed to resolve local notes ref '%s'", o->local_ref);
579         else if (!check_refname_format(o->local_ref, 0) &&
580                 is_null_sha1(local_sha1))
581                 local = NULL; /* local_sha1 == null_sha1 indicates unborn ref */
582         else if (!(local = lookup_commit_reference(local_sha1)))
583                 die("Could not parse local commit %s (%s)",
584                     sha1_to_hex(local_sha1), o->local_ref);
585         trace_printf("\tlocal commit: %.7s\n", sha1_to_hex(local_sha1));
586
587         /* Dereference o->remote_ref into remote_sha1 */
588         if (get_sha1(o->remote_ref, remote_sha1)) {
589                 /*
590                  * Failed to get remote_sha1. If o->remote_ref looks like an
591                  * unborn ref, perform the merge using an empty notes tree.
592                  */
593                 if (!check_refname_format(o->remote_ref, 0)) {
594                         hashclr(remote_sha1);
595                         remote = NULL;
596                 } else {
597                         die("Failed to resolve remote notes ref '%s'",
598                             o->remote_ref);
599                 }
600         } else if (!(remote = lookup_commit_reference(remote_sha1))) {
601                 die("Could not parse remote commit %s (%s)",
602                     sha1_to_hex(remote_sha1), o->remote_ref);
603         }
604         trace_printf("\tremote commit: %.7s\n", sha1_to_hex(remote_sha1));
605
606         if (!local && !remote)
607                 die("Cannot merge empty notes ref (%s) into empty notes ref "
608                     "(%s)", o->remote_ref, o->local_ref);
609         if (!local) {
610                 /* result == remote commit */
611                 hashcpy(result_sha1, remote_sha1);
612                 goto found_result;
613         }
614         if (!remote) {
615                 /* result == local commit */
616                 hashcpy(result_sha1, local_sha1);
617                 goto found_result;
618         }
619         assert(local && remote);
620
621         /* Find merge bases */
622         bases = get_merge_bases(local, remote, 1);
623         if (!bases) {
624                 base_sha1 = null_sha1;
625                 base_tree_sha1 = EMPTY_TREE_SHA1_BIN;
626                 if (o->verbosity >= 4)
627                         printf("No merge base found; doing history-less merge\n");
628         } else if (!bases->next) {
629                 base_sha1 = bases->item->object.sha1;
630                 base_tree_sha1 = bases->item->tree->object.sha1;
631                 if (o->verbosity >= 4)
632                         printf("One merge base found (%.7s)\n",
633                                 sha1_to_hex(base_sha1));
634         } else {
635                 /* TODO: How to handle multiple merge-bases? */
636                 base_sha1 = bases->item->object.sha1;
637                 base_tree_sha1 = bases->item->tree->object.sha1;
638                 if (o->verbosity >= 3)
639                         printf("Multiple merge bases found. Using the first "
640                                 "(%.7s)\n", sha1_to_hex(base_sha1));
641         }
642
643         if (o->verbosity >= 4)
644                 printf("Merging remote commit %.7s into local commit %.7s with "
645                         "merge-base %.7s\n", sha1_to_hex(remote->object.sha1),
646                         sha1_to_hex(local->object.sha1),
647                         sha1_to_hex(base_sha1));
648
649         if (!hashcmp(remote->object.sha1, base_sha1)) {
650                 /* Already merged; result == local commit */
651                 if (o->verbosity >= 2)
652                         printf("Already up-to-date!\n");
653                 hashcpy(result_sha1, local->object.sha1);
654                 goto found_result;
655         }
656         if (!hashcmp(local->object.sha1, base_sha1)) {
657                 /* Fast-forward; result == remote commit */
658                 if (o->verbosity >= 2)
659                         printf("Fast-forward\n");
660                 hashcpy(result_sha1, remote->object.sha1);
661                 goto found_result;
662         }
663
664         result = merge_from_diffs(o, base_tree_sha1, local->tree->object.sha1,
665                                   remote->tree->object.sha1, local_tree);
666
667         if (result != 0) { /* non-trivial merge (with or without conflicts) */
668                 /* Commit (partial) result */
669                 struct commit_list *parents = NULL;
670                 commit_list_insert(remote, &parents); /* LIFO order */
671                 commit_list_insert(local, &parents);
672                 create_notes_commit(local_tree, parents, &o->commit_msg,
673                                     result_sha1);
674         }
675
676 found_result:
677         free_commit_list(bases);
678         strbuf_release(&(o->commit_msg));
679         trace_printf("notes_merge(): result = %i, result_sha1 = %.7s\n",
680                result, sha1_to_hex(result_sha1));
681         return result;
682 }
683
684 int notes_merge_commit(struct notes_merge_options *o,
685                        struct notes_tree *partial_tree,
686                        struct commit *partial_commit,
687                        unsigned char *result_sha1)
688 {
689         /*
690          * Iterate through files in .git/NOTES_MERGE_WORKTREE and add all
691          * found notes to 'partial_tree'. Write the updated notes tree to
692          * the DB, and commit the resulting tree object while reusing the
693          * commit message and parents from 'partial_commit'.
694          * Finally store the new commit object SHA1 into 'result_sha1'.
695          */
696         DIR *dir;
697         struct dirent *e;
698         struct strbuf path = STRBUF_INIT;
699         char *msg = strstr(partial_commit->buffer, "\n\n");
700         struct strbuf sb_msg = STRBUF_INIT;
701         int baselen;
702
703         strbuf_addstr(&path, git_path(NOTES_MERGE_WORKTREE));
704         if (o->verbosity >= 3)
705                 printf("Committing notes in notes merge worktree at %s\n",
706                         path.buf);
707
708         if (!msg || msg[2] == '\0')
709                 die("partial notes commit has empty message");
710         msg += 2;
711
712         dir = opendir(path.buf);
713         if (!dir)
714                 die_errno("could not open %s", path.buf);
715
716         strbuf_addch(&path, '/');
717         baselen = path.len;
718         while ((e = readdir(dir)) != NULL) {
719                 struct stat st;
720                 unsigned char obj_sha1[20], blob_sha1[20];
721
722                 if (is_dot_or_dotdot(e->d_name))
723                         continue;
724
725                 if (strlen(e->d_name) != 40 || get_sha1_hex(e->d_name, obj_sha1)) {
726                         if (o->verbosity >= 3)
727                                 printf("Skipping non-SHA1 entry '%s%s'\n",
728                                         path.buf, e->d_name);
729                         continue;
730                 }
731
732                 strbuf_addstr(&path, e->d_name);
733                 /* write file as blob, and add to partial_tree */
734                 if (stat(path.buf, &st))
735                         die_errno("Failed to stat '%s'", path.buf);
736                 if (index_path(blob_sha1, path.buf, &st, HASH_WRITE_OBJECT))
737                         die("Failed to write blob object from '%s'", path.buf);
738                 if (add_note(partial_tree, obj_sha1, blob_sha1, NULL))
739                         die("Failed to add resolved note '%s' to notes tree",
740                             path.buf);
741                 if (o->verbosity >= 4)
742                         printf("Added resolved note for object %s: %s\n",
743                                 sha1_to_hex(obj_sha1), sha1_to_hex(blob_sha1));
744                 strbuf_setlen(&path, baselen);
745         }
746
747         strbuf_attach(&sb_msg, msg, strlen(msg), strlen(msg) + 1);
748         create_notes_commit(partial_tree, partial_commit->parents, &sb_msg,
749                             result_sha1);
750         if (o->verbosity >= 4)
751                 printf("Finalized notes merge commit: %s\n",
752                         sha1_to_hex(result_sha1));
753         strbuf_release(&path);
754         closedir(dir);
755         return 0;
756 }
757
758 int notes_merge_abort(struct notes_merge_options *o)
759 {
760         /*
761          * Remove all files within .git/NOTES_MERGE_WORKTREE. We do not remove
762          * the .git/NOTES_MERGE_WORKTREE directory itself, since it might be
763          * the current working directory of the user.
764          */
765         struct strbuf buf = STRBUF_INIT;
766         int ret;
767
768         strbuf_addstr(&buf, git_path(NOTES_MERGE_WORKTREE));
769         if (o->verbosity >= 3)
770                 printf("Removing notes merge worktree at %s/*\n", buf.buf);
771         ret = remove_dir_recursively(&buf, REMOVE_DIR_KEEP_TOPLEVEL);
772         strbuf_release(&buf);
773         return ret;
774 }