#include "reset_subcommand.hpp"

// #include "../wrapper/index_wrapper.hpp"

#include <stdexcept>

#include "../wrapper/repository_wrapper.hpp"

enum class reset_type

{

GIT_RESET_SOFT = 1,

GIT_RESET_MIXED = 2,

GIT_RESET_HARD = 3

};

reset_subcommand::reset_subcommand(const libgit2_object&, CLI::App& app)

{

auto* sub = app.add_subcommand("reset", "Reset current HEAD to the specified state");

sub->add_option("<commit>", m_commit, "The ID of the commit that will become HEAD");

sub->add_flag("--soft", m_soft_flag, "");

sub->add_flag("--mixed", m_mixed_flag, "");

sub->add_flag("--hard", m_hard_flag, "");

sub->callback(

[this]()

{

this->run();

}

);

};

void reset_subcommand::run()

{

auto directory = get_current_git_path();

auto repo = repository_wrapper::open(directory);

auto target = repo.revparse_single(m_commit);

if (!target)

{

throw std::runtime_error("Target revision not found.");

}

git_checkout_options options;

git_checkout_options_init(&options, GIT_CHECKOUT_OPTIONS_VERSION);

git_reset_t reset_type;

if (m_soft_flag)

{

reset_type = GIT_RESET_SOFT;

}

if (m_mixed_flag)

{

reset_type = GIT_RESET_MIXED;

}

if (m_hard_flag)

{

reset_type = GIT_RESET_HARD;

if (m_commit.empty())

{

m_commit = "HEAD";

}

}

repo.reset(target.value(), reset_type, options);

}