1 files changed, 57 insertions, 18 deletions

diff --git a/launcher/minecraft/auth/AccountTask.cpp b/launcher/minecraft/auth/AccountTask.cpp
index 25d753de..98d8d94d 100644
--- a/launcher/minecraft/auth/AccountTask.cpp
+++ b/launcher/minecraft/auth/AccountTask.cpp
@@ -28,40 +28,79 @@
 AccountTask::AccountTask(AccountData *data, QObject *parent)
     : Task(parent), m_data(data)
 {
-    changeState(STATE_CREATED);
+    changeState(AccountTaskState::STATE_CREATED);
 }
 
 QString AccountTask::getStateMessage() const
 {
-    switch (m_accountState)
+    switch (m_taskState)
     {
-    case STATE_CREATED:
+    case AccountTaskState::STATE_CREATED:
         return "Waiting...";
-    case STATE_WORKING:
+    case AccountTaskState::STATE_WORKING:
         return tr("Sending request to auth servers...");
-    case STATE_SUCCEEDED:
+    case AccountTaskState::STATE_SUCCEEDED:
         return tr("Authentication task succeeded.");
-    case STATE_FAILED_SOFT:
+    case AccountTaskState::STATE_OFFLINE:
         return tr("Failed to contact the authentication server.");
-    case STATE_FAILED_HARD:
-        return tr("Failed to authenticate.");
-    case STATE_FAILED_GONE:
+    case AccountTaskState::STATE_FAILED_SOFT:
+        return tr("Encountered an error during authentication.");
+    case AccountTaskState::STATE_FAILED_HARD:
+        return tr("Failed to authenticate. The session has expired.");
+    case AccountTaskState::STATE_FAILED_GONE:
         return tr("Failed to authenticate. The account no longer exists.");
     default:
         return tr("...");
     }
 }
 
-void AccountTask::changeState(AccountTask::State newState, QString reason)
+bool AccountTask::changeState(AccountTaskState newState, QString reason)
 {
-    m_accountState = newState;
+    m_taskState = newState;
     setStatus(getStateMessage());
-    if (newState == STATE_SUCCEEDED)
-    {
-        emitSucceeded();
-    }
-    else if (newState == STATE_FAILED_HARD || newState == STATE_FAILED_SOFT || newState == STATE_FAILED_GONE)
-    {
-        emitFailed(reason);
+    switch(newState) {
+        case AccountTaskState::STATE_CREATED: {
+            m_data->errorString.clear();
+            return true;
+        }
+        case AccountTaskState::STATE_WORKING: {
+            m_data->accountState = AccountState::Working;
+            return true;
+        }
+        case AccountTaskState::STATE_SUCCEEDED: {
+            m_data->accountState = AccountState::Online;
+            emitSucceeded();
+            return false;
+        }
+        case AccountTaskState::STATE_OFFLINE: {
+            m_data->errorString = reason;
+            m_data->accountState = AccountState::Offline;
+            emitFailed(reason);
+            return false;
+        }
+        case AccountTaskState::STATE_FAILED_SOFT: {
+            m_data->errorString = reason;
+            m_data->accountState = AccountState::Errored;
+            emitFailed(reason);
+            return false;
+        }
+        case AccountTaskState::STATE_FAILED_HARD: {
+            m_data->errorString = reason;
+            m_data->accountState = AccountState::Expired;
+            emitFailed(reason);
+            return false;
+        }
+        case AccountTaskState::STATE_FAILED_GONE: {
+            m_data->errorString = reason;
+            m_data->accountState = AccountState::Gone;
+            emitFailed(reason);
+            return false;
+        }
+        default: {
+            QString error = tr("Unknown account task state: %1").arg(int(newState));
+            m_data->accountState = AccountState::Errored;
+            emitFailed(error);
+            return false;
+        }
     }
 }