// SPDX-License-Identifier: GPL-3.0-only
/*
 *  Prism Launcher - Minecraft Launcher
 *  Copyright (c) 2022 Jamie Mansfield <jmansfield@cadixdev.org>
 *  Copyright (C) 2022 Sefa Eyeoglu <contact@scrumplex.net>
 *  Copyright (C) 2022 TheKodeToad <TheKodeToad@proton.me>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, version 3.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 * This file incorporates work covered by the following copyright and
 * permission notice:
 *
 *      Copyright 2013-2021 MultiMC Contributors
 *
 *      Licensed under the Apache License, Version 2.0 (the "License");
 *      you may not use this file except in compliance with the License.
 *      You may obtain a copy of the License at
 *
 *          http://www.apache.org/licenses/LICENSE-2.0
 *
 *      Unless required by applicable law or agreed to in writing, software
 *      distributed under the License is distributed on an "AS IS" BASIS,
 *      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *      See the License for the specific language governing permissions and
 *      limitations under the License.
 */

#include "MultiWorldListPage.h"
#include "minecraft/MultiWorldList.h"
#include "ui/dialogs/CustomMessageBox.h"
#include "ui_MultiWorldListPage.h"

#include <ui/widgets/PageContainer.h>
#include <QClipboard>
#include <QDialogButtonBox>
#include <QEvent>
#include <QInputDialog>
#include <QKeyEvent>
#include <QMenu>
#include <QMessageBox>
#include <QPushButton>
#include <QSortFilterProxyModel>
#include <QTreeView>
#include <Qt>

#include "FileSystem.h"
#include "tools/MCEditTool.h"

#include "DesktopServices.h"
#include "ui/GuiUtil.h"

#include "Application.h"
#include "icons/IconList.h"
#include "pages/instance/DataPackPage.h"

class MultiWorldListProxyModel : public QSortFilterProxyModel {
    Q_OBJECT

   public:
    MultiWorldListProxyModel(QObject* parent) : QSortFilterProxyModel(parent) {}

    virtual QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const
    {
        QModelIndex sourceIndex = mapToSource(index);

        if (index.column() == 0 && role == Qt::DecorationRole) {
            MultiWorldList* worlds = qobject_cast<MultiWorldList*>(sourceModel());
            auto iconFile = worlds->data(sourceIndex, MultiWorldList::IconFileRole).toString();
            if (iconFile.isNull()) {
                // NOTE: Minecraft uses the same placeholder for servers AND worlds
                return QIcon::fromTheme("unknown_server");
            }
            return QIcon(iconFile);
        }

        if (index.column() == 1 && role == Qt::DecorationRole) {
            MultiWorldList* worlds = qobject_cast<MultiWorldList*>(sourceModel());
            auto icon = worlds->data(sourceIndex, MultiWorldList::InstanceIconFileRole).value<QIcon>();
            return icon.pixmap(24, 24);
        }

        return sourceIndex.data(role);
    }
};

MultiWorldListPage::MultiWorldListPage(MultiWorldList* worlds, QWidget* parent)
    : QMainWindow(parent), ui(new Ui::MultiWorldListPage), m_worlds(worlds)
{
    ui->setupUi(this);

    ui->toolBar->insertSpacer(ui->actionRefresh);
    ui->actionJoin->setEnabled(true);

    auto* proxy = new MultiWorldListProxyModel(this);
    proxy->setSortCaseSensitivity(Qt::CaseInsensitive);
    proxy->setSourceModel(m_worlds);
    proxy->setSortRole(Qt::UserRole);
    ui->worldTreeView->setSortingEnabled(true);
    ui->worldTreeView->setModel(proxy);
    ui->worldTreeView->installEventFilter(this);
    ui->worldTreeView->setContextMenuPolicy(Qt::CustomContextMenu);
    ui->worldTreeView->setIconSize(QSize(64, 64));
    connect(ui->worldTreeView, &QTreeView::customContextMenuRequested, this, &MultiWorldListPage::ShowContextMenu);

    auto head = ui->worldTreeView->header();
    head->setSectionResizeMode(0, QHeaderView::Stretch);
    head->setSectionResizeMode(1, QHeaderView::ResizeToContents);
    head->setSectionResizeMode(4, QHeaderView::ResizeToContents);

    connect(ui->worldTreeView->selectionModel(), &QItemSelectionModel::currentChanged, this, &MultiWorldListPage::worldChanged);
    connect(ui->worldTreeView, &QAbstractItemView::doubleClicked, this, &MultiWorldListPage::worldDoubleClicked);
    worldChanged(QModelIndex(), QModelIndex());

    connect(m_worlds, &MultiWorldList::fileDropped, this, &MultiWorldListPage::fileDropped);
}

void MultiWorldListPage::openedImpl()
{
    m_worlds->startWatching();

    auto const setting_name = QString("WideBarVisibility_%1").arg(id());
    m_wide_bar_setting = APPLICATION->settings()->getOrRegisterSetting(setting_name);

    ui->toolBar->setVisibilityState(QByteArray::fromBase64(m_wide_bar_setting->get().toString().toUtf8()));
}

void MultiWorldListPage::closedImpl()
{
    m_worlds->stopWatching();

    m_wide_bar_setting->set(QString::fromUtf8(ui->toolBar->getVisibilityState().toBase64()));
}

MultiWorldListPage::~MultiWorldListPage()
{
    m_worlds->stopWatching();
    delete ui;
}

void MultiWorldListPage::ShowContextMenu(const QPoint& pos)
{
    auto menu = ui->toolBar->createContextMenu(this, tr("Context menu"));
    menu->exec(ui->worldTreeView->mapToGlobal(pos));
    delete menu;
}

QMenu* MultiWorldListPage::createPopupMenu()
{
    QMenu* filteredMenu = QMainWindow::createPopupMenu();
    filteredMenu->removeAction(ui->toolBar->toggleViewAction());
    return filteredMenu;
}

bool MultiWorldListPage::shouldDisplay() const
{
    return true;
}

void MultiWorldListPage::retranslate()
{
    ui->retranslateUi(this);
}

bool MultiWorldListPage::worldListFilter(QKeyEvent* keyEvent)
{
    if (keyEvent->key() == Qt::Key_Delete) {
        on_actionRemove_triggered();
        return true;
    }
    return QWidget::eventFilter(ui->worldTreeView, keyEvent);
}

bool MultiWorldListPage::eventFilter(QObject* obj, QEvent* ev)
{
    if (ev->type() != QEvent::KeyPress) {
        return QWidget::eventFilter(obj, ev);
    }
    QKeyEvent* keyEvent = static_cast<QKeyEvent*>(ev);
    if (obj == ui->worldTreeView)
        return worldListFilter(keyEvent);
    return QWidget::eventFilter(obj, ev);
}

void MultiWorldListPage::on_actionRemove_triggered()
{
    auto proxiedIndex = getSelectedWorld();

    if (!proxiedIndex.isValid())
        return;

    auto result = CustomMessageBox::selectable(this, tr("Confirm Deletion"),
                                               tr("You are about to delete \"%1\".\n"
                                                  "The world may be gone forever (A LONG TIME).\n\n"
                                                  "Are you sure?")
                                                   .arg(m_worlds->allWorlds().at(proxiedIndex.row()).world.name()),
                                               QMessageBox::Warning, QMessageBox::Yes | QMessageBox::No, QMessageBox::No)
                      ->exec();

    if (result != QMessageBox::Yes) {
        return;
    }
    m_worlds->stopWatching();
    m_worlds->deleteWorld(proxiedIndex.row());
    m_worlds->startWatching();
}

void MultiWorldListPage::on_actionView_Folder_triggered()
{
    QModelIndex index = getSelectedWorld();
    if (!index.isValid()) {
        return;
    }

    auto worldVariant = m_worlds->data(index, MultiWorldList::ObjectRole);
    auto world = (World*)worldVariant.value<void*>();

    DesktopServices::openPath(world->canonicalFilePath(), true);
}

void MultiWorldListPage::on_actionData_Packs_triggered()
{
    QModelIndex index = getSelectedWorld();

    if (!index.isValid()) {
        return;
    }

    if (!worldSafetyNagQuestion(tr("Manage Data Packs")))
        return;

    const QString fullPath = m_worlds->data(index, MultiWorldList::FolderRole).toString();
    const QString folder = FS::PathCombine(fullPath, "datapacks");

    auto dialog = new QDialog(this);
    dialog->setWindowTitle(tr("Data packs for %1").arg(m_worlds->data(index, MultiWorldList::NameRole).toString()));
    dialog->setWindowModality(Qt::WindowModal);

    dialog->resize(static_cast<int>(std::max(0.5 * window()->width(), 400.0)),
                   static_cast<int>(std::max(0.75 * window()->height(), 400.0)));
    dialog->restoreGeometry(QByteArray::fromBase64(APPLICATION->settings()->get("DataPackDownloadGeometry").toByteArray()));

    GenericPageProvider provider(dialog->windowTitle());

    auto instance = (static_cast<InstanceWorld*>(m_worlds->data(index, MultiWorldList::ObjectRole).value<void*>()))->instance;

    bool isIndexed = !APPLICATION->settings()->get("ModMetadataDisabled").toBool();
    m_datapackModel.reset(new DataPackFolderModel(folder, instance, isIndexed, true));

    provider.addPageCreator([this, instance] { return new DataPackPage(instance, m_datapackModel.get(), this); });

    auto layout = new QVBoxLayout(dialog);

    auto focusStealer = new QPushButton(dialog);
    layout->addWidget(focusStealer);
    focusStealer->setDefault(true);
    focusStealer->hide();

    auto pageContainer = new PageContainer(&provider, {}, dialog);
    pageContainer->hidePageList();
    layout->addWidget(pageContainer);

    auto buttonBox = new QDialogButtonBox(QDialogButtonBox::Close | QDialogButtonBox::Help);
    connect(buttonBox, &QDialogButtonBox::rejected, dialog, &QDialog::reject);
    connect(buttonBox, &QDialogButtonBox::helpRequested, pageContainer, &PageContainer::help);
    layout->addWidget(buttonBox);

    dialog->setLayout(layout);

    dialog->setAttribute(Qt::WA_DeleteOnClose);

    connect(dialog, &QDialog::finished, this,
            [dialog]() { APPLICATION->settings()->set("DataPackDownloadGeometry", dialog->saveGeometry().toBase64()); });

    dialog->open();
}

void MultiWorldListPage::on_actionReset_Icon_triggered()
{
    auto proxiedIndex = getSelectedWorld();

    if (!proxiedIndex.isValid())
        return;

    if (m_worlds->resetIcon(proxiedIndex.row())) {
        ui->actionReset_Icon->setEnabled(false);
    }
}

QModelIndex MultiWorldListPage::getSelectedWorld()
{
    auto index = ui->worldTreeView->selectionModel()->currentIndex();

    auto proxy = (QSortFilterProxyModel*)ui->worldTreeView->model();
    return proxy->mapToSource(index);
}

void MultiWorldListPage::on_actionCopy_Seed_triggered()
{
    QModelIndex index = getSelectedWorld();

    if (!index.isValid()) {
        return;
    }
    int64_t seed = m_worlds->data(index, MultiWorldList::SeedRole).toLongLong();
    APPLICATION->clipboard()->setText(QString::number(seed));
}

void MultiWorldListPage::on_actionMCEdit_triggered()
{
    if (m_mceditStarting)
        return;

    auto mcedit = APPLICATION->mcedit();

    const QString mceditPath = mcedit->path();

    QModelIndex index = getSelectedWorld();

    if (!index.isValid()) {
        return;
    }

    if (!worldSafetyNagQuestion(tr("Open World in MCEdit")))
        return;

    auto fullPath = m_worlds->data(index, MultiWorldList::FolderRole).toString();

    auto program = mcedit->getProgramPath();
    if (program.size()) {
#ifdef Q_OS_WIN32
        if (!QProcess::startDetached(program, { fullPath }, mceditPath)) {
            mceditError();
        }
#else
        m_mceditProcess.reset(new LoggedProcess());
        m_mceditProcess->setDetachable(true);
        connect(m_mceditProcess.get(), &LoggedProcess::stateChanged, this, &MultiWorldListPage::mceditState);
        m_mceditProcess->start(program, { fullPath });
        m_mceditProcess->setWorkingDirectory(mceditPath);
        m_mceditStarting = true;
#endif
    } else {
        QMessageBox::warning(this->parentWidget(), tr("No MCEdit found or set up!"),
                             tr("You do not have MCEdit set up or it was moved.\nYou can set it up in the global settings."));
    }
}

void MultiWorldListPage::mceditError()
{
    QMessageBox::warning(this->parentWidget(), tr("MCEdit failed to start!"),
                         tr("MCEdit failed to start.\nIt may be necessary to reinstall it."));
}

void MultiWorldListPage::mceditState(LoggedProcess::State state)
{
    bool failed = false;
    switch (state) {
        case LoggedProcess::NotRunning:
        case LoggedProcess::Starting:
            return;
        case LoggedProcess::FailedToStart:
        case LoggedProcess::Crashed:
        case LoggedProcess::Aborted: {
            failed = true;
        }
        /* fallthrough */
        case LoggedProcess::Running:
        case LoggedProcess::Finished: {
            m_mceditStarting = false;
            break;
        }
    }
    if (failed) {
        mceditError();
    }
}

void MultiWorldListPage::worldChanged([[maybe_unused]] const QModelIndex& current, [[maybe_unused]] const QModelIndex& previous)
{
    QModelIndex index = getSelectedWorld();
    bool enable = index.isValid();
    ui->actionCopy_Seed->setEnabled(enable);
    ui->actionMCEdit->setEnabled(enable);
    ui->actionRemove->setEnabled(enable);
    ui->actionCopy->setEnabled(enable);
    ui->actionRename->setEnabled(enable);
    ui->actionData_Packs->setEnabled(enable);
    ui->actionView_Folder->setEnabled(enable);
    ui->actionJoin->setEnabled(enable);
    ui->actionInstance_Settings->setEnabled(enable);
    bool hasIcon = !index.data(MultiWorldList::IconFileRole).isNull();
    ui->actionReset_Icon->setEnabled(enable && hasIcon);
}

void MultiWorldListPage::on_actionAdd_triggered()
{
    auto list = GuiUtil::BrowseForFiles(displayName(), tr("Select a Minecraft world zip"), tr("Minecraft World Zip File") + " (*.zip)",
                                        QString(), this->parentWidget());
    if (!list.empty()) {
        m_worlds->stopWatching();
        for (auto filename : list) {
            auto *instance = selectInstance(tr("Select instance to add world '%1' to.").arg(QFileInfo(filename).fileName()));

            if (instance != nullptr) {
                m_worlds->installWorld(instance, QFileInfo(filename));
            }
        }
        m_worlds->startWatching();
    }
}

bool MultiWorldListPage::isWorldSafe(QModelIndex index)
{
    return !static_cast<InstanceWorld*>(m_worlds->data(index, MultiWorldList::ObjectRole).value<void*>())->instance->isRunning();
}

bool MultiWorldListPage::worldSafetyNagQuestion(const QString& actionType)
{
    if (!isWorldSafe(getSelectedWorld())) {
        auto result = QMessageBox::question(
            this, actionType, tr("Changing a world while Minecraft is running is potentially unsafe.\nDo you wish to proceed?"));
        if (result == QMessageBox::No) {
            return false;
        }
    }
    return true;
}

void MultiWorldListPage::on_actionCopy_triggered()
{
    QModelIndex index = getSelectedWorld();
    if (!index.isValid()) {
        return;
    }

    if (!worldSafetyNagQuestion(tr("Copy World")))
        return;

    auto worldVariant = m_worlds->data(index, MultiWorldList::ObjectRole);
    auto *world = static_cast<InstanceWorld*>(worldVariant.value<void*>());

    bool ok = false;
    QString name =
        QInputDialog::getText(this, tr("World name"), tr("Enter a new name for the copy."), QLineEdit::Normal, world->world.name(), &ok);

    if (ok && name.length() > 0) {
        auto *instance = selectInstance(tr("Select instance to copy world to."), world->instance);

        if (instance != nullptr) {
            world->world.install((QDir(instance->worldDir())).absolutePath(), name);
            m_worlds->update();
        }
    }
}

Q_DECLARE_METATYPE(BaseInstance*);

MinecraftInstance* MultiWorldListPage::selectInstance(const QString& message, BaseInstance* preselectedInstance)
{
    auto *dialog = new QDialog(this);
    dialog->setWindowTitle(tr("Select Instance"));

    dialog->resize(static_cast<int>(std::max(0.5 * window()->width(), 400.0)),
                   static_cast<int>(std::max(0.75 * window()->height(), 400.0)));
    dialog->restoreGeometry(QByteArray::fromBase64(APPLICATION->settings()->get("SelectInstanceGeometry").toByteArray()));

    auto layout = new QVBoxLayout(dialog);

    layout->addWidget(new QLabel(message));

    auto instanceList = new QListWidget(dialog);

    for (auto instance : m_worlds->getInstances()) {
        auto *item = new QListWidgetItem(instanceList);
        item->setText(instance->name());
        item->setIcon(APPLICATION->icons()->getIcon(instance->iconKey()));
        item->setData(Qt::UserRole, QVariant::fromValue(instance));
        if (instance == preselectedInstance) {
            instanceList->setCurrentItem(item);
        }
    }

    instanceList->sortItems(Qt::AscendingOrder);

    layout->addWidget(instanceList);

    auto buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel);
    connect(buttonBox, &QDialogButtonBox::rejected, dialog, &QDialog::reject);
    connect(buttonBox, &QDialogButtonBox::accepted, dialog, &QDialog::accept);
    layout->addWidget(buttonBox);

    dialog->setLayout(layout);

    connect(dialog, &QDialog::finished, this,
            [dialog]() { APPLICATION->settings()->set("SelectInstanceGeometry", dialog->saveGeometry().toBase64()); });

    if (dialog->exec() == QDialog::Accepted) {
        return static_cast<MinecraftInstance*>(instanceList->currentItem()->data(Qt::UserRole).value<BaseInstance*>());
    }

    return nullptr;
}

void MultiWorldListPage::on_actionRename_triggered()
{
    QModelIndex index = getSelectedWorld();
    if (!index.isValid()) {
        return;
    }

    if (!worldSafetyNagQuestion(tr("Rename World")))
        return;

    auto worldVariant = m_worlds->data(index, MultiWorldList::ObjectRole);
    auto world = (World*)worldVariant.value<void*>();

    bool ok = false;
    QString name = QInputDialog::getText(this, tr("World name"), tr("Enter a new world name."), QLineEdit::Normal, world->name(), &ok);

    if (ok && name.length() > 0) {
        world->rename(name);
    }
}

void MultiWorldListPage::on_actionInstance_Settings_triggered()
{
    QModelIndex index = getSelectedWorld();
    if (!index.isValid()) {
        return;
    }
    auto worldVariant = m_worlds->data(index, MultiWorldList::ObjectRole);
    auto *world = static_cast<InstanceWorld*>(worldVariant.value<void*>());

    if (world->instance->canEdit()) {
        APPLICATION->showInstanceWindow(world->instance);
    } else {
        CustomMessageBox::selectable(this, tr("Instance not editable"),
                                     tr("This instance is not editable. It may be broken, invalid, or too old. Check logs for details."),
                                     QMessageBox::Critical)
            ->show();
    }
}

void MultiWorldListPage::on_actionRefresh_triggered()
{
    m_worlds->update();
}

void MultiWorldListPage::worldDoubleClicked(const QModelIndex& index)
{
    auto proxy = (QSortFilterProxyModel*)ui->worldTreeView->model();
    join(proxy->mapToSource(index));
}

void MultiWorldListPage::fileDropped(const QFileInfo& worldInfo)
{
    auto *instance = selectInstance(tr("Select instance to add world '%1' to.").arg(worldInfo.fileName()));

    if (instance != nullptr) {
        if (!QDir(instance->worldDir()).entryInfoList().contains(worldInfo)) {
            m_worlds->installWorld(instance, worldInfo);
        }
    }
}

void MultiWorldListPage::on_actionJoin_triggered()
{
    QModelIndex index = getSelectedWorld();
    join(index);
}

void MultiWorldListPage::join(QModelIndex index)
{
    if (!index.isValid()) {
        return;
    }
    auto worldVariant = m_worlds->data(index, MultiWorldList::ObjectRole);
    auto *world = static_cast<InstanceWorld*>(worldVariant.value<void*>());
    APPLICATION->launch(world->instance, LaunchMode::Normal, std::make_shared<MinecraftTarget>(MinecraftTarget::parse(world->world.folderName(), true)));
    emit worldJoined();
}

#include "MultiWorldListPage.moc"