1 files changed, 30 insertions, 0 deletions
diff --git a/jb-ui/src/models/appSlice.ts b/jb-ui/src/models/appSlice.ts
index 46c6a2f..42a776e 100644
--- a/jb-ui/src/models/appSlice.ts
+++ b/jb-ui/src/models/appSlice.ts
@@ -4,6 +4,8 @@ import {
   UserLedgerTransactionsResponseTransactionsInner,
   UserLedgerTransactionTypesSummary,
   UserWalletBalance,
+  UserDisplayedWalletBalance,
+  UserLedgerWallet,
   OfferwallReason,
 } from "@/api_fsb";
 import { App } from "@/models/app";
@@ -26,6 +28,12 @@ const initialState: App = {
   userLedgerTxCount: undefined,
   userLedgerTxs: [],
 
+  // --- new approach
+  userDisplayedWalletBalance: undefined,
+  userWallet: undefined,
+  userAttemptCreditWallet: undefined,
+  // ---
+
   txPagination: { pageIndex: 0, pageSize: 10 },
   txTotalItems: undefined,
   txTotalPages: undefined,
@@ -91,6 +99,24 @@ const appSlice = createSlice({
       state.userWalletBalance = action.payload;
     },
 
+    // --- New approach ---
+    setUserDisplayedWalletBalance(
+      state,
+      action: PayloadAction<UserDisplayedWalletBalance>,
+    ) {
+      state.userDisplayedWalletBalance = action.payload;
+    },
+
+    setUserWallet(state, action: PayloadAction<UserLedgerWallet>) {
+      state.userWallet = action.payload;
+    },
+
+    setUserAttemptCreditWallet(state, action: PayloadAction<UserLedgerWallet>) {
+      state.userAttemptCreditWallet = action.payload;
+    },
+
+    // ---  ---
+
     setUserLedgerSummary(
       state,
       action: PayloadAction<UserLedgerTransactionTypesSummary>,
@@ -138,6 +164,10 @@ export const {
   setUserLedgerSummary,
   setUserLedgerTxs,
 
+  setUserDisplayedWalletBalance,
+  setUserWallet,
+  setUserAttemptCreditWallet,
+
   setTxPagination,
   setTxTotalItems,
   setTxTotalPages,