AA Support

config/groups/stake.ts

@@ -19,7 +19,7 @@ export const STAKE_GASLIMIT_FALLBACK = BigInt(
   ),
 );
 
-export const LIDO_ADDRESS = '0x11D00000000000000000000000000000000011D0';
+export const LIDO_ADDRESS = '0x11d00000000000000000000000000000000011d0';
 
 export const STAKE_FALLBACK_REFERRAL_ADDRESS = preConfig.ipfsMode
   ? IPFS_REFERRAL_ADDRESS

config/groups/web3.ts

@@ -4,6 +4,8 @@ export const PROVIDER_POLLING_INTERVAL = 12_000;
 export const PROVIDER_BATCH_TIME = 150;
 // max batch
 export const PROVIDER_MAX_BATCH = 20;
+// AA transaction polling timeout(ms)
+export const AA_TX_POLLING_TIMEOUT = 60_000;
 
 // account for gas estimation
 // will always have:

consts/token-addresses.ts

@@ -25,7 +25,7 @@ export const TOKENS_BY_NETWORK: {
   },
 };
 
-export const getRateTokenAddress = (
+export const getTokenAddress = (
   chainId: CHAINS,
   token: TOKENS,
 ): Address | undefined => {

features/settings/settings-form/settings-form.tsx

@@ -1,7 +1,7 @@
 import { useCallback } from 'react';
 import { useForm } from 'react-hook-form';
 
-import type { LIDO_CONTRACT_NAMES } from '@lidofinance/lido-ethereum-sdk/common';
+import { LIDO_CONTRACT_NAMES } from '@lidofinance/lido-ethereum-sdk/common';
 import { Button, ToastSuccess, Block, Input } from '@lidofinance/lido-ui';
 
 import { useUserConfig } from 'config/user-config';
@@ -34,9 +34,7 @@ export const SettingsForm = () => {
     },
   });
 
-  const { data: stethAddress } = useContractAddress(
-    'lido' as LIDO_CONTRACT_NAMES,
-  );
+  const { data: stethAddress } = useContractAddress(LIDO_CONTRACT_NAMES.lido);
 
   const {
     formState,

features/stake/stake-form/hooks/use-tx-modal-stages-stake.tsx

@@ -32,11 +32,12 @@ const getTxModalStagesStake = (transitStage: TransactionModalTransitStage) => ({
       />,
     ),
 
-  pending: (amount: bigint, txHash?: Hash) =>
+  pending: (amount: bigint, txHash?: Hash, isAA?: boolean) =>
     transitStage(
       <TxStageSignOperationAmount
         {...STAGE_OPERATION_ARGS}
         amount={amount}
+        isAA={isAA}
         willReceive={amount}
         isPending
         txHash={txHash}

features/stake/stake-form/stake-form-context/stake-form-context.tsx

@@ -24,7 +24,7 @@ import {
 } from 'shared/hook-form/form-controller';
 import { useTokenMaxAmount } from 'shared/hooks/use-token-max-amount';
 import { useStakingLimitInfo } from 'shared/hooks/useStakingLimitInfo';
-import { useIsMultisig, useMaxGasPrice } from 'modules/web3';
+import { useIsSmartAccount, useMaxGasPrice } from 'modules/web3';
 import { useFormControllerRetry } from 'shared/hook-form/form-controller/use-form-controller-retry-delegate';
 
 import {
@@ -63,7 +63,8 @@ const useStakeFormNetworkData = (): StakeFormNetworkData => {
     refetch: updateStethBalance,
     isLoading: isStethBalanceLoading,
   } = useStethBalance();
-  const { isMultisig, isLoading: isMultisigLoading } = useIsMultisig();
+  const { isSmartAccount, isLoading: isSmartAccountLoading } =
+    useIsSmartAccount();
   const gasLimit = useStethSubmitGasLimit();
   const { maxGasPrice, isLoading: isMaxGasPriceLoading } = useMaxGasPrice();
 
@@ -97,10 +98,10 @@ const useStakeFormNetworkData = (): StakeFormNetworkData => {
   const maxAmount = useTokenMaxAmount({
     balance: etherBalance,
     limit: stakingLimitInfo?.currentStakeLimit,
-    isPadded: !isMultisig,
+    isPadded: !isSmartAccount,
     gasLimit: gasLimit,
     padding: BALANCE_PADDING,
-    isLoading: isMultisigLoading,
+    isLoading: isSmartAccountLoading,
   });
 
   const revalidate = useCallback(async () => {
@@ -114,14 +115,14 @@ const useStakeFormNetworkData = (): StakeFormNetworkData => {
   const loading = useMemo(
     () => ({
       isStethBalanceLoading,
-      isMultisigLoading,
+      isSmartAccountLoading,
       isMaxGasPriceLoading,
       isEtherBalanceLoading,
       isStakeableEtherLoading: isStakingLimitIsLoading || isEtherBalanceLoading,
     }),
     [
       isStethBalanceLoading,
-      isMultisigLoading,
+      isSmartAccountLoading,
       isMaxGasPriceLoading,
       isEtherBalanceLoading,
       isStakingLimitIsLoading,
@@ -131,7 +132,7 @@ const useStakeFormNetworkData = (): StakeFormNetworkData => {
   return {
     stethBalance,
     etherBalance,
-    isMultisig: isMultisigLoading ? undefined : isMultisig,
+    isSmartAccount,
     stakeableEther,
     stakingLimitInfo,
     gasCost,

features/stake/stake-form/stake-form-context/types.ts

@@ -10,15 +10,15 @@ export type StakeFormInput = {
 
 export type StakeFormLoading = {
   isStethBalanceLoading: boolean;
-  isMultisigLoading: boolean;
+  isSmartAccountLoading: boolean;
   isMaxGasPriceLoading: boolean;
   isEtherBalanceLoading: boolean;
   isStakeableEtherLoading: boolean;
 };
 
 export type StakeFormNetworkData = {
   etherBalance?: bigint;
-  isMultisig?: boolean;
+  isSmartAccount?: boolean;
   stethBalance?: bigint;
   stakeableEther?: bigint;
   stakingLimitInfo?: StakeLimitFullInfo;
@@ -35,5 +35,6 @@ export type StakeFormValidationContext = {
   currentStakeLimit: bigint;
   gasCost: bigint;
   etherBalance: bigint;
-  isMultisig: boolean;
+  isSmartAccount: boolean;
+  shouldValidateEtherBalance: boolean;
 };

features/stake/stake-form/stake-form-context/validation.ts

@@ -2,7 +2,7 @@ import { useMemo } from 'react';
 import type { Resolver } from 'react-hook-form';
 import invariant from 'tiny-invariant';
 
-import { useDappStatus } from 'modules/web3';
+import { useAA, useDappStatus } from 'modules/web3';
 import { VALIDATION_CONTEXT_TIMEOUT } from 'features/withdrawals/withdrawals-constants';
 import { TOKENS_TO_WRAP } from 'features/wsteth/shared/types';
 import { useAwaiter } from 'shared/hooks/use-awaiter';
@@ -36,7 +36,8 @@ export const stakeFormValidationResolver: Resolver<
       currentStakeLimit,
       etherBalance,
       gasCost,
-      isMultisig,
+      isSmartAccount,
+      shouldValidateEtherBalance,
     } = await awaitWithTimeout(
       validationContextPromise,
       VALIDATION_CONTEXT_TIMEOUT,
@@ -49,8 +50,9 @@ export const stakeFormValidationResolver: Resolver<
       stakingLimitLevel,
       currentStakeLimit,
       etherBalance,
+      shouldValidateEtherBalance,
       gasCost,
-      isMultisig,
+      isSmartAccount,
     });
 
     if (!isWalletActive) {
@@ -73,7 +75,9 @@ export const useStakeFormValidationContext = (
   networkData: StakeFormNetworkData,
 ): Promise<StakeFormValidationContext> => {
   const { isDappActive } = useDappStatus();
-  const { stakingLimitInfo, etherBalance, isMultisig, gasCost } = networkData;
+  const { areAuxiliaryFundsSupported } = useAA();
+  const { stakingLimitInfo, etherBalance, isSmartAccount, gasCost } =
+    networkData;
 
   const validationContextAwaited = useMemo(() => {
     if (
@@ -82,20 +86,28 @@ export const useStakeFormValidationContext = (
       (!isDappActive ||
         (etherBalance !== undefined &&
           gasCost !== undefined &&
-          isMultisig !== undefined))
+          isSmartAccount !== undefined))
     ) {
       return {
         isWalletActive: isDappActive,
         stakingLimitLevel: stakingLimitInfo.stakeLimitLevel,
         currentStakeLimit: stakingLimitInfo.currentStakeLimit,
+        shouldValidateEtherBalance: areAuxiliaryFundsSupported,
         // condition above guaranties stubs will only be passed when isDappActive = false
         etherBalance: etherBalance ?? 0n,
         gasCost: gasCost ?? 0n,
-        isMultisig: isMultisig ?? false,
+        isSmartAccount: isSmartAccount ?? false,
       };
     }
     return undefined;
-  }, [isDappActive, etherBalance, gasCost, isMultisig, stakingLimitInfo]);
+  }, [
+    stakingLimitInfo,
+    isDappActive,
+    etherBalance,
+    gasCost,
+    isSmartAccount,
+    areAuxiliaryFundsSupported,
+  ]);
 
   return useAwaiter(validationContextAwaited).awaiter;
 };

features/stake/stake-form/use-stake.ts

@@ -9,7 +9,13 @@ import {
 } from '@lidofinance/lido-ethereum-sdk';
 
 import { config } from 'config';
-import { applyRoundUpGasLimit, useDappStatus, useLidoSDK } from 'modules/web3';
+import {
+  applyRoundUpGasLimit,
+  useAA,
+  useDappStatus,
+  useLidoSDK,
+  useSendAACalls,
+} from 'modules/web3';
 
 import { MockLimitReachedError, getRefferalAddress } from './utils';
 import { useTxModalStagesStake } from './hooks/use-tx-modal-stages-stake';
@@ -26,6 +32,8 @@ type StakeOptions = {
 
 export const useStake = ({ onConfirm, onRetry }: StakeOptions) => {
   const { address } = useDappStatus();
+  const { isAA } = useAA();
+  const sendAACalls = useSendAACalls();
   const { stake, stETH } = useLidoSDK();
   const { txModalStages } = useTxModalStagesStake();
 
@@ -42,11 +50,44 @@ export const useStake = ({ onConfirm, onRetry }: StakeOptions) => {
           throw new MockLimitReachedError('Stake limit reached');
         }
 
-        // eslint-disable-next-line @typescript-eslint/no-unused-vars
         const referralAddress = referral
           ? await getRefferalAddress(referral, stake.core.rpcProvider)
           : config.STAKE_FALLBACK_REFERRAL_ADDRESS;
 
+        //
+        // ERC5792 flow
+        //
+        if (isAA) {
+          const calls: unknown[] = [];
+          const steth = await stake.getContractStETH();
+          calls.push({
+            to: steth.address,
+            abi: steth.abi,
+            functionName: 'submit',
+            args: [referralAddress],
+            value: amount,
+          });
+
+          txModalStages.sign(amount);
+          const { txHash } = await sendAACalls(calls, (props) => {
+            if (props.stage === 'sent')
+              txModalStages.pending(amount, props.callId as Hash, isAA);
+          });
+
+          const [, balance] = await Promise.all([
+            onConfirm?.(),
+            stETH.balance(address),
+          ]);
+
+          txModalStages.success(balance, txHash);
+
+          return true;
+        }
+
+        //
+        // Legacy flow
+        //
+
         let txHash: Hash | undefined = undefined;
         const txCallback: TransactionCallback = async ({ stage, payload }) => {
           switch (stage) {
@@ -92,6 +133,15 @@ export const useStake = ({ onConfirm, onRetry }: StakeOptions) => {
         return false;
       }
     },
-    [address, stake, txModalStages, onConfirm, stETH, onRetry],
+    [
+      address,
+      stake,
+      isAA,
+      txModalStages,
+      sendAACalls,
+      onConfirm,
+      stETH,
+      onRetry,
+    ],
   );
 };

features/withdrawals/claim/form/requests-list/request-item.tsx

@@ -7,7 +7,7 @@ import {
   External,
   Tooltip,
 } from '@lidofinance/lido-ui';
-import type { LIDO_CONTRACT_NAMES } from '@lidofinance/lido-ethereum-sdk/common';
+import { LIDO_CONTRACT_NAMES } from '@lidofinance/lido-ethereum-sdk/common';
 
 import { FormatToken } from 'shared/formatters';
 import { useDappStatus, useContractAddress } from 'modules/web3';
@@ -36,7 +36,7 @@ export const RequestItem = forwardRef<HTMLInputElement, RequestItemProps>(
     });
 
     const { data: withdrawalQueueAddress } = useContractAddress(
-      'withdrawalQueue' as LIDO_CONTRACT_NAMES,
+      LIDO_CONTRACT_NAMES.withdrawalQueue,
     );
 
     const isDisabled =

features/withdrawals/claim/transaction-modal-claim/use-tx-modal-stages-claim.tsx

@@ -25,11 +25,12 @@ const getTxModalStagesClaim = (transitStage: TransactionModalTransitStage) => ({
       <TxStageSignOperationAmount {...STAGE_OPERATION_ARGS} amount={amount} />,
     ),
 
-  pending: (amount: bigint, txHash?: Hash) =>
+  pending: (amount: bigint, txHash?: Hash, isAA?: boolean) =>
     transitStage(
       <TxStageSignOperationAmount
         {...STAGE_OPERATION_ARGS}
         amount={amount}
+        isAA={isAA}
         txHash={txHash}
         isPending
       />,