1 files changed, 66 insertions, 0 deletions
diff --git a/launcher/pages/modplatform/atlauncher/AtlOptionalModDialog.h b/launcher/pages/modplatform/atlauncher/AtlOptionalModDialog.h
new file mode 100644
index 00000000..a1df43f6
--- /dev/null
+++ b/launcher/pages/modplatform/atlauncher/AtlOptionalModDialog.h
@@ -0,0 +1,66 @@
+#pragma once
+
+#include <QDialog>
+#include <QAbstractListModel>
+
+#include "modplatform/atlauncher/ATLPackIndex.h"
+
+namespace Ui {
+class AtlOptionalModDialog;
+}
+
+class AtlOptionalModListModel : public QAbstractListModel {
+    Q_OBJECT
+
+public:
+    enum Columns
+    {
+        EnabledColumn = 0,
+        NameColumn,
+        DescriptionColumn,
+    };
+
+    AtlOptionalModListModel(QWidget *parent, QVector<ATLauncher::VersionMod> mods);
+
+    QVector<QString> getResult();
+
+    int rowCount(const QModelIndex &parent) const override;
+    int columnCount(const QModelIndex &parent) const override;
+
+    QVariant data(const QModelIndex &index, int role) const override;
+    bool setData(const QModelIndex &index, const QVariant &value, int role) override;
+    QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
+
+    Qt::ItemFlags flags(const QModelIndex &index) const override;
+
+public slots:
+    void selectRecommended();
+    void clearAll();
+
+private:
+    void toggleMod(ATLauncher::VersionMod mod, int index);
+    void setMod(ATLauncher::VersionMod mod, int index, bool enable, bool shouldEmit = true);
+
+private:
+    QVector<ATLauncher::VersionMod> m_mods;
+    QMap<QString, bool> m_selection;
+    QMap<QString, int> m_index;
+    QMap<QString, QVector<QString>> m_dependants;
+};
+
+class AtlOptionalModDialog : public QDialog {
+    Q_OBJECT
+
+public:
+    AtlOptionalModDialog(QWidget *parent, QVector<ATLauncher::VersionMod> mods);
+    ~AtlOptionalModDialog() override;
+
+    QVector<QString> getResult() {
+        return listModel->getResult();
+    }
+
+private:
+    Ui::AtlOptionalModDialog *ui;
+
+    AtlOptionalModListModel *listModel;
+};