fix: userBalance wasn't always passed to queries from form

Author: 0xAlunara

apps/main/src/dex/entities/gauge/model/estimate-gas-options.ts

@@ -10,13 +10,14 @@ import { gaugeKeys } from './query-keys'
 import { depositRewardAvailable, depositRewardIsApproved } from './query-options'
 
 export const estimateGasDepositRewardApprove = queryFactory({
-  queryKey: ({ rewardTokenId, amount, ...gaugeParams }: DepositRewardApproveParams) =>
+  queryKey: ({ rewardTokenId, amount, userBalance, ...gaugeParams }: DepositRewardApproveParams) =>
     [
       ...rootKeys.gauge({ ...gaugeParams }),
       ...gaugeKeys.estimateGas(),
       'depositRewardApprove',
       { rewardTokenId },
       { amount },
+      { userBalance },
     ] as const,
   queryFn: api.queryEstimateGasDepositRewardApprove,
   refetchInterval: '1m',
@@ -43,14 +44,15 @@ export const estimateGasAddRewardToken = queryFactory({
 })
 
 export const estimateGasDepositReward = queryFactory({
-  queryKey: ({ rewardTokenId, amount, epoch, ...gaugeParams }: DepositRewardParams) =>
+  queryKey: ({ rewardTokenId, amount, epoch, userBalance, ...gaugeParams }: DepositRewardParams) =>
     [
       ...rootKeys.gauge({ ...gaugeParams }),
       ...gaugeKeys.estimateGas(),
       'depositReward',
       { rewardTokenId },
       { amount },
       { epoch },
+      { userBalance },
     ] as const,
   queryFn: api.queryEstimateGasDepositReward,
   refetchInterval: '1m',

apps/main/src/dex/features/deposit-gauge-reward/ui/ActionsStepper.tsx

@@ -32,6 +32,7 @@ export const DepositStepper = ({ chainId, poolId }: { chainId: ChainId; poolId:
   const amount = watch('amount')
   const rewardTokenId = watch('rewardTokenId')
   const step = watch('step')
+  const userBalance = watch('userBalance')
 
   const { mutate: depositRewardApprove, isPending: isPendingDepositRewardApprove } = useDepositRewardApprove({
     chainId,
@@ -59,6 +60,7 @@ export const DepositStepper = ({ chainId, poolId }: { chainId: ChainId; poolId:
       {
         rewardTokenId: getValues('rewardTokenId'),
         amount: getValues('amount'),
+        userBalance: getValues('userBalance'),
       },
       { onSuccess: onApproveSuccess, onError: onApproveError },
     )
@@ -70,6 +72,7 @@ export const DepositStepper = ({ chainId, poolId }: { chainId: ChainId; poolId:
         rewardTokenId: getValues('rewardTokenId'),
         amount: getValues('amount'),
         epoch: getValues('epoch'),
+        userBalance: getValues('userBalance'),
       },
       {
         onSuccess: (data: string) => {
@@ -91,6 +94,7 @@ export const DepositStepper = ({ chainId, poolId }: { chainId: ChainId; poolId:
     poolId,
     rewardTokenId,
     amount,
+    userBalance,
   })
 
   useLayoutEffect(() => {

apps/main/src/dex/features/deposit-gauge-reward/ui/GasEstimation.tsx

@@ -14,13 +14,14 @@ export const GasEstimation = ({ chainId, poolId }: { chainId: ChainId; poolId: s
   const amount = watch('amount')
   const epoch = watch('epoch')
   const step = watch('step')
+  const userBalance = watch('userBalance')
 
   const {
     data: estimatedGasDepositRewardApprove,
     isLoading: isLoadingGasEstimateDepositRewardApprove,
     isFetching: isFetchingGasEstimateDepositRewardApprove,
   } = useEstimateGasDepositRewardApprove(
-    { chainId, poolId, rewardTokenId, amount },
+    { chainId, poolId, rewardTokenId, amount, userBalance },
     step === DepositRewardStep.APPROVAL && isValid,
   )
 
@@ -35,6 +36,7 @@ export const GasEstimation = ({ chainId, poolId }: { chainId: ChainId; poolId: s
       rewardTokenId,
       amount,
       epoch,
+      userBalance,
     },
     step === DepositRewardStep.DEPOSIT && isValid,
   )