chore: full restructure of the unity sample app

darrenmelvison1 · darrenmelvison1 · commit fb4e6cad7d72 · 2025-05-16T14:22:59.000+10:00
diff --git a/sample/Assets/Scenes/Passport/SelectAuthMethod.unity b/sample/Assets/Scenes/Passport/SelectAuthMethod.unity
@@ -1367,8 +1367,8 @@ MonoBehaviour:
   m_TargetGraphic: {fileID: 167431872}
   m_HandleRect: {fileID: 167431871}
   m_Direction: 2
-  m_Value: 0
-  m_Size: 1
+  m_Value: 1
+  m_Size: 0.99999994
   m_NumberOfSteps: 0
   m_OnValueChanged:
     m_PersistentCalls:
@@ -1717,8 +1717,8 @@ RectTransform:
   m_LocalEulerAnglesHint: {x: 0, y: 0, z: 0}
   m_AnchorMin: {x: 0, y: 0}
   m_AnchorMax: {x: 0, y: 0}
-  m_AnchoredPosition: {x: 845.9838, y: 0}
-  m_SizeDelta: {x: 1651.9677, y: 0}
+  m_AnchoredPosition: {x: 797.1642, y: 0}
+  m_SizeDelta: {x: 1554.3284, y: 0}
   m_Pivot: {x: 0.5, y: 0.5}
 --- !u!114 &1661390145
 MonoBehaviour:
@@ -1812,8 +1812,8 @@ RectTransform:
   m_LocalEulerAnglesHint: {x: 0, y: 0, z: 0}
   m_AnchorMin: {x: 0, y: 0}
   m_AnchorMax: {x: 0, y: 0}
-  m_AnchoredPosition: {x: 845.9838, y: 0}
-  m_SizeDelta: {x: 1651.9677, y: 0}
+  m_AnchoredPosition: {x: 797.1642, y: 0}
+  m_SizeDelta: {x: 1554.3284, y: 0}
   m_Pivot: {x: 0.5, y: 0.5}
 --- !u!114 &1688502349
 MonoBehaviour:
diff --git a/sample/Assets/Scenes/Passport/UnauthenticatedScene.unity b/sample/Assets/Scenes/Passport/UnauthenticatedScene.unity
@@ -765,10 +765,9 @@ MonoBehaviour:
   m_OnClick:
     m_PersistentCalls:
       m_Calls:
-      - m_Target: {fileID: 333006419}
-        m_TargetAssemblyTypeName: Immutable.Passport.Sample.PassportFeatures.ZkEvmConnectScript,
-          Assembly-CSharp
-        m_MethodName: ConnectZkEvm
+      - m_Target: {fileID: 333006421}
+        m_TargetAssemblyTypeName: ImxConnectScript, Assembly-CSharp
+        m_MethodName: ConnectImx
         m_Mode: 1
         m_Arguments:
           m_ObjectArgument: {fileID: 0}
@@ -924,28 +923,14 @@ GameObject:
   serializedVersion: 6
   m_Component:
   - component: {fileID: 333006420}
-  - component: {fileID: 333006419}
+  - component: {fileID: 333006421}
   m_Layer: 0
   m_Name: Connect
   m_TagString: Untagged
   m_Icon: {fileID: 0}
   m_NavMeshLayer: 0
   m_StaticEditorFlags: 0
   m_IsActive: 1
---- !u!114 &333006419
-MonoBehaviour:
-  m_ObjectHideFlags: 0
-  m_CorrespondingSourceObject: {fileID: 0}
-  m_PrefabInstance: {fileID: 0}
-  m_PrefabAsset: {fileID: 0}
-  m_GameObject: {fileID: 333006418}
-  m_Enabled: 1
-  m_EditorHideFlags: 0
-  m_Script: {fileID: 11500000, guid: 033bac63c98e94dbd8324bca7d37ed2b, type: 3}
-  m_Name: 
-  m_EditorClassIdentifier: 
-  connectButton: {fileID: 291915222}
-  output: {fileID: 1615533291}
 --- !u!4 &333006420
 Transform:
   m_ObjectHideFlags: 0
@@ -961,6 +946,19 @@ Transform:
   m_Father: {fileID: 0}
   m_RootOrder: 8
   m_LocalEulerAnglesHint: {x: 0, y: 0, z: 0}
+--- !u!114 &333006421
+MonoBehaviour:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 333006418}
+  m_Enabled: 1
+  m_EditorHideFlags: 0
+  m_Script: {fileID: 11500000, guid: 8f90530ca8e134504b30c2444bf20a66, type: 3}
+  m_Name: 
+  m_EditorClassIdentifier: 
+  Output: {fileID: 1615533291}
 --- !u!1 &495074501
 GameObject:
   m_ObjectHideFlags: 0
@@ -1045,7 +1043,7 @@ MonoBehaviour:
   m_TargetGraphic: {fileID: 1541372192}
   m_HandleRect: {fileID: 1541372191}
   m_Direction: 0
-  m_Value: 0
+  m_Value: 1
   m_Size: 1
   m_NumberOfSteps: 0
   m_OnValueChanged:
@@ -2679,8 +2677,8 @@ MonoBehaviour:
   m_TargetGraphic: {fileID: 1495172582}
   m_HandleRect: {fileID: 1495172581}
   m_Direction: 2
-  m_Value: 0
-  m_Size: 1
+  m_Value: 1
+  m_Size: 0.99999994
   m_NumberOfSteps: 0
   m_OnValueChanged:
     m_PersistentCalls:
diff --git a/sample/Assets/Scripts/Passport/AuthenticatedSceneManager.cs b/sample/Assets/Scripts/Passport/AuthenticatedSceneManager.cs
@@ -1,5 +1,6 @@
 using UnityEngine;
 using UnityEngine.UI;
+using System;
 
 public class AuthenticatedSceneManager : MonoBehaviour
 {
@@ -18,6 +19,13 @@ public class AuthenticatedSceneManager : MonoBehaviour
     [SerializeField] private Button GetTransactionReceiptButton;
     [SerializeField] private Button SignTypedDataButton;
 
+    public Action OnImxConnected;
+
+    void Awake()
+    {
+        OnImxConnected = () => { UpdateImxButtonStates(); };
+    }
+
     void Start()
     {
         UpdateImxButtonStates();
diff --git a/sample/Assets/Scripts/Passport/DeviceCodeLogin/DeviceCodeLoginScript.cs b/sample/Assets/Scripts/Passport/DeviceCodeLogin/DeviceCodeLoginScript.cs
@@ -23,7 +23,7 @@ public void StartDeviceCodeLogin()
                 Debug.LogWarning("[DeviceCodeLoginScript] Passport.Instance is null. Initialization must complete before device code login.");
                 return;
             }
-            StartDeviceCodeLoginAsync().Forget();
+            StartDeviceCodeLoginAsync();
         }
 
         private async UniTaskVoid StartDeviceCodeLoginAsync()
diff --git a/sample/Assets/Scripts/Passport/GetAccessToken/GetAccessTokenScript.cs b/sample/Assets/Scripts/Passport/GetAccessToken/GetAccessTokenScript.cs
@@ -12,7 +12,7 @@ public class GetAccessTokenScript : MonoBehaviour
     /// </summary>
     public void GetAccessToken()
     {
-        GetAccessTokenAsync().Forget();
+        GetAccessTokenAsync();
     }
 
     private async UniTaskVoid GetAccessTokenAsync()
diff --git a/sample/Assets/Scripts/Passport/GetIdToken/GetIdTokenScript.cs b/sample/Assets/Scripts/Passport/GetIdToken/GetIdTokenScript.cs
@@ -12,7 +12,7 @@ public class GetIdTokenScript : MonoBehaviour
     /// </summary>
     public void GetIdToken()
     {
-        GetIdTokenAsync().Forget();
+        GetIdTokenAsync();
     }
 
     private async UniTaskVoid GetIdTokenAsync()
diff --git a/sample/Assets/Scripts/Passport/GetLinkedAddresses/GetLinkedAddressesScript.cs b/sample/Assets/Scripts/Passport/GetLinkedAddresses/GetLinkedAddressesScript.cs
@@ -13,20 +13,26 @@ public class GetLinkedAddressesScript : MonoBehaviour
     /// </summary>
     public void GetLinkedAddresses()
     {
-        GetLinkedAddressesAsync().Forget();
+        Debug.Log("E2E TEST DEBUG: GetLinkedAddresses() called (public method)");
+        GetLinkedAddressesAsync();
     }
 
     private async UniTaskVoid GetLinkedAddressesAsync()
     {
+        Debug.Log("E2E TEST DEBUG: GetLinkedAddressesAsync() started");
         if (Passport.Instance == null)
         {
+            Debug.Log("E2E TEST DEBUG: Passport instance is null in GetLinkedAddressesAsync");
             ShowOutput("Passport instance is null");
             return;
         }
         try
         {
+            Debug.Log("E2E TEST DEBUG: About to await Passport.Instance.GetLinkedAddresses()");
             List<string> addresses = await Passport.Instance.GetLinkedAddresses();
-            ShowOutput(addresses.Count > 0 ? string.Join(", ", addresses) : "No linked addresses");
+            string outputMessage = addresses.Count > 0 ? string.Join(", ", addresses) : "No linked addresses";
+            Debug.Log($"E2E TEST DEBUG: Got addresses, output will be: {outputMessage}");
+            ShowOutput(outputMessage);
         }
         catch (System.Exception ex)
         {
diff --git a/sample/Assets/Scripts/Passport/GetUserInfo/GetUserInfoScript.cs b/sample/Assets/Scripts/Passport/GetUserInfo/GetUserInfoScript.cs
@@ -12,7 +12,7 @@ public class GetUserInfoScript : MonoBehaviour
     /// </summary>
     public void GetEmail()
     {
-        GetEmailAsync().Forget();
+        GetEmailAsync();
     }
 
     private async UniTaskVoid GetEmailAsync()
@@ -38,7 +38,7 @@ private async UniTaskVoid GetEmailAsync()
     /// </summary>
     public void GetPassportId()
     {
-        GetPassportIdAsync().Forget();
+        GetPassportIdAsync();
     }
 
     private async UniTaskVoid GetPassportIdAsync()
diff --git a/sample/Assets/Scripts/Passport/ImxConnect/ImxConnectScript.cs b/sample/Assets/Scripts/Passport/ImxConnect/ImxConnectScript.cs
@@ -12,7 +12,7 @@ public class ImxConnectScript : MonoBehaviour
     /// </summary>
     public void ConnectImx()
     {
-        ConnectImxAsync().Forget();
+        ConnectImxAsync();
     }
 
     private async UniTaskVoid ConnectImxAsync()
@@ -29,20 +29,24 @@ private async UniTaskVoid ConnectImxAsync()
         {
             await Passport.Instance.ConnectImx();
             
-            // Add these lines to update connection state and refresh UI
             SampleAppManager.IsConnectedToImx = true;
-            var sceneManager = FindObjectOfType<AuthenticatedSceneManager>();
-            if (sceneManager != null)
+            ShowOutput("Connected to IMX"); // Show success early
+
+            // Try to find UnauthenticatedSceneManager first
+            var unauthSceneManager = FindObjectOfType<UnauthenticatedSceneManager>();
+            if (unauthSceneManager != null)
             {
-                sceneManager.UpdateImxButtonStates();
-                Debug.Log("Updated IMX button states after connection");
+                unauthSceneManager.OnImxConnected?.Invoke();
+                return; 
             }
-            else
+
+            var authSceneManager = FindObjectOfType<AuthenticatedSceneManager>();
+            if (authSceneManager != null)
             {
-                Debug.LogWarning("Could not find AuthenticatedSceneManager to update button states");
+                authSceneManager.UpdateImxButtonStates();
+                authSceneManager.OnImxConnected?.Invoke(); 
+                return;
             }
-            
-            ShowOutput("Connected to IMX");
         }
         catch (System.Exception ex)
         {
diff --git a/sample/Assets/Scripts/Passport/ImxIsRegisteredOffchain/ImxIsRegisteredOffchainScript.cs b/sample/Assets/Scripts/Passport/ImxIsRegisteredOffchain/ImxIsRegisteredOffchainScript.cs
@@ -14,7 +14,7 @@ public class ImxIsRegisteredOffchainScript : MonoBehaviour
 
         public void CheckIsRegisteredOffchain()
         {
-            CheckIsRegisteredOffchainAsync().Forget();
+            CheckIsRegisteredOffchainAsync();
         }
 
         private async UniTaskVoid CheckIsRegisteredOffchainAsync()
diff --git a/sample/Assets/Scripts/Passport/ImxNftTransfer/ImxNftTransferScript.cs b/sample/Assets/Scripts/Passport/ImxNftTransfer/ImxNftTransferScript.cs
@@ -21,7 +21,7 @@ public class ImxNftTransferScript : MonoBehaviour
     /// </summary>
     public void Transfer()
     {
-        TransferAsync().Forget();
+        TransferAsync();
     }
 
     private async UniTaskVoid TransferAsync()
diff --git a/sample/Assets/Scripts/Passport/Logout/LogoutScript.cs b/sample/Assets/Scripts/Passport/Logout/LogoutScript.cs
@@ -10,7 +10,7 @@ public class LogoutScript : MonoBehaviour
     /// </summary>
     public void Logout()
     {
-        LogoutAsync().Forget();
+        LogoutAsync();
     }
 
     private async UniTaskVoid LogoutAsync()
diff --git a/sample/Assets/Scripts/Passport/PkceAuth/PkceAuthScript.cs b/sample/Assets/Scripts/Passport/PkceAuth/PkceAuthScript.cs
@@ -22,7 +22,7 @@ public void StartPkceAuth()
                 Debug.LogWarning("[PkceAuthScript] Passport.Instance is null. Initialization must complete before PKCE auth.");
                 return;
             }
-            StartPkceAuthAsync().Forget();
+            StartPkceAuthAsync();
         }
 
         private async UniTaskVoid StartPkceAuthAsync()
diff --git a/sample/Assets/Scripts/Passport/Reconnect/ReconnectScript.cs b/sample/Assets/Scripts/Passport/Reconnect/ReconnectScript.cs
@@ -13,7 +13,7 @@ public class ReconnectScript : MonoBehaviour
     /// </summary>
     public void Reconnect()
     {
-        ReconnectAsync().Forget();
+        ReconnectAsync();
     }
 
     private async UniTaskVoid ReconnectAsync()
diff --git a/sample/Assets/Scripts/Passport/Relogin/ReloginScript.cs b/sample/Assets/Scripts/Passport/Relogin/ReloginScript.cs
@@ -13,7 +13,7 @@ public class ReloginScript : MonoBehaviour
     /// </summary>
     public void Relogin()
     {
-        ReloginAsync().Forget();
+        ReloginAsync();
     }
 
     private async UniTaskVoid ReloginAsync()
diff --git a/sample/Assets/Scripts/Passport/UnauthenticatedSceneManager.cs b/sample/Assets/Scripts/Passport/UnauthenticatedSceneManager.cs
@@ -2,13 +2,16 @@
 using UnityEngine.UI;
 using Cysharp.Threading.Tasks;
 using Immutable.Passport;
+using System;
 
 public class UnauthenticatedSceneManager : MonoBehaviour
 {
     [SerializeField] private GameObject LoginButtons;
     [SerializeField] private GameObject ReloginButtons;
     [SerializeField] private InputField DeviceCodeTimeoutMs;
 
+    public Action OnImxConnected;
+
     private async void Start()
     {
         if (Passport.Instance != null)
@@ -23,4 +26,9 @@ private async void Start()
             Debug.LogError("[UnauthenticatedSceneManager] Passport.Instance is null");
         }
     }
+
+    void Awake()
+    {
+        OnImxConnected = () => { UnityEngine.SceneManagement.SceneManager.LoadScene("AuthenticatedScene"); };
+    }
 } 
diff --git a/sample/Assets/Scripts/Passport/ZkEvmConnect/ZkEvmConnectScript.cs b/sample/Assets/Scripts/Passport/ZkEvmConnect/ZkEvmConnectScript.cs
@@ -14,7 +14,7 @@ public class ZkEvmConnectScript : MonoBehaviour
 
         public void ConnectZkEvm()
         {
-            ConnectZkEvmAsync().Forget();
+            ConnectZkEvmAsync();
         }
 
         private async UniTaskVoid ConnectZkEvmAsync()
diff --git a/sample/Assets/Scripts/Passport/ZkEvmGetBalance/ZkEvmGetBalanceScript.cs b/sample/Assets/Scripts/Passport/ZkEvmGetBalance/ZkEvmGetBalanceScript.cs
@@ -13,7 +13,7 @@ public class ZkEvmGetBalanceScript : MonoBehaviour
 
     public void GetBalance()
     {
-        GetBalanceAsync().Forget();
+        GetBalanceAsync();
     }
 
     private async UniTaskVoid GetBalanceAsync()
diff --git a/sample/Assets/Scripts/Passport/ZkEvmGetTransactionReceipt/ZkEvmGetTransactionReceiptScript.cs b/sample/Assets/Scripts/Passport/ZkEvmGetTransactionReceipt/ZkEvmGetTransactionReceiptScript.cs
@@ -12,7 +12,7 @@ public class ZkEvmGetTransactionReceiptScript : MonoBehaviour
 
     public void GetZkEvmTransactionReceipt()
     {
-        GetZkEvmTransactionReceiptAsync().Forget();
+        GetZkEvmTransactionReceiptAsync();
     }
 
     private async UniTaskVoid GetZkEvmTransactionReceiptAsync()
diff --git a/sample/Assets/Scripts/Passport/ZkEvmRequestAccounts/ZkEvmRequestAccountsScript.cs b/sample/Assets/Scripts/Passport/ZkEvmRequestAccounts/ZkEvmRequestAccountsScript.cs
@@ -9,7 +9,7 @@ public class ZkEvmRequestAccountsScript : MonoBehaviour
 
     public void RequestAccounts()
     {
-        RequestAccountsAsync().Forget();
+        RequestAccountsAsync();
     }
 
     private async UniTaskVoid RequestAccountsAsync()
diff --git a/sample/Assets/Scripts/Passport/ZkEvmSendTransaction/ZkEvmSendTransactionScript.cs b/sample/Assets/Scripts/Passport/ZkEvmSendTransaction/ZkEvmSendTransactionScript.cs
@@ -31,7 +31,7 @@ void Start()
 
     public void SendTransaction()
     {
-        SendTransactionAsync().Forget();
+        SendTransactionAsync();
     }
 
     private async UniTaskVoid SendTransactionAsync()
diff --git a/sample/Assets/Scripts/Passport/ZkEvmSignTypedData/ZkEvmSignTypedDataScript.cs b/sample/Assets/Scripts/Passport/ZkEvmSignTypedData/ZkEvmSignTypedDataScript.cs
@@ -12,7 +12,7 @@ public class ZkEvmSignTypedDataScript : MonoBehaviour
 
     public void SignTypedData()
     {
-        SignTypedDataAsync().Forget();
+        SignTypedDataAsync();
     }
 
     private async UniTaskVoid SignTypedDataAsync()
diff --git a/sample/Tests/test/test.py b/sample/Tests/test/test.py
@@ -71,7 +71,7 @@ def test_1_passport_functions(self):
 
         # Get linked addresses
         self.altdriver.find_object(By.NAME, "GetLinkedAddresses").tap()
-        time.sleep(7)
+        time.sleep(1)
         text = output.get_text()
         print(f"GetLinkedAddresses output: {text}")
         self.assertEqual("No linked addresses", text)

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ public void StartDeviceCodeLogin()`
`23`	`23`	`Debug.LogWarning("[DeviceCodeLoginScript] Passport.Instance is null. Initialization must complete before device code login.");`
`24`	`24`	`return;`
`25`	`25`	`}`
`26`		`- StartDeviceCodeLoginAsync().Forget();`
	`26`	`+ StartDeviceCodeLoginAsync();`
`27`	`27`	`}`
`28`	`28`
`29`	`29`	`private async UniTaskVoid StartDeviceCodeLoginAsync()`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ public class GetAccessTokenScript : MonoBehaviour`
`12`	`12`	`/// </summary>`
`13`	`13`	`public void GetAccessToken()`
`14`	`14`	`{`
`15`		`- GetAccessTokenAsync().Forget();`
	`15`	`+ GetAccessTokenAsync();`
`16`	`16`	`}`
`17`	`17`
`18`	`18`	`private async UniTaskVoid GetAccessTokenAsync()`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ public class GetIdTokenScript : MonoBehaviour`
`12`	`12`	`/// </summary>`
`13`	`13`	`public void GetIdToken()`
`14`	`14`	`{`
`15`		`- GetIdTokenAsync().Forget();`
	`15`	`+ GetIdTokenAsync();`
`16`	`16`	`}`
`17`	`17`
`18`	`18`	`private async UniTaskVoid GetIdTokenAsync()`
Original file line number	Diff line number	Diff line change
`@@ -13,20 +13,26 @@ public class GetLinkedAddressesScript : MonoBehaviour`
`13`	`13`	`/// </summary>`
`14`	`14`	`public void GetLinkedAddresses()`
`15`	`15`	`{`
`16`		`- GetLinkedAddressesAsync().Forget();`
	`16`	`+ Debug.Log("E2E TEST DEBUG: GetLinkedAddresses() called (public method)");`
	`17`	`+ GetLinkedAddressesAsync();`
`17`	`18`	`}`
`18`	`19`
`19`	`20`	`private async UniTaskVoid GetLinkedAddressesAsync()`
`20`	`21`	`{`
	`22`	`+ Debug.Log("E2E TEST DEBUG: GetLinkedAddressesAsync() started");`
`21`	`23`	`if (Passport.Instance == null)`
`22`	`24`	`{`
	`25`	`+ Debug.Log("E2E TEST DEBUG: Passport instance is null in GetLinkedAddressesAsync");`
`23`	`26`	`ShowOutput("Passport instance is null");`
`24`	`27`	`return;`
`25`	`28`	`}`
`26`	`29`	`try`
`27`	`30`	`{`
	`31`	`+ Debug.Log("E2E TEST DEBUG: About to await Passport.Instance.GetLinkedAddresses()");`
`28`	`32`	`List<string> addresses = await Passport.Instance.GetLinkedAddresses();`
`29`		`- ShowOutput(addresses.Count > 0 ? string.Join(", ", addresses) : "No linked addresses");`
	`33`	`+ string outputMessage = addresses.Count > 0 ? string.Join(", ", addresses) : "No linked addresses";`
	`34`	`+ Debug.Log($"E2E TEST DEBUG: Got addresses, output will be: {outputMessage}");`
	`35`	`+ ShowOutput(outputMessage);`
`30`	`36`	`}`
`31`	`37`	`catch (System.Exception ex)`
`32`	`38`	`{`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ public class GetUserInfoScript : MonoBehaviour`
`12`	`12`	`/// </summary>`
`13`	`13`	`public void GetEmail()`
`14`	`14`	`{`
`15`		`- GetEmailAsync().Forget();`
	`15`	`+ GetEmailAsync();`
`16`	`16`	`}`
`17`	`17`
`18`	`18`	`private async UniTaskVoid GetEmailAsync()`
`@@ -38,7 +38,7 @@ private async UniTaskVoid GetEmailAsync()`
`38`	`38`	`/// </summary>`
`39`	`39`	`public void GetPassportId()`
`40`	`40`	`{`
`41`		`- GetPassportIdAsync().Forget();`
	`41`	`+ GetPassportIdAsync();`
`42`	`42`	`}`
`43`	`43`
`44`	`44`	`private async UniTaskVoid GetPassportIdAsync()`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ public class ImxConnectScript : MonoBehaviour`
`12`	`12`	`/// </summary>`
`13`	`13`	`public void ConnectImx()`
`14`	`14`	`{`
`15`		`- ConnectImxAsync().Forget();`
	`15`	`+ ConnectImxAsync();`
`16`	`16`	`}`
`17`	`17`
`18`	`18`	`private async UniTaskVoid ConnectImxAsync()`
`@@ -29,20 +29,24 @@ private async UniTaskVoid ConnectImxAsync()`
`29`	`29`	`{`
`30`	`30`	`await Passport.Instance.ConnectImx();`
`31`	`31`
`32`		`- // Add these lines to update connection state and refresh UI`
`33`	`32`	`SampleAppManager.IsConnectedToImx = true;`
`34`		`- var sceneManager = FindObjectOfType<AuthenticatedSceneManager>();`
`35`		`- if (sceneManager != null)`
	`33`	`+ ShowOutput("Connected to IMX"); // Show success early`
	`34`	`+`
	`35`	`+ // Try to find UnauthenticatedSceneManager first`
	`36`	`+ var unauthSceneManager = FindObjectOfType<UnauthenticatedSceneManager>();`
	`37`	`+ if (unauthSceneManager != null)`
`36`	`38`	`{`
`37`		`- sceneManager.UpdateImxButtonStates();`
`38`		`- Debug.Log("Updated IMX button states after connection");`
	`39`	`+ unauthSceneManager.OnImxConnected?.Invoke();`
	`40`	`+ return;`
`39`	`41`	`}`
`40`		`- else`
	`42`	`+`
	`43`	`+ var authSceneManager = FindObjectOfType<AuthenticatedSceneManager>();`
	`44`	`+ if (authSceneManager != null)`
`41`	`45`	`{`
`42`		`- Debug.LogWarning("Could not find AuthenticatedSceneManager to update button states");`
	`46`	`+ authSceneManager.UpdateImxButtonStates();`
	`47`	`+ authSceneManager.OnImxConnected?.Invoke();`
	`48`	`+ return;`
`43`	`49`	`}`
`44`		`-`
`45`		`- ShowOutput("Connected to IMX");`
`46`	`50`	`}`
`47`	`51`	`catch (System.Exception ex)`
`48`	`52`	`{`
Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@ public class ImxIsRegisteredOffchainScript : MonoBehaviour`
`14`	`14`
`15`	`15`	`public void CheckIsRegisteredOffchain()`
`16`	`16`	`{`
`17`		`- CheckIsRegisteredOffchainAsync().Forget();`
	`17`	`+ CheckIsRegisteredOffchainAsync();`
`18`	`18`	`}`
`19`	`19`
`20`	`20`	`private async UniTaskVoid CheckIsRegisteredOffchainAsync()`