1 files changed, 94 insertions, 0 deletions
diff --git a/launcher/minecraft/auth/flows/AuthContext.h b/launcher/minecraft/auth/flows/AuthContext.h
new file mode 100644
index 00000000..5f99dba3
--- /dev/null
+++ b/launcher/minecraft/auth/flows/AuthContext.h
@@ -0,0 +1,94 @@
+#pragma once
+
+#include <QObject>
+#include <QList>
+#include <QVector>
+#include <QNetworkReply>
+#include <QImage>
+
+#include <katabasis/OAuth2.h>
+#include "Yggdrasil.h"
+#include "../AccountData.h"
+#include "../AccountTask.h"
+
+class AuthContext : public AccountTask
+{
+    Q_OBJECT
+
+public:
+    explicit AuthContext(AccountData * data, QObject *parent = 0);
+
+    bool isBusy() {
+        return m_activity != Katabasis::Activity::Idle;
+    };
+    Katabasis::Validity validity() {
+        return m_data->validity_;
+    };
+
+    //bool signOut();
+
+    QString getStateMessage() const override;
+
+signals:
+    void activityChanged(Katabasis::Activity activity);
+
+private slots:
+// OAuth-specific callbacks
+    void onOAuthLinkingSucceeded();
+    void onOAuthLinkingFailed();
+    void onOpenBrowser(const QUrl &url);
+    void onCloseBrowser();
+    void onOAuthActivityChanged(Katabasis::Activity activity);
+
+// Yggdrasil specific callbacks
+    void onMojangSucceeded();
+    void onMojangFailed();
+
+protected:
+    void initMSA();
+    void initMojang();
+
+    void doUserAuth();
+    Q_SLOT void onUserAuthDone(int, QNetworkReply::NetworkError, QByteArray, QList<QNetworkReply::RawHeaderPair>);
+
+    void doSTSAuthMinecraft();
+    Q_SLOT void onSTSAuthMinecraftDone(int, QNetworkReply::NetworkError, QByteArray, QList<QNetworkReply::RawHeaderPair>);
+    void doMinecraftAuth();
+    Q_SLOT void onMinecraftAuthDone(int, QNetworkReply::NetworkError, QByteArray, QList<QNetworkReply::RawHeaderPair>);
+
+    void doSTSAuthGeneric();
+    Q_SLOT void onSTSAuthGenericDone(int, QNetworkReply::NetworkError, QByteArray, QList<QNetworkReply::RawHeaderPair>);
+    void doXBoxProfile();
+    Q_SLOT void onXBoxProfileDone(int, QNetworkReply::NetworkError, QByteArray, QList<QNetworkReply::RawHeaderPair>);
+
+    void doMinecraftProfile();
+    Q_SLOT void onMinecraftProfileDone(int, QNetworkReply::NetworkError, QByteArray, QList<QNetworkReply::RawHeaderPair>);
+
+    void doGetSkin();
+    Q_SLOT void onSkinDone(int, QNetworkReply::NetworkError, QByteArray, QList<QNetworkReply::RawHeaderPair>);
+
+    void checkResult();
+
+protected:
+    void beginActivity(Katabasis::Activity activity);
+    void finishActivity();
+    void clearTokens();
+
+protected:
+    Katabasis::OAuth2 *m_oauth2 = nullptr;
+    Yggdrasil *m_yggdrasil = nullptr;
+
+    int m_requestsDone = 0;
+    bool m_xboxProfileSucceeded = false;
+    bool m_mcAuthSucceeded = false;
+    Katabasis::Activity m_activity = Katabasis::Activity::Idle;
+    enum class MSAStage {
+        Idle,
+        UserAuth,
+        XboxAuth,
+        MinecraftProfile,
+        Skin
+    } m_stage = MSAStage::Idle;
+
+    QNetworkAccessManager *mgr = nullptr;
+};