1 #define USE_THE_INDEX_COMPATIBILITY_MACROS
2 #include "builtin.h"
3 #include "advice.h"
4 #include "blob.h"
5 #include "branch.h"
6 #include "cache-tree.h"
7 #include "checkout.h"
8 #include "commit.h"
9 #include "config.h"
10 #include "diff.h"
11 #include "dir.h"
12 #include "ll-merge.h"
13 #include "lockfile.h"
14 #include "merge-recursive.h"
15 #include "object-store.h"
16 #include "parse-options.h"
17 #include "refs.h"
18 #include "remote.h"
19 #include "resolve-undo.h"
20 #include "revision.h"
21 #include "run-command.h"
22 #include "submodule.h"
23 #include "submodule-config.h"
24 #include "tree.h"
25 #include "tree-walk.h"
26 #include "unpack-trees.h"
27 #include "wt-status.h"
28 #include "xdiff-interface.h"
29 
30 static const char * const checkout_usage[] = {
31 	N_("git checkout [<options>] <branch>"),
32 	N_("git checkout [<options>] [<branch>] -- <file>..."),
33 	NULL,
34 };
35 
36 static const char * const switch_branch_usage[] = {
37 	N_("git switch [<options>] [<branch>]"),
38 	NULL,
39 };
40 
41 static const char * const restore_usage[] = {
42 	N_("git restore [<options>] [--source=<branch>] <file>..."),
43 	NULL,
44 };
45 
46 struct checkout_opts {
47 	int patch_mode;
48 	int quiet;
49 	int merge;
50 	int force;
51 	int force_detach;
52 	int implicit_detach;
53 	int writeout_stage;
54 	int overwrite_ignore;
55 	int ignore_skipworktree;
56 	int ignore_other_worktrees;
57 	int show_progress;
58 	int count_checkout_paths;
59 	int overlay_mode;
60 	int dwim_new_local_branch;
61 	int discard_changes;
62 	int accept_ref;
63 	int accept_pathspec;
64 	int switch_branch_doing_nothing_is_ok;
65 	int only_merge_on_switching_branches;
66 	int can_switch_when_in_progress;
67 	int orphan_from_empty_tree;
68 	int empty_pathspec_ok;
69 	int checkout_index;
70 	int checkout_worktree;
71 	const char *ignore_unmerged_opt;
72 	int ignore_unmerged;
73 	int pathspec_file_nul;
74 	const char *pathspec_from_file;
75 
76 	const char *new_branch;
77 	const char *new_branch_force;
78 	const char *new_orphan_branch;
79 	int new_branch_log;
80 	enum branch_track track;
81 	struct diff_options diff_options;
82 	char *conflict_style;
83 
84 	int branch_exists;
85 	const char *prefix;
86 	struct pathspec pathspec;
87 	const char *from_treeish;
88 	struct tree *source_tree;
89 };
90 
post_checkout_hook(struct commit * old_commit,struct commit * new_commit,int changed)91 static int post_checkout_hook(struct commit *old_commit, struct commit *new_commit,
92 			      int changed)
93 {
94 	return run_hook_le(NULL, "post-checkout",
95 			   oid_to_hex(old_commit ? &old_commit->object.oid : &null_oid),
96 			   oid_to_hex(new_commit ? &new_commit->object.oid : &null_oid),
97 			   changed ? "1" : "0", NULL);
98 	/* "new_commit" can be NULL when checking out from the index before
99 	   a commit exists. */
100 
101 }
102 
update_some(const struct object_id * oid,struct strbuf * base,const char * pathname,unsigned mode,int stage,void * context)103 static int update_some(const struct object_id *oid, struct strbuf *base,
104 		const char *pathname, unsigned mode, int stage, void *context)
105 {
106 	int len;
107 	struct cache_entry *ce;
108 	int pos;
109 
110 	if (S_ISDIR(mode))
111 		return READ_TREE_RECURSIVE;
112 
113 	len = base->len + strlen(pathname);
114 	ce = make_empty_cache_entry(&the_index, len);
115 	oidcpy(&ce->oid, oid);
116 	memcpy(ce->name, base->buf, base->len);
117 	memcpy(ce->name + base->len, pathname, len - base->len);
118 	ce->ce_flags = create_ce_flags(0) | CE_UPDATE;
119 	ce->ce_namelen = len;
120 	ce->ce_mode = create_ce_mode(mode);
121 
122 	/*
123 	 * If the entry is the same as the current index, we can leave the old
124 	 * entry in place. Whether it is UPTODATE or not, checkout_entry will
125 	 * do the right thing.
126 	 */
127 	pos = cache_name_pos(ce->name, ce->ce_namelen);
128 	if (pos >= 0) {
129 		struct cache_entry *old = active_cache[pos];
130 		if (ce->ce_mode == old->ce_mode &&
131 		    !ce_intent_to_add(old) &&
132 		    oideq(&ce->oid, &old->oid)) {
133 			old->ce_flags |= CE_UPDATE;
134 			discard_cache_entry(ce);
135 			return 0;
136 		}
137 	}
138 
139 	add_cache_entry(ce, ADD_CACHE_OK_TO_ADD | ADD_CACHE_OK_TO_REPLACE);
140 	return 0;
141 }
142 
read_tree_some(struct tree * tree,const struct pathspec * pathspec)143 static int read_tree_some(struct tree *tree, const struct pathspec *pathspec)
144 {
145 	read_tree_recursive(the_repository, tree, "", 0, 0,
146 			    pathspec, update_some, NULL);
147 
148 	/* update the index with the given tree's info
149 	 * for all args, expanding wildcards, and exit
150 	 * with any non-zero return code.
151 	 */
152 	return 0;
153 }
154 
skip_same_name(const struct cache_entry * ce,int pos)155 static int skip_same_name(const struct cache_entry *ce, int pos)
156 {
157 	while (++pos < active_nr &&
158 	       !strcmp(active_cache[pos]->name, ce->name))
159 		; /* skip */
160 	return pos;
161 }
162 
check_stage(int stage,const struct cache_entry * ce,int pos,int overlay_mode)163 static int check_stage(int stage, const struct cache_entry *ce, int pos,
164 		       int overlay_mode)
165 {
166 	while (pos < active_nr &&
167 	       !strcmp(active_cache[pos]->name, ce->name)) {
168 		if (ce_stage(active_cache[pos]) == stage)
169 			return 0;
170 		pos++;
171 	}
172 	if (!overlay_mode)
173 		return 0;
174 	if (stage == 2)
175 		return error(_("path '%s' does not have our version"), ce->name);
176 	else
177 		return error(_("path '%s' does not have their version"), ce->name);
178 }
179 
check_stages(unsigned stages,const struct cache_entry * ce,int pos)180 static int check_stages(unsigned stages, const struct cache_entry *ce, int pos)
181 {
182 	unsigned seen = 0;
183 	const char *name = ce->name;
184 
185 	while (pos < active_nr) {
186 		ce = active_cache[pos];
187 		if (strcmp(name, ce->name))
188 			break;
189 		seen |= (1 << ce_stage(ce));
190 		pos++;
191 	}
192 	if ((stages & seen) != stages)
193 		return error(_("path '%s' does not have all necessary versions"),
194 			     name);
195 	return 0;
196 }
197 
checkout_stage(int stage,const struct cache_entry * ce,int pos,const struct checkout * state,int * nr_checkouts,int overlay_mode)198 static int checkout_stage(int stage, const struct cache_entry *ce, int pos,
199 			  const struct checkout *state, int *nr_checkouts,
200 			  int overlay_mode)
201 {
202 	while (pos < active_nr &&
203 	       !strcmp(active_cache[pos]->name, ce->name)) {
204 		if (ce_stage(active_cache[pos]) == stage)
205 			return checkout_entry(active_cache[pos], state,
206 					      NULL, nr_checkouts);
207 		pos++;
208 	}
209 	if (!overlay_mode) {
210 		unlink_entry(ce);
211 		return 0;
212 	}
213 	if (stage == 2)
214 		return error(_("path '%s' does not have our version"), ce->name);
215 	else
216 		return error(_("path '%s' does not have their version"), ce->name);
217 }
218 
checkout_merged(int pos,const struct checkout * state,int * nr_checkouts)219 static int checkout_merged(int pos, const struct checkout *state, int *nr_checkouts)
220 {
221 	struct cache_entry *ce = active_cache[pos];
222 	const char *path = ce->name;
223 	mmfile_t ancestor, ours, theirs;
224 	int status;
225 	struct object_id oid;
226 	mmbuffer_t result_buf;
227 	struct object_id threeway[3];
228 	unsigned mode = 0;
229 
230 	memset(threeway, 0, sizeof(threeway));
231 	while (pos < active_nr) {
232 		int stage;
233 		stage = ce_stage(ce);
234 		if (!stage || strcmp(path, ce->name))
235 			break;
236 		oidcpy(&threeway[stage - 1], &ce->oid);
237 		if (stage == 2)
238 			mode = create_ce_mode(ce->ce_mode);
239 		pos++;
240 		ce = active_cache[pos];
241 	}
242 	if (is_null_oid(&threeway[1]) || is_null_oid(&threeway[2]))
243 		return error(_("path '%s' does not have necessary versions"), path);
244 
245 	read_mmblob(&ancestor, &threeway[0]);
246 	read_mmblob(&ours, &threeway[1]);
247 	read_mmblob(&theirs, &threeway[2]);
248 
249 	/*
250 	 * NEEDSWORK: re-create conflicts from merges with
251 	 * merge.renormalize set, too
252 	 */
253 	status = ll_merge(&result_buf, path, &ancestor, "base",
254 			  &ours, "ours", &theirs, "theirs",
255 			  state->istate, NULL);
256 	free(ancestor.ptr);
257 	free(ours.ptr);
258 	free(theirs.ptr);
259 	if (status < 0 || !result_buf.ptr) {
260 		free(result_buf.ptr);
261 		return error(_("path '%s': cannot merge"), path);
262 	}
263 
264 	/*
265 	 * NEEDSWORK:
266 	 * There is absolutely no reason to write this as a blob object
267 	 * and create a phony cache entry.  This hack is primarily to get
268 	 * to the write_entry() machinery that massages the contents to
269 	 * work-tree format and writes out which only allows it for a
270 	 * cache entry.  The code in write_entry() needs to be refactored
271 	 * to allow us to feed a <buffer, size, mode> instead of a cache
272 	 * entry.  Such a refactoring would help merge_recursive as well
273 	 * (it also writes the merge result to the object database even
274 	 * when it may contain conflicts).
275 	 */
276 	if (write_object_file(result_buf.ptr, result_buf.size, blob_type, &oid))
277 		die(_("Unable to add merge result for '%s'"), path);
278 	free(result_buf.ptr);
279 	ce = make_transient_cache_entry(mode, &oid, path, 2);
280 	if (!ce)
281 		die(_("make_cache_entry failed for path '%s'"), path);
282 	status = checkout_entry(ce, state, NULL, nr_checkouts);
283 	discard_cache_entry(ce);
284 	return status;
285 }
286 
mark_ce_for_checkout_overlay(struct cache_entry * ce,char * ps_matched,const struct checkout_opts * opts)287 static void mark_ce_for_checkout_overlay(struct cache_entry *ce,
288 					 char *ps_matched,
289 					 const struct checkout_opts *opts)
290 {
291 	ce->ce_flags &= ~CE_MATCHED;
292 	if (!opts->ignore_skipworktree && ce_skip_worktree(ce))
293 		return;
294 	if (opts->source_tree && !(ce->ce_flags & CE_UPDATE))
295 		/*
296 		 * "git checkout tree-ish -- path", but this entry
297 		 * is in the original index but is not in tree-ish
298 		 * or does not match the pathspec; it will not be
299 		 * checked out to the working tree.  We will not do
300 		 * anything to this entry at all.
301 		 */
302 		return;
303 	/*
304 	 * Either this entry came from the tree-ish we are
305 	 * checking the paths out of, or we are checking out
306 	 * of the index.
307 	 *
308 	 * If it comes from the tree-ish, we already know it
309 	 * matches the pathspec and could just stamp
310 	 * CE_MATCHED to it from update_some(). But we still
311 	 * need ps_matched and read_tree_recursive (and
312 	 * eventually tree_entry_interesting) cannot fill
313 	 * ps_matched yet. Once it can, we can avoid calling
314 	 * match_pathspec() for _all_ entries when
315 	 * opts->source_tree != NULL.
316 	 */
317 	if (ce_path_match(&the_index, ce, &opts->pathspec, ps_matched))
318 		ce->ce_flags |= CE_MATCHED;
319 }
320 
mark_ce_for_checkout_no_overlay(struct cache_entry * ce,char * ps_matched,const struct checkout_opts * opts)321 static void mark_ce_for_checkout_no_overlay(struct cache_entry *ce,
322 					    char *ps_matched,
323 					    const struct checkout_opts *opts)
324 {
325 	ce->ce_flags &= ~CE_MATCHED;
326 	if (!opts->ignore_skipworktree && ce_skip_worktree(ce))
327 		return;
328 	if (ce_path_match(&the_index, ce, &opts->pathspec, ps_matched)) {
329 		ce->ce_flags |= CE_MATCHED;
330 		if (opts->source_tree && !(ce->ce_flags & CE_UPDATE))
331 			/*
332 			 * In overlay mode, but the path is not in
333 			 * tree-ish, which means we should remove it
334 			 * from the index and the working tree.
335 			 */
336 			ce->ce_flags |= CE_REMOVE | CE_WT_REMOVE;
337 	}
338 }
339 
checkout_worktree(const struct checkout_opts * opts)340 static int checkout_worktree(const struct checkout_opts *opts)
341 {
342 	struct checkout state = CHECKOUT_INIT;
343 	int nr_checkouts = 0, nr_unmerged = 0;
344 	int errs = 0;
345 	int pos;
346 
347 	state.force = 1;
348 	state.refresh_cache = 1;
349 	state.istate = &the_index;
350 
351 	enable_delayed_checkout(&state);
352 	for (pos = 0; pos < active_nr; pos++) {
353 		struct cache_entry *ce = active_cache[pos];
354 		if (ce->ce_flags & CE_MATCHED) {
355 			if (!ce_stage(ce)) {
356 				errs |= checkout_entry(ce, &state,
357 						       NULL, &nr_checkouts);
358 				continue;
359 			}
360 			if (opts->writeout_stage)
361 				errs |= checkout_stage(opts->writeout_stage,
362 						       ce, pos,
363 						       &state,
364 						       &nr_checkouts, opts->overlay_mode);
365 			else if (opts->merge)
366 				errs |= checkout_merged(pos, &state,
367 							&nr_unmerged);
368 			pos = skip_same_name(ce, pos) - 1;
369 		}
370 	}
371 	remove_marked_cache_entries(&the_index, 1);
372 	remove_scheduled_dirs();
373 	errs |= finish_delayed_checkout(&state, &nr_checkouts);
374 
375 	if (opts->count_checkout_paths) {
376 		if (nr_unmerged)
377 			fprintf_ln(stderr, Q_("Recreated %d merge conflict",
378 					      "Recreated %d merge conflicts",
379 					      nr_unmerged),
380 				   nr_unmerged);
381 		if (opts->source_tree)
382 			fprintf_ln(stderr, Q_("Updated %d path from %s",
383 					      "Updated %d paths from %s",
384 					      nr_checkouts),
385 				   nr_checkouts,
386 				   find_unique_abbrev(&opts->source_tree->object.oid,
387 						      DEFAULT_ABBREV));
388 		else if (!nr_unmerged || nr_checkouts)
389 			fprintf_ln(stderr, Q_("Updated %d path from the index",
390 					      "Updated %d paths from the index",
391 					      nr_checkouts),
392 				   nr_checkouts);
393 	}
394 
395 	return errs;
396 }
397 
checkout_paths(const struct checkout_opts * opts,const char * revision)398 static int checkout_paths(const struct checkout_opts *opts,
399 			  const char *revision)
400 {
401 	int pos;
402 	static char *ps_matched;
403 	struct object_id rev;
404 	struct commit *head;
405 	int errs = 0;
406 	struct lock_file lock_file = LOCK_INIT;
407 	int checkout_index;
408 
409 	trace2_cmd_mode(opts->patch_mode ? "patch" : "path");
410 
411 	if (opts->track != BRANCH_TRACK_UNSPECIFIED)
412 		die(_("'%s' cannot be used with updating paths"), "--track");
413 
414 	if (opts->new_branch_log)
415 		die(_("'%s' cannot be used with updating paths"), "-l");
416 
417 	if (opts->ignore_unmerged && opts->patch_mode)
418 		die(_("'%s' cannot be used with updating paths"),
419 		    opts->ignore_unmerged_opt);
420 
421 	if (opts->force_detach)
422 		die(_("'%s' cannot be used with updating paths"), "--detach");
423 
424 	if (opts->merge && opts->patch_mode)
425 		die(_("'%s' cannot be used with %s"), "--merge", "--patch");
426 
427 	if (opts->ignore_unmerged && opts->merge)
428 		die(_("'%s' cannot be used with %s"),
429 		    opts->ignore_unmerged_opt, "-m");
430 
431 	if (opts->new_branch)
432 		die(_("Cannot update paths and switch to branch '%s' at the same time."),
433 		    opts->new_branch);
434 
435 	if (!opts->checkout_worktree && !opts->checkout_index)
436 		die(_("neither '%s' or '%s' is specified"),
437 		    "--staged", "--worktree");
438 
439 	if (!opts->checkout_worktree && !opts->from_treeish)
440 		die(_("'%s' must be used when '%s' is not specified"),
441 		    "--worktree", "--source");
442 
443 	if (opts->checkout_index && !opts->checkout_worktree &&
444 	    opts->writeout_stage)
445 		die(_("'%s' or '%s' cannot be used with %s"),
446 		    "--ours", "--theirs", "--staged");
447 
448 	if (opts->checkout_index && !opts->checkout_worktree &&
449 	    opts->merge)
450 		die(_("'%s' or '%s' cannot be used with %s"),
451 		    "--merge", "--conflict", "--staged");
452 
453 	if (opts->patch_mode) {
454 		const char *patch_mode;
455 
456 		if (opts->checkout_index && opts->checkout_worktree)
457 			patch_mode = "--patch=checkout";
458 		else if (opts->checkout_index && !opts->checkout_worktree)
459 			patch_mode = "--patch=reset";
460 		else if (!opts->checkout_index && opts->checkout_worktree)
461 			patch_mode = "--patch=worktree";
462 		else
463 			BUG("either flag must have been set, worktree=%d, index=%d",
464 			    opts->checkout_worktree, opts->checkout_index);
465 		return run_add_interactive(revision, patch_mode, &opts->pathspec);
466 	}
467 
468 	repo_hold_locked_index(the_repository, &lock_file, LOCK_DIE_ON_ERROR);
469 	if (read_cache_preload(&opts->pathspec) < 0)
470 		return error(_("index file corrupt"));
471 
472 	if (opts->source_tree)
473 		read_tree_some(opts->source_tree, &opts->pathspec);
474 
475 	ps_matched = xcalloc(opts->pathspec.nr, 1);
476 
477 	/*
478 	 * Make sure all pathspecs participated in locating the paths
479 	 * to be checked out.
480 	 */
481 	for (pos = 0; pos < active_nr; pos++)
482 		if (opts->overlay_mode)
483 			mark_ce_for_checkout_overlay(active_cache[pos],
484 						     ps_matched,
485 						     opts);
486 		else
487 			mark_ce_for_checkout_no_overlay(active_cache[pos],
488 							ps_matched,
489 							opts);
490 
491 	if (report_path_error(ps_matched, &opts->pathspec)) {
492 		free(ps_matched);
493 		return 1;
494 	}
495 	free(ps_matched);
496 
497 	/* "checkout -m path" to recreate conflicted state */
498 	if (opts->merge)
499 		unmerge_marked_index(&the_index);
500 
501 	/* Any unmerged paths? */
502 	for (pos = 0; pos < active_nr; pos++) {
503 		const struct cache_entry *ce = active_cache[pos];
504 		if (ce->ce_flags & CE_MATCHED) {
505 			if (!ce_stage(ce))
506 				continue;
507 			if (opts->ignore_unmerged) {
508 				if (!opts->quiet)
509 					warning(_("path '%s' is unmerged"), ce->name);
510 			} else if (opts->writeout_stage) {
511 				errs |= check_stage(opts->writeout_stage, ce, pos, opts->overlay_mode);
512 			} else if (opts->merge) {
513 				errs |= check_stages((1<<2) | (1<<3), ce, pos);
514 			} else {
515 				errs = 1;
516 				error(_("path '%s' is unmerged"), ce->name);
517 			}
518 			pos = skip_same_name(ce, pos) - 1;
519 		}
520 	}
521 	if (errs)
522 		return 1;
523 
524 	/* Now we are committed to check them out */
525 	if (opts->checkout_worktree)
526 		errs |= checkout_worktree(opts);
527 	else
528 		remove_marked_cache_entries(&the_index, 1);
529 
530 	/*
531 	 * Allow updating the index when checking out from the index.
532 	 * This is to save new stat info.
533 	 */
534 	if (opts->checkout_worktree && !opts->checkout_index && !opts->source_tree)
535 		checkout_index = 1;
536 	else
537 		checkout_index = opts->checkout_index;
538 
539 	if (checkout_index) {
540 		if (write_locked_index(&the_index, &lock_file, COMMIT_LOCK))
541 			die(_("unable to write new index file"));
542 	} else {
543 		/*
544 		 * NEEDSWORK: if --worktree is not specified, we
545 		 * should save stat info of checked out files in the
546 		 * index to avoid the next (potentially costly)
547 		 * refresh. But it's a bit tricker to do...
548 		 */
549 		rollback_lock_file(&lock_file);
550 	}
551 
552 	read_ref_full("HEAD", 0, &rev, NULL);
553 	head = lookup_commit_reference_gently(the_repository, &rev, 1);
554 
555 	errs |= post_checkout_hook(head, head, 0);
556 	return errs;
557 }
558 
show_local_changes(struct object * head,const struct diff_options * opts)559 static void show_local_changes(struct object *head,
560 			       const struct diff_options *opts)
561 {
562 	struct rev_info rev;
563 	/* I think we want full paths, even if we're in a subdirectory. */
564 	repo_init_revisions(the_repository, &rev, NULL);
565 	rev.diffopt.flags = opts->flags;
566 	rev.diffopt.output_format |= DIFF_FORMAT_NAME_STATUS;
567 	diff_setup_done(&rev.diffopt);
568 	add_pending_object(&rev, head, NULL);
569 	run_diff_index(&rev, 0);
570 }
571 
describe_detached_head(const char * msg,struct commit * commit)572 static void describe_detached_head(const char *msg, struct commit *commit)
573 {
574 	struct strbuf sb = STRBUF_INIT;
575 
576 	if (!parse_commit(commit))
577 		pp_commit_easy(CMIT_FMT_ONELINE, commit, &sb);
578 	if (print_sha1_ellipsis()) {
579 		fprintf(stderr, "%s %s... %s\n", msg,
580 			find_unique_abbrev(&commit->object.oid, DEFAULT_ABBREV), sb.buf);
581 	} else {
582 		fprintf(stderr, "%s %s %s\n", msg,
583 			find_unique_abbrev(&commit->object.oid, DEFAULT_ABBREV), sb.buf);
584 	}
585 	strbuf_release(&sb);
586 }
587 
reset_tree(struct tree * tree,const struct checkout_opts * o,int worktree,int * writeout_error)588 static int reset_tree(struct tree *tree, const struct checkout_opts *o,
589 		      int worktree, int *writeout_error)
590 {
591 	struct unpack_trees_options opts;
592 	struct tree_desc tree_desc;
593 
594 	memset(&opts, 0, sizeof(opts));
595 	opts.head_idx = -1;
596 	opts.update = worktree;
597 	opts.skip_unmerged = !worktree;
598 	opts.reset = 1;
599 	opts.merge = 1;
600 	opts.fn = oneway_merge;
601 	opts.verbose_update = o->show_progress;
602 	opts.src_index = &the_index;
603 	opts.dst_index = &the_index;
604 	parse_tree(tree);
605 	init_tree_desc(&tree_desc, tree->buffer, tree->size);
606 	switch (unpack_trees(1, &tree_desc, &opts)) {
607 	case -2:
608 		*writeout_error = 1;
609 		/*
610 		 * We return 0 nevertheless, as the index is all right
611 		 * and more importantly we have made best efforts to
612 		 * update paths in the work tree, and we cannot revert
613 		 * them.
614 		 */
615 		/* fallthrough */
616 	case 0:
617 		return 0;
618 	default:
619 		return 128;
620 	}
621 }
622 
623 struct branch_info {
624 	const char *name; /* The short name used */
625 	const char *path; /* The full name of a real branch */
626 	struct commit *commit; /* The named commit */
627 	/*
628 	 * if not null the branch is detached because it's already
629 	 * checked out in this checkout
630 	 */
631 	char *checkout;
632 };
633 
setup_branch_path(struct branch_info * branch)634 static void setup_branch_path(struct branch_info *branch)
635 {
636 	struct strbuf buf = STRBUF_INIT;
637 
638 	strbuf_branchname(&buf, branch->name, INTERPRET_BRANCH_LOCAL);
639 	if (strcmp(buf.buf, branch->name))
640 		branch->name = xstrdup(buf.buf);
641 	strbuf_splice(&buf, 0, 0, "refs/heads/", 11);
642 	branch->path = strbuf_detach(&buf, NULL);
643 }
644 
merge_working_tree(const struct checkout_opts * opts,struct branch_info * old_branch_info,struct branch_info * new_branch_info,int * writeout_error)645 static int merge_working_tree(const struct checkout_opts *opts,
646 			      struct branch_info *old_branch_info,
647 			      struct branch_info *new_branch_info,
648 			      int *writeout_error)
649 {
650 	int ret;
651 	struct lock_file lock_file = LOCK_INIT;
652 	struct tree *new_tree;
653 
654 	hold_locked_index(&lock_file, LOCK_DIE_ON_ERROR);
655 	if (read_cache_preload(NULL) < 0)
656 		return error(_("index file corrupt"));
657 
658 	resolve_undo_clear();
659 	if (opts->new_orphan_branch && opts->orphan_from_empty_tree) {
660 		if (new_branch_info->commit)
661 			BUG("'switch --orphan' should never accept a commit as starting point");
662 		new_tree = parse_tree_indirect(the_hash_algo->empty_tree);
663 	} else
664 		new_tree = get_commit_tree(new_branch_info->commit);
665 	if (opts->discard_changes) {
666 		ret = reset_tree(new_tree, opts, 1, writeout_error);
667 		if (ret)
668 			return ret;
669 	} else {
670 		struct tree_desc trees[2];
671 		struct tree *tree;
672 		struct unpack_trees_options topts;
673 
674 		memset(&topts, 0, sizeof(topts));
675 		topts.head_idx = -1;
676 		topts.src_index = &the_index;
677 		topts.dst_index = &the_index;
678 
679 		setup_unpack_trees_porcelain(&topts, "checkout");
680 
681 		refresh_cache(REFRESH_QUIET);
682 
683 		if (unmerged_cache()) {
684 			error(_("you need to resolve your current index first"));
685 			return 1;
686 		}
687 
688 		/* 2-way merge to the new branch */
689 		topts.initial_checkout = is_cache_unborn();
690 		topts.update = 1;
691 		topts.merge = 1;
692 		topts.quiet = opts->merge && old_branch_info->commit;
693 		topts.verbose_update = opts->show_progress;
694 		topts.fn = twoway_merge;
695 		if (opts->overwrite_ignore) {
696 			topts.dir = xcalloc(1, sizeof(*topts.dir));
697 			topts.dir->flags |= DIR_SHOW_IGNORED;
698 			setup_standard_excludes(topts.dir);
699 		}
700 		tree = parse_tree_indirect(old_branch_info->commit ?
701 					   &old_branch_info->commit->object.oid :
702 					   the_hash_algo->empty_tree);
703 		init_tree_desc(&trees[0], tree->buffer, tree->size);
704 		parse_tree(new_tree);
705 		tree = new_tree;
706 		init_tree_desc(&trees[1], tree->buffer, tree->size);
707 
708 		ret = unpack_trees(2, trees, &topts);
709 		clear_unpack_trees_porcelain(&topts);
710 		if (ret == -1) {
711 			/*
712 			 * Unpack couldn't do a trivial merge; either
713 			 * give up or do a real merge, depending on
714 			 * whether the merge flag was used.
715 			 */
716 			struct tree *work;
717 			struct tree *old_tree;
718 			struct merge_options o;
719 			struct strbuf sb = STRBUF_INIT;
720 			struct strbuf old_commit_shortname = STRBUF_INIT;
721 
722 			if (!opts->merge)
723 				return 1;
724 
725 			/*
726 			 * Without old_branch_info->commit, the below is the same as
727 			 * the two-tree unpack we already tried and failed.
728 			 */
729 			if (!old_branch_info->commit)
730 				return 1;
731 			old_tree = get_commit_tree(old_branch_info->commit);
732 
733 			if (repo_index_has_changes(the_repository, old_tree, &sb))
734 				die(_("cannot continue with staged changes in "
735 				      "the following files:\n%s"), sb.buf);
736 			strbuf_release(&sb);
737 
738 			/* Do more real merge */
739 
740 			/*
741 			 * We update the index fully, then write the
742 			 * tree from the index, then merge the new
743 			 * branch with the current tree, with the old
744 			 * branch as the base. Then we reset the index
745 			 * (but not the working tree) to the new
746 			 * branch, leaving the working tree as the
747 			 * merged version, but skipping unmerged
748 			 * entries in the index.
749 			 */
750 
751 			add_files_to_cache(NULL, NULL, 0);
752 			/*
753 			 * NEEDSWORK: carrying over local changes
754 			 * when branches have different end-of-line
755 			 * normalization (or clean+smudge rules) is
756 			 * a pain; plumb in an option to set
757 			 * o.renormalize?
758 			 */
759 			init_merge_options(&o, the_repository);
760 			o.verbosity = 0;
761 			work = write_in_core_index_as_tree(the_repository);
762 
763 			ret = reset_tree(new_tree,
764 					 opts, 1,
765 					 writeout_error);
766 			if (ret)
767 				return ret;
768 			o.ancestor = old_branch_info->name;
769 			if (old_branch_info->name == NULL) {
770 				strbuf_add_unique_abbrev(&old_commit_shortname,
771 							 &old_branch_info->commit->object.oid,
772 							 DEFAULT_ABBREV);
773 				o.ancestor = old_commit_shortname.buf;
774 			}
775 			o.branch1 = new_branch_info->name;
776 			o.branch2 = "local";
777 			ret = merge_trees(&o,
778 					  new_tree,
779 					  work,
780 					  old_tree);
781 			if (ret < 0)
782 				exit(128);
783 			ret = reset_tree(new_tree,
784 					 opts, 0,
785 					 writeout_error);
786 			strbuf_release(&o.obuf);
787 			strbuf_release(&old_commit_shortname);
788 			if (ret)
789 				return ret;
790 		}
791 	}
792 
793 	if (!active_cache_tree)
794 		active_cache_tree = cache_tree();
795 
796 	if (!cache_tree_fully_valid(active_cache_tree))
797 		cache_tree_update(&the_index, WRITE_TREE_SILENT | WRITE_TREE_REPAIR);
798 
799 	if (write_locked_index(&the_index, &lock_file, COMMIT_LOCK))
800 		die(_("unable to write new index file"));
801 
802 	if (!opts->discard_changes && !opts->quiet && new_branch_info->commit)
803 		show_local_changes(&new_branch_info->commit->object, &opts->diff_options);
804 
805 	return 0;
806 }
807 
report_tracking(struct branch_info * new_branch_info)808 static void report_tracking(struct branch_info *new_branch_info)
809 {
810 	struct strbuf sb = STRBUF_INIT;
811 	struct branch *branch = branch_get(new_branch_info->name);
812 
813 	if (!format_tracking_info(branch, &sb, AHEAD_BEHIND_FULL))
814 		return;
815 	fputs(sb.buf, stdout);
816 	strbuf_release(&sb);
817 }
818 
update_refs_for_switch(const struct checkout_opts * opts,struct branch_info * old_branch_info,struct branch_info * new_branch_info)819 static void update_refs_for_switch(const struct checkout_opts *opts,
820 				   struct branch_info *old_branch_info,
821 				   struct branch_info *new_branch_info)
822 {
823 	struct strbuf msg = STRBUF_INIT;
824 	const char *old_desc, *reflog_msg;
825 	if (opts->new_branch) {
826 		if (opts->new_orphan_branch) {
827 			char *refname;
828 
829 			refname = mkpathdup("refs/heads/%s", opts->new_orphan_branch);
830 			if (opts->new_branch_log &&
831 			    !should_autocreate_reflog(refname)) {
832 				int ret;
833 				struct strbuf err = STRBUF_INIT;
834 
835 				ret = safe_create_reflog(refname, 1, &err);
836 				if (ret) {
837 					fprintf(stderr, _("Can not do reflog for '%s': %s\n"),
838 						opts->new_orphan_branch, err.buf);
839 					strbuf_release(&err);
840 					free(refname);
841 					return;
842 				}
843 				strbuf_release(&err);
844 			}
845 			free(refname);
846 		}
847 		else
848 			create_branch(the_repository,
849 				      opts->new_branch, new_branch_info->name,
850 				      opts->new_branch_force ? 1 : 0,
851 				      opts->new_branch_force ? 1 : 0,
852 				      opts->new_branch_log,
853 				      opts->quiet,
854 				      opts->track);
855 		new_branch_info->name = opts->new_branch;
856 		setup_branch_path(new_branch_info);
857 	}
858 
859 	old_desc = old_branch_info->name;
860 	if (!old_desc && old_branch_info->commit)
861 		old_desc = oid_to_hex(&old_branch_info->commit->object.oid);
862 
863 	reflog_msg = getenv("GIT_REFLOG_ACTION");
864 	if (!reflog_msg)
865 		strbuf_addf(&msg, "checkout: moving from %s to %s",
866 			old_desc ? old_desc : "(invalid)", new_branch_info->name);
867 	else
868 		strbuf_insert(&msg, 0, reflog_msg, strlen(reflog_msg));
869 
870 	if (!strcmp(new_branch_info->name, "HEAD") && !new_branch_info->path && !opts->force_detach) {
871 		/* Nothing to do. */
872 	} else if (opts->force_detach || !new_branch_info->path) {	/* No longer on any branch. */
873 		update_ref(msg.buf, "HEAD", &new_branch_info->commit->object.oid, NULL,
874 			   REF_NO_DEREF, UPDATE_REFS_DIE_ON_ERR);
875 		if (!opts->quiet) {
876 			if (old_branch_info->path &&
877 			    advice_detached_head && !opts->force_detach)
878 				detach_advice(new_branch_info->name);
879 			describe_detached_head(_("HEAD is now at"), new_branch_info->commit);
880 		}
881 	} else if (new_branch_info->path) {	/* Switch branches. */
882 		if (create_symref("HEAD", new_branch_info->path, msg.buf) < 0)
883 			die(_("unable to update HEAD"));
884 		if (!opts->quiet) {
885 			if (old_branch_info->path && !strcmp(new_branch_info->path, old_branch_info->path)) {
886 				if (opts->new_branch_force)
887 					fprintf(stderr, _("Reset branch '%s'\n"),
888 						new_branch_info->name);
889 				else
890 					fprintf(stderr, _("Already on '%s'\n"),
891 						new_branch_info->name);
892 			} else if (opts->new_branch) {
893 				if (opts->branch_exists)
894 					fprintf(stderr, _("Switched to and reset branch '%s'\n"), new_branch_info->name);
895 				else
896 					fprintf(stderr, _("Switched to a new branch '%s'\n"), new_branch_info->name);
897 			} else {
898 				fprintf(stderr, _("Switched to branch '%s'\n"),
899 					new_branch_info->name);
900 			}
901 		}
902 		if (old_branch_info->path && old_branch_info->name) {
903 			if (!ref_exists(old_branch_info->path) && reflog_exists(old_branch_info->path))
904 				delete_reflog(old_branch_info->path);
905 		}
906 	}
907 	remove_branch_state(the_repository, !opts->quiet);
908 	strbuf_release(&msg);
909 	if (!opts->quiet &&
910 	    (new_branch_info->path || (!opts->force_detach && !strcmp(new_branch_info->name, "HEAD"))))
911 		report_tracking(new_branch_info);
912 }
913 
add_pending_uninteresting_ref(const char * refname,const struct object_id * oid,int flags,void * cb_data)914 static int add_pending_uninteresting_ref(const char *refname,
915 					 const struct object_id *oid,
916 					 int flags, void *cb_data)
917 {
918 	add_pending_oid(cb_data, refname, oid, UNINTERESTING);
919 	return 0;
920 }
921 
describe_one_orphan(struct strbuf * sb,struct commit * commit)922 static void describe_one_orphan(struct strbuf *sb, struct commit *commit)
923 {
924 	strbuf_addstr(sb, "  ");
925 	strbuf_add_unique_abbrev(sb, &commit->object.oid, DEFAULT_ABBREV);
926 	strbuf_addch(sb, ' ');
927 	if (!parse_commit(commit))
928 		pp_commit_easy(CMIT_FMT_ONELINE, commit, sb);
929 	strbuf_addch(sb, '\n');
930 }
931 
932 #define ORPHAN_CUTOFF 4
suggest_reattach(struct commit * commit,struct rev_info * revs)933 static void suggest_reattach(struct commit *commit, struct rev_info *revs)
934 {
935 	struct commit *c, *last = NULL;
936 	struct strbuf sb = STRBUF_INIT;
937 	int lost = 0;
938 	while ((c = get_revision(revs)) != NULL) {
939 		if (lost < ORPHAN_CUTOFF)
940 			describe_one_orphan(&sb, c);
941 		last = c;
942 		lost++;
943 	}
944 	if (ORPHAN_CUTOFF < lost) {
945 		int more = lost - ORPHAN_CUTOFF;
946 		if (more == 1)
947 			describe_one_orphan(&sb, last);
948 		else
949 			strbuf_addf(&sb, _(" ... and %d more.\n"), more);
950 	}
951 
952 	fprintf(stderr,
953 		Q_(
954 		/* The singular version */
955 		"Warning: you are leaving %d commit behind, "
956 		"not connected to\n"
957 		"any of your branches:\n\n"
958 		"%s\n",
959 		/* The plural version */
960 		"Warning: you are leaving %d commits behind, "
961 		"not connected to\n"
962 		"any of your branches:\n\n"
963 		"%s\n",
964 		/* Give ngettext() the count */
965 		lost),
966 		lost,
967 		sb.buf);
968 	strbuf_release(&sb);
969 
970 	if (advice_detached_head)
971 		fprintf(stderr,
972 			Q_(
973 			/* The singular version */
974 			"If you want to keep it by creating a new branch, "
975 			"this may be a good time\nto do so with:\n\n"
976 			" git branch <new-branch-name> %s\n\n",
977 			/* The plural version */
978 			"If you want to keep them by creating a new branch, "
979 			"this may be a good time\nto do so with:\n\n"
980 			" git branch <new-branch-name> %s\n\n",
981 			/* Give ngettext() the count */
982 			lost),
983 			find_unique_abbrev(&commit->object.oid, DEFAULT_ABBREV));
984 }
985 
986 /*
987  * We are about to leave commit that was at the tip of a detached
988  * HEAD.  If it is not reachable from any ref, this is the last chance
989  * for the user to do so without resorting to reflog.
990  */
orphaned_commit_warning(struct commit * old_commit,struct commit * new_commit)991 static void orphaned_commit_warning(struct commit *old_commit, struct commit *new_commit)
992 {
993 	struct rev_info revs;
994 	struct object *object = &old_commit->object;
995 
996 	repo_init_revisions(the_repository, &revs, NULL);
997 	setup_revisions(0, NULL, &revs, NULL);
998 
999 	object->flags &= ~UNINTERESTING;
1000 	add_pending_object(&revs, object, oid_to_hex(&object->oid));
1001 
1002 	for_each_ref(add_pending_uninteresting_ref, &revs);
1003 	if (new_commit)
1004 		add_pending_oid(&revs, "HEAD",
1005 				&new_commit->object.oid,
1006 				UNINTERESTING);
1007 
1008 	if (prepare_revision_walk(&revs))
1009 		die(_("internal error in revision walk"));
1010 	if (!(old_commit->object.flags & UNINTERESTING))
1011 		suggest_reattach(old_commit, &revs);
1012 	else
1013 		describe_detached_head(_("Previous HEAD position was"), old_commit);
1014 
1015 	/* Clean up objects used, as they will be reused. */
1016 	clear_commit_marks_all(ALL_REV_FLAGS);
1017 }
1018 
switch_branches(const struct checkout_opts * opts,struct branch_info * new_branch_info)1019 static int switch_branches(const struct checkout_opts *opts,
1020 			   struct branch_info *new_branch_info)
1021 {
1022 	int ret = 0;
1023 	struct branch_info old_branch_info;
1024 	void *path_to_free;
1025 	struct object_id rev;
1026 	int flag, writeout_error = 0;
1027 	int do_merge = 1;
1028 
1029 	trace2_cmd_mode("branch");
1030 
1031 	memset(&old_branch_info, 0, sizeof(old_branch_info));
1032 	old_branch_info.path = path_to_free = resolve_refdup("HEAD", 0, &rev, &flag);
1033 	if (old_branch_info.path)
1034 		old_branch_info.commit = lookup_commit_reference_gently(the_repository, &rev, 1);
1035 	if (!(flag & REF_ISSYMREF))
1036 		old_branch_info.path = NULL;
1037 
1038 	if (old_branch_info.path)
1039 		skip_prefix(old_branch_info.path, "refs/heads/", &old_branch_info.name);
1040 
1041 	if (opts->new_orphan_branch && opts->orphan_from_empty_tree) {
1042 		if (new_branch_info->name)
1043 			BUG("'switch --orphan' should never accept a commit as starting point");
1044 		new_branch_info->commit = NULL;
1045 		new_branch_info->name = "(empty)";
1046 		do_merge = 1;
1047 	}
1048 
1049 	if (!new_branch_info->name) {
1050 		new_branch_info->name = "HEAD";
1051 		new_branch_info->commit = old_branch_info.commit;
1052 		if (!new_branch_info->commit)
1053 			die(_("You are on a branch yet to be born"));
1054 		parse_commit_or_die(new_branch_info->commit);
1055 
1056 		if (opts->only_merge_on_switching_branches)
1057 			do_merge = 0;
1058 	}
1059 
1060 	if (do_merge) {
1061 		ret = merge_working_tree(opts, &old_branch_info, new_branch_info, &writeout_error);
1062 		if (ret) {
1063 			free(path_to_free);
1064 			return ret;
1065 		}
1066 	}
1067 
1068 	if (!opts->quiet && !old_branch_info.path && old_branch_info.commit && new_branch_info->commit != old_branch_info.commit)
1069 		orphaned_commit_warning(old_branch_info.commit, new_branch_info->commit);
1070 
1071 	update_refs_for_switch(opts, &old_branch_info, new_branch_info);
1072 
1073 	ret = post_checkout_hook(old_branch_info.commit, new_branch_info->commit, 1);
1074 	free(path_to_free);
1075 	return ret || writeout_error;
1076 }
1077 
git_checkout_config(const char * var,const char * value,void * cb)1078 static int git_checkout_config(const char *var, const char *value, void *cb)
1079 {
1080 	if (!strcmp(var, "diff.ignoresubmodules")) {
1081 		struct checkout_opts *opts = cb;
1082 		handle_ignore_submodules_arg(&opts->diff_options, value);
1083 		return 0;
1084 	}
1085 
1086 	if (starts_with(var, "submodule."))
1087 		return git_default_submodule_config(var, value, NULL);
1088 
1089 	return git_xmerge_config(var, value, NULL);
1090 }
1091 
setup_new_branch_info_and_source_tree(struct branch_info * new_branch_info,struct checkout_opts * opts,struct object_id * rev,const char * arg)1092 static void setup_new_branch_info_and_source_tree(
1093 	struct branch_info *new_branch_info,
1094 	struct checkout_opts *opts,
1095 	struct object_id *rev,
1096 	const char *arg)
1097 {
1098 	struct tree **source_tree = &opts->source_tree;
1099 	struct object_id branch_rev;
1100 
1101 	new_branch_info->name = arg;
1102 	setup_branch_path(new_branch_info);
1103 
1104 	if (!check_refname_format(new_branch_info->path, 0) &&
1105 	    !read_ref(new_branch_info->path, &branch_rev))
1106 		oidcpy(rev, &branch_rev);
1107 	else
1108 		new_branch_info->path = NULL; /* not an existing branch */
1109 
1110 	new_branch_info->commit = lookup_commit_reference_gently(the_repository, rev, 1);
1111 	if (!new_branch_info->commit) {
1112 		/* not a commit */
1113 		*source_tree = parse_tree_indirect(rev);
1114 	} else {
1115 		parse_commit_or_die(new_branch_info->commit);
1116 		*source_tree = get_commit_tree(new_branch_info->commit);
1117 	}
1118 }
1119 
parse_branchname_arg(int argc,const char ** argv,int dwim_new_local_branch_ok,struct branch_info * new_branch_info,struct checkout_opts * opts,struct object_id * rev,int * dwim_remotes_matched)1120 static int parse_branchname_arg(int argc, const char **argv,
1121 				int dwim_new_local_branch_ok,
1122 				struct branch_info *new_branch_info,
1123 				struct checkout_opts *opts,
1124 				struct object_id *rev,
1125 				int *dwim_remotes_matched)
1126 {
1127 	const char **new_branch = &opts->new_branch;
1128 	int argcount = 0;
1129 	const char *arg;
1130 	int dash_dash_pos;
1131 	int has_dash_dash = 0;
1132 	int i;
1133 
1134 	/*
1135 	 * case 1: git checkout <ref> -- [<paths>]
1136 	 *
1137 	 *   <ref> must be a valid tree, everything after the '--' must be
1138 	 *   a path.
1139 	 *
1140 	 * case 2: git checkout -- [<paths>]
1141 	 *
1142 	 *   everything after the '--' must be paths.
1143 	 *
1144 	 * case 3: git checkout <something> [--]
1145 	 *
1146 	 *   (a) If <something> is a commit, that is to
1147 	 *       switch to the branch or detach HEAD at it.  As a special case,
1148 	 *       if <something> is A...B (missing A or B means HEAD but you can
1149 	 *       omit at most one side), and if there is a unique merge base
1150 	 *       between A and B, A...B names that merge base.
1151 	 *
1152 	 *   (b) If <something> is _not_ a commit, either "--" is present
1153 	 *       or <something> is not a path, no -t or -b was given, and
1154 	 *       and there is a tracking branch whose name is <something>
1155 	 *       in one and only one remote (or if the branch exists on the
1156 	 *       remote named in checkout.defaultRemote), then this is a
1157 	 *       short-hand to fork local <something> from that
1158 	 *       remote-tracking branch.
1159 	 *
1160 	 *   (c) Otherwise, if "--" is present, treat it like case (1).
1161 	 *
1162 	 *   (d) Otherwise :
1163 	 *       - if it's a reference, treat it like case (1)
1164 	 *       - else if it's a path, treat it like case (2)
1165 	 *       - else: fail.
1166 	 *
1167 	 * case 4: git checkout <something> <paths>
1168 	 *
1169 	 *   The first argument must not be ambiguous.
1170 	 *   - If it's *only* a reference, treat it like case (1).
1171 	 *   - If it's only a path, treat it like case (2).
1172 	 *   - else: fail.
1173 	 *
1174 	 */
1175 	if (!argc)
1176 		return 0;
1177 
1178 	if (!opts->accept_pathspec) {
1179 		if (argc > 1)
1180 			die(_("only one reference expected"));
1181 		has_dash_dash = 1; /* helps disambiguate */
1182 	}
1183 
1184 	arg = argv[0];
1185 	dash_dash_pos = -1;
1186 	for (i = 0; i < argc; i++) {
1187 		if (opts->accept_pathspec && !strcmp(argv[i], "--")) {
1188 			dash_dash_pos = i;
1189 			break;
1190 		}
1191 	}
1192 	if (dash_dash_pos == 0)
1193 		return 1; /* case (2) */
1194 	else if (dash_dash_pos == 1)
1195 		has_dash_dash = 1; /* case (3) or (1) */
1196 	else if (dash_dash_pos >= 2)
1197 		die(_("only one reference expected, %d given."), dash_dash_pos);
1198 	opts->count_checkout_paths = !opts->quiet && !has_dash_dash;
1199 
1200 	if (!strcmp(arg, "-"))
1201 		arg = "@{-1}";
1202 
1203 	if (get_oid_mb(arg, rev)) {
1204 		/*
1205 		 * Either case (3) or (4), with <something> not being
1206 		 * a commit, or an attempt to use case (1) with an
1207 		 * invalid ref.
1208 		 *
1209 		 * It's likely an error, but we need to find out if
1210 		 * we should auto-create the branch, case (3).(b).
1211 		 */
1212 		int recover_with_dwim = dwim_new_local_branch_ok;
1213 
1214 		int could_be_checkout_paths = !has_dash_dash &&
1215 			check_filename(opts->prefix, arg);
1216 
1217 		if (!has_dash_dash && !no_wildcard(arg))
1218 			recover_with_dwim = 0;
1219 
1220 		/*
1221 		 * Accept "git checkout foo", "git checkout foo --"
1222 		 * and "git switch foo" as candidates for dwim.
1223 		 */
1224 		if (!(argc == 1 && !has_dash_dash) &&
1225 		    !(argc == 2 && has_dash_dash) &&
1226 		    opts->accept_pathspec)
1227 			recover_with_dwim = 0;
1228 
1229 		if (recover_with_dwim) {
1230 			const char *remote = unique_tracking_name(arg, rev,
1231 								  dwim_remotes_matched);
1232 			if (remote) {
1233 				if (could_be_checkout_paths)
1234 					die(_("'%s' could be both a local file and a tracking branch.\n"
1235 					      "Please use -- (and optionally --no-guess) to disambiguate"),
1236 					    arg);
1237 				*new_branch = arg;
1238 				arg = remote;
1239 				/* DWIMmed to create local branch, case (3).(b) */
1240 			} else {
1241 				recover_with_dwim = 0;
1242 			}
1243 		}
1244 
1245 		if (!recover_with_dwim) {
1246 			if (has_dash_dash)
1247 				die(_("invalid reference: %s"), arg);
1248 			return argcount;
1249 		}
1250 	}
1251 
1252 	/* we can't end up being in (2) anymore, eat the argument */
1253 	argcount++;
1254 	argv++;
1255 	argc--;
1256 
1257 	setup_new_branch_info_and_source_tree(new_branch_info, opts, rev, arg);
1258 
1259 	if (!opts->source_tree)                   /* case (1): want a tree */
1260 		die(_("reference is not a tree: %s"), arg);
1261 
1262 	if (!has_dash_dash) {	/* case (3).(d) -> (1) */
1263 		/*
1264 		 * Do not complain the most common case
1265 		 *	git checkout branch
1266 		 * even if there happen to be a file called 'branch';
1267 		 * it would be extremely annoying.
1268 		 */
1269 		if (argc)
1270 			verify_non_filename(opts->prefix, arg);
1271 	} else if (opts->accept_pathspec) {
1272 		argcount++;
1273 		argv++;
1274 		argc--;
1275 	}
1276 
1277 	return argcount;
1278 }
1279 
switch_unborn_to_new_branch(const struct checkout_opts * opts)1280 static int switch_unborn_to_new_branch(const struct checkout_opts *opts)
1281 {
1282 	int status;
1283 	struct strbuf branch_ref = STRBUF_INIT;
1284 
1285 	trace2_cmd_mode("unborn");
1286 
1287 	if (!opts->new_branch)
1288 		die(_("You are on a branch yet to be born"));
1289 	strbuf_addf(&branch_ref, "refs/heads/%s", opts->new_branch);
1290 	status = create_symref("HEAD", branch_ref.buf, "checkout -b");
1291 	strbuf_release(&branch_ref);
1292 	if (!opts->quiet)
1293 		fprintf(stderr, _("Switched to a new branch '%s'\n"),
1294 			opts->new_branch);
1295 	return status;
1296 }
1297 
die_expecting_a_branch(const struct branch_info * branch_info)1298 static void die_expecting_a_branch(const struct branch_info *branch_info)
1299 {
1300 	struct object_id oid;
1301 	char *to_free;
1302 
1303 	if (dwim_ref(branch_info->name, strlen(branch_info->name), &oid, &to_free) == 1) {
1304 		const char *ref = to_free;
1305 
1306 		if (skip_prefix(ref, "refs/tags/", &ref))
1307 			die(_("a branch is expected, got tag '%s'"), ref);
1308 		if (skip_prefix(ref, "refs/remotes/", &ref))
1309 			die(_("a branch is expected, got remote branch '%s'"), ref);
1310 		die(_("a branch is expected, got '%s'"), ref);
1311 	}
1312 	if (branch_info->commit)
1313 		die(_("a branch is expected, got commit '%s'"), branch_info->name);
1314 	/*
1315 	 * This case should never happen because we already die() on
1316 	 * non-commit, but just in case.
1317 	 */
1318 	die(_("a branch is expected, got '%s'"), branch_info->name);
1319 }
1320 
die_if_some_operation_in_progress(void)1321 static void die_if_some_operation_in_progress(void)
1322 {
1323 	struct wt_status_state state;
1324 
1325 	memset(&state, 0, sizeof(state));
1326 	wt_status_get_state(the_repository, &state, 0);
1327 
1328 	if (state.merge_in_progress)
1329 		die(_("cannot switch branch while merging\n"
1330 		      "Consider \"git merge --quit\" "
1331 		      "or \"git worktree add\"."));
1332 	if (state.am_in_progress)
1333 		die(_("cannot switch branch in the middle of an am session\n"
1334 		      "Consider \"git am --quit\" "
1335 		      "or \"git worktree add\"."));
1336 	if (state.rebase_interactive_in_progress || state.rebase_in_progress)
1337 		die(_("cannot switch branch while rebasing\n"
1338 		      "Consider \"git rebase --quit\" "
1339 		      "or \"git worktree add\"."));
1340 	if (state.cherry_pick_in_progress)
1341 		die(_("cannot switch branch while cherry-picking\n"
1342 		      "Consider \"git cherry-pick --quit\" "
1343 		      "or \"git worktree add\"."));
1344 	if (state.revert_in_progress)
1345 		die(_("cannot switch branch while reverting\n"
1346 		      "Consider \"git revert --quit\" "
1347 		      "or \"git worktree add\"."));
1348 	if (state.bisect_in_progress)
1349 		warning(_("you are switching branch while bisecting"));
1350 }
1351 
checkout_branch(struct checkout_opts * opts,struct branch_info * new_branch_info)1352 static int checkout_branch(struct checkout_opts *opts,
1353 			   struct branch_info *new_branch_info)
1354 {
1355 	if (opts->pathspec.nr)
1356 		die(_("paths cannot be used with switching branches"));
1357 
1358 	if (opts->patch_mode)
1359 		die(_("'%s' cannot be used with switching branches"),
1360 		    "--patch");
1361 
1362 	if (opts->overlay_mode != -1)
1363 		die(_("'%s' cannot be used with switching branches"),
1364 		    "--[no]-overlay");
1365 
1366 	if (opts->writeout_stage)
1367 		die(_("'%s' cannot be used with switching branches"),
1368 		    "--ours/--theirs");
1369 
1370 	if (opts->force && opts->merge)
1371 		die(_("'%s' cannot be used with '%s'"), "-f", "-m");
1372 
1373 	if (opts->discard_changes && opts->merge)
1374 		die(_("'%s' cannot be used with '%s'"), "--discard-changes", "--merge");
1375 
1376 	if (opts->force_detach && opts->new_branch)
1377 		die(_("'%s' cannot be used with '%s'"),
1378 		    "--detach", "-b/-B/--orphan");
1379 
1380 	if (opts->new_orphan_branch) {
1381 		if (opts->track != BRANCH_TRACK_UNSPECIFIED)
1382 			die(_("'%s' cannot be used with '%s'"), "--orphan", "-t");
1383 		if (opts->orphan_from_empty_tree && new_branch_info->name)
1384 			die(_("'%s' cannot take <start-point>"), "--orphan");
1385 	} else if (opts->force_detach) {
1386 		if (opts->track != BRANCH_TRACK_UNSPECIFIED)
1387 			die(_("'%s' cannot be used with '%s'"), "--detach", "-t");
1388 	} else if (opts->track == BRANCH_TRACK_UNSPECIFIED)
1389 		opts->track = git_branch_track;
1390 
1391 	if (new_branch_info->name && !new_branch_info->commit)
1392 		die(_("Cannot switch branch to a non-commit '%s'"),
1393 		    new_branch_info->name);
1394 
1395 	if (!opts->switch_branch_doing_nothing_is_ok &&
1396 	    !new_branch_info->name &&
1397 	    !opts->new_branch &&
1398 	    !opts->force_detach)
1399 		die(_("missing branch or commit argument"));
1400 
1401 	if (!opts->implicit_detach &&
1402 	    !opts->force_detach &&
1403 	    !opts->new_branch &&
1404 	    !opts->new_branch_force &&
1405 	    new_branch_info->name &&
1406 	    !new_branch_info->path)
1407 		die_expecting_a_branch(new_branch_info);
1408 
1409 	if (!opts->can_switch_when_in_progress)
1410 		die_if_some_operation_in_progress();
1411 
1412 	if (new_branch_info->path && !opts->force_detach && !opts->new_branch &&
1413 	    !opts->ignore_other_worktrees) {
1414 		int flag;
1415 		char *head_ref = resolve_refdup("HEAD", 0, NULL, &flag);
1416 		if (head_ref &&
1417 		    (!(flag & REF_ISSYMREF) || strcmp(head_ref, new_branch_info->path)))
1418 			die_if_checked_out(new_branch_info->path, 1);
1419 		free(head_ref);
1420 	}
1421 
1422 	if (!new_branch_info->commit && opts->new_branch) {
1423 		struct object_id rev;
1424 		int flag;
1425 
1426 		if (!read_ref_full("HEAD", 0, &rev, &flag) &&
1427 		    (flag & REF_ISSYMREF) && is_null_oid(&rev))
1428 			return switch_unborn_to_new_branch(opts);
1429 	}
1430 	return switch_branches(opts, new_branch_info);
1431 }
1432 
add_common_options(struct checkout_opts * opts,struct option * prevopts)1433 static struct option *add_common_options(struct checkout_opts *opts,
1434 					 struct option *prevopts)
1435 {
1436 	struct option options[] = {
1437 		OPT__QUIET(&opts->quiet, N_("suppress progress reporting")),
1438 		{ OPTION_CALLBACK, 0, "recurse-submodules", NULL,
1439 			    "checkout", "control recursive updating of submodules",
1440 			    PARSE_OPT_OPTARG, option_parse_recurse_submodules_worktree_updater },
1441 		OPT_BOOL(0, "progress", &opts->show_progress, N_("force progress reporting")),
1442 		OPT_BOOL('m', "merge", &opts->merge, N_("perform a 3-way merge with the new branch")),
1443 		OPT_STRING(0, "conflict", &opts->conflict_style, N_("style"),
1444 			   N_("conflict style (merge or diff3)")),
1445 		OPT_END()
1446 	};
1447 	struct option *newopts = parse_options_concat(prevopts, options);
1448 	free(prevopts);
1449 	return newopts;
1450 }
1451 
add_common_switch_branch_options(struct checkout_opts * opts,struct option * prevopts)1452 static struct option *add_common_switch_branch_options(
1453 	struct checkout_opts *opts, struct option *prevopts)
1454 {
1455 	struct option options[] = {
1456 		OPT_BOOL('d', "detach", &opts->force_detach, N_("detach HEAD at named commit")),
1457 		OPT_SET_INT('t', "track",  &opts->track, N_("set upstream info for new branch"),
1458 			BRANCH_TRACK_EXPLICIT),
1459 		OPT__FORCE(&opts->force, N_("force checkout (throw away local modifications)"),
1460 			   PARSE_OPT_NOCOMPLETE),
1461 		OPT_STRING(0, "orphan", &opts->new_orphan_branch, N_("new-branch"), N_("new unparented branch")),
1462 		OPT_BOOL_F(0, "overwrite-ignore", &opts->overwrite_ignore,
1463 			   N_("update ignored files (default)"),
1464 			   PARSE_OPT_NOCOMPLETE),
1465 		OPT_BOOL(0, "ignore-other-worktrees", &opts->ignore_other_worktrees,
1466 			 N_("do not check if another worktree is holding the given ref")),
1467 		OPT_END()
1468 	};
1469 	struct option *newopts = parse_options_concat(prevopts, options);
1470 	free(prevopts);
1471 	return newopts;
1472 }
1473 
add_checkout_path_options(struct checkout_opts * opts,struct option * prevopts)1474 static struct option *add_checkout_path_options(struct checkout_opts *opts,
1475 						struct option *prevopts)
1476 {
1477 	struct option options[] = {
1478 		OPT_SET_INT_F('2', "ours", &opts->writeout_stage,
1479 			      N_("checkout our version for unmerged files"),
1480 			      2, PARSE_OPT_NONEG),
1481 		OPT_SET_INT_F('3', "theirs", &opts->writeout_stage,
1482 			      N_("checkout their version for unmerged files"),
1483 			      3, PARSE_OPT_NONEG),
1484 		OPT_BOOL('p', "patch", &opts->patch_mode, N_("select hunks interactively")),
1485 		OPT_BOOL(0, "ignore-skip-worktree-bits", &opts->ignore_skipworktree,
1486 			 N_("do not limit pathspecs to sparse entries only")),
1487 		OPT_PATHSPEC_FROM_FILE(&opts->pathspec_from_file),
1488 		OPT_PATHSPEC_FILE_NUL(&opts->pathspec_file_nul),
1489 		OPT_END()
1490 	};
1491 	struct option *newopts = parse_options_concat(prevopts, options);
1492 	free(prevopts);
1493 	return newopts;
1494 }
1495 
checkout_main(int argc,const char ** argv,const char * prefix,struct checkout_opts * opts,struct option * options,const char * const usagestr[])1496 static int checkout_main(int argc, const char **argv, const char *prefix,
1497 			 struct checkout_opts *opts, struct option *options,
1498 			 const char * const usagestr[])
1499 {
1500 	struct branch_info new_branch_info;
1501 	int dwim_remotes_matched = 0;
1502 	int parseopt_flags = 0;
1503 
1504 	memset(&new_branch_info, 0, sizeof(new_branch_info));
1505 	opts->overwrite_ignore = 1;
1506 	opts->prefix = prefix;
1507 	opts->show_progress = -1;
1508 
1509 	git_config(git_checkout_config, opts);
1510 
1511 	opts->track = BRANCH_TRACK_UNSPECIFIED;
1512 
1513 	if (!opts->accept_pathspec && !opts->accept_ref)
1514 		BUG("make up your mind, you need to take _something_");
1515 	if (opts->accept_pathspec && opts->accept_ref)
1516 		parseopt_flags = PARSE_OPT_KEEP_DASHDASH;
1517 
1518 	argc = parse_options(argc, argv, prefix, options,
1519 			     usagestr, parseopt_flags);
1520 
1521 	if (opts->show_progress < 0) {
1522 		if (opts->quiet)
1523 			opts->show_progress = 0;
1524 		else
1525 			opts->show_progress = isatty(2);
1526 	}
1527 
1528 	if (opts->conflict_style) {
1529 		opts->merge = 1; /* implied */
1530 		git_xmerge_config("merge.conflictstyle", opts->conflict_style, NULL);
1531 	}
1532 	if (opts->force) {
1533 		opts->discard_changes = 1;
1534 		opts->ignore_unmerged_opt = "--force";
1535 		opts->ignore_unmerged = 1;
1536 	}
1537 
1538 	if ((!!opts->new_branch + !!opts->new_branch_force + !!opts->new_orphan_branch) > 1)
1539 		die(_("-b, -B and --orphan are mutually exclusive"));
1540 
1541 	if (opts->overlay_mode == 1 && opts->patch_mode)
1542 		die(_("-p and --overlay are mutually exclusive"));
1543 
1544 	if (opts->checkout_index >= 0 || opts->checkout_worktree >= 0) {
1545 		if (opts->checkout_index < 0)
1546 			opts->checkout_index = 0;
1547 		if (opts->checkout_worktree < 0)
1548 			opts->checkout_worktree = 0;
1549 	} else {
1550 		if (opts->checkout_index < 0)
1551 			opts->checkout_index = -opts->checkout_index - 1;
1552 		if (opts->checkout_worktree < 0)
1553 			opts->checkout_worktree = -opts->checkout_worktree - 1;
1554 	}
1555 	if (opts->checkout_index < 0 || opts->checkout_worktree < 0)
1556 		BUG("these flags should be non-negative by now");
1557 	/*
1558 	 * convenient shortcut: "git restore --staged" equals
1559 	 * "git restore --staged --source HEAD"
1560 	 */
1561 	if (!opts->from_treeish && opts->checkout_index && !opts->checkout_worktree)
1562 		opts->from_treeish = "HEAD";
1563 
1564 	/*
1565 	 * From here on, new_branch will contain the branch to be checked out,
1566 	 * and new_branch_force and new_orphan_branch will tell us which one of
1567 	 * -b/-B/--orphan is being used.
1568 	 */
1569 	if (opts->new_branch_force)
1570 		opts->new_branch = opts->new_branch_force;
1571 
1572 	if (opts->new_orphan_branch)
1573 		opts->new_branch = opts->new_orphan_branch;
1574 
1575 	/* --track without -b/-B/--orphan should DWIM */
1576 	if (opts->track != BRANCH_TRACK_UNSPECIFIED && !opts->new_branch) {
1577 		const char *argv0 = argv[0];
1578 		if (!argc || !strcmp(argv0, "--"))
1579 			die(_("--track needs a branch name"));
1580 		skip_prefix(argv0, "refs/", &argv0);
1581 		skip_prefix(argv0, "remotes/", &argv0);
1582 		argv0 = strchr(argv0, '/');
1583 		if (!argv0 || !argv0[1])
1584 			die(_("missing branch name; try -b"));
1585 		opts->new_branch = argv0 + 1;
1586 	}
1587 
1588 	/*
1589 	 * Extract branch name from command line arguments, so
1590 	 * all that is left is pathspecs.
1591 	 *
1592 	 * Handle
1593 	 *
1594 	 *  1) git checkout <tree> -- [<paths>]
1595 	 *  2) git checkout -- [<paths>]
1596 	 *  3) git checkout <something> [<paths>]
1597 	 *
1598 	 * including "last branch" syntax and DWIM-ery for names of
1599 	 * remote branches, erroring out for invalid or ambiguous cases.
1600 	 */
1601 	if (argc && opts->accept_ref) {
1602 		struct object_id rev;
1603 		int dwim_ok =
1604 			!opts->patch_mode &&
1605 			opts->dwim_new_local_branch &&
1606 			opts->track == BRANCH_TRACK_UNSPECIFIED &&
1607 			!opts->new_branch;
1608 		int n = parse_branchname_arg(argc, argv, dwim_ok,
1609 					     &new_branch_info, opts, &rev,
1610 					     &dwim_remotes_matched);
1611 		argv += n;
1612 		argc -= n;
1613 	} else if (!opts->accept_ref && opts->from_treeish) {
1614 		struct object_id rev;
1615 
1616 		if (get_oid_mb(opts->from_treeish, &rev))
1617 			die(_("could not resolve %s"), opts->from_treeish);
1618 
1619 		setup_new_branch_info_and_source_tree(&new_branch_info,
1620 						      opts, &rev,
1621 						      opts->from_treeish);
1622 
1623 		if (!opts->source_tree)
1624 			die(_("reference is not a tree: %s"), opts->from_treeish);
1625 	}
1626 
1627 	if (argc) {
1628 		parse_pathspec(&opts->pathspec, 0,
1629 			       opts->patch_mode ? PATHSPEC_PREFIX_ORIGIN : 0,
1630 			       prefix, argv);
1631 
1632 		if (!opts->pathspec.nr)
1633 			die(_("invalid path specification"));
1634 
1635 		/*
1636 		 * Try to give more helpful suggestion.
1637 		 * new_branch && argc > 1 will be caught later.
1638 		 */
1639 		if (opts->new_branch && argc == 1)
1640 			die(_("'%s' is not a commit and a branch '%s' cannot be created from it"),
1641 				argv[0], opts->new_branch);
1642 
1643 		if (opts->force_detach)
1644 			die(_("git checkout: --detach does not take a path argument '%s'"),
1645 			    argv[0]);
1646 	}
1647 
1648 	if (opts->pathspec_from_file) {
1649 		if (opts->pathspec.nr)
1650 			die(_("--pathspec-from-file is incompatible with pathspec arguments"));
1651 
1652 		if (opts->force_detach)
1653 			die(_("--pathspec-from-file is incompatible with --detach"));
1654 
1655 		if (opts->patch_mode)
1656 			die(_("--pathspec-from-file is incompatible with --patch"));
1657 
1658 		parse_pathspec_file(&opts->pathspec, 0,
1659 				    0,
1660 				    prefix, opts->pathspec_from_file, opts->pathspec_file_nul);
1661 	} else if (opts->pathspec_file_nul) {
1662 		die(_("--pathspec-file-nul requires --pathspec-from-file"));
1663 	}
1664 
1665 	if (opts->pathspec.nr) {
1666 		if (1 < !!opts->writeout_stage + !!opts->force + !!opts->merge)
1667 			die(_("git checkout: --ours/--theirs, --force and --merge are incompatible when\n"
1668 			      "checking out of the index."));
1669 	} else {
1670 		if (opts->accept_pathspec && !opts->empty_pathspec_ok &&
1671 		    !opts->patch_mode)	/* patch mode is special */
1672 			die(_("you must specify path(s) to restore"));
1673 	}
1674 
1675 	if (opts->new_branch) {
1676 		struct strbuf buf = STRBUF_INIT;
1677 
1678 		if (opts->new_branch_force)
1679 			opts->branch_exists = validate_branchname(opts->new_branch, &buf);
1680 		else
1681 			opts->branch_exists =
1682 				validate_new_branchname(opts->new_branch, &buf, 0);
1683 		strbuf_release(&buf);
1684 	}
1685 
1686 	UNLEAK(opts);
1687 	if (opts->patch_mode || opts->pathspec.nr) {
1688 		int ret = checkout_paths(opts, new_branch_info.name);
1689 		if (ret && dwim_remotes_matched > 1 &&
1690 		    advice_checkout_ambiguous_remote_branch_name)
1691 			advise(_("'%s' matched more than one remote tracking branch.\n"
1692 				 "We found %d remotes with a reference that matched. So we fell back\n"
1693 				 "on trying to resolve the argument as a path, but failed there too!\n"
1694 				 "\n"
1695 				 "If you meant to check out a remote tracking branch on, e.g. 'origin',\n"
1696 				 "you can do so by fully qualifying the name with the --track option:\n"
1697 				 "\n"
1698 				 "    git checkout --track origin/<name>\n"
1699 				 "\n"
1700 				 "If you'd like to always have checkouts of an ambiguous <name> prefer\n"
1701 				 "one remote, e.g. the 'origin' remote, consider setting\n"
1702 				 "checkout.defaultRemote=origin in your config."),
1703 			       argv[0],
1704 			       dwim_remotes_matched);
1705 		return ret;
1706 	} else {
1707 		return checkout_branch(opts, &new_branch_info);
1708 	}
1709 }
1710 
cmd_checkout(int argc,const char ** argv,const char * prefix)1711 int cmd_checkout(int argc, const char **argv, const char *prefix)
1712 {
1713 	struct checkout_opts opts;
1714 	struct option *options;
1715 	struct option checkout_options[] = {
1716 		OPT_STRING('b', NULL, &opts.new_branch, N_("branch"),
1717 			   N_("create and checkout a new branch")),
1718 		OPT_STRING('B', NULL, &opts.new_branch_force, N_("branch"),
1719 			   N_("create/reset and checkout a branch")),
1720 		OPT_BOOL('l', NULL, &opts.new_branch_log, N_("create reflog for new branch")),
1721 		OPT_BOOL(0, "guess", &opts.dwim_new_local_branch,
1722 			 N_("second guess 'git checkout <no-such-branch>' (default)")),
1723 		OPT_BOOL(0, "overlay", &opts.overlay_mode, N_("use overlay mode (default)")),
1724 		OPT_END()
1725 	};
1726 	int ret;
1727 
1728 	memset(&opts, 0, sizeof(opts));
1729 	opts.dwim_new_local_branch = 1;
1730 	opts.switch_branch_doing_nothing_is_ok = 1;
1731 	opts.only_merge_on_switching_branches = 0;
1732 	opts.accept_ref = 1;
1733 	opts.accept_pathspec = 1;
1734 	opts.implicit_detach = 1;
1735 	opts.can_switch_when_in_progress = 1;
1736 	opts.orphan_from_empty_tree = 0;
1737 	opts.empty_pathspec_ok = 1;
1738 	opts.overlay_mode = -1;
1739 	opts.checkout_index = -2;    /* default on */
1740 	opts.checkout_worktree = -2; /* default on */
1741 
1742 	if (argc == 3 && !strcmp(argv[1], "-b")) {
1743 		/*
1744 		 * User ran 'git checkout -b <branch>' and expects
1745 		 * the same behavior as 'git switch -c <branch>'.
1746 		 */
1747 		opts.switch_branch_doing_nothing_is_ok = 0;
1748 		opts.only_merge_on_switching_branches = 1;
1749 	}
1750 
1751 	options = parse_options_dup(checkout_options);
1752 	options = add_common_options(&opts, options);
1753 	options = add_common_switch_branch_options(&opts, options);
1754 	options = add_checkout_path_options(&opts, options);
1755 
1756 	ret = checkout_main(argc, argv, prefix, &opts,
1757 			    options, checkout_usage);
1758 	FREE_AND_NULL(options);
1759 	return ret;
1760 }
1761 
cmd_switch(int argc,const char ** argv,const char * prefix)1762 int cmd_switch(int argc, const char **argv, const char *prefix)
1763 {
1764 	struct checkout_opts opts;
1765 	struct option *options = NULL;
1766 	struct option switch_options[] = {
1767 		OPT_STRING('c', "create", &opts.new_branch, N_("branch"),
1768 			   N_("create and switch to a new branch")),
1769 		OPT_STRING('C', "force-create", &opts.new_branch_force, N_("branch"),
1770 			   N_("create/reset and switch to a branch")),
1771 		OPT_BOOL(0, "guess", &opts.dwim_new_local_branch,
1772 			 N_("second guess 'git switch <no-such-branch>'")),
1773 		OPT_BOOL(0, "discard-changes", &opts.discard_changes,
1774 			 N_("throw away local modifications")),
1775 		OPT_END()
1776 	};
1777 	int ret;
1778 
1779 	memset(&opts, 0, sizeof(opts));
1780 	opts.dwim_new_local_branch = 1;
1781 	opts.accept_ref = 1;
1782 	opts.accept_pathspec = 0;
1783 	opts.switch_branch_doing_nothing_is_ok = 0;
1784 	opts.only_merge_on_switching_branches = 1;
1785 	opts.implicit_detach = 0;
1786 	opts.can_switch_when_in_progress = 0;
1787 	opts.orphan_from_empty_tree = 1;
1788 	opts.overlay_mode = -1;
1789 
1790 	options = parse_options_dup(switch_options);
1791 	options = add_common_options(&opts, options);
1792 	options = add_common_switch_branch_options(&opts, options);
1793 
1794 	ret = checkout_main(argc, argv, prefix, &opts,
1795 			    options, switch_branch_usage);
1796 	FREE_AND_NULL(options);
1797 	return ret;
1798 }
1799 
cmd_restore(int argc,const char ** argv,const char * prefix)1800 int cmd_restore(int argc, const char **argv, const char *prefix)
1801 {
1802 	struct checkout_opts opts;
1803 	struct option *options;
1804 	struct option restore_options[] = {
1805 		OPT_STRING('s', "source", &opts.from_treeish, "<tree-ish>",
1806 			   N_("which tree-ish to checkout from")),
1807 		OPT_BOOL('S', "staged", &opts.checkout_index,
1808 			   N_("restore the index")),
1809 		OPT_BOOL('W', "worktree", &opts.checkout_worktree,
1810 			   N_("restore the working tree (default)")),
1811 		OPT_BOOL(0, "ignore-unmerged", &opts.ignore_unmerged,
1812 			 N_("ignore unmerged entries")),
1813 		OPT_BOOL(0, "overlay", &opts.overlay_mode, N_("use overlay mode")),
1814 		OPT_END()
1815 	};
1816 	int ret;
1817 
1818 	memset(&opts, 0, sizeof(opts));
1819 	opts.accept_ref = 0;
1820 	opts.accept_pathspec = 1;
1821 	opts.empty_pathspec_ok = 0;
1822 	opts.overlay_mode = 0;
1823 	opts.checkout_index = -1;    /* default off */
1824 	opts.checkout_worktree = -2; /* default on */
1825 	opts.ignore_unmerged_opt = "--ignore-unmerged";
1826 
1827 	options = parse_options_dup(restore_options);
1828 	options = add_common_options(&opts, options);
1829 	options = add_checkout_path_options(&opts, options);
1830 
1831 	ret = checkout_main(argc, argv, prefix, &opts,
1832 			    options, restore_usage);
1833 	FREE_AND_NULL(options);
1834 	return ret;
1835 }
1836