3 #include "parse-options.h"
8 #include "unpack-trees.h"
10 #include "run-command.h"
11 #include "merge-recursive.h"
17 #include "xdiff-interface.h"
20 static const char * const checkout_usage
[] = {
21 "git checkout [options] <branch>",
22 "git checkout [options] [<branch>] -- <file>...",
26 struct checkout_opts
{
33 const char *new_branch
;
35 enum branch_track track
;
38 static int post_checkout_hook(struct commit
*old
, struct commit
*new,
41 struct child_process proc
;
42 const char *name
= git_path("hooks/post-checkout");
45 if (access(name
, X_OK
) < 0)
48 memset(&proc
, 0, sizeof(proc
));
50 argv
[1] = xstrdup(sha1_to_hex(old
->object
.sha1
));
51 argv
[2] = xstrdup(sha1_to_hex(new->object
.sha1
));
52 argv
[3] = changed
? "1" : "0";
56 proc
.stdout_to_stderr
= 1;
57 return run_command(&proc
);
60 static int update_some(const unsigned char *sha1
, const char *base
, int baselen
,
61 const char *pathname
, unsigned mode
, int stage
, void *context
)
64 struct cache_entry
*ce
;
66 if (S_ISGITLINK(mode
))
70 return READ_TREE_RECURSIVE
;
72 len
= baselen
+ strlen(pathname
);
73 ce
= xcalloc(1, cache_entry_size(len
));
74 hashcpy(ce
->sha1
, sha1
);
75 memcpy(ce
->name
, base
, baselen
);
76 memcpy(ce
->name
+ baselen
, pathname
, len
- baselen
);
77 ce
->ce_flags
= create_ce_flags(len
, 0);
78 ce
->ce_mode
= create_ce_mode(mode
);
79 add_cache_entry(ce
, ADD_CACHE_OK_TO_ADD
| ADD_CACHE_OK_TO_REPLACE
);
83 static int read_tree_some(struct tree
*tree
, const char **pathspec
)
85 read_tree_recursive(tree
, "", 0, 0, pathspec
, update_some
, NULL
);
87 /* update the index with the given tree's info
88 * for all args, expanding wildcards, and exit
89 * with any non-zero return code.
94 static int skip_same_name(struct cache_entry
*ce
, int pos
)
96 while (++pos
< active_nr
&&
97 !strcmp(active_cache
[pos
]->name
, ce
->name
))
102 static int check_stage(int stage
, struct cache_entry
*ce
, int pos
)
104 while (pos
< active_nr
&&
105 !strcmp(active_cache
[pos
]->name
, ce
->name
)) {
106 if (ce_stage(active_cache
[pos
]) == stage
)
110 return error("path '%s' does not have %s version",
112 (stage
== 2) ? "our" : "their");
115 static int check_all_stages(struct cache_entry
*ce
, int pos
)
117 if (ce_stage(ce
) != 1 ||
118 active_nr
<= pos
+ 2 ||
119 strcmp(active_cache
[pos
+1]->name
, ce
->name
) ||
120 ce_stage(active_cache
[pos
+1]) != 2 ||
121 strcmp(active_cache
[pos
+2]->name
, ce
->name
) ||
122 ce_stage(active_cache
[pos
+2]) != 3)
123 return error("path '%s' does not have all three versions",
128 static int checkout_stage(int stage
, struct cache_entry
*ce
, int pos
,
129 struct checkout
*state
)
131 while (pos
< active_nr
&&
132 !strcmp(active_cache
[pos
]->name
, ce
->name
)) {
133 if (ce_stage(active_cache
[pos
]) == stage
)
134 return checkout_entry(active_cache
[pos
], state
, NULL
);
137 return error("path '%s' does not have %s version",
139 (stage
== 2) ? "our" : "their");
142 /* NEEDSWORK: share with merge-recursive */
143 static void fill_mm(const unsigned char *sha1
, mmfile_t
*mm
)
146 enum object_type type
;
148 if (!hashcmp(sha1
, null_sha1
)) {
149 mm
->ptr
= xstrdup("");
154 mm
->ptr
= read_sha1_file(sha1
, &type
, &size
);
155 if (!mm
->ptr
|| type
!= OBJ_BLOB
)
156 die("unable to read blob object %s", sha1_to_hex(sha1
));
160 static int checkout_merged(int pos
, struct checkout
*state
)
162 struct cache_entry
*ce
= active_cache
[pos
];
163 const char *path
= ce
->name
;
164 mmfile_t ancestor
, ours
, theirs
;
166 unsigned char sha1
[20];
167 mmbuffer_t result_buf
;
169 if (ce_stage(ce
) != 1 ||
170 active_nr
<= pos
+ 2 ||
171 strcmp(active_cache
[pos
+1]->name
, path
) ||
172 ce_stage(active_cache
[pos
+1]) != 2 ||
173 strcmp(active_cache
[pos
+2]->name
, path
) ||
174 ce_stage(active_cache
[pos
+2]) != 3)
175 return error("path '%s' does not have all 3 versions", path
);
177 fill_mm(active_cache
[pos
]->sha1
, &ancestor
);
178 fill_mm(active_cache
[pos
+1]->sha1
, &ours
);
179 fill_mm(active_cache
[pos
+2]->sha1
, &theirs
);
181 status
= ll_merge(&result_buf
, path
, &ancestor
,
182 &ours
, "ours", &theirs
, "theirs", 1);
186 if (status
< 0 || !result_buf
.ptr
) {
187 free(result_buf
.ptr
);
188 return error("path '%s': cannot merge", path
);
193 * There is absolutely no reason to write this as a blob object
194 * and create a phoney cache entry just to leak. This hack is
195 * primarily to get to the write_entry() machinery that massages
196 * the contents to work-tree format and writes out which only
197 * allows it for a cache entry. The code in write_entry() needs
198 * to be refactored to allow us to feed a <buffer, size, mode>
199 * instead of a cache entry. Such a refactoring would help
200 * merge_recursive as well (it also writes the merge result to the
201 * object database even when it may contain conflicts).
203 if (write_sha1_file(result_buf
.ptr
, result_buf
.size
,
205 die("Unable to add merge result for '%s'", path
);
206 ce
= make_cache_entry(create_ce_mode(active_cache
[pos
+1]->ce_mode
),
209 status
= checkout_entry(ce
, state
, NULL
);
213 static int checkout_paths(struct tree
*source_tree
, const char **pathspec
,
214 struct checkout_opts
*opts
)
217 struct checkout state
;
218 static char *ps_matched
;
219 unsigned char rev
[20];
223 int stage
= opts
->writeout_stage
;
224 int merge
= opts
->merge
;
226 struct lock_file
*lock_file
= xcalloc(1, sizeof(struct lock_file
));
228 newfd
= hold_locked_index(lock_file
, 1);
232 read_tree_some(source_tree
, pathspec
);
234 for (pos
= 0; pathspec
[pos
]; pos
++)
236 ps_matched
= xcalloc(1, pos
);
238 for (pos
= 0; pos
< active_nr
; pos
++) {
239 struct cache_entry
*ce
= active_cache
[pos
];
240 pathspec_match(pathspec
, ps_matched
, ce
->name
, 0);
243 if (report_path_error(ps_matched
, pathspec
, 0))
246 /* Any unmerged paths? */
247 for (pos
= 0; pos
< active_nr
; pos
++) {
248 struct cache_entry
*ce
= active_cache
[pos
];
249 if (pathspec_match(pathspec
, NULL
, ce
->name
, 0)) {
253 warning("path '%s' is unmerged", ce
->name
);
255 errs
|= check_stage(stage
, ce
, pos
);
256 } else if (opts
->merge
) {
257 errs
|= check_all_stages(ce
, pos
);
260 error("path '%s' is unmerged", ce
->name
);
262 pos
= skip_same_name(ce
, pos
) - 1;
268 /* Now we are committed to check them out */
269 memset(&state
, 0, sizeof(state
));
271 state
.refresh_cache
= 1;
272 for (pos
= 0; pos
< active_nr
; pos
++) {
273 struct cache_entry
*ce
= active_cache
[pos
];
274 if (pathspec_match(pathspec
, NULL
, ce
->name
, 0)) {
276 errs
|= checkout_entry(ce
, &state
, NULL
);
280 errs
|= checkout_stage(stage
, ce
, pos
, &state
);
282 errs
|= checkout_merged(pos
, &state
);
283 pos
= skip_same_name(ce
, pos
) - 1;
287 if (write_cache(newfd
, active_cache
, active_nr
) ||
288 commit_locked_index(lock_file
))
289 die("unable to write new index file");
291 resolve_ref("HEAD", rev
, 0, &flag
);
292 head
= lookup_commit_reference_gently(rev
, 1);
294 errs
|= post_checkout_hook(head
, head
, 0);
298 static void show_local_changes(struct object
*head
)
301 /* I think we want full paths, even if we're in a subdirectory. */
302 init_revisions(&rev
, NULL
);
304 rev
.diffopt
.output_format
|= DIFF_FORMAT_NAME_STATUS
;
305 add_pending_object(&rev
, head
, NULL
);
306 run_diff_index(&rev
, 0);
309 static void describe_detached_head(char *msg
, struct commit
*commit
)
313 parse_commit(commit
);
314 pretty_print_commit(CMIT_FMT_ONELINE
, commit
, &sb
, 0, NULL
, NULL
, 0, 0);
315 fprintf(stderr
, "%s %s... %s\n", msg
,
316 find_unique_abbrev(commit
->object
.sha1
, DEFAULT_ABBREV
), sb
.buf
);
320 static int reset_tree(struct tree
*tree
, struct checkout_opts
*o
, int worktree
)
322 struct unpack_trees_options opts
;
323 struct tree_desc tree_desc
;
325 memset(&opts
, 0, sizeof(opts
));
327 opts
.update
= worktree
;
328 opts
.skip_unmerged
= !worktree
;
331 opts
.fn
= oneway_merge
;
332 opts
.verbose_update
= !o
->quiet
;
333 opts
.src_index
= &the_index
;
334 opts
.dst_index
= &the_index
;
336 init_tree_desc(&tree_desc
, tree
->buffer
, tree
->size
);
337 switch (unpack_trees(1, &tree_desc
, &opts
)) {
339 o
->writeout_error
= 1;
341 * We return 0 nevertheless, as the index is all right
342 * and more importantly we have made best efforts to
343 * update paths in the work tree, and we cannot revert
354 const char *name
; /* The short name used */
355 const char *path
; /* The full name of a real branch */
356 struct commit
*commit
; /* The named commit */
359 static void setup_branch_path(struct branch_info
*branch
)
362 strbuf_init(&buf
, 0);
363 strbuf_addstr(&buf
, "refs/heads/");
364 strbuf_addstr(&buf
, branch
->name
);
365 branch
->path
= strbuf_detach(&buf
, NULL
);
368 static int merge_working_tree(struct checkout_opts
*opts
,
369 struct branch_info
*old
, struct branch_info
*new)
372 struct lock_file
*lock_file
= xcalloc(1, sizeof(struct lock_file
));
373 int newfd
= hold_locked_index(lock_file
, 1);
377 ret
= reset_tree(new->commit
->tree
, opts
, 1);
381 struct tree_desc trees
[2];
383 struct unpack_trees_options topts
;
385 memset(&topts
, 0, sizeof(topts
));
387 topts
.src_index
= &the_index
;
388 topts
.dst_index
= &the_index
;
390 topts
.msgs
.not_uptodate_file
= "You have local changes to '%s'; cannot switch branches.";
392 refresh_cache(REFRESH_QUIET
);
394 if (unmerged_cache()) {
395 error("you need to resolve your current index first");
399 /* 2-way merge to the new branch */
400 topts
.initial_checkout
= (!active_nr
&&
401 (old
->commit
== new->commit
));
404 topts
.gently
= opts
->merge
;
405 topts
.verbose_update
= !opts
->quiet
;
406 topts
.fn
= twoway_merge
;
407 topts
.dir
= xcalloc(1, sizeof(*topts
.dir
));
408 topts
.dir
->show_ignored
= 1;
409 topts
.dir
->exclude_per_dir
= ".gitignore";
410 tree
= parse_tree_indirect(old
->commit
->object
.sha1
);
411 init_tree_desc(&trees
[0], tree
->buffer
, tree
->size
);
412 tree
= parse_tree_indirect(new->commit
->object
.sha1
);
413 init_tree_desc(&trees
[1], tree
->buffer
, tree
->size
);
415 ret
= unpack_trees(2, trees
, &topts
);
418 * Unpack couldn't do a trivial merge; either
419 * give up or do a real merge, depending on
420 * whether the merge flag was used.
424 struct merge_options o
;
427 parse_commit(old
->commit
);
429 /* Do more real merge */
432 * We update the index fully, then write the
433 * tree from the index, then merge the new
434 * branch with the current tree, with the old
435 * branch as the base. Then we reset the index
436 * (but not the working tree) to the new
437 * branch, leaving the working tree as the
438 * merged version, but skipping unmerged
439 * entries in the index.
442 add_files_to_cache(NULL
, NULL
, 0);
443 init_merge_options(&o
);
445 work
= write_tree_from_memory(&o
);
447 ret
= reset_tree(new->commit
->tree
, opts
, 1);
450 o
.branch1
= new->name
;
452 merge_trees(&o
, new->commit
->tree
, work
,
453 old
->commit
->tree
, &result
);
454 ret
= reset_tree(new->commit
->tree
, opts
, 0);
460 if (write_cache(newfd
, active_cache
, active_nr
) ||
461 commit_locked_index(lock_file
))
462 die("unable to write new index file");
464 if (!opts
->force
&& !opts
->quiet
)
465 show_local_changes(&new->commit
->object
);
470 static void report_tracking(struct branch_info
*new)
472 struct strbuf sb
= STRBUF_INIT
;
473 struct branch
*branch
= branch_get(new->name
);
475 if (!format_tracking_info(branch
, &sb
))
477 fputs(sb
.buf
, stdout
);
481 static void update_refs_for_switch(struct checkout_opts
*opts
,
482 struct branch_info
*old
,
483 struct branch_info
*new)
486 const char *old_desc
;
487 if (opts
->new_branch
) {
488 create_branch(old
->name
, opts
->new_branch
, new->name
, 0,
489 opts
->new_branch_log
, opts
->track
);
490 new->name
= opts
->new_branch
;
491 setup_branch_path(new);
494 strbuf_init(&msg
, 0);
495 old_desc
= old
->name
;
497 old_desc
= sha1_to_hex(old
->commit
->object
.sha1
);
498 strbuf_addf(&msg
, "checkout: moving from %s to %s",
499 old_desc
, new->name
);
502 create_symref("HEAD", new->path
, msg
.buf
);
504 if (old
->path
&& !strcmp(new->path
, old
->path
))
505 fprintf(stderr
, "Already on \"%s\"\n",
508 fprintf(stderr
, "Switched to%s branch \"%s\"\n",
509 opts
->new_branch
? " a new" : "",
512 } else if (strcmp(new->name
, "HEAD")) {
513 update_ref(msg
.buf
, "HEAD", new->commit
->object
.sha1
, NULL
,
514 REF_NODEREF
, DIE_ON_ERR
);
517 fprintf(stderr
, "Note: moving to \"%s\" which isn't a local branch\nIf you want to create a new branch from this checkout, you may do so\n(now or later) by using -b with the checkout command again. Example:\n git checkout -b <new_branch_name>\n", new->name
);
518 describe_detached_head("HEAD is now at", new->commit
);
521 remove_branch_state();
522 strbuf_release(&msg
);
523 if (!opts
->quiet
&& (new->path
|| !strcmp(new->name
, "HEAD")))
524 report_tracking(new);
527 static int switch_branches(struct checkout_opts
*opts
, struct branch_info
*new)
530 struct branch_info old
;
531 unsigned char rev
[20];
533 memset(&old
, 0, sizeof(old
));
534 old
.path
= resolve_ref("HEAD", rev
, 0, &flag
);
535 old
.commit
= lookup_commit_reference_gently(rev
, 1);
536 if (!(flag
& REF_ISSYMREF
))
539 if (old
.path
&& !prefixcmp(old
.path
, "refs/heads/"))
540 old
.name
= old
.path
+ strlen("refs/heads/");
544 new->commit
= old
.commit
;
546 die("You are on a branch yet to be born");
547 parse_commit(new->commit
);
551 * If we were on a detached HEAD, but we are now moving to
552 * a new commit, we want to mention the old commit once more
553 * to remind the user that it might be lost.
555 if (!opts
->quiet
&& !old
.path
&& new->commit
!= old
.commit
)
556 describe_detached_head("Previous HEAD position was", old
.commit
);
560 fprintf(stderr
, "warning: You appear to be on a branch yet to be born.\n");
561 fprintf(stderr
, "warning: Forcing checkout of %s.\n", new->name
);
566 ret
= merge_working_tree(opts
, &old
, new);
570 update_refs_for_switch(opts
, &old
, new);
572 ret
= post_checkout_hook(old
.commit
, new->commit
, 1);
573 return ret
|| opts
->writeout_error
;
576 static int git_checkout_config(const char *var
, const char *value
, void *cb
)
578 return git_xmerge_config(var
, value
, cb
);
581 int cmd_checkout(int argc
, const char **argv
, const char *prefix
)
583 struct checkout_opts opts
;
584 unsigned char rev
[20];
586 struct branch_info
new;
587 struct tree
*source_tree
= NULL
;
588 char *conflict_style
= NULL
;
589 struct option options
[] = {
590 OPT__QUIET(&opts
.quiet
),
591 OPT_STRING('b', NULL
, &opts
.new_branch
, "new branch", "branch"),
592 OPT_BOOLEAN('l', NULL
, &opts
.new_branch_log
, "log for new branch"),
593 OPT_SET_INT('t', "track", &opts
.track
, "track",
594 BRANCH_TRACK_EXPLICIT
),
595 OPT_SET_INT('2', "ours", &opts
.writeout_stage
, "stage",
597 OPT_SET_INT('3', "theirs", &opts
.writeout_stage
, "stage",
599 OPT_BOOLEAN('f', NULL
, &opts
.force
, "force"),
600 OPT_BOOLEAN('m', "merge", &opts
.merge
, "merge"),
601 OPT_STRING(0, "conflict", &conflict_style
, "style",
602 "conflict style (merge or diff3)"),
607 memset(&opts
, 0, sizeof(opts
));
608 memset(&new, 0, sizeof(new));
610 git_config(git_checkout_config
, NULL
);
612 opts
.track
= BRANCH_TRACK_UNSPECIFIED
;
614 argc
= parse_options(argc
, argv
, options
, checkout_usage
,
615 PARSE_OPT_KEEP_DASHDASH
);
617 /* --track without -b should DWIM */
618 if (0 < opts
.track
&& !opts
.new_branch
) {
619 const char *argv0
= argv
[0];
620 if (!argc
|| !strcmp(argv0
, "--"))
621 die ("--track needs a branch name");
622 if (!prefixcmp(argv0
, "refs/"))
624 if (!prefixcmp(argv0
, "remotes/"))
626 argv0
= strchr(argv0
, '/');
627 if (!argv0
|| !argv0
[1])
628 die ("Missing branch name; try -b");
629 opts
.new_branch
= argv0
+ 1;
632 if (opts
.track
== BRANCH_TRACK_UNSPECIFIED
)
633 opts
.track
= git_branch_track
;
634 if (conflict_style
) {
635 opts
.merge
= 1; /* implied */
636 git_xmerge_config("merge.conflictstyle", conflict_style
, NULL
);
639 if (!opts
.new_branch
&& (opts
.track
!= git_branch_track
))
640 die("git checkout: --track and --no-track require -b");
642 if (opts
.force
&& opts
.merge
)
643 die("git checkout: -f and -m are incompatible");
646 * case 1: git checkout <ref> -- [<paths>]
648 * <ref> must be a valid tree, everything after the '--' must be
651 * case 2: git checkout -- [<paths>]
653 * everything after the '--' must be paths.
655 * case 3: git checkout <something> [<paths>]
657 * With no paths, if <something> is a commit, that is to
658 * switch to the branch or detach HEAD at it.
660 * Otherwise <something> shall not be ambiguous.
661 * - If it's *only* a reference, treat it like case (1).
662 * - If it's only a path, treat it like case (2).
667 if (!strcmp(argv
[0], "--")) { /* case (2) */
674 has_dash_dash
= (argc
> 1) && !strcmp(argv
[1], "--");
676 if (get_sha1(arg
, rev
)) {
677 if (has_dash_dash
) /* case (1) */
678 die("invalid reference: %s", arg
);
679 goto no_reference
; /* case (3 -> 2) */
682 /* we can't end up being in (2) anymore, eat the argument */
686 if ((new.commit
= lookup_commit_reference_gently(rev
, 1))) {
688 setup_branch_path(&new);
689 if (resolve_ref(new.path
, rev
, 1, NULL
))
690 new.commit
= lookup_commit_reference(rev
);
693 parse_commit(new.commit
);
694 source_tree
= new.commit
->tree
;
696 source_tree
= parse_tree_indirect(rev
);
698 if (!source_tree
) /* case (1): want a tree */
699 die("reference is not a tree: %s", arg
);
700 if (!has_dash_dash
) {/* case (3 -> 1) */
702 * Do not complain the most common case
703 * git checkout branch
704 * even if there happen to be a file called 'branch';
705 * it would be extremely annoying.
708 verify_non_filename(NULL
, arg
);
718 const char **pathspec
= get_pathspec(prefix
, argv
);
721 die("invalid path specification");
724 if (opts
.new_branch
) {
726 die("git checkout: updating paths is incompatible with switching branches.\nDid you intend to checkout '%s' which can not be resolved as commit?", argv
[0]);
728 die("git checkout: updating paths is incompatible with switching branches.");
732 if (1 < !!opts
.writeout_stage
+ !!opts
.force
+ !!opts
.merge
)
733 die("git checkout: --ours/--theirs, --force and --merge are incompatible when\nchecking out of the index.");
735 return checkout_paths(source_tree
, pathspec
, &opts
);
738 if (opts
.new_branch
) {
740 strbuf_init(&buf
, 0);
741 strbuf_addstr(&buf
, "refs/heads/");
742 strbuf_addstr(&buf
, opts
.new_branch
);
743 if (!get_sha1(buf
.buf
, rev
))
744 die("git checkout: branch %s already exists", opts
.new_branch
);
745 if (check_ref_format(buf
.buf
))
746 die("git checkout: we do not like '%s' as a branch name.", opts
.new_branch
);
747 strbuf_release(&buf
);
750 if (new.name
&& !new.commit
) {
751 die("Cannot switch branch to a non-commit.");
753 if (opts
.writeout_stage
)
754 die("--ours/--theirs is incompatible with switching branches.");
756 return switch_branches(&opts
, &new);