Add stash push to the helper.

Signed-off-by: Paul-Sebastian Ungureanu <ungureanupaulsebast...@gmail.com>
---
 builtin/stash--helper.c | 229 ++++++++++++++++++++++++++++++++++++++++
 git-stash.sh            |   6 +-
 2 files changed, 233 insertions(+), 2 deletions(-)

diff --git a/builtin/stash--helper.c b/builtin/stash--helper.c
index ce360a569d..23670321d8 100644
--- a/builtin/stash--helper.c
+++ b/builtin/stash--helper.c
@@ -21,6 +21,9 @@ static const char * const git_stash_helper_usage[] = {
        N_("git stash--helper ( pop | apply ) [--index] [-q|--quiet] 
[<stash>]"),
        N_("git stash--helper branch <branchname> [<stash>]"),
        N_("git stash--helper clear"),
+       N_("git stash--helper [push [-p|--patch] [-k|--[no-]keep-index] 
[-q|--quiet]\n"
+          "          [-u|--include-untracked] [-a|--all] [-m|--message 
<message>]\n"
+          "          [--] [<pathspec>...]]"),
        NULL
 };
 
@@ -69,6 +72,13 @@ static const char * const git_stash_helper_create_usage[] = {
        NULL
 };
 
+static const char * const git_stash_helper_push_usage[] = {
+       N_("git stash--helper [push [-p|--patch] [-k|--[no-]keep-index] 
[-q|--quiet]\n"
+          "          [-u|--include-untracked] [-a|--all] [-m|--message 
<message>]\n"
+          "          [--] [<pathspec>...]]"),
+       NULL
+};
+
 static const char *ref_stash = "refs/stash";
 static int quiet;
 static struct strbuf stash_index_path = STRBUF_INIT;
@@ -1204,6 +1214,223 @@ static int create_stash(int argc, const char **argv, 
const char *prefix)
        return ret < 0;
 }
 
+static void add_ps_items_to_argv_array(struct argv_array *args,
+                                      struct pathspec ps) {
+       int i;
+       for (i = 0; i < ps.nr; ++i)
+               argv_array_push(args, ps.items[i].match);
+}
+
+static int do_push_stash(struct pathspec ps, const char *stash_msg, int quiet,
+                        int keep_index, int patch_mode, int include_untracked)
+{
+       int ret = 0;
+       struct stash_info info;
+       if (patch_mode && keep_index == -1)
+               keep_index = 1;
+
+       if (patch_mode && include_untracked) {
+               fprintf_ln(stderr, _("Can't use --patch and --include-untracked 
or --all at the same time"));
+               return -1;
+       }
+
+       read_cache_preload(NULL);
+       if (!include_untracked && ps.nr) {
+               int i;
+               char *ps_matched = xcalloc(ps.nr, 1);
+
+               for (i = 0; i < active_nr; ++i) {
+                       const struct cache_entry *ce = active_cache[i];
+                       ce_path_match(&the_index, ce, &ps, ps_matched);
+               }
+
+               if (report_path_error(ps_matched, &ps, NULL)) {
+                       fprintf_ln(stderr, _("Did you forget to 'git add'?"));
+                       return -1;
+               }
+               free(ps_matched);
+       }
+
+       if (refresh_cache(REFRESH_QUIET))
+               return -1;
+
+       if (!check_changes(ps, include_untracked)) {
+               printf_ln(_("No local changes to save"));
+               return 0;
+       }
+
+       if (!reflog_exists(ref_stash) && do_clear_stash()) {
+               fprintf_ln(stderr, _("Cannot initialize stash"));
+               return -1;
+       }
+
+       if (do_create_stash(ps, &stash_msg, include_untracked, patch_mode,
+                           &info)) {
+               ret = -1;
+               goto done;
+       }
+
+       if (do_store_stash(oid_to_hex(&info.w_commit), stash_msg, 1)) {
+               fprintf(stderr, _("Cannot save the current status"));
+               ret = -1;
+               goto done;
+       }
+
+       printf_ln(_("Saved working directory and index state %s"), stash_msg);
+
+       if (!patch_mode) {
+               if (include_untracked && !ps.nr) {
+                       struct child_process cp = CHILD_PROCESS_INIT;
+
+                       cp.git_cmd = 1;
+                       argv_array_pushl(&cp.args, "clean", "--force",
+                                        "--quiet", "-d", NULL);
+                       if (include_untracked == 2)
+                               argv_array_push(&cp.args, "-x");
+                       if (run_command(&cp)) {
+                               ret = -1;
+                               goto done;
+                       }
+               }
+               if (ps.nr) {
+                       struct child_process cp1 = CHILD_PROCESS_INIT;
+                       struct child_process cp2 = CHILD_PROCESS_INIT;
+                       struct child_process cp3 = CHILD_PROCESS_INIT;
+                       struct strbuf out = STRBUF_INIT;
+
+                       cp1.git_cmd = 1;
+                       argv_array_push(&cp1.args, "add");
+                       if (!include_untracked)
+                               argv_array_push(&cp1.args, "-u");
+                       if (include_untracked == 2)
+                               argv_array_push(&cp1.args, "--force");
+                       argv_array_push(&cp1.args, "--");
+                       add_ps_items_to_argv_array(&cp1.args, ps);
+                       if (run_command(&cp1)) {
+                               ret = -1;
+                               goto done;
+                       }
+
+                       cp2.git_cmd = 1;
+                       argv_array_pushl(&cp2.args, "diff-index", "-p",
+                                        "--cached", "--binary", "HEAD", "--",
+                                        NULL);
+                       add_ps_items_to_argv_array(&cp2.args, ps);
+                       if (pipe_command(&cp2, NULL, 0, &out, 0, NULL, 0)) {
+                               ret = -1;
+                               goto done;
+                       }
+
+                       cp3.git_cmd = 1;
+                       argv_array_pushl(&cp3.args, "apply", "--index", "-R",
+                                        NULL);
+                       if (pipe_command(&cp3, out.buf, out.len, NULL, 0, NULL,
+                                        0)) {
+                               ret = -1;
+                               goto done;
+                       }
+               } else {
+                       struct child_process cp = CHILD_PROCESS_INIT;
+                       cp.git_cmd = 1;
+                       argv_array_pushl(&cp.args, "reset", "--hard", "-q",
+                                        NULL);
+                       if (run_command(&cp)) {
+                               ret = -1;
+                               goto done;
+                       }
+               }
+
+               if (keep_index == 1 && !is_null_oid(&info.i_tree)) {
+                       struct child_process cp1 = CHILD_PROCESS_INIT;
+                       struct child_process cp2 = CHILD_PROCESS_INIT;
+                       struct strbuf out = STRBUF_INIT;
+
+                       if (reset_tree(&info.i_tree, 0, 1)) {
+                               ret = -1;
+                               goto done;
+                       }
+
+                       cp1.git_cmd = 1;
+                       argv_array_pushl(&cp1.args, "ls-files", "-z",
+                                        "--modified", "--", NULL);
+                       add_ps_items_to_argv_array(&cp1.args, ps);
+                       if (pipe_command(&cp1, NULL, 0, &out, 0, NULL, 0)) {
+                               ret = -1;
+                               goto done;
+                       }
+
+                       cp2.git_cmd = 1;
+                       argv_array_pushl(&cp2.args, "checkout-index", "-z",
+                                        "--force", "--stdin", NULL);
+                       if (pipe_command(&cp2, out.buf, out.len, NULL, 0, NULL,
+                                        0)) {
+                               ret = -1;
+                               goto done;
+                       }
+               }
+       } else {
+               struct child_process cp = CHILD_PROCESS_INIT;
+
+               cp.git_cmd = 1;
+               argv_array_pushl(&cp.args, "apply", "-R", NULL);
+
+               if (pipe_command(&cp, patch.buf, patch.len, NULL, 0, NULL, 0)) {
+                       fprintf_ln(stderr, _("Cannot remove worktree changes"));
+                       ret = -1;
+                       goto done;
+               }
+
+               if (keep_index < 1) {
+                       int i;
+                       struct child_process cp = CHILD_PROCESS_INIT;
+
+                       cp.git_cmd = 1;
+                       argv_array_pushl(&cp.args, "reset", "-q", "--", NULL);
+                       for (i = 0; i < ps.nr; ++i)
+                               argv_array_push(&cp.args, ps.items[i].match);
+                       if (run_command(&cp)) {
+                               ret = -1;
+                               goto done;
+                       }
+               }
+       }
+done:
+       free((char *) stash_msg);
+       return ret;
+}
+
+static int push_stash(int argc, const char **argv, const char *prefix)
+{
+       int keep_index = -1;
+       int patch_mode = 0;
+       int include_untracked = 0;
+       int quiet = 0;
+       const char *stash_msg = NULL;
+       struct pathspec ps;
+       struct option options[] = {
+               OPT_SET_INT('k', "keep-index", &keep_index,
+                       N_("keep index"), 1),
+               OPT_BOOL('p', "patch", &patch_mode,
+                       N_("stash in patch mode")),
+               OPT__QUIET(&quiet, N_("quiet mode")),
+               OPT_BOOL('u', "include-untracked", &include_untracked,
+                        N_("include untracked files in stash")),
+               OPT_SET_INT('a', "all", &include_untracked,
+                           N_("include ignore files"), 2),
+               OPT_STRING('m', "message", &stash_msg, N_("message"),
+                        N_("stash message")),
+               OPT_END()
+       };
+
+       argc = parse_options(argc, argv, prefix, options,
+                            git_stash_helper_push_usage,
+                            0);
+
+       parse_pathspec(&ps, 0, PATHSPEC_PREFER_FULL, prefix, argv);
+       return do_push_stash(ps, stash_msg, quiet, keep_index, patch_mode,
+                            include_untracked);
+}
+
 int cmd_stash__helper(int argc, const char **argv, const char *prefix)
 {
        pid_t pid = getpid();
@@ -1242,6 +1469,8 @@ int cmd_stash__helper(int argc, const char **argv, const 
char *prefix)
                return !!store_stash(argc, argv, prefix);
        else if (!strcmp(argv[0], "create"))
                return !!create_stash(argc, argv, prefix);
+       else if (!strcmp(argv[0], "push"))
+               return !!push_stash(argc, argv, prefix);
 
        usage_msg_opt(xstrfmt(_("unknown subcommand: %s"), argv[0]),
                      git_stash_helper_usage, options);
diff --git a/git-stash.sh b/git-stash.sh
index ab06e4ffb8..c3146f62ab 100755
--- a/git-stash.sh
+++ b/git-stash.sh
@@ -412,7 +412,8 @@ save)
        ;;
 push)
        shift
-       push_stash "$@"
+       cd "$START_DIR"
+       git stash--helper push "$@"
        ;;
 apply)
        shift
@@ -448,7 +449,8 @@ branch)
 *)
        case $# in
        0)
-               push_stash &&
+               cd "$START_DIR"
+               git stash--helper push &&
                say "$(gettext "(To restore them type \"git stash apply\")")"
                ;;
        *)
-- 
2.19.0.rc0.22.gc26283d74e

Reply via email to