#include "pch.h"
#include "vcpkg_Commands.h"
#include "vcpkglib.h"
#include "vcpkg_System.h"
#include "vcpkg_Input.h"
#include "vcpkg_Dependencies.h"

namespace vcpkg::Commands::Remove
{
    using Dependencies::package_spec_with_remove_plan;
    using Dependencies::remove_plan_type;

    static const std::string OPTION_PURGE = "--purge";
    static const std::string OPTION_RECURSIVE = "--recursive";

    static void delete_directory(const fs::path& directory)
    {
        std::error_code ec;
        fs::remove_all(directory, ec);
        if (!ec)
        {
            System::println(System::color::success, "Cleaned up %s", directory.string());
        }
        if (fs::exists(directory))
        {
            System::println(System::color::warning, "Some files in %s were unable to be removed. Close any editors operating in this directory and retry.", directory.string());
        }
    }

    static void remove_package(const vcpkg_paths& paths, StatusParagraph& pkg)
    {
        pkg.want = want_t::purge;
        pkg.state = install_state_t::half_installed;
        write_update(paths, pkg);

        std::fstream listfile(paths.listfile_path(pkg.package), std::ios_base::in | std::ios_base::binary);
        if (listfile)
        {
            std::vector<fs::path> dirs_touched;
            std::string suffix;
            while (std::getline(listfile, suffix))
            {
                if (!suffix.empty() && suffix.back() == '\r')
                    suffix.pop_back();

                std::error_code ec;

                auto target = paths.installed / suffix;

                auto status = fs::status(target, ec);
                if (ec)
                {
                    System::println(System::color::error, "failed: %s", ec.message());
                    continue;
                }

                if (fs::is_directory(status))
                {
                    dirs_touched.push_back(target);
                }
                else if (fs::is_regular_file(status))
                {
                    fs::remove(target, ec);
                    if (ec)
                    {
                        System::println(System::color::error, "failed: %s: %s", target.u8string(), ec.message());
                    }
                }
                else if (!fs::status_known(status))
                {
                    System::println(System::color::warning, "Warning: unknown status: %s", target.u8string());
                }
                else
                {
                    System::println(System::color::warning, "Warning: %s: cannot handle file type", target.u8string());
                }
            }

            auto b = dirs_touched.rbegin();
            auto e = dirs_touched.rend();
            for (; b != e; ++b)
            {
                if (fs::directory_iterator(*b) == fs::directory_iterator())
                {
                    std::error_code ec;
                    fs::remove(*b, ec);
                    if (ec)
                    {
                        System::println(System::color::error, "failed: %s", ec.message());
                    }
                }
            }

            listfile.close();
            fs::remove(paths.listfile_path(pkg.package));
        }

        pkg.state = install_state_t::not_installed;
        write_update(paths, pkg);
    }

    static void sort_packages_by_name(std::vector<const package_spec_with_remove_plan*>* packages)
    {
        std::sort(packages->begin(), packages->end(), [](const package_spec_with_remove_plan* left, const package_spec_with_remove_plan* right) -> bool
                  {
                      return left->spec.name() < right->spec.name();
                  });
    }

    static void print_plan(const std::vector<package_spec_with_remove_plan>& plan)
    {
        std::vector<const package_spec_with_remove_plan*> not_installed;
        std::vector<const package_spec_with_remove_plan*> remove;

        for (const package_spec_with_remove_plan& i : plan)
        {
            if (i.plan.type == remove_plan_type::NOT_INSTALLED)
            {
                not_installed.push_back(&i);
                continue;
            }

            if (i.plan.type == remove_plan_type::REMOVE_AUTO_SELECTED || i.plan.type == remove_plan_type::REMOVE_USER_REQUESTED)
            {
                remove.push_back(&i);
                continue;
            }

            Checks::unreachable();
        }

        if (!not_installed.empty())
        {
            sort_packages_by_name(&not_installed);
            System::println("The following packages are not installed, so not removed:\n%s",
                            Strings::Joiner::on("\n    ").prefix("    ").join(not_installed, [](const package_spec_with_remove_plan* p)
                                                                              {
                                                                                  return p->spec.toString();
                                                                              }));
        }

        if (!remove.empty())
        {
            sort_packages_by_name(&remove);
            System::println("The following packages will be removed:\n%s",
                            Strings::Joiner::on("\n").join(remove, [](const package_spec_with_remove_plan* p)
                                                           {
                                                               switch (p->plan.type)
                                                               {
                                                                   case remove_plan_type::REMOVE_USER_REQUESTED:
                                                                       return "    " + p->spec.toString();
                                                                   case remove_plan_type::REMOVE_AUTO_SELECTED:
                                                                       return "  * " + p->spec.toString();
                                                                   default:
                                                                       Checks::unreachable();
                                                               }
                                                           }));
        }
    }

    void perform_and_exit(const vcpkg_cmd_arguments& args, const vcpkg_paths& paths, const triplet& default_target_triplet)
    {
        static const std::string example = Commands::Help::create_example_string("remove zlib zlib:x64-windows curl boost");
        args.check_min_arg_count(1, example);

        const std::unordered_set<std::string> options = args.check_and_get_optional_command_arguments({OPTION_PURGE, OPTION_RECURSIVE});
        auto status_db = database_load_check(paths);

        std::vector<package_spec> specs = Input::check_and_get_package_specs(args.command_arguments, default_target_triplet, example);
        Input::check_triplets(specs, paths);
        const bool alsoRemoveFolderFromPackages = options.find(OPTION_PURGE) != options.end();
        const bool isRecursive = options.find(OPTION_RECURSIVE) != options.end();

        const std::vector<package_spec_with_remove_plan> remove_plan = Dependencies::create_remove_plan(paths, specs, status_db);
        Checks::check_exit(!remove_plan.empty(), "Remove plan cannot be empty");

        print_plan(remove_plan);

        const bool has_non_user_requested_packages = std::find_if(remove_plan.cbegin(), remove_plan.cend(), [](const package_spec_with_remove_plan& package)-> bool
                                                                  {
                                                                      return package.plan.type == remove_plan_type::REMOVE_AUTO_SELECTED;
                                                                  }) != remove_plan.cend();

        if (has_non_user_requested_packages && !isRecursive)
        {
            System::println(System::color::warning,
                            "Additional packages (*) need to be removed to complete this operation.\n"
                            "If you are sure you want to remove them, run the command with the --recursive option");
            exit(EXIT_FAILURE);
        }

        for (const package_spec_with_remove_plan& action : remove_plan)
        {
            if (action.plan.type == remove_plan_type::NOT_INSTALLED)
            {
                System::println(System::color::success, "Package %s is not installed", action.spec);
            }
            else if (action.plan.type == remove_plan_type::REMOVE_AUTO_SELECTED || action.plan.type == remove_plan_type::REMOVE_USER_REQUESTED)
            {
                const std::string display_name = action.spec.display_name();
                System::println("Removing package %s... ", display_name);
                remove_package(paths, *action.plan.status_pgh);
                System::println(System::color::success, "Removing package %s... done", display_name);

                if (alsoRemoveFolderFromPackages)
                {
                    const fs::path spec_package_dir = paths.packages / action.spec.dir();
                    System::println("Purging package %s... ", display_name);
                    delete_directory(spec_package_dir);
                    System::println(System::color::success, "Purging package %s... done", display_name);
                }
            }
            else
                Checks::unreachable();
        }

        exit(EXIT_SUCCESS);
    }
}