git: restore - support --source

carapace-sh · Oct 23, 2024 · 84d3896 · 84d3896
1 parent 50e3839
commit 84d3896
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/completers/git_completer/cmd/restore.go b/completers/git_completer/cmd/restore.go
@@ -46,6 +46,10 @@ func init() {
 			if restoreCmd.Flag("staged").Changed {
 				return git.ActionChanges(git.ChangeOpts{Staged: true}).FilterArgs()
 			}
+
+			if f := restoreCmd.Flag("source"); f.Changed {
+				return git.ActionRefDiffs(f.Value.String())
+			}
 			return git.ActionChanges(git.ChangeOpts{Unstaged: true}).FilterArgs()
 		}),
 	)