refactor: add optional chaining to store state access for improved sa…

…fety
valory-xyz · Dec 20, 2024 · 9a551e6 · 9a551e6
1 parent 311e20e
commit 9a551e6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/frontend/components/MainPage/sections/AlertSections/LowFunds/LowOperatingBalanceAlert.tsx b/frontend/components/MainPage/sections/AlertSections/LowFunds/LowOperatingBalanceAlert.tsx
@@ -36,7 +36,7 @@ export const LowOperatingBalanceAlert = () => {
 
   if (!isBalanceLoaded) return null;
   if (!masterThresholds) return null;
-  if (!storeState?.[selectedAgentType].isInitialFunded) return;
+  if (!storeState?.[selectedAgentType]?.isInitialFunded) return;
   if (!isLowBalance) return null;
 
   return (

diff --git a/frontend/components/MainPage/sections/GasBalanceSection.tsx b/frontend/components/MainPage/sections/GasBalanceSection.tsx
@@ -55,7 +55,7 @@ const BalanceStatus = () => {
   useEffect(() => {
     if (!isBalanceLoaded || !isServicesLoaded) return;
     if (!showNotification) return;
-    if (!storeState?.[selectedAgentType].isInitialFunded) return;
+    if (!storeState?.[selectedAgentType]?.isInitialFunded) return;
 
     if (isMasterSafeLowOnNativeGas && !isLowBalanceNotificationShown) {
       showNotification('Operating balance is too low.');

diff --git a/frontend/hooks/useNeedsFunds.ts b/frontend/hooks/useNeedsFunds.ts
@@ -27,7 +27,7 @@ export const useNeedsFunds = (stakingProgramId: Maybe<StakingProgramId>) => {
   const { isLoaded: isBalanceLoaded } = useBalanceContext();
   const { masterSafeBalances } = useMasterBalances();
 
-  const isInitialFunded = storeState?.[selectedAgentType].isInitialFunded;
+  const isInitialFunded = storeState?.[selectedAgentType]?.isInitialFunded;
   const serviceFundRequirements = useMemo<{
     [chainId: number]: {
       [tokenSymbol: string]: number;