1 files changed, 107 insertions, 16 deletions
diff --git a/launcher/ui/pages/modplatform/modrinth/ModrinthModel.h b/launcher/ui/pages/modplatform/modrinth/ModrinthModel.h
index 45a6090a..14aa6747 100644
--- a/launcher/ui/pages/modplatform/modrinth/ModrinthModel.h
+++ b/launcher/ui/pages/modplatform/modrinth/ModrinthModel.h
@@ -1,25 +1,116 @@
+// SPDX-License-Identifier: GPL-3.0-only
+/*
+ *  PolyMC - Minecraft Launcher
+ *  Copyright (c) 2022 flowln <flowlnlnln@gmail.com>
+ *
+ *  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.
+ */
+
 #pragma once
 
-#include "ModrinthPage.h"
+#include <QAbstractListModel>
+
+#include "modplatform/modrinth/ModrinthPackManifest.h"
+#include "ui/pages/modplatform/modrinth/ModrinthPage.h"
+
+class ModPage;
+class Version;
 
 namespace Modrinth {
 
-class ListModel : public ModPlatform::ListModel {
+using LogoMap = QMap<QString, QIcon>;
+using LogoCallback = std::function<void (QString)>;
+
+class ModpackListModel : public QAbstractListModel {
     Q_OBJECT
 
    public:
-    ListModel(ModrinthPage* parent) : ModPlatform::ListModel(parent){};
-    ~ListModel() override = default;
-
-   private:
-    void loadIndexedPack(ModPlatform::IndexedPack& m, QJsonObject& obj) override;
-    void loadIndexedPackVersions(ModPlatform::IndexedPack& m, QJsonArray& arr) override;
-    
-    auto documentToArray(QJsonDocument& obj) const -> QJsonArray override;
-
-    // NOLINTNEXTLINE(modernize-avoid-c-arrays)
-    static const char* sorts[5];
-    inline auto getSorts() const -> const char** override { return sorts; };
-};
+    ModpackListModel(ModrinthPage* parent);
+    ~ModpackListModel() override = default;
+
+    inline auto rowCount(const QModelIndex& parent) const -> int override { return modpacks.size(); };
+    inline auto columnCount(const QModelIndex& parent) const -> int override { return 1; };
+    inline auto flags(const QModelIndex& index) const -> Qt::ItemFlags override { return QAbstractListModel::flags(index); };
+
+    auto debugName() const -> QString;
+
+    /* Retrieve information from the model at a given index with the given role */
+    auto data(const QModelIndex& index, int role) const -> QVariant override;
+
+    inline void setActiveJob(NetJob::Ptr ptr) { jobPtr = ptr; }
+
+    /* Ask the API for more information */
+    void fetchMore(const QModelIndex& parent) override;
+    void refresh();
+    void searchWithTerm(const QString& term, const int sort);
+
+    void getLogo(const QString& logo, const QString& logoUrl, LogoCallback callback);
+
+    inline auto canFetchMore(const QModelIndex& parent) const -> bool override { return searchState == CanPossiblyFetchMore; };
+
+   public slots:
+    void searchRequestFinished(QJsonDocument& doc_all);
+    void searchRequestFailed(QString reason);
 
-}  // namespace Modrinth
+   protected slots:
+
+    void logoFailed(QString logo);
+    void logoLoaded(QString logo, QIcon out);
+
+    void performPaginatedSearch();
+
+   protected:
+    void requestLogo(QString file, QString url);
+
+    inline auto getMineVersions() const -> std::list<Version>;
+
+   protected:
+    ModrinthPage* m_parent;
+
+    QList<Modrinth::Modpack> modpacks;
+
+    LogoMap m_logoMap;
+    QMap<QString, LogoCallback> waitingCallbacks;
+    QStringList m_failedLogos;
+    QStringList m_loadingLogos;
+
+    QString currentSearchTerm;
+    QString currentSort;
+    int nextSearchOffset = 0;
+    enum SearchState { None, CanPossiblyFetchMore, ResetRequested, Finished } searchState = None;
+
+    NetJob::Ptr jobPtr;
+
+    QByteArray m_all_response;
+    QByteArray m_specific_response;
+
+    int m_modpacks_per_page = 20;
+};
+}  // namespace ModPlatform