chore(bonsai-core): sync local address to store for use in bonsai (#1395

)
dydxprotocol · Jan 7, 2025 · df32a1a · df32a1a
1 parent f6c6b53
commit df32a1a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/src/abacus-ts/socketSelectors.ts b/src/abacus-ts/socketSelectors.ts
@@ -3,7 +3,6 @@ import { ENVIRONMENT_CONFIG_MAP } from '@/constants/networks';
 import { EndpointsConfig } from '@/hooks/useEndpointsConfig';
 
 import { type RootState } from '@/state/_store';
-import { getUserSubaccountNumber, getUserWalletAddress } from '@/state/accountSelectors';
 import { getSelectedNetwork } from '@/state/appSelectors';
 import { createAppSelector } from '@/state/appTypes';
 
@@ -18,9 +17,10 @@ export const selectIndexerUrl = createAppSelector([getSelectedNetwork], (network
   return `${endpointsConfig.indexers[0]!.api}`;
 });
 
+// TODO allow configurable parent subaccount number
 export const selectParentSubaccountInfo = createAppSelector(
-  [getUserWalletAddress, getUserSubaccountNumber],
-  (wallet, subaccount) => ({ wallet, subaccount })
+  [(state) => state.wallet.localWallet?.address],
+  (wallet) => ({ wallet, subaccount: 0 })
 );
 
 export const selectIndexerReady = createAppSelector(

diff --git a/src/hooks/useAccounts.tsx b/src/hooks/useAccounts.tsx
@@ -18,7 +18,7 @@ import { setOnboardingGuard, setOnboardingState } from '@/state/account';
 import { getGeo, getHasSubaccount } from '@/state/accountSelectors';
 import { getSelectedDydxChainId } from '@/state/appSelectors';
 import { useAppDispatch, useAppSelector } from '@/state/appTypes';
-import { clearSavedEncryptedSignature } from '@/state/wallet';
+import { clearSavedEncryptedSignature, setLocalWallet } from '@/state/wallet';
 import { getSourceAccount } from '@/state/walletSelectors';
 
 import abacusStateManager from '@/lib/abacus';
@@ -135,6 +135,10 @@ const useAccountsContext = () => {
     [localDydxWallet]
   );
 
+  useEffect(() => {
+    dispatch(setLocalWallet({ address: dydxAddress }));
+  }, [dispatch, dydxAddress]);
+
   const nobleAddress = useMemo(() => {
     return localNobleWallet?.address;
   }, [localNobleWallet]);

diff --git a/src/state/wallet.ts b/src/state/wallet.ts
@@ -13,6 +13,9 @@ export type SourceAccount = {
 // NOTE: This app slice is persisted via redux-persist. Changes to this type may require migrations.
 export interface WalletState {
   sourceAccount: SourceAccount;
+  localWallet?: {
+    address?: string;
+  };
 }
 
 const initialState: WalletState = {
@@ -58,6 +61,9 @@ export const walletSlice = createSlice({
     clearSavedEncryptedSignature: (state) => {
       state.sourceAccount.encryptedSignature = undefined;
     },
+    setLocalWallet: (state, { payload }: PayloadAction<{ address?: string }>) => {
+      state.localWallet = payload;
+    },
     clearSourceAccount: (state) => {
       state.sourceAccount = {
         address: undefined,
@@ -75,4 +81,5 @@ export const {
   setSavedEncryptedSignature,
   clearSavedEncryptedSignature,
   clearSourceAccount,
+  setLocalWallet,
 } = walletSlice.actions;