fixed stake amount bug

Author: aalavandhan

frontend/src/components/GeyserFirst/GeyserStakeView.tsx

@@ -70,13 +70,18 @@ export const GeyserStakeView = () => {
   useEffect(() => {
     refreshInputAmount()
     if (geyserAction === GeyserAction.STAKE && stakingTokenInfo.price > 0) {
-      const initialStakeAmountUSD = 1000
-      const stakeAmt = Math.max(initialStakeAmountUSD / stakingTokenInfo.price, 0.000001)
-      const stakeAmtFP = parseUnits(stakeAmt.toFixed(stakingTokenInfo.decimals), stakingTokenInfo.decimals)
-      setUserInput(stakeAmt)
-      setParsedUserInput(BigNumber.from(stakeAmtFP))
+      if (stakableAmount.eq(0)) {
+        const initialStakeAmountUSD = 1000
+        const stakeAmt = Math.max(initialStakeAmountUSD / stakingTokenInfo.price, 0.000001)
+        const stakeAmtFP = parseUnits(stakeAmt.toFixed(stakingTokenInfo.decimals), stakingTokenInfo.decimals)
+        setUserInput(stakeAmt)
+        setParsedUserInput(BigNumber.from(stakeAmtFP))
+      } else {
+        setUserInput(formatUnits(stakableAmount, stakingTokenDecimals))
+        setParsedUserInput(stakableAmount)
+      }
     }
-  }, [geyserAction])
+  }, [geyserAction, stakingTokenBalance, currentStakeable])
 
   const handleGeyserInteraction = () => {
     if (geyserAction === GeyserAction.STAKE) {
@@ -182,7 +187,7 @@ export const GeyserStakeView = () => {
         <EstimatedRewards parsedUserInput={parsedUserInput} />
         {!address && <ConnectWalletWarning onClick={() => connectWallet()} />}
         {address && parsedUserInput.gt(0) && (
-          <StakeWarning link={poolAddress} balance={stakableAmount} staked={currentStakeAmount} />
+          <StakeWarning link={poolAddress} balance={stakableAmount.sub(parsedUserInput)} staked={currentStakeAmount} />
         )}
         <GeyserInteractionButton
           disabled={!address || parsedUserInput.isZero() || parsedUserInput.gt(stakableAmount)}

frontend/src/components/GeyserFirst/StakeWarning.tsx

@@ -10,15 +10,16 @@ interface Props {
 }
 
 export const StakeWarning: React.FC<Props> = ({ link, balance, staked }) => {
-  if (balance.gt(0) && staked.lte(0)) {
+  if (balance.gte(0)) {
     return <></>
   }
+
   return (
     <StakeWarningContainer>
       <ColoredDiv />
       <Content>
         <MessageContainer>
-          <Message>{staked.lte(0) ? 'Insufficient balance' : 'Entire balance staked'}</Message>
+          <Message>Insufficient balance</Message>
         </MessageContainer>
         <ButtonWrapper>
           <Button onClick={() => window.open(link, '_blank')}>{staked.lte(0) ? 'Get LP' : 'Get more'}</Button>

frontend/src/context/GeyserContext.tsx

@@ -166,16 +166,24 @@ export const GeyserContextProvider: React.FC = ({ children }) => {
     })()
   }
 
+  const clearSatsCache = (selectedGeyser: Geyser | null, selectedVault: Vault | null) => {
+    if (selectedGeyser) {
+      clearGeyserStatsCache(selectedGeyser)
+    }
+    if (selectedGeyser && selectedVault) {
+      clearUserStatsCache(selectedGeyser, selectedVault)
+    }
+  }
+
   const handleStakeUnstake = async (selectedVault: Vault | null, parsedAmount: BigNumber) => {
-    let r: any
+    clearSatsCache(selectedGeyserInfo?.geyser, selectedVault)
     if (geyserAction === GeyserAction.STAKE) {
-      r = await handleStake(selectedVault, parsedAmount)
+      return handleStake(selectedVault, parsedAmount)
     } else if (geyserAction === GeyserAction.UNSTAKE) {
-      r = await handleUnstake(selectedVault, parsedAmount)
+      return handleUnstake(selectedVault, parsedAmount)
+    } else {
+      return undefined
     }
-    clearGeyserStatsCache(selectedGeyser)
-    clearUserStatsCache(selectedGeyser, selectedVault)
-    return r
   }
 
   const handleWrapping = async (