#include "rm_subcommand.hpp"

#include <filesystem>
#include <ranges>

#include "../utils/common.hpp"
#include "../utils/git_exception.hpp"
#include "../wrapper/index_wrapper.hpp"
#include "../wrapper/repository_wrapper.hpp"

namespace fs = std::filesystem;

rm_subcommand::rm_subcommand(const libgit2_object&, CLI::App& app)
{
    auto* rm = app.add_subcommand("rm", "Remove files from the working tree and from the index");
    rm->add_option("<pathspec>", m_pathspec, "Files to remove");
    rm->add_flag("-r", m_recursive, "Allow recursive removal when a leading directory name is given");

    rm->callback(
        [this]()
        {
            this->run();
        }
    );
}

void rm_subcommand::run()
{
    auto directory = get_current_git_path();
    auto repo = repository_wrapper::open(directory);

    index_wrapper index = repo.make_index();

    std::vector<std::string> files;
    std::vector<std::string> directories;

    std::ranges::for_each(
        m_pathspec,
        [&](const std::string& path)
        {
            if (!fs::exists(path))
            {
                std::string msg = "fatal: pathspec '" + path + "' did not math any file";
                throw git_exception(msg, git2cpp_error_code::FILESYSTEM_ERROR);
            }
            if (fs::is_directory(path))
            {
                directories.push_back(path);
            }
            else
            {
                if (!repo.does_track(path))
                {
                    std::string msg = "fatal: pathsspec '" + path + "'is not tracked";
                    throw git_exception(msg, git2cpp_error_code::FILESYSTEM_ERROR);
                }
                files.push_back(path);
            }
        }
    );

    if (!directories.empty() && !m_recursive)
    {
        std::string msg = "fatal: not removing '" + directories.front() + "' recursively without -r";
        throw git_exception(msg, git2cpp_error_code::FILESYSTEM_ERROR);
    }

    index.remove_entries(files);
    index.remove_directories(directories);
    index.write();

    std::ranges::for_each(
        files,
        [](const std::string& path)
        {
            fs::remove(path);
        }
    );
    std::ranges::for_each(
        directories,
        [](const std::string& path)
        {
            fs::remove_all(path);
        }
    );
}