The combination of "git clean" and fat fingers can some times cause data-loss, which can be frustrating. So let's add a flag that imports the files to be deleted into the object-database, in a way similar to what git-stash does. Maintain a reflog of the previously backed up clean-runs. Signed-off-by: Erik Faye-Lund <kusmabite@xxxxxxxxx> --- I've had a similar patch laying around for quite a while, but since f538a91 ("git-clean: Display more accurate delete messages"), this patch is a lot less nasty than before. So here you go, perhaps someone else has fat fingers and hate to lose work? Documentation/config.txt | 4 ++ Documentation/git-clean.txt | 4 ++ builtin/clean.c | 111 +++++++++++++++++++++++++++++++++++++++++++- t/t7300-clean.sh | 12 +++++ 4 files changed, 130 insertions(+), 1 deletion(-) diff --git a/Documentation/config.txt b/Documentation/config.txt index 1932e9b..d58fe31 100644 --- a/Documentation/config.txt +++ b/Documentation/config.txt @@ -797,6 +797,10 @@ clean.requireForce:: A boolean to make git-clean do nothing unless given -f, -i or -n. Defaults to true. +clean.backup:: + A boolean to make git-clean back up files before they are + deleted. Defaults to false. + color.branch:: A boolean to enable/disable color in the output of linkgit:git-branch[1]. May be set to `always`, diff --git a/Documentation/git-clean.txt b/Documentation/git-clean.txt index 8997922..bc9d703 100644 --- a/Documentation/git-clean.txt +++ b/Documentation/git-clean.txt @@ -41,6 +41,10 @@ OPTIONS --interactive:: Show what would be done and clean files interactively. See ``Interactive mode'' for details. +-b:: +--backup:: + Back up files to a reflog before deleting them. The tree of + backed up files are stored in the reflog for refs/clean-backup. -n:: --dry-run:: diff --git a/builtin/clean.c b/builtin/clean.c index 9a91515..96fb4b2 100644 --- a/builtin/clean.c +++ b/builtin/clean.c @@ -16,9 +16,12 @@ #include "column.h" #include "color.h" #include "pathspec.h" +#include "tree-walk.h" +#include "unpack-trees.h" +#include "cache-tree.h" static int force = -1; /* unset */ -static int interactive; +static int interactive, backup; static struct string_list del_list = STRING_LIST_INIT_DUP; static unsigned int colopts; @@ -120,6 +123,11 @@ static int git_clean_config(const char *var, const char *value, void *cb) return 0; } + if (!strcmp(var, "clean.backup")) { + backup = git_config_bool(var, value); + return 0; + } + if (!strcmp(var, "clean.requireforce")) { force = !git_config_bool(var, value); return 0; @@ -148,6 +156,93 @@ static int exclude_cb(const struct option *opt, const char *arg, int unset) return 0; } +static int backed_up_anything; + +static void backup_file(const char *path, struct stat *st) +{ + if (add_to_cache(path, st, 0)) + die(_("backing up '%s' failed"), path); + backed_up_anything = 1; +} + +static struct commit_list *parents; + +static void prepare_backup(void) +{ + struct unpack_trees_options opts; + unsigned char sha1[20]; + struct tree *tree; + struct commit *parent; + struct tree_desc t; + + if (get_sha1("HEAD", sha1)) + die(_("You do not have the initial commit yet")); + + /* prepare parent-list */ + parent = lookup_commit_or_die(sha1, "HEAD"); + commit_list_insert(parent, &parents); + + /* load HEAD into the index */ + + tree = parse_tree_indirect(sha1); + if (!tree) + die(_("Failed to unpack tree object %s"), sha1); + + parse_tree(tree); + init_tree_desc(&t, tree->buffer, tree->size); + + memset(&opts, 0, sizeof(opts)); + opts.head_idx = -1; + opts.src_index = &the_index; + opts.dst_index = &the_index; + opts.index_only = 1; + + if (unpack_trees(1, &t, &opts)) { + /* We've already reported the error, finish dying */ + exit(128); + } +} + +static void finish_backup(void) +{ + const char *ref = "refs/clean-backup"; + unsigned char commit_sha1[20]; + struct strbuf msg = STRBUF_INIT; + char logfile[PATH_MAX]; + struct stat st; + + if (!backed_up_anything) + return; + + if (!active_cache_tree) + active_cache_tree = cache_tree(); + + if (!cache_tree_fully_valid(active_cache_tree)) { + if (cache_tree_update(active_cache_tree, + (const struct cache_entry * const *)active_cache, + active_nr, 0) < 0) + die("failed to update cache"); + } + + strbuf_addstr(&msg, "Automatically committed by git-clean"); + + /* create a reflog, if there isn't one */ + git_snpath(logfile, sizeof(logfile), "logs/%s", ref); + if (stat(logfile, &st)) { + FILE *fp = fopen(logfile, "w"); + if (!fp) + warning(_("Can not do reflog for '%s'\n"), ref); + else + fclose(fp); + } + + if (commit_tree(&msg, active_cache_tree->sha1, parents, commit_sha1, + NULL, NULL)) + die("failed to commit :("); + + update_ref(msg.buf, ref, commit_sha1, NULL, 0, DIE_ON_ERR); +} + static int remove_dirs(struct strbuf *path, const char *prefix, int force_flag, int dry_run, int quiet, int *dir_gone) { @@ -207,6 +302,8 @@ static int remove_dirs(struct strbuf *path, const char *prefix, int force_flag, *dir_gone = 0; continue; } else { + if (backup && !dry_run) + backup_file(path->buf, &st); res = dry_run ? 0 : unlink(path->buf); if (!res) { quote_path_relative(path->buf, prefix, "ed); @@ -878,6 +975,8 @@ int cmd_clean(int argc, const char **argv, const char *prefix) OPT_BOOL('i', "interactive", &interactive, N_("interactive cleaning")), OPT_BOOL('d', NULL, &remove_directories, N_("remove whole directories")), + OPT_BOOL('b', "backup", &backup, + N_("back up files to a reflog before deleting them")), { OPTION_CALLBACK, 'e', "exclude", &exclude_list, N_("pattern"), N_("add <pattern> to ignore rules"), PARSE_OPT_NONEG, exclude_cb }, OPT_BOOL('x', NULL, &ignored, N_("remove ignored files, too")), @@ -922,6 +1021,9 @@ int cmd_clean(int argc, const char **argv, const char *prefix) if (!ignored) setup_standard_excludes(&dir); + if (backup && !dry_run) + prepare_backup(); + el = add_exclude_list(&dir, EXC_CMDL, "--exclude option"); for (i = 0; i < exclude_list.nr; i++) add_exclude(exclude_list.items[i].string, "", 0, el, -(i+1)); @@ -985,6 +1087,9 @@ int cmd_clean(int argc, const char **argv, const char *prefix) printf(dry_run ? _(msg_would_remove) : _(msg_remove), qname); } } else { + if (backup && !dry_run) + backup_file(abs_path.buf, &st); + res = dry_run ? 0 : unlink(abs_path.buf); if (res) { qname = quote_path_relative(item->string, NULL, &buf); @@ -1002,5 +1107,9 @@ int cmd_clean(int argc, const char **argv, const char *prefix) strbuf_release(&buf); string_list_clear(&del_list, 0); string_list_clear(&exclude_list, 0); + + if (backup && !dry_run) + finish_backup(); + return (errors != 0); } diff --git a/t/t7300-clean.sh b/t/t7300-clean.sh index 74de814..27d1d74 100755 --- a/t/t7300-clean.sh +++ b/t/t7300-clean.sh @@ -527,4 +527,16 @@ test_expect_success 'git clean -d respects pathspecs (pathspec is prefix of dir) test_path_is_dir foobar ' +test_expect_success 'git clean -b' ' + git reset --hard HEAD && + git clean -dfx && + mkdir -p foobar && + echo "bar" > bar && + echo "baz" > foobar/baz && + git clean -d -f -b && + git diff --name-only refs/clean-backup@{0}^ refs/clean-backup@{0} >actual && + grep bar actual && + grep foobar/baz actual +' + test_done -- 1.9.2.msysgit.0.161.g83227c1 -- To unsubscribe from this list: send the line "unsubscribe git" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html