feat: implement server-side control for withdraw and claim on Earn pages (#7519)

* feat: enable local address label in history list OK-38806 (#7509)

* feat: replace address with local address label in history list

* feat: query address with cache

* feat: update address label when wallet/account/addressbook changed

* fix: lint

* fix: prevent Android reset from freezing. (#7515)

* feat: implement server-side control for withdraw and claim on Earn pages

* Revert "fix: prevent Android reset from freezing. (#7515)"

This reverts commit 8c8f2f4.

* Revert "feat: enable local address label in history list OK-38806 (#7509)"

This reverts commit 184746b.

* Update EarnHome.tsx

* Update byte-test-web.yml

---------

Co-authored-by: weatherstar <[email protected]>

Author: huhuanming

packages/kit/src/views/Earn/EarnHome.tsx

@@ -60,7 +60,10 @@ import { ListItem } from '../../components/ListItem';
 import { TabPageHeader } from '../../components/TabPageHeader';
 import useAppNavigation from '../../hooks/useAppNavigation';
 import { usePromiseResult } from '../../hooks/usePromiseResult';
-import { useActiveAccount } from '../../states/jotai/contexts/accountSelector';
+import {
+  useAccountSelectorActions,
+  useActiveAccount,
+} from '../../states/jotai/contexts/accountSelector';
 import { useEarnActions, useEarnAtom } from '../../states/jotai/contexts/earn';
 
 import { EARN_PAGE_MAX_WIDTH, EARN_RIGHT_PANEL_WIDTH } from './EarnConfig';
@@ -847,6 +850,7 @@ function BasicEarnHome() {
   ];
 
   const navigation = useAppNavigation();
+  const accountSelectorActions = useAccountSelectorActions();
 
   const onBannerPress = useCallback(
     async ({
@@ -896,9 +900,16 @@ function BasicEarnHome() {
         } else {
           openUrlInApp(href);
         }
+      } else {
+        await accountSelectorActions.current.showAccountSelector({
+          navigation,
+          activeWallet: undefined,
+          num: 0,
+          sceneName: EAccountSelectorSceneName.home,
+        });
       }
     },
-    [account, indexedAccount, navigation],
+    [account, accountSelectorActions, indexedAccount, navigation],
   );
 
   const banners = useMemo(() => {

packages/kit/src/views/Earn/earnUtils.ts

@@ -33,7 +33,8 @@ export const EarnNavigation = {
       params: {
         accountId: earnAccount?.accountId || accountId || '',
         networkId,
-        indexedAccountId: earnAccount?.account.indexedAccountId,
+        indexedAccountId:
+          earnAccount?.account.indexedAccountId || indexedAccountId,
         symbol,
         provider,
         vault,

packages/kit/src/views/Staking/components/ProtocolDetails/EarnActionIcon.tsx

@@ -1,4 +1,4 @@
-import { memo, useCallback } from 'react';
+import { memo, useCallback, useState } from 'react';
 
 import { StyleSheet } from 'react-native';
 
@@ -197,12 +197,18 @@ function BasicClaimActionIcon({
     accountId: protocolInfo?.earnAccount?.accountId || '',
     networkId: tokenInfo?.networkId || '',
   });
+  const [loading, setLoading] = useState(false);
   return (
     <Button
       size="small"
       variant="primary"
-      disabled={actionIcon?.disabled}
+      loading={loading}
+      disabled={loading || actionIcon?.disabled}
       onPress={async () => {
+        setLoading(true);
+        setTimeout(() => {
+          setLoading(false);
+        }, 10 * 1000);
         const claimAmount =
           protocolInfo?.claimable || actionIcon.data?.balance || '0';
         const isMorphoClaim = !!(
@@ -212,6 +218,7 @@ function BasicClaimActionIcon({
           })
         );
         await handleClaim({
+          claimType: actionIcon.type,
           symbol: protocolInfo?.symbol || '',
           protocolInfo,
           tokenInfo: tokenInfo
@@ -236,6 +243,7 @@ function BasicClaimActionIcon({
             tags: protocolInfo?.stakeTag ? [protocolInfo.stakeTag] : [],
           },
         });
+        setLoading(false);
       }}
     >
       {typeof actionIcon.text === 'string'

packages/kit/src/views/Staking/pages/HistoryList/index.tsx

@@ -47,6 +47,7 @@ type IHistoryItemProps = {
   networks?: IStakeHistoriesResponse['networks'];
   token?: IToken;
   provider?: string;
+  tokenMap?: IStakeHistoriesResponse['tokenMap'];
 };
 
 const HistoryItem = ({
@@ -55,22 +56,37 @@ const HistoryItem = ({
   token,
   network,
   networks,
+  tokenMap,
 }: IHistoryItemProps) => {
   const navigation = useAppNavigation();
   const route = useAppRoute<
     IModalStakingParamList,
     EModalStakingRoutes.HistoryList
   >();
-  const { accountId, networkId } = route.params;
+  const { accountId } = route.params;
   const logoURI = useMemo(() => {
     if (token?.logoURI) {
       return token.logoURI;
     }
     if (networks?.length) {
       return networks.find((o) => o.networkId === item.networkId)?.logoURI;
     }
+    if (tokenMap && item.type === 'stake') {
+      const uri = tokenMap[item.tokenAddress]?.logoURI;
+      if (uri) {
+        return uri;
+      }
+    }
     return network?.logoURI;
-  }, [token?.logoURI, networks, network?.logoURI, item.networkId]);
+  }, [
+    item.type,
+    item.tokenAddress,
+    item.networkId,
+    token?.logoURI,
+    networks,
+    network?.logoURI,
+    tokenMap,
+  ]);
   const onPress = useCallback(() => {
     navigation.push(EModalAssetDetailRoutes.HistoryDetails, {
       networkId: item.networkId,
@@ -121,6 +137,7 @@ type IHistoryContentProps = {
   onFilterTypeChange: (type: string) => void;
   network?: { networkId: string; name: string; logoURI: string };
   networks?: IStakeHistoriesResponse['networks'];
+  tokenMap?: IStakeHistoriesResponse['tokenMap'];
   provider?: string;
 };
 
@@ -136,6 +153,7 @@ const HistoryContent = ({
   filter,
   filterType,
   networks,
+  tokenMap,
   onFilterTypeChange,
 }: IHistoryContentProps) => {
   const renderItem = useCallback(
@@ -144,11 +162,12 @@ const HistoryContent = ({
         item={item}
         network={network}
         token={item.token?.info}
+        tokenMap={tokenMap}
         provider={provider}
         networks={networks}
       />
     ),
-    [network, networks, provider],
+    [network, networks, provider, tokenMap],
   );
 
   const renderSectionHeader = useCallback(
@@ -327,6 +346,7 @@ function HistoryList() {
       return {
         network: historyResp.network,
         networks: historyResp.networks,
+        tokenMap: historyResp.tokenMap,
         sections,
         filter: historyResp.filter || {},
       };
@@ -362,6 +382,7 @@ function HistoryList() {
               sections={result.sections}
               network={result.network}
               networks={result.networks}
+              tokenMap={result.tokenMap}
               filter={result.filter}
               provider={provider}
               onFilterTypeChange={setFilterType}

packages/kit/src/views/Staking/pages/ProtocolDetails/useHandleActions.ts

@@ -5,7 +5,7 @@ import useAppNavigation from '@onekeyhq/kit/src/hooks/useAppNavigation';
 import { MorphoBundlerContract } from '@onekeyhq/shared/src/consts/addresses';
 import { EModalStakingRoutes } from '@onekeyhq/shared/src/routes';
 import accountUtils from '@onekeyhq/shared/src/utils/accountUtils';
-import { EApproveType } from '@onekeyhq/shared/types/staking';
+import { EApproveType, EWithdrawType } from '@onekeyhq/shared/types/staking';
 import type {
   IEarnTokenInfo,
   IProtocolInfo,
@@ -15,6 +15,7 @@ export const useHandleWithdraw = () => {
   const appNavigation = useAppNavigation();
   return useCallback(
     async ({
+      withdrawType,
       tokenInfo,
       protocolInfo,
       accountId,
@@ -23,6 +24,7 @@ export const useHandleWithdraw = () => {
       provider,
       onSuccess,
     }: {
+      withdrawType: EWithdrawType;
       protocolInfo?: IProtocolInfo;
       tokenInfo?: IEarnTokenInfo;
       accountId?: string;
@@ -41,7 +43,10 @@ export const useHandleWithdraw = () => {
       if (!stakingConfig) {
         throw new Error('Staking config not found');
       }
-      if (stakingConfig.withdrawWithTx) {
+      if (
+        withdrawType === EWithdrawType.WithdrawOrder ||
+        stakingConfig.withdrawWithTx
+      ) {
         appNavigation.push(EModalStakingRoutes.WithdrawOptions, {
           accountId,
           networkId,

packages/kit/src/views/Staking/pages/ProtocolDetails/useHandleClaim.ts

@@ -12,6 +12,7 @@ import type {
   IProtocolInfo,
   IStakingInfo,
 } from '@onekeyhq/shared/types/staking';
+import { EClaimType } from '@onekeyhq/shared/types/staking';
 
 import { useUniversalClaim } from '../../hooks/useUniversalHooks';
 
@@ -32,6 +33,7 @@ export const useHandleClaim = ({
   });
   return useCallback(
     async ({
+      claimType,
       protocolInfo,
       tokenInfo,
       symbol,
@@ -42,6 +44,7 @@ export const useHandleClaim = ({
       stakingInfo,
       onSuccess,
     }: {
+      claimType: EClaimType;
       protocolInfo?: IProtocolInfo;
       tokenInfo?: IEarnTokenInfo;
       symbol: string;
@@ -71,7 +74,7 @@ export const useHandleClaim = ({
           provider,
           stakingInfo,
           claimTokenAddress,
-          vault: protocolInfo?.approve?.approveTarget || '',
+          vault,
         });
         return;
       }
@@ -102,7 +105,22 @@ export const useHandleClaim = ({
         });
         return;
       }
-      if (stakingConfig.claimWithTx) {
+      if (
+        provider.toLowerCase() === 'everstake' &&
+        symbol.toLowerCase() === 'apt'
+      ) {
+        appNavigation.push(EModalStakingRoutes.Claim, {
+          accountId,
+          networkId,
+          protocolInfo,
+          tokenInfo,
+          onSuccess,
+          amount: stakingConfig.claimWithAmount ? claimAmount : undefined,
+        });
+        return;
+      }
+
+      if (claimType === EClaimType.ClaimOrder) {
         appNavigation.push(EModalStakingRoutes.ClaimOptions, {
           accountId,
           networkId,
@@ -114,16 +132,29 @@ export const useHandleClaim = ({
         return;
       }
       if (
-        provider.toLowerCase() === 'everstake' &&
-        symbol.toLowerCase() === 'apt'
+        claimType === EClaimType.Claim &&
+        claimAmount &&
+        Number(claimAmount) > 0
       ) {
-        appNavigation.push(EModalStakingRoutes.Claim, {
+        await handleUniversalClaim({
+          amount: claimAmount,
+          symbol,
+          provider,
+          claimTokenAddress,
+          stakingInfo,
+          vault,
+        });
+        return;
+      }
+
+      if (stakingConfig.claimWithTx) {
+        appNavigation.push(EModalStakingRoutes.ClaimOptions, {
           accountId,
           networkId,
           protocolInfo,
           tokenInfo,
-          onSuccess,
-          amount: stakingConfig.claimWithAmount ? claimAmount : undefined,
+          symbol,
+          provider,
         });
         return;
       }