#include "builtin.h"
#include "cache.h"
#include "config.h"
#include "parse-options.h"
#include "sequencer.h"
#include "rebase-interactive.h"
#include "argv-array.h"
#include "refs.h"
#include "rerere.h"
#include "run-command.h"

static GIT_PATH_FUNC(path_state_dir, "rebase-merge/")
static GIT_PATH_FUNC(path_squash_onto, "rebase-merge/squash-onto")
static GIT_PATH_FUNC(path_interactive, "rebase-merge/interactive")

static int get_revision_ranges(const char *upstream, const char *onto,
                               const char **head_hash,
                               char **revisions, char **shortrevisions)
{
        const char *base_rev = upstream ? upstream : onto, *shorthead;
        struct object_id orig_head;

        if (get_oid("HEAD", &orig_head))
                return error(_("no HEAD?"));

        *head_hash = find_unique_abbrev(&orig_head, GIT_MAX_HEXSZ);
        *revisions = xstrfmt("%s...%s", base_rev, *head_hash);

        shorthead = find_unique_abbrev(&orig_head, DEFAULT_ABBREV);

        if (upstream) {
                const char *shortrev;
                struct object_id rev_oid;

                get_oid(base_rev, &rev_oid);
                shortrev = find_unique_abbrev(&rev_oid, DEFAULT_ABBREV);

                *shortrevisions = xstrfmt("%s..%s", shortrev, shorthead);
        } else
                *shortrevisions = xstrdup(shorthead);

        return 0;
}

static int init_basic_state(struct replay_opts *opts, const char *head_name,
                            const char *onto, const char *orig_head)
{
        FILE *interactive;

        if (!is_directory(path_state_dir()) && mkdir_in_gitdir(path_state_dir()))
                return error_errno(_("could not create temporary %s"), path_state_dir());

        delete_reflog("REBASE_HEAD");

        interactive = fopen(path_interactive(), "w");
        if (!interactive)
                return error_errno(_("could not mark as interactive"));
        fclose(interactive);

        return write_basic_state(opts, head_name, onto, orig_head);
}

static int do_interactive_rebase(struct replay_opts *opts, unsigned flags,
                                 const char *switch_to, const char *upstream,
                                 const char *onto, const char *onto_name,
                                 const char *squash_onto, const char *head_name,
                                 const char *restrict_revision, char *raw_strategies,
                                 const char *cmd, unsigned autosquash)
{
        int ret;
        const char *head_hash = NULL;
        char *revisions = NULL, *shortrevisions = NULL;
        struct argv_array make_script_args = ARGV_ARRAY_INIT;
        FILE *todo_list;

        if (prepare_branch_to_be_rebased(opts, switch_to))
                return -1;

        if (get_revision_ranges(upstream, onto, &head_hash,
                                &revisions, &shortrevisions))
                return -1;

        if (raw_strategies)
                parse_strategy_opts(opts, raw_strategies);

        if (init_basic_state(opts, head_name, onto, head_hash)) {
                free(revisions);
                free(shortrevisions);

                return -1;
        }

        if (!upstream && squash_onto)
                write_file(path_squash_onto(), "%s\n", squash_onto);

        todo_list = fopen(rebase_path_todo(), "w");
        if (!todo_list) {
                free(revisions);
                free(shortrevisions);

                return error_errno(_("could not open %s"), rebase_path_todo());
        }

        argv_array_pushl(&make_script_args, "", revisions, NULL);
        if (restrict_revision)
                argv_array_push(&make_script_args, restrict_revision);

        ret = sequencer_make_script(todo_list,
                                    make_script_args.argc, make_script_args.argv,
                                    flags);
        fclose(todo_list);

        if (ret)
                error(_("could not generate todo list"));
        else {
                discard_cache();
                ret = complete_action(opts, flags, shortrevisions, onto_name, onto,
                                      head_hash, cmd, autosquash);
        }

        free(revisions);
        free(shortrevisions);
        argv_array_clear(&make_script_args);

        return ret;
}

static const char * const builtin_rebase_interactive_usage[] = {
        N_("git rebase--interactive [<options>]"),
        NULL
};

int cmd_rebase__interactive(int argc, const char **argv, const char *prefix)
{
        struct replay_opts opts = REPLAY_OPTS_INIT;
        unsigned flags = 0, keep_empty = 0, rebase_merges = 0, autosquash = 0;
        int abbreviate_commands = 0, rebase_cousins = -1;
        const char *onto = NULL, *onto_name = NULL, *restrict_revision = NULL,
                *squash_onto = NULL, *upstream = NULL, *head_name = NULL,
                *switch_to = NULL, *cmd = NULL;
        char *raw_strategies = NULL;
        struct option options[] = {
                OPT_BOOL(0, "ff", &opts.allow_ff, N_("allow fast-forward")),
                OPT_BOOL(0, "keep-empty", &keep_empty, N_("keep empty commits")),
                OPT_BOOL(0, "allow-empty-message", &opts.allow_empty_message,
                         N_("allow commits with empty messages")),
                OPT_BOOL(0, "rebase-merges", &rebase_merges, N_("rebase merge commits")),
                OPT_BOOL(0, "rebase-cousins", &rebase_cousins,
                         N_("keep original branch points of cousins")),
                OPT_BOOL(0, "autosquash", &autosquash,
                         N_("move commits that begin with squash!/fixup!")),
                OPT_BOOL(0, "signoff", &opts.signoff, N_("sign commits")),
                OPT__VERBOSE(&opts.verbose, N_("be verbose")),
                OPT_STRING(0, "onto", &onto, N_("onto"), N_("onto")),
                OPT_STRING(0, "restrict-revision", &restrict_revision,
                           N_("restrict-revision"), N_("restrict revision")),
                OPT_STRING(0, "squash-onto", &squash_onto, N_("squash-onto"),
                           N_("squash onto")),
                OPT_STRING(0, "upstream", &upstream, N_("upstream"),
                           N_("the upstream commit")),
                OPT_STRING(0, "head-name", &head_name, N_("head-name"), N_("head name")),
                { OPTION_STRING, 'S', "gpg-sign", &opts.gpg_sign, N_("key-id"),
                        N_("GPG-sign commits"),
                        PARSE_OPT_OPTARG, NULL, (intptr_t) "" },
                OPT_STRING(0, "strategy", &opts.strategy, N_("strategy"),
                           N_("rebase strategy")),
                OPT_STRING(0, "strategy-opts", &raw_strategies, N_("strategy-opts"),
                           N_("strategy options")),
                OPT_STRING(0, "switch-to", &switch_to, N_("switch-to"),
                           N_("the branch or commit to checkout")),
                OPT_STRING(0, "onto-name", &onto_name, N_("onto-name"), N_("onto name")),
                OPT_STRING(0, "cmd", &cmd, N_("cmd"), N_("the command to run")),
                OPT_RERERE_AUTOUPDATE(&opts.allow_rerere_auto),
                OPT_END()
        };

        sequencer_init_config(&opts);
        git_config_get_bool("rebase.abbreviatecommands", &abbreviate_commands);

        opts.action = REPLAY_INTERACTIVE_REBASE;
        opts.allow_ff = 1;
        opts.allow_empty = 1;

        if (argc == 1)
                usage_with_options(builtin_rebase_interactive_usage, options);

        argc = parse_options(argc, argv, NULL, options,
                        builtin_rebase_interactive_usage, PARSE_OPT_KEEP_ARGV0);

        opts.gpg_sign = xstrdup_or_null(opts.gpg_sign);

        flags |= keep_empty ? TODO_LIST_KEEP_EMPTY : 0;
        flags |= abbreviate_commands ? TODO_LIST_ABBREVIATE_CMDS : 0;
        flags |= rebase_merges ? TODO_LIST_REBASE_MERGES : 0;
        flags |= rebase_cousins > 0 ? TODO_LIST_REBASE_COUSINS : 0;

        if (rebase_cousins >= 0 && !rebase_merges)
                warning(_("--[no-]rebase-cousins has no effect without "
                          "--rebase-merges"));

        if (!onto && !upstream)
                die(_("a base commit must be provided with --upstream or --onto"));

        return !!do_interactive_rebase(&opts, flags, switch_to, upstream, onto,
                                       onto_name, squash_onto, head_name, restrict_revision,
                                       raw_strategies, cmd, autosquash);
}