1 files changed, 148 insertions, 0 deletions
diff --git a/launcher/ui/dialogs/InstallLoaderDialog.cpp b/launcher/ui/dialogs/InstallLoaderDialog.cpp
new file mode 100644
index 00000000..6e1ad1c0
--- /dev/null
+++ b/launcher/ui/dialogs/InstallLoaderDialog.cpp
@@ -0,0 +1,148 @@
+// SPDX-License-Identifier: GPL-3.0-only
+/*
+ *  Prism Launcher - Minecraft Launcher
+ *  Copyright (C) 2023 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/>.
+ */
+
+#include "InstallLoaderDialog.h"
+
+#include <QDialogButtonBox>
+#include <QPushButton>
+#include <QVBoxLayout>
+#include "Application.h"
+#include "BuildConfig.h"
+#include "DesktopServices.h"
+#include "meta/Index.h"
+#include "minecraft/MinecraftInstance.h"
+#include "minecraft/PackProfile.h"
+#include "ui/widgets/PageContainer.h"
+#include "ui/widgets/VersionSelectWidget.h"
+
+class LoaderPage : public VersionSelectWidget, public BasePage {
+   public:
+    LoaderPage(const QString& id,
+               const QString& icon,
+               const QString& name,
+               // "lightweight" loaders are independent to any game version
+               const bool lightweight,
+               const std::shared_ptr<PackProfile> profile,
+               QWidget* parent = nullptr)
+        : VersionSelectWidget(parent), m_id(id), m_icon(icon), m_name(name)
+    {
+        const QString minecraftVersion = profile->getComponentVersion("net.minecraft");
+        setEmptyString(tr("No versions are currently available for Minecraft %1").arg(minecraftVersion));
+        if (!lightweight)
+            setExactFilter(BaseVersionList::ParentVersionRole, minecraftVersion);
+
+        if (const QString currentVersion = profile->getComponentVersion(id); !currentVersion.isNull())
+            setCurrentVersion(currentVersion);
+    }
+
+    QString id() const override { return m_id; }
+    QString displayName() const override { return m_name; }
+    QIcon icon() const override { return APPLICATION->getThemedIcon(m_icon); }
+
+    void openedImpl() override
+    {
+        if (m_loaded)
+            return;
+
+        const auto versions = APPLICATION->metadataIndex()->get(m_id);
+        if (!versions)
+            return;
+
+        initialize(versions.get());
+        m_loaded = true;
+    }
+
+   private:
+    const QString m_id;
+    const QString m_icon;
+    const QString m_name;
+    bool m_loaded = false;
+};
+
+InstallLoaderDialog::InstallLoaderDialog(std::shared_ptr<PackProfile> profile, QWidget* parent)
+    : QDialog(parent), m_profile(profile), m_container(new PageContainer(this)), m_buttons(new QDialogButtonBox(this))
+{
+    auto layout = new QVBoxLayout(this);
+
+    m_container->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Expanding);
+    layout->addWidget(m_container);
+
+    auto buttonLayout = new QHBoxLayout(this);
+
+    auto refreshButton = new QPushButton(tr("&Refresh"), this);
+    connect(refreshButton, &QPushButton::pressed, this, [this] {
+        LoaderPage* page = dynamic_cast<LoaderPage*>(m_container->selectedPage());
+        Q_ASSERT(page != nullptr);
+        page->loadList();
+    });
+    buttonLayout->addWidget(refreshButton);
+
+    m_buttons->setOrientation(Qt::Horizontal);
+    m_buttons->setStandardButtons(QDialogButtonBox::Cancel | QDialogButtonBox::Ok);
+    connect(m_buttons, &QDialogButtonBox::accepted, this, &QDialog::accept);
+    connect(m_buttons, &QDialogButtonBox::rejected, this, &QDialog::reject);
+    buttonLayout->addWidget(m_buttons);
+
+    layout->addLayout(buttonLayout);
+
+    setWindowTitle(dialogTitle());
+    resize(650, 400);
+
+    connect(m_container, &PageContainer::selectedPageChanged, this,
+            [this](BasePage* previous, BasePage* selected) { updateAcceptButton(selected); });
+    updateAcceptButton(m_container->selectedPage());
+}
+
+QList<BasePage*> InstallLoaderDialog::getPages()
+{
+    return { // Forge
+             new LoaderPage("net.minecraftforge", "forge", tr("Forge"), false, m_profile, this),
+             // Fabric
+             new LoaderPage("net.fabricmc.fabric-loader", "fabricmc-small", tr("Fabric"), true, m_profile, this),
+             // Quilt
+             new LoaderPage("org.quiltmc.quilt-loader", "quiltmc", tr("Quilt"), true, m_profile, this),
+             // LiteLoader
+             new LoaderPage("com.mumfrey.liteloader", "liteloader", tr("LiteLoader"), false, m_profile, this)
+    };
+}
+
+QString InstallLoaderDialog::dialogTitle()
+{
+    return tr("Install Loader");
+}
+
+void InstallLoaderDialog::updateAcceptButton(const BasePage* page)
+{
+    auto installed = !m_profile->getComponentVersion(page->id()).isNull();
+    m_buttons->button(QDialogButtonBox::Ok)->setText(installed ? tr("&Update") : tr("&Install"));
+}
+
+void InstallLoaderDialog::done(int result)
+{
+    if (result == Accepted) {
+        LoaderPage* page = dynamic_cast<LoaderPage*>(m_container->selectedPage());
+        Q_ASSERT(page != nullptr);
+
+        if (page->selectedVersion()) {
+            m_profile->setComponentVersion(page->id(), page->selectedVersion()->descriptor());
+            m_profile->resolve(Net::Mode::Online);
+        }
+    }
+
+    QDialog::done(result);
+}