builtin / rebase--interactive.con commit worktree: fix worktree add race (7af01f2)
   1#define USE_THE_INDEX_COMPATIBILITY_MACROS
   2#include "builtin.h"
   3#include "cache.h"
   4#include "config.h"
   5#include "parse-options.h"
   6#include "sequencer.h"
   7#include "rebase-interactive.h"
   8#include "argv-array.h"
   9#include "refs.h"
  10#include "rerere.h"
  11#include "run-command.h"
  12
  13static GIT_PATH_FUNC(path_state_dir, "rebase-merge/")
  14static GIT_PATH_FUNC(path_squash_onto, "rebase-merge/squash-onto")
  15static GIT_PATH_FUNC(path_interactive, "rebase-merge/interactive")
  16
  17static int get_revision_ranges(const char *upstream, const char *onto,
  18                               const char **head_hash,
  19                               char **revisions, char **shortrevisions)
  20{
  21        const char *base_rev = upstream ? upstream : onto, *shorthead;
  22        struct object_id orig_head;
  23
  24        if (get_oid("HEAD", &orig_head))
  25                return error(_("no HEAD?"));
  26
  27        *head_hash = find_unique_abbrev(&orig_head, GIT_MAX_HEXSZ);
  28        *revisions = xstrfmt("%s...%s", base_rev, *head_hash);
  29
  30        shorthead = find_unique_abbrev(&orig_head, DEFAULT_ABBREV);
  31
  32        if (upstream) {
  33                const char *shortrev;
  34                struct object_id rev_oid;
  35
  36                get_oid(base_rev, &rev_oid);
  37                shortrev = find_unique_abbrev(&rev_oid, DEFAULT_ABBREV);
  38
  39                *shortrevisions = xstrfmt("%s..%s", shortrev, shorthead);
  40        } else
  41                *shortrevisions = xstrdup(shorthead);
  42
  43        return 0;
  44}
  45
  46static int init_basic_state(struct replay_opts *opts, const char *head_name,
  47                            const char *onto, const char *orig_head)
  48{
  49        FILE *interactive;
  50
  51        if (!is_directory(path_state_dir()) && mkdir_in_gitdir(path_state_dir()))
  52                return error_errno(_("could not create temporary %s"), path_state_dir());
  53
  54        delete_reflog("REBASE_HEAD");
  55
  56        interactive = fopen(path_interactive(), "w");
  57        if (!interactive)
  58                return error_errno(_("could not mark as interactive"));
  59        fclose(interactive);
  60
  61        return write_basic_state(opts, head_name, onto, orig_head);
  62}
  63
  64static int do_interactive_rebase(struct replay_opts *opts, unsigned flags,
  65                                 const char *switch_to, const char *upstream,
  66                                 const char *onto, const char *onto_name,
  67                                 const char *squash_onto, const char *head_name,
  68                                 const char *restrict_revision, char *raw_strategies,
  69                                 const char *cmd, unsigned autosquash)
  70{
  71        int ret;
  72        const char *head_hash = NULL;
  73        char *revisions = NULL, *shortrevisions = NULL;
  74        struct argv_array make_script_args = ARGV_ARRAY_INIT;
  75        FILE *todo_list;
  76
  77        if (prepare_branch_to_be_rebased(opts, switch_to))
  78                return -1;
  79
  80        if (get_revision_ranges(upstream, onto, &head_hash,
  81                                &revisions, &shortrevisions))
  82                return -1;
  83
  84        if (raw_strategies)
  85                parse_strategy_opts(opts, raw_strategies);
  86
  87        if (init_basic_state(opts, head_name, onto, head_hash)) {
  88                free(revisions);
  89                free(shortrevisions);
  90
  91                return -1;
  92        }
  93
  94        if (!upstream && squash_onto)
  95                write_file(path_squash_onto(), "%s\n", squash_onto);
  96
  97        todo_list = fopen(rebase_path_todo(), "w");
  98        if (!todo_list) {
  99                free(revisions);
 100                free(shortrevisions);
 101
 102                return error_errno(_("could not open %s"), rebase_path_todo());
 103        }
 104
 105        argv_array_pushl(&make_script_args, "", revisions, NULL);
 106        if (restrict_revision)
 107                argv_array_push(&make_script_args, restrict_revision);
 108
 109        ret = sequencer_make_script(the_repository, todo_list,
 110                                    make_script_args.argc, make_script_args.argv,
 111                                    flags);
 112        fclose(todo_list);
 113
 114        if (ret)
 115                error(_("could not generate todo list"));
 116        else {
 117                discard_cache();
 118                ret = complete_action(the_repository, opts, flags,
 119                                      shortrevisions, onto_name, onto,
 120                                      head_hash, cmd, autosquash);
 121        }
 122
 123        free(revisions);
 124        free(shortrevisions);
 125        argv_array_clear(&make_script_args);
 126
 127        return ret;
 128}
 129
 130static const char * const builtin_rebase_interactive_usage[] = {
 131        N_("git rebase--interactive [<options>]"),
 132        NULL
 133};
 134
 135int cmd_rebase__interactive(int argc, const char **argv, const char *prefix)
 136{
 137        struct replay_opts opts = REPLAY_OPTS_INIT;
 138        unsigned flags = 0, keep_empty = 0, rebase_merges = 0, autosquash = 0;
 139        int abbreviate_commands = 0, rebase_cousins = -1, ret = 0;
 140        const char *onto = NULL, *onto_name = NULL, *restrict_revision = NULL,
 141                *squash_onto = NULL, *upstream = NULL, *head_name = NULL,
 142                *switch_to = NULL, *cmd = NULL;
 143        char *raw_strategies = NULL;
 144        enum {
 145                NONE = 0, CONTINUE, SKIP, EDIT_TODO, SHOW_CURRENT_PATCH,
 146                SHORTEN_OIDS, EXPAND_OIDS, CHECK_TODO_LIST, REARRANGE_SQUASH, ADD_EXEC
 147        } command = 0;
 148        struct option options[] = {
 149                OPT_BOOL(0, "ff", &opts.allow_ff, N_("allow fast-forward")),
 150                OPT_BOOL(0, "keep-empty", &keep_empty, N_("keep empty commits")),
 151                OPT_BOOL(0, "allow-empty-message", &opts.allow_empty_message,
 152                         N_("allow commits with empty messages")),
 153                OPT_BOOL(0, "rebase-merges", &rebase_merges, N_("rebase merge commits")),
 154                OPT_BOOL(0, "rebase-cousins", &rebase_cousins,
 155                         N_("keep original branch points of cousins")),
 156                OPT_BOOL(0, "autosquash", &autosquash,
 157                         N_("move commits that begin with squash!/fixup!")),
 158                OPT_BOOL(0, "signoff", &opts.signoff, N_("sign commits")),
 159                OPT__VERBOSE(&opts.verbose, N_("be verbose")),
 160                OPT_CMDMODE(0, "continue", &command, N_("continue rebase"),
 161                            CONTINUE),
 162                OPT_CMDMODE(0, "skip", &command, N_("skip commit"), SKIP),
 163                OPT_CMDMODE(0, "edit-todo", &command, N_("edit the todo list"),
 164                            EDIT_TODO),
 165                OPT_CMDMODE(0, "show-current-patch", &command, N_("show the current patch"),
 166                            SHOW_CURRENT_PATCH),
 167                OPT_CMDMODE(0, "shorten-ids", &command,
 168                        N_("shorten commit ids in the todo list"), SHORTEN_OIDS),
 169                OPT_CMDMODE(0, "expand-ids", &command,
 170                        N_("expand commit ids in the todo list"), EXPAND_OIDS),
 171                OPT_CMDMODE(0, "check-todo-list", &command,
 172                        N_("check the todo list"), CHECK_TODO_LIST),
 173                OPT_CMDMODE(0, "rearrange-squash", &command,
 174                        N_("rearrange fixup/squash lines"), REARRANGE_SQUASH),
 175                OPT_CMDMODE(0, "add-exec-commands", &command,
 176                        N_("insert exec commands in todo list"), ADD_EXEC),
 177                OPT_STRING(0, "onto", &onto, N_("onto"), N_("onto")),
 178                OPT_STRING(0, "restrict-revision", &restrict_revision,
 179                           N_("restrict-revision"), N_("restrict revision")),
 180                OPT_STRING(0, "squash-onto", &squash_onto, N_("squash-onto"),
 181                           N_("squash onto")),
 182                OPT_STRING(0, "upstream", &upstream, N_("upstream"),
 183                           N_("the upstream commit")),
 184                OPT_STRING(0, "head-name", &head_name, N_("head-name"), N_("head name")),
 185                { OPTION_STRING, 'S', "gpg-sign", &opts.gpg_sign, N_("key-id"),
 186                        N_("GPG-sign commits"),
 187                        PARSE_OPT_OPTARG, NULL, (intptr_t) "" },
 188                OPT_STRING(0, "strategy", &opts.strategy, N_("strategy"),
 189                           N_("rebase strategy")),
 190                OPT_STRING(0, "strategy-opts", &raw_strategies, N_("strategy-opts"),
 191                           N_("strategy options")),
 192                OPT_STRING(0, "switch-to", &switch_to, N_("switch-to"),
 193                           N_("the branch or commit to checkout")),
 194                OPT_STRING(0, "onto-name", &onto_name, N_("onto-name"), N_("onto name")),
 195                OPT_STRING(0, "cmd", &cmd, N_("cmd"), N_("the command to run")),
 196                OPT_RERERE_AUTOUPDATE(&opts.allow_rerere_auto),
 197                OPT_BOOL(0, "reschedule-failed-exec", &opts.reschedule_failed_exec,
 198                         N_("automatically re-schedule any `exec` that fails")),
 199                OPT_END()
 200        };
 201
 202        sequencer_init_config(&opts);
 203        git_config_get_bool("rebase.abbreviatecommands", &abbreviate_commands);
 204
 205        opts.action = REPLAY_INTERACTIVE_REBASE;
 206        opts.allow_ff = 1;
 207        opts.allow_empty = 1;
 208
 209        if (argc == 1)
 210                usage_with_options(builtin_rebase_interactive_usage, options);
 211
 212        argc = parse_options(argc, argv, NULL, options,
 213                        builtin_rebase_interactive_usage, PARSE_OPT_KEEP_ARGV0);
 214
 215        opts.gpg_sign = xstrdup_or_null(opts.gpg_sign);
 216
 217        flags |= keep_empty ? TODO_LIST_KEEP_EMPTY : 0;
 218        flags |= abbreviate_commands ? TODO_LIST_ABBREVIATE_CMDS : 0;
 219        flags |= rebase_merges ? TODO_LIST_REBASE_MERGES : 0;
 220        flags |= rebase_cousins > 0 ? TODO_LIST_REBASE_COUSINS : 0;
 221        flags |= command == SHORTEN_OIDS ? TODO_LIST_SHORTEN_IDS : 0;
 222
 223        if (rebase_cousins >= 0 && !rebase_merges)
 224                warning(_("--[no-]rebase-cousins has no effect without "
 225                          "--rebase-merges"));
 226
 227        switch (command) {
 228        case NONE:
 229                if (!onto && !upstream)
 230                        die(_("a base commit must be provided with --upstream or --onto"));
 231
 232                ret = do_interactive_rebase(&opts, flags, switch_to, upstream, onto,
 233                                            onto_name, squash_onto, head_name, restrict_revision,
 234                                            raw_strategies, cmd, autosquash);
 235                break;
 236        case SKIP: {
 237                struct string_list merge_rr = STRING_LIST_INIT_DUP;
 238
 239                rerere_clear(the_repository, &merge_rr);
 240                /* fallthrough */
 241        case CONTINUE:
 242                ret = sequencer_continue(the_repository, &opts);
 243                break;
 244        }
 245        case EDIT_TODO:
 246                ret = edit_todo_list(the_repository, flags);
 247                break;
 248        case SHOW_CURRENT_PATCH: {
 249                struct child_process cmd = CHILD_PROCESS_INIT;
 250
 251                cmd.git_cmd = 1;
 252                argv_array_pushl(&cmd.args, "show", "REBASE_HEAD", "--", NULL);
 253                ret = run_command(&cmd);
 254
 255                break;
 256        }
 257        case SHORTEN_OIDS:
 258        case EXPAND_OIDS:
 259                ret = transform_todos(the_repository, flags);
 260                break;
 261        case CHECK_TODO_LIST:
 262                ret = check_todo_list(the_repository);
 263                break;
 264        case REARRANGE_SQUASH:
 265                ret = rearrange_squash(the_repository);
 266                break;
 267        case ADD_EXEC:
 268                ret = sequencer_add_exec_commands(the_repository, cmd);
 269                break;
 270        default:
 271                BUG("invalid command '%d'", command);
 272        }
 273
 274        return !!ret;
 275}