From 550c65d6f536c04edce4f384b989ca4c59f262d1 Mon Sep 17 00:00:00 2001
From: Flashfyre <flashfireex@gmail.com>
Date: Sun, 4 Feb 2024 00:49:57 -0500
Subject: [PATCH] Fix save data not loading on login

---
 src/battle-phases.ts    | 8 ++++++--
 src/system/game-data.ts | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/battle-phases.ts b/src/battle-phases.ts
index e1da7928c3c..c4d3513f84b 100644
--- a/src/battle-phases.ts
+++ b/src/battle-phases.ts
@@ -69,19 +69,23 @@ export class LoginPhase extends BattlePhase {
           this.scene.ui.showText('Log in or create an account to start. No email required!');
   
         this.scene.playSound('menu_open');
+
+        const loadData = () => {
+          this.scene.gameData.loadSystem().then(() => this.end());
+        };
     
         this.scene.ui.setMode(Mode.LOGIN_FORM, {
           buttonActions: [
             () => {
               this.scene.ui.playSelect();
-              this.end();
+              loadData();
             }, () => {
               this.scene.playSound('menu_open');
               this.scene.ui.setMode(Mode.REGISTRATION_FORM, {
                 buttonActions: [
                   () => {
                     this.scene.ui.playSelect();
-                    this.end();
+                    loadData();
                   }, () => {
                     this.scene.unshiftPhase(new LoginPhase(this.scene, false));
                     this.end();
diff --git a/src/system/game-data.ts b/src/system/game-data.ts
index b9596744ecc..5f0b3b85c38 100644
--- a/src/system/game-data.ts
+++ b/src/system/game-data.ts
@@ -220,7 +220,7 @@ export class GameData {
     });
   }
 
-  private loadSystem(): Promise<boolean> {
+  public loadSystem(): Promise<boolean> {
     return new Promise<boolean>(resolve => {
       if (bypassLogin && !localStorage.hasOwnProperty('data'))
         return false;