feat: update balances on dismiss of add funds view

immutable · Dec 16, 2024 · 05995fa · 05995fa
1 parent 9138bd8
commit 05995fa
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/Assets/Shared/Scripts/UI/AddFunds/AddFunds.cs b/Assets/Shared/Scripts/UI/AddFunds/AddFunds.cs
@@ -1,3 +1,4 @@
+using System;
 using UnityEngine;
 using UnityEngine.UI;
 using System.Collections.Generic;
@@ -19,8 +20,12 @@ public class AddFunds : MonoBehaviour
 
         [SerializeField] TransakView m_TransakView;
 
-        public void Show()
+        private Action m_OnDismiss;
+
+        public void Show(Action onDismiss)
         {
+            m_OnDismiss = onDismiss;
+
             m_FiatButton.onClick.RemoveListener(OnFiatButtonClicked);
             m_FiatButton.onClick.AddListener(OnFiatButtonClicked);
 
@@ -61,6 +66,7 @@ private void OnCancelButtonClicked()
             gameObject.SetActive(false);
             m_Options.gameObject.SetActive(true);
             m_TransakView.gameObject.SetActive(false);
+            m_OnDismiss();
         }
     }
 }
diff --git a/Assets/Shared/Scripts/UI/BalanceObject.cs b/Assets/Shared/Scripts/UI/BalanceObject.cs
@@ -79,7 +79,9 @@ private void TogglePanel()
         /// </summary>
         private void ShowAddFunds()
         {
-            m_AddFunds.Show();
+#pragma warning disable CS4014
+            m_AddFunds.Show(() => UpdateBalance());
+#pragma warning restore CS4014
         }
 
         /// <summary>