1 files changed, 54 insertions, 0 deletions
diff --git a/launcher/modplatform/EnsureMetadataTask.h b/launcher/modplatform/EnsureMetadataTask.h
new file mode 100644
index 00000000..79db6976
--- /dev/null
+++ b/launcher/modplatform/EnsureMetadataTask.h
@@ -0,0 +1,54 @@
+#pragma once
+
+#include "ModIndex.h"
+#include "tasks/SequentialTask.h"
+#include "net/NetJob.h"
+
+class Mod;
+class QDir;
+class MultipleOptionsTask;
+
+class EnsureMetadataTask : public Task {
+    Q_OBJECT
+
+   public:
+    EnsureMetadataTask(Mod*, QDir, ModPlatform::Provider = ModPlatform::Provider::MODRINTH);
+    EnsureMetadataTask(QList<Mod*>&, QDir, ModPlatform::Provider = ModPlatform::Provider::MODRINTH);
+
+    ~EnsureMetadataTask() = default;
+
+   public slots:
+    bool abort() override;
+   protected slots:
+    void executeTask() override;
+
+   private:
+    // FIXME: Move to their own namespace
+    auto modrinthVersionsTask() -> NetJob::Ptr;
+    auto modrinthProjectsTask() -> NetJob::Ptr;
+
+    auto flameVersionsTask() -> NetJob::Ptr;
+    auto flameProjectsTask() -> NetJob::Ptr;
+
+    // Helpers
+    void emitReady(Mod*);
+    void emitFail(Mod*);
+
+    auto getHash(Mod*) -> QString;
+
+   private slots:
+    void modrinthCallback(ModPlatform::IndexedPack& pack, ModPlatform::IndexedVersion& ver, Mod*);
+    void flameCallback(ModPlatform::IndexedPack& pack, ModPlatform::IndexedVersion& ver, Mod*);
+
+   signals:
+    void metadataReady(Mod*);
+    void metadataFailed(Mod*);
+
+   private:
+    QHash<QString, Mod*> m_mods;
+    QDir m_index_dir;
+    ModPlatform::Provider m_provider;
+
+    QHash<QString, ModPlatform::IndexedVersion> m_temp_versions;
+    NetJob* m_current_task;
+};