#include "drop_install_modal.hpp"

#include "dusk/mods/loader/loader.hpp"
#include "dusk/mods/loader/packages.hpp"
#include "dusk/mods/queue.hpp"
#include "format.hpp"
#include "package_row.hpp"
#include "queue_window.hpp"

#include <borealis/io.hpp>
#include <borealis/update.hpp>
#include <fmt/format.h>

#include <algorithm>

namespace dusk::ui {
namespace {

size_t valid_count(const std::vector<DropPackage>& packages) {
    return std::ranges::count(packages, true, &DropPackage::valid);
}

std::vector<DropPackage> prepare_packages(std::vector<DropPackage> packages) {
    std::vector<std::string> batchIds;
    for (auto& package : packages) {
        if (!package.error.empty()) {
            package.status = package.error;
        } else if (std::ranges::find(batchIds, package.metadata.id) != batchIds.end()) {
            package.status = "Duplicate package in this drop";
        } else if (!borealis::update::parse_version(package.metadata.version)) {
            package.status = "Invalid package version";
        } else if (package.hasNative && !mods::EnableCodeMods) {
            package.status = "Native mods cannot be installed on this platform";
        } else if (const auto queued = mods::queue::find_by_mod_id(package.metadata.id);
            queued && !mods::queue::is_terminal(queued->state))
        {
            package.status = "Already in the install queue";
        } else if (const auto* installed =
                       mods::ModLoader::instance().find_mod(package.metadata.id))
        {
            if (!mods::ModLoader::instance().can_update(*installed)) {
                package.status = "A development directory cannot be replaced";
            } else if (mods::compare_package_versions(
                           package.metadata.version, installed->metadata.version) < 0) {
                package.status = "A newer version is already installed";
            } else if (mods::compare_package_versions(
                           package.metadata.version, installed->metadata.version) == 0) {
                package.status = fmt::format("Reinstall {}", package.metadata.version);
                package.valid = true;
            } else {
                package.status = fmt::format("Update from {}", installed->metadata.version);
                package.valid = true;
            }
        } else {
            package.status = "New";
            package.valid = true;
        }
        batchIds.push_back(package.metadata.id);
    }
    return packages;
}

}  // namespace

std::vector<DropPackage> inspect_drop_packages(
    const std::vector<std::filesystem::path>& paths, borealis::TaskContext& context) {
    std::vector<DropPackage> packages;
    packages.reserve(paths.size());
    for (const auto& path : paths) {
        if (context.cancel_requested()) {
            break;
        }
        DropPackage package{.path = path};
        std::error_code error;
        package.size = std::filesystem::file_size(path, error);
        if (error) {
            package.error = fmt::format("Could not read package: {}", error.message());
        } else if (!mods::inspect_mod_bundle(
                       path, package.metadata, package.error, &package.hasNative))
        {
            package.error = fmt::format("Invalid package: {}", package.error);
        }
        packages.push_back(std::move(package));
        context.report_progress(packages.size(), paths.size());
    }
    return packages;
}

DropInstallModal::DropInstallModal(std::vector<DropPackage> packages)
    : DropInstallModal{prepare_packages(std::move(packages)), PreparedTag{}} {}

DropInstallModal::DropInstallModal(std::vector<DropPackage> packages, PreparedTag)
    : Modal{Props{
          .title = "Install mods?",
          .bodyText = "Only install mods from trusted authors.",
          .actions =
              {
                  ModalAction{
                      .label = "Cancel",
                      .onPressed = [](Modal& modal) { modal.pop(); },
                      .isDisabled = {},
                  },
                  ModalAction{
                      .label = fmt::format("Install {}", valid_count(packages)),
                      .onPressed = [this](Modal&) { install(); },
                      .isDisabled = [this] { return valid_count(mPackages) == 0; },
                  },
              },
          .variant = "drop-install",
      }},
      mPackages{std::move(packages)} {
    auto& pane = content_pane();
    for (auto& package : mPackages) {
        auto& row = pane.add_child<PackageRow>();
        const auto name = package.metadata.name.empty() ?
                              borealis::io::fs_path_to_string(package.path.filename()) :
                              package.metadata.name;
        const auto version =
            package.metadata.name.empty() ? std::string{} : package.metadata.version;
        const auto detail =
            package.metadata.author.empty() ?
                format_bytes(package.size) :
                fmt::format("{} · {}", package.metadata.author, format_bytes(package.size));
        row.set_package(name, version, package.status, detail, package.valid ? "queued" : "failed");
        row.set_disabled(!package.valid);
    }
}

void DropInstallModal::install() {
    std::string firstKey;
    for (const auto& package : mPackages) {
        if (!package.valid) {
            continue;
        }
        std::string key;
        if (mods::queue::enqueue(
                {
                    .id = package.metadata.id,
                    .name = package.metadata.name,
                    .version = package.metadata.version,
                    .source = mods::queue::LocalFile{package.path},
                },
                &key) &&
            firstKey.empty())
        {
            firstKey = std::move(key);
        }
    }
    pop();
    if (!firstKey.empty()) {
        if (auto* current = top_document()) {
            current->cover();
        }
        push_document(std::make_unique<QueueWindow>(std::move(firstKey)));
    }
}

}  // namespace dusk::ui