Skip to content

Commit 5e4e84c

Browse files
HaraldNordgrenHaraldNordgren
committed
checkout: add --autostash option for branch switching
When switching branches, local modifications in the working tree can prevent the checkout from succeeding. While "git rebase" and "git merge" already support --autostash to handle this case automatically, "git checkout" and "git switch" require users to manually stash and unstash their changes. Teach "git checkout" and "git switch" to accept --autostash and --no-autostash options that automatically create a temporary stash entry before the branch switch begins and apply it after the switch completes. If the stash application results in conflicts, the stash entry is saved to the stash list so the user can resolve them later. Also add a checkout.autoStash configuration option that enables this behavior by default, which can be overridden with --no-autostash on the command line. Signed-off-by: Harald Nordgren <haraldnordgren@gmail.com> Made-with: Cursor
1 parent d181b93 commit 5e4e84c

File tree

7 files changed

+203
-0
lines changed

7 files changed

+203
-0
lines changed

‎Documentation/config/checkout.adoc‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,18 @@ with a small number of cores, the default sequential checkout often performs
3636
better. The size and compression level of a repository might also influence how
3737
well the parallel version performs.
3838

39+
`checkout.autoStash`::
40+
When set to true, automatically create a temporary stash entry
41+
before the operation begins, and apply it after the operation
42+
ends. This means that you can run `git checkout` or `git switch`
43+
on a dirty worktree. However, use with care: the final stash
44+
application after a successful branch switch might result in
45+
non-trivial conflicts.
46+
This option can be overridden by the `--no-autostash` and
47+
`--autostash` options of linkgit:git-checkout[1] and
48+
linkgit:git-switch[1].
49+
Defaults to false.
50+
3951
`checkout.thresholdForParallelism`::
4052
When running parallel checkout with a small number of files, the cost
4153
of subprocess spawning and inter-process communication might outweigh

‎Documentation/git-checkout.adoc‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -272,6 +272,15 @@ When switching branches with `--merge`, staged changes may be lost.
272272
`merge.conflictStyle` configuration variable. Possible values are
273273
`merge` (default), `diff3`, and `zdiff3`.
274274
275+
`--autostash`::
276+
`--no-autostash`::
277+
When switching branches, automatically create a temporary stash
278+
entry before the operation begins, and apply it after the
279+
operation ends. This means that you can switch branches on a
280+
dirty worktree. However, use with care: the final stash
281+
application after a successful branch switch might result in
282+
non-trivial conflicts.
283+
275284
`-p`::
276285
`--patch`::
277286
Interactively select hunks in the difference between the

‎Documentation/git-switch.adoc‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,15 @@ should result in deletion of the path).
142142
`merge.conflictStyle` configuration variable. Possible values are
143143
`merge` (default), `diff3`, and `zdiff3`.
144144

145+
`--autostash`::
146+
`--no-autostash`::
147+
Automatically create a temporary stash entry before the
148+
operation begins, and apply it after the operation ends.
149+
This means that you can switch branches on a dirty worktree.
150+
However, use with care: the final stash application after a
151+
successful branch switch might result in non-trivial
152+
conflicts.
153+
145154
`-q`::
146155
`--quiet`::
147156
Quiet, suppress feedback messages.

‎builtin/checkout.c‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
#include "repo-settings.h"
3131
#include "resolve-undo.h"
3232
#include "revision.h"
33+
#include "sequencer.h"
3334
#include "setup.h"
3435
#include "submodule.h"
3536
#include "symlinks.h"
@@ -68,6 +69,7 @@ struct checkout_opts {
6869
int only_merge_on_switching_branches;
6970
int can_switch_when_in_progress;
7071
int orphan_from_empty_tree;
72+
int autostash;
7173
int empty_pathspec_ok;
7274
int checkout_index;
7375
int checkout_worktree;
@@ -1202,9 +1204,16 @@ static int switch_branches(const struct checkout_opts *opts,
12021204
do_merge = 0;
12031205
}
12041206

1207+
if (opts->autostash) {
1208+
if (repo_read_index(the_repository) < 0)
1209+
die(_("index file corrupt"));
1210+
create_autostash_ref(the_repository, "CHECKOUT_AUTOSTASH");
1211+
}
1212+
12051213
if (do_merge) {
12061214
ret = merge_working_tree(opts, &old_branch_info, new_branch_info, &writeout_error);
12071215
if (ret) {
1216+
apply_autostash_ref(the_repository, "CHECKOUT_AUTOSTASH");
12081217
branch_info_release(&old_branch_info);
12091218
return ret;
12101219
}
@@ -1215,6 +1224,8 @@ static int switch_branches(const struct checkout_opts *opts,
12151224

12161225
update_refs_for_switch(opts, &old_branch_info, new_branch_info);
12171226

1227+
apply_autostash_ref(the_repository, "CHECKOUT_AUTOSTASH");
1228+
12181229
ret = post_checkout_hook(old_branch_info.commit, new_branch_info->commit, 1);
12191230
branch_info_release(&old_branch_info);
12201231

@@ -1236,6 +1247,10 @@ static int git_checkout_config(const char *var, const char *value,
12361247
opts->dwim_new_local_branch = git_config_bool(var, value);
12371248
return 0;
12381249
}
1250+
if (!strcmp(var, "checkout.autostash")) {
1251+
opts->autostash = git_config_bool(var, value);
1252+
return 0;
1253+
}
12391254

12401255
if (starts_with(var, "submodule."))
12411256
return git_default_submodule_config(var, value, NULL);
@@ -1745,6 +1760,7 @@ static struct option *add_common_switch_branch_options(
17451760
PARSE_OPT_NOCOMPLETE),
17461761
OPT_BOOL(0, "ignore-other-worktrees", &opts->ignore_other_worktrees,
17471762
N_("do not check if another worktree is using this branch")),
1763+
OPT_AUTOSTASH(&opts->autostash),
17481764
OPT_END()
17491765
};
17501766
struct option *newopts = parse_options_concat(prevopts, options);

‎t/meson.build‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -275,6 +275,7 @@ integration_tests = [
275275
't2030-unresolve-info.sh',
276276
't2050-git-dir-relative.sh',
277277
't2060-switch.sh',
278+
't2061-switch-autostash.sh',
278279
't2070-restore.sh',
279280
't2071-restore-patch.sh',
280281
't2072-restore-pathspec-file.sh',

‎t/t2061-switch-autostash.sh‎

Lines changed: 155 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,155 @@
1+
#!/bin/sh
2+
3+
test_description='checkout/switch --autostash tests'
4+
5+
GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME=main
6+
export GIT_TEST_DEFAULT_INITIAL_BRANCH_NAME
7+
8+
. ./test-lib.sh
9+
10+
test_expect_success 'setup' '
11+
echo file0content >file0 &&
12+
echo file1content >file1 &&
13+
git add . &&
14+
test_tick &&
15+
git commit -m "initial commit" &&
16+
git branch other-branch &&
17+
echo file1main >file1 &&
18+
git add . &&
19+
test_tick &&
20+
git commit -m "modify file1 on main" &&
21+
git checkout other-branch &&
22+
echo file1other >file1 &&
23+
git add . &&
24+
test_tick &&
25+
git commit -m "modify file1 on other-branch" &&
26+
echo file2content >file2 &&
27+
git add . &&
28+
test_tick &&
29+
git commit -m "add file2 on other-branch" &&
30+
git checkout main
31+
'
32+
33+
test_expect_success 'switch --autostash on dirty worktree' '
34+
echo dirty >file0 &&
35+
git switch --autostash other-branch >actual 2>&1 &&
36+
test_grep "Created autostash" actual &&
37+
test_grep "Applied autostash" actual &&
38+
echo dirty >expected &&
39+
test_cmp expected file0 &&
40+
git switch main
41+
'
42+
43+
test_expect_success 'checkout --autostash on dirty worktree' '
44+
echo dirty >file0 &&
45+
git checkout --autostash other-branch >actual 2>&1 &&
46+
test_grep "Created autostash" actual &&
47+
test_grep "Applied autostash" actual &&
48+
echo dirty >expected &&
49+
test_cmp expected file0 &&
50+
git checkout main
51+
'
52+
53+
test_expect_success 'switch: checkout.autostash config' '
54+
echo dirty >file0 &&
55+
test_config checkout.autostash true &&
56+
git switch other-branch >actual 2>&1 &&
57+
test_grep "Created autostash" actual &&
58+
test_grep "Applied autostash" actual &&
59+
echo dirty >expected &&
60+
test_cmp expected file0 &&
61+
git switch main
62+
'
63+
64+
test_expect_success 'checkout: checkout.autostash config' '
65+
echo dirty >file0 &&
66+
test_config checkout.autostash true &&
67+
git checkout other-branch >actual 2>&1 &&
68+
test_grep "Created autostash" actual &&
69+
test_grep "Applied autostash" actual &&
70+
echo dirty >expected &&
71+
test_cmp expected file0 &&
72+
git checkout main
73+
'
74+
75+
test_expect_success '--no-autostash overrides checkout.autostash' '
76+
echo dirty >file1 &&
77+
test_config checkout.autostash true &&
78+
test_must_fail git switch --no-autostash other-branch 2>stderr &&
79+
test_grep ! "Created autostash" stderr &&
80+
git checkout -- file1
81+
'
82+
83+
test_expect_success '--autostash overrides checkout.autostash=false' '
84+
echo dirty >file0 &&
85+
test_config checkout.autostash false &&
86+
git switch --autostash other-branch >actual 2>&1 &&
87+
test_grep "Created autostash" actual &&
88+
test_grep "Applied autostash" actual &&
89+
echo dirty >expected &&
90+
test_cmp expected file0 &&
91+
git switch main
92+
'
93+
94+
test_expect_success 'autostash with dirty index' '
95+
echo dirty-index >file0 &&
96+
git add file0 &&
97+
git switch --autostash other-branch >actual 2>&1 &&
98+
test_grep "Created autostash" actual &&
99+
test_grep "Applied autostash" actual &&
100+
echo dirty-index >expected &&
101+
test_cmp expected file0 &&
102+
git checkout -- file0 &&
103+
git switch main
104+
'
105+
106+
test_expect_success 'autostash bypasses local changes would be overwritten error' '
107+
echo dirty >file1 &&
108+
test_must_fail git switch other-branch 2>stderr &&
109+
test_grep "Your local changes" stderr &&
110+
git switch --autostash other-branch >actual 2>&1 &&
111+
test_grep "Created autostash" actual &&
112+
test_grep "Applying autostash resulted in conflicts" actual &&
113+
git stash drop &&
114+
git reset --hard &&
115+
git switch main
116+
'
117+
118+
test_expect_success 'autostash with conflicting stash apply saves to stash list' '
119+
echo dirty-conflicting >file1 &&
120+
git switch --autostash other-branch >actual 2>&1 &&
121+
test_grep "Created autostash" actual &&
122+
test_grep "Applying autostash resulted in conflicts" actual &&
123+
test_grep "Your changes are safe in the stash" actual &&
124+
git stash drop &&
125+
git reset --hard &&
126+
git switch main
127+
'
128+
129+
test_expect_success 'autostash is a no-op with clean worktree' '
130+
git switch --autostash other-branch >actual 2>&1 &&
131+
test_grep ! "Created autostash" actual &&
132+
git switch main
133+
'
134+
135+
test_expect_success '--autostash with --merge stashes and switches' '
136+
echo dirty >file0 &&
137+
git switch --autostash --merge other-branch >actual 2>&1 &&
138+
test_grep "Created autostash" actual &&
139+
test_grep "Applied autostash" actual &&
140+
echo dirty >expected &&
141+
test_cmp expected file0 &&
142+
git switch main
143+
'
144+
145+
test_expect_success 'autostash with staged changes switches successfully' '
146+
echo staged-change >file1 &&
147+
git add file1 &&
148+
git switch --autostash other-branch >actual 2>&1 &&
149+
test_grep "Created autostash" actual &&
150+
git stash drop 2>/dev/null &&
151+
git reset --hard &&
152+
git switch main
153+
'
154+
155+
test_done

‎t/t9902-completion.sh‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2602,6 +2602,7 @@ test_expect_success 'double dash "git checkout"' '
26022602
--ignore-other-worktrees Z
26032603
--recurse-submodules Z
26042604
--auto-advance Z
2605+
--autostash Z
26052606
--progress Z
26062607
--guess Z
26072608
--no-guess Z

0 commit comments

Comments
 (0)