Andrew's git - gitweb.git/blob - builtin/rebase--interactive.c

   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 add_exec_commands(struct string_list *commands)
  18{
  19        const char *todo_file = rebase_path_todo();
  20        struct todo_list todo_list = TODO_LIST_INIT;
  21        int res;
  22
  23        if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0)
  24                return error_errno(_("could not read '%s'."), todo_file);
  25
  26        if (todo_list_parse_insn_buffer(the_repository, todo_list.buf.buf,
  27                                        &todo_list)) {
  28                todo_list_release(&todo_list);
  29                return error(_("unusable todo list: '%s'"), todo_file);
  30        }
  31
  32        todo_list_add_exec_commands(&todo_list, commands);
  33        res = todo_list_write_to_file(the_repository, &todo_list,
  34                                      todo_file, NULL, NULL, -1, 0);
  35        todo_list_release(&todo_list);
  36
  37        if (res)
  38                return error_errno(_("could not write '%s'."), todo_file);
  39        return 0;
  40}
  41
  42static int rearrange_squash_in_todo_file(void)
  43{
  44        const char *todo_file = rebase_path_todo();
  45        struct todo_list todo_list = TODO_LIST_INIT;
  46        int res = 0;
  47
  48        if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0)
  49                return error_errno(_("could not read '%s'."), todo_file);
  50        if (todo_list_parse_insn_buffer(the_repository, todo_list.buf.buf,
  51                                        &todo_list)) {
  52                todo_list_release(&todo_list);
  53                return error(_("unusable todo list: '%s'"), todo_file);
  54        }
  55
  56        res = todo_list_rearrange_squash(&todo_list);
  57        if (!res)
  58                res = todo_list_write_to_file(the_repository, &todo_list,
  59                                              todo_file, NULL, NULL, -1, 0);
  60
  61        todo_list_release(&todo_list);
  62
  63        if (res)
  64                return error_errno(_("could not write '%s'."), todo_file);
  65        return 0;
  66}
  67
  68static int transform_todo_file(unsigned flags)
  69{
  70        const char *todo_file = rebase_path_todo();
  71        struct todo_list todo_list = TODO_LIST_INIT;
  72        int res;
  73
  74        if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0)
  75                return error_errno(_("could not read '%s'."), todo_file);
  76
  77        if (todo_list_parse_insn_buffer(the_repository, todo_list.buf.buf,
  78                                        &todo_list)) {
  79                todo_list_release(&todo_list);
  80                return error(_("unusable todo list: '%s'"), todo_file);
  81        }
  82
  83        res = todo_list_write_to_file(the_repository, &todo_list, todo_file,
  84                                      NULL, NULL, -1, flags);
  85        todo_list_release(&todo_list);
  86
  87        if (res)
  88                return error_errno(_("could not write '%s'."), todo_file);
  89        return 0;
  90}
  91
  92static int edit_todo_file(unsigned flags)
  93{
  94        const char *todo_file = rebase_path_todo();
  95        struct todo_list todo_list = TODO_LIST_INIT,
  96                new_todo = TODO_LIST_INIT;
  97        int res = 0;
  98
  99        if (strbuf_read_file(&todo_list.buf, todo_file, 0) < 0)
 100                return error_errno(_("could not read '%s'."), todo_file);
 101
 102        strbuf_stripspace(&todo_list.buf, 1);
 103        res = edit_todo_list(the_repository, &todo_list, &new_todo, NULL, NULL, flags);
 104        if (!res && todo_list_write_to_file(the_repository, &new_todo, todo_file,
 105                                            NULL, NULL, -1, flags & ~(TODO_LIST_SHORTEN_IDS)))
 106                res = error_errno(_("could not write '%s'"), todo_file);
 107
 108        todo_list_release(&todo_list);
 109        todo_list_release(&new_todo);
 110
 111        return res;
 112}
 113
 114static int get_revision_ranges(const char *upstream, const char *onto,
 115                               const char **head_hash,
 116                               char **revisions, char **shortrevisions)
 117{
 118        const char *base_rev = upstream ? upstream : onto, *shorthead;
 119        struct object_id orig_head;
 120
 121        if (get_oid("HEAD", &orig_head))
 122                return error(_("no HEAD?"));
 123
 124        *head_hash = find_unique_abbrev(&orig_head, GIT_MAX_HEXSZ);
 125        *revisions = xstrfmt("%s...%s", base_rev, *head_hash);
 126
 127        shorthead = find_unique_abbrev(&orig_head, DEFAULT_ABBREV);
 128
 129        if (upstream) {
 130                const char *shortrev;
 131                struct object_id rev_oid;
 132
 133                get_oid(base_rev, &rev_oid);
 134                shortrev = find_unique_abbrev(&rev_oid, DEFAULT_ABBREV);
 135
 136                *shortrevisions = xstrfmt("%s..%s", shortrev, shorthead);
 137        } else
 138                *shortrevisions = xstrdup(shorthead);
 139
 140        return 0;
 141}
 142
 143static int init_basic_state(struct replay_opts *opts, const char *head_name,
 144                            const char *onto, const char *orig_head)
 145{
 146        FILE *interactive;
 147
 148        if (!is_directory(path_state_dir()) && mkdir_in_gitdir(path_state_dir()))
 149                return error_errno(_("could not create temporary %s"), path_state_dir());
 150
 151        delete_reflog("REBASE_HEAD");
 152
 153        interactive = fopen(path_interactive(), "w");
 154        if (!interactive)
 155                return error_errno(_("could not mark as interactive"));
 156        fclose(interactive);
 157
 158        return write_basic_state(opts, head_name, onto, orig_head);
 159}
 160
 161static int do_interactive_rebase(struct replay_opts *opts, unsigned flags,
 162                                 const char *switch_to, const char *upstream,
 163                                 const char *onto, const char *onto_name,
 164                                 const char *squash_onto, const char *head_name,
 165                                 const char *restrict_revision, char *raw_strategies,
 166                                 struct string_list *commands, unsigned autosquash)
 167{
 168        int ret;
 169        const char *head_hash = NULL;
 170        char *revisions = NULL, *shortrevisions = NULL;
 171        struct argv_array make_script_args = ARGV_ARRAY_INIT;
 172        struct todo_list todo_list = TODO_LIST_INIT;
 173
 174        if (prepare_branch_to_be_rebased(opts, switch_to))
 175                return -1;
 176
 177        if (get_revision_ranges(upstream, onto, &head_hash,
 178                                &revisions, &shortrevisions))
 179                return -1;
 180
 181        if (raw_strategies)
 182                parse_strategy_opts(opts, raw_strategies);
 183
 184        if (init_basic_state(opts, head_name, onto, head_hash)) {
 185                free(revisions);
 186                free(shortrevisions);
 187
 188                return -1;
 189        }
 190
 191        if (!upstream && squash_onto)
 192                write_file(path_squash_onto(), "%s\n", squash_onto);
 193
 194        argv_array_pushl(&make_script_args, "", revisions, NULL);
 195        if (restrict_revision)
 196                argv_array_push(&make_script_args, restrict_revision);
 197
 198        ret = sequencer_make_script(the_repository, &todo_list.buf,
 199                                    make_script_args.argc, make_script_args.argv,
 200                                    flags);
 201
 202        if (ret)
 203                error(_("could not generate todo list"));
 204        else {
 205                discard_cache();
 206                if (todo_list_parse_insn_buffer(the_repository, todo_list.buf.buf,
 207                                                &todo_list))
 208                        BUG("unusable todo list");
 209
 210                ret = complete_action(the_repository, opts, flags, shortrevisions, onto_name,
 211                                      onto, head_hash, commands, autosquash, &todo_list);
 212        }
 213
 214        free(revisions);
 215        free(shortrevisions);
 216        todo_list_release(&todo_list);
 217        argv_array_clear(&make_script_args);
 218
 219        return ret;
 220}
 221
 222static const char * const builtin_rebase_interactive_usage[] = {
 223        N_("git rebase--interactive [<options>]"),
 224        NULL
 225};
 226
 227int cmd_rebase__interactive(int argc, const char **argv, const char *prefix)
 228{
 229        struct replay_opts opts = REPLAY_OPTS_INIT;
 230        unsigned flags = 0, keep_empty = 0, rebase_merges = 0, autosquash = 0;
 231        int abbreviate_commands = 0, rebase_cousins = -1, ret = 0;
 232        const char *onto = NULL, *onto_name = NULL, *restrict_revision = NULL,
 233                *squash_onto = NULL, *upstream = NULL, *head_name = NULL,
 234                *switch_to = NULL, *cmd = NULL;
 235        struct string_list commands = STRING_LIST_INIT_DUP;
 236        char *raw_strategies = NULL;
 237        enum {
 238                NONE = 0, CONTINUE, SKIP, EDIT_TODO, SHOW_CURRENT_PATCH,
 239                SHORTEN_OIDS, EXPAND_OIDS, CHECK_TODO_LIST, REARRANGE_SQUASH, ADD_EXEC
 240        } command = 0;
 241        struct option options[] = {
 242                OPT_BOOL(0, "ff", &opts.allow_ff, N_("allow fast-forward")),
 243                OPT_BOOL(0, "keep-empty", &keep_empty, N_("keep empty commits")),
 244                OPT_BOOL(0, "allow-empty-message", &opts.allow_empty_message,
 245                         N_("allow commits with empty messages")),
 246                OPT_BOOL(0, "rebase-merges", &rebase_merges, N_("rebase merge commits")),
 247                OPT_BOOL(0, "rebase-cousins", &rebase_cousins,
 248                         N_("keep original branch points of cousins")),
 249                OPT_BOOL(0, "autosquash", &autosquash,
 250                         N_("move commits that begin with squash!/fixup!")),
 251                OPT_BOOL(0, "signoff", &opts.signoff, N_("sign commits")),
 252                OPT__VERBOSE(&opts.verbose, N_("be verbose")),
 253                OPT_CMDMODE(0, "continue", &command, N_("continue rebase"),
 254                            CONTINUE),
 255                OPT_CMDMODE(0, "skip", &command, N_("skip commit"), SKIP),
 256                OPT_CMDMODE(0, "edit-todo", &command, N_("edit the todo list"),
 257                            EDIT_TODO),
 258                OPT_CMDMODE(0, "show-current-patch", &command, N_("show the current patch"),
 259                            SHOW_CURRENT_PATCH),
 260                OPT_CMDMODE(0, "shorten-ids", &command,
 261                        N_("shorten commit ids in the todo list"), SHORTEN_OIDS),
 262                OPT_CMDMODE(0, "expand-ids", &command,
 263                        N_("expand commit ids in the todo list"), EXPAND_OIDS),
 264                OPT_CMDMODE(0, "check-todo-list", &command,
 265                        N_("check the todo list"), CHECK_TODO_LIST),
 266                OPT_CMDMODE(0, "rearrange-squash", &command,
 267                        N_("rearrange fixup/squash lines"), REARRANGE_SQUASH),
 268                OPT_CMDMODE(0, "add-exec-commands", &command,
 269                        N_("insert exec commands in todo list"), ADD_EXEC),
 270                OPT_STRING(0, "onto", &onto, N_("onto"), N_("onto")),
 271                OPT_STRING(0, "restrict-revision", &restrict_revision,
 272                           N_("restrict-revision"), N_("restrict revision")),
 273                OPT_STRING(0, "squash-onto", &squash_onto, N_("squash-onto"),
 274                           N_("squash onto")),
 275                OPT_STRING(0, "upstream", &upstream, N_("upstream"),
 276                           N_("the upstream commit")),
 277                OPT_STRING(0, "head-name", &head_name, N_("head-name"), N_("head name")),
 278                { OPTION_STRING, 'S', "gpg-sign", &opts.gpg_sign, N_("key-id"),
 279                        N_("GPG-sign commits"),
 280                        PARSE_OPT_OPTARG, NULL, (intptr_t) "" },
 281                OPT_STRING(0, "strategy", &opts.strategy, N_("strategy"),
 282                           N_("rebase strategy")),
 283                OPT_STRING(0, "strategy-opts", &raw_strategies, N_("strategy-opts"),
 284                           N_("strategy options")),
 285                OPT_STRING(0, "switch-to", &switch_to, N_("switch-to"),
 286                           N_("the branch or commit to checkout")),
 287                OPT_STRING(0, "onto-name", &onto_name, N_("onto-name"), N_("onto name")),
 288                OPT_STRING(0, "cmd", &cmd, N_("cmd"), N_("the command to run")),
 289                OPT_RERERE_AUTOUPDATE(&opts.allow_rerere_auto),
 290                OPT_BOOL(0, "reschedule-failed-exec", &opts.reschedule_failed_exec,
 291                         N_("automatically re-schedule any `exec` that fails")),
 292                OPT_END()
 293        };
 294
 295        sequencer_init_config(&opts);
 296        git_config_get_bool("rebase.abbreviatecommands", &abbreviate_commands);
 297
 298        opts.action = REPLAY_INTERACTIVE_REBASE;
 299        opts.allow_ff = 1;
 300        opts.allow_empty = 1;
 301
 302        if (argc == 1)
 303                usage_with_options(builtin_rebase_interactive_usage, options);
 304
 305        argc = parse_options(argc, argv, NULL, options,
 306                        builtin_rebase_interactive_usage, PARSE_OPT_KEEP_ARGV0);
 307
 308        opts.gpg_sign = xstrdup_or_null(opts.gpg_sign);
 309
 310        flags |= keep_empty ? TODO_LIST_KEEP_EMPTY : 0;
 311        flags |= abbreviate_commands ? TODO_LIST_ABBREVIATE_CMDS : 0;
 312        flags |= rebase_merges ? TODO_LIST_REBASE_MERGES : 0;
 313        flags |= rebase_cousins > 0 ? TODO_LIST_REBASE_COUSINS : 0;
 314        flags |= command == SHORTEN_OIDS ? TODO_LIST_SHORTEN_IDS : 0;
 315
 316        if (rebase_cousins >= 0 && !rebase_merges)
 317                warning(_("--[no-]rebase-cousins has no effect without "
 318                          "--rebase-merges"));
 319
 320        if (cmd && *cmd) {
 321                string_list_split(&commands, cmd, '\n', -1);
 322
 323                /* rebase.c adds a new line to cmd after every command,
 324                 * so here the last command is always empty */
 325                string_list_remove_empty_items(&commands, 0);
 326        }
 327
 328        switch (command) {
 329        case NONE:
 330                if (!onto && !upstream)
 331                        die(_("a base commit must be provided with --upstream or --onto"));
 332
 333                ret = do_interactive_rebase(&opts, flags, switch_to, upstream, onto,
 334                                            onto_name, squash_onto, head_name, restrict_revision,
 335                                            raw_strategies, &commands, autosquash);
 336                break;
 337        case SKIP: {
 338                struct string_list merge_rr = STRING_LIST_INIT_DUP;
 339
 340                rerere_clear(the_repository, &merge_rr);
 341                /* fallthrough */
 342        case CONTINUE:
 343                ret = sequencer_continue(the_repository, &opts);
 344                break;
 345        }
 346        case EDIT_TODO:
 347                ret = edit_todo_file(flags);
 348                break;
 349        case SHOW_CURRENT_PATCH: {
 350                struct child_process cmd = CHILD_PROCESS_INIT;
 351
 352                cmd.git_cmd = 1;
 353                argv_array_pushl(&cmd.args, "show", "REBASE_HEAD", "--", NULL);
 354                ret = run_command(&cmd);
 355
 356                break;
 357        }
 358        case SHORTEN_OIDS:
 359        case EXPAND_OIDS:
 360                ret = transform_todo_file(flags);
 361                break;
 362        case CHECK_TODO_LIST:
 363                ret = check_todo_list_from_file(the_repository);
 364                break;
 365        case REARRANGE_SQUASH:
 366                ret = rearrange_squash_in_todo_file();
 367                break;
 368        case ADD_EXEC:
 369                ret = add_exec_commands(&commands);
 370                break;
 371        default:
 372                BUG("invalid command '%d'", command);
 373        }
 374
 375        string_list_clear(&commands, 0);
 376        return !!ret;
 377}