@@ -10,54 +10,91 @@
*/
#include "cache.h"
+#include "cache-tree.h"
#include "builtin.h"
-#include "run-command.h"
+#include "lockfile.h"
+#include "merge-strategies.h"
+#include "unpack-trees.h"
+
+static int add_tree(const struct object_id *oid, struct tree_desc *t)
+{
+ struct tree *tree;
+
+ tree = parse_tree_indirect(oid);
+ if (parse_tree(tree))
+ return -1;
+
+ init_tree_desc(t, tree->buffer, tree->size);
+ return 0;
+}
static int merge_resolve(struct commit_list *bases, const char *head_arg,
struct commit_list *remote)
{
+ int i = 0;
+ struct lock_file lock = LOCK_INIT;
+ struct tree_desc t[MAX_UNPACK_TREES];
+ struct unpack_trees_options opts;
+ struct object_id head, oid;
struct commit_list *j;
- struct child_process cp_update = CHILD_PROCESS_INIT,
- cp_read = CHILD_PROCESS_INIT,
- cp_write = CHILD_PROCESS_INIT;
-
- cp_update.git_cmd = 1;
- argv_array_pushl(&cp_update.args, "update-index", "-q", "--refresh", NULL);
- run_command(&cp_update);
-
- cp_read.git_cmd = 1;
- argv_array_pushl(&cp_read.args, "read-tree", "-u", "-m", "--aggressive", NULL);
-
- for (j = bases; j && j->item; j = j->next)
- argv_array_push(&cp_read.args, oid_to_hex(&j->item->object.oid));
if (head_arg)
- argv_array_push(&cp_read.args, head_arg);
- if (remote && remote->item)
- argv_array_push(&cp_read.args, oid_to_hex(&remote->item->object.oid));
+ get_oid(head_arg, &head);
- if (run_command(&cp_read))
- return 2;
+ repo_hold_locked_index(the_repository, &lock, LOCK_DIE_ON_ERROR);
+ refresh_index(the_repository->index, 0, NULL, NULL, NULL);
+
+ memset(&opts, 0, sizeof(opts));
+ opts.head_idx = 1;
+ opts.src_index = the_repository->index;
+ opts.dst_index = the_repository->index;
+ opts.update = 1;
+ opts.merge = 1;
+ opts.aggressive = 1;
+
+ for (j = bases; j; j = j->next) {
+ if (add_tree(&j->item->object.oid, t + (i++)))
+ goto out;
+ }
+
+ if (head_arg && add_tree(&head, t + (i++)))
+ goto out;
+ if (remote && add_tree(&remote->item->object.oid, t + (i++)))
+ goto out;
+
+ if (i == 1)
+ opts.fn = oneway_merge;
+ else if (i == 2) {
+ opts.fn = twoway_merge;
+ opts.initial_checkout = is_index_unborn(the_repository->index);
+ } else if (i >= 3) {
+ opts.fn = threeway_merge;
+ opts.head_idx = i - 1;
+ }
+
+ if (unpack_trees(i, t, &opts))
+ goto out;
puts("Trying simple merge.");
+ write_locked_index(the_repository->index, &lock, COMMIT_LOCK);
- cp_write.git_cmd = 1;
- cp_write.no_stdout = 1;
- cp_write.no_stderr = 1;
- argv_array_push(&cp_write.args, "write-tree");
- if (run_command(&cp_write)) {
- struct child_process cp_merge = CHILD_PROCESS_INIT;
+ if (write_index_as_tree(&oid, the_repository->index,
+ the_repository->index_file, 0, NULL)) {
+ int ret;
- puts("Simple merge failed, trying Automatic merge.");
+ repo_hold_locked_index(the_repository, &lock, LOCK_DIE_ON_ERROR);
+ ret = merge_all(the_repository->index, 0, 0,
+ merge_one_file_cb, the_repository);
- cp_merge.git_cmd = 1;
- argv_array_pushl(&cp_merge.args, "merge-index", "-o",
- "git-merge-one-file", "-a", NULL);
- if (run_command(&cp_merge))
- return 1;
+ write_locked_index(the_repository->index, &lock, COMMIT_LOCK);
+ return !!ret;
}
return 0;
+
+ out:
+ rollback_lock_file(&lock);
+ return 2;
}
static const char builtin_merge_resolve_usage[] =
@@ -73,6 +110,10 @@ int cmd_merge_resolve(int argc, const char **argv, const char *prefix)
if (argc < 5)
usage(builtin_merge_resolve_usage);
+ setup_work_tree();
+ if (repo_read_index(the_repository) < 0)
+ die("invalid index");
+
/* The first parameters up to -- are merge bases; the rest are
* heads. */
for (i = 1; i < argc; i++) {
@@ -191,6 +191,17 @@ int merge_strategies_one_file(struct repository *r,
return 0;
}
+int merge_one_file_cb(const struct object_id *orig_blob,
+ const struct object_id *our_blob,
+ const struct object_id *their_blob, const char *path,
+ unsigned int orig_mode, unsigned int our_mode, unsigned int their_mode,
+ void *data)
+{
+ return merge_strategies_one_file((struct repository *)data,
+ orig_blob, our_blob, their_blob, path,
+ orig_mode, our_mode, their_mode);
+}
+
int merge_program_cb(const struct object_id *orig_blob,
const struct object_id *our_blob,
const struct object_id *their_blob, const char *path,
@@ -16,6 +16,12 @@ typedef int (*merge_cb)(const struct object_id *orig_blob,
unsigned int orig_mode, unsigned int our_mode, unsigned int their_mode,
void *data);
+int merge_one_file_cb(const struct object_id *orig_blob,
+ const struct object_id *our_blob,
+ const struct object_id *their_blob, const char *path,
+ unsigned int orig_mode, unsigned int our_mode, unsigned int their_mode,
+ void *data);
+
int merge_program_cb(const struct object_id *orig_blob,
const struct object_id *our_blob,
const struct object_id *their_blob, const char *path,
This removes calls to external processes to avoid reading and writing the index over and over again. - The call to `update-index -q --refresh' is replaced by a call to refresh_index(). - The call to `read-tree' is replaced by a call to unpack_trees() (and all the setup needed). - The call to `write-tree' is replaced by a call to write_index_as_tree(). - The call to `merge-index', needed to invoke `git merge-one-file', is replaced by a call to the new merge_all() function. A callback function, merge_one_file_cb(), is added to allow it to call merge_one_file() without forking. Here too, the index is read in cmd_merge_resolve(), but merge_resolve() takes care of writing it back to the disk. Signed-off-by: Alban Gruin <alban.gruin@gmail.com> --- builtin/merge-resolve.c | 103 ++++++++++++++++++++++++++++------------ merge-strategies.c | 11 +++++ merge-strategies.h | 6 +++ 3 files changed, 89 insertions(+), 31 deletions(-)