save

pancakeswap · Jan 9, 2025 · 45ab95c · 45ab95c
1 parent 23db1ad
commit 45ab95c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/apps/web/src/views/Ifos/components/IfoVesting/VestingPeriod/Claim.tsx b/apps/web/src/views/Ifos/components/IfoVesting/VestingPeriod/Claim.tsx
@@ -9,6 +9,7 @@ import { useCallback, useMemo } from 'react'
 import { Address } from 'viem'
 import { VestingData } from 'views/Ifos/hooks/vesting/fetchUserWalletIfoData'
 
+import { isVestingEnd } from 'views/Ifos/hooks/vesting/useFetchVestingData'
 import { SwitchNetworkTips } from '../../IfoFoldableCard/IfoPoolCard/SwitchNetworkTips'
 
 interface Props {
@@ -34,6 +35,7 @@ const ClaimButton: React.FC<React.PropsWithChildren<Props>> = ({
   const { address, token, chainId } = data.ifo
   const contract = useIfoV3Contract(address)
   const { fetchWithCatchTxError, loading: isPending } = useCatchTxError()
+  const isEnd = isVestingEnd(data)
 
   const isReady = useMemo(() => {
     const checkClaimableAmount = isVestingInitialized ? claimableAmount === '0' : false
@@ -87,7 +89,7 @@ const ClaimButton: React.FC<React.PropsWithChildren<Props>> = ({
       width="100%"
       onClick={handleClaim}
       isLoading={isPending}
-      disabled={isReady && !enabled}
+      disabled={isReady && !enabled && !isEnd}
       endIcon={isPending ? <AutoRenewIcon spin color="currentColor" /> : null}
     >
       {t('Claim %symbol%', { symbol: token.symbol })}

diff --git a/apps/web/src/views/Ifos/hooks/vesting/useFetchVestingData.ts b/apps/web/src/views/Ifos/hooks/vesting/useFetchVestingData.ts
@@ -30,15 +30,20 @@ const isPoolEligible = (poolId: PoolIds, userVestingData: UserVestingData) => {
   return vestingEndTime.gte(currentTimeStamp)
 }
 
+export const isVestingEnd = (vestingData: VestingData) => {
+  const pool = vestingData.userVestingData[PoolIds.poolBasic] || vestingData.userVestingData[PoolIds.poolUnlimited]
+  const currentTimeStamp = Date.now()
+  const vestingEndTime = (vestingData.userVestingData.vestingStartTime + pool.vestingInformationDuration) * 1000
+  return vestingEndTime < currentTimeStamp
+}
+
 const isEligibleVestingData = (ifo: VestingData) => {
   const { userVestingData } = ifo
   return POOLS.some((poolId) => isPoolEligible(poolId, userVestingData))
 }
 
 const useFetchVestingData = () => {
-  let { address: account } = useAccount()
-  // For demo purposes, overriding the account here
-  account = '0xa9c60777fD1A95602D6c080A72Ff02324373F609'
+  const { address: account } = useAccount()
 
   const { chainId } = useActiveChainId()
   const configs = useIfoConfigsAcrossChains()