refactor: remove partial withdrawal menu button

Author: ev-d

features/validators/const.ts

@@ -1,8 +1,7 @@
 import { WEI_PER_ETHER } from 'consts/tx';
 
 export enum VALIDATOR_MODALS {
-  partialWithdrawal = 'partialWithdrawal',
-  fullWithdrawal = 'fullWithdrawal',
+  withdrawalToVault = 'withdrawalToVault',
   topUpValidator = 'topUpValidator',
 }
 

features/validators/modals/withdraw-to-vault-modal/components/withdrawal-type/withdrawal-type.tsx

@@ -1,40 +1,31 @@
 import { type FC, useCallback } from 'react';
 
-import {
-  type ModalData,
-  useValidatorModal,
-} from 'features/validators/contexts';
-import { VALIDATOR_MODALS } from 'features/validators/const';
-
 import { SwitcherItemStyled, SwitcherStyled } from './styles';
 
+export type WithdrawalVariant = 'partial' | 'full';
+
 type WithdrawalTypeProps = {
-  modalData: ModalData;
+  value: WithdrawalVariant;
+  onChange: (value: WithdrawalVariant) => void;
 };
 
-export const WithdrawalType: FC<WithdrawalTypeProps> = ({ modalData }) => {
-  const { openModal } = useValidatorModal();
-  const { currentModal } = modalData;
-  const isPartial = VALIDATOR_MODALS.partialWithdrawal === currentModal;
-  const isFull = VALIDATOR_MODALS.fullWithdrawal === currentModal;
+export const WithdrawalType: FC<WithdrawalTypeProps> = ({
+  value,
+  onChange,
+}) => {
+  const isFull = value === 'full';
 
   const openPartialWithdrawal = useCallback(() => {
-    openModal({
-      ...modalData,
-      currentModal: VALIDATOR_MODALS.partialWithdrawal,
-    });
-  }, [openModal, modalData]);
+    onChange('partial');
+  }, [onChange]);
 
   const openFullWithdrawal = useCallback(() => {
-    openModal({
-      ...modalData,
-      currentModal: VALIDATOR_MODALS.fullWithdrawal,
-    });
-  }, [openModal, modalData]);
+    onChange('full');
+  }, [onChange]);
 
   return (
     <SwitcherStyled>
-      <SwitcherItemStyled active={isPartial} onClick={openPartialWithdrawal}>
+      <SwitcherItemStyled active={!isFull} onClick={openPartialWithdrawal}>
         Partial withdrawal
       </SwitcherItemStyled>
       <SwitcherItemStyled active={isFull} onClick={openFullWithdrawal}>

features/validators/modals/withdraw-to-vault-modal/withdraw-to-vault-modal.tsx

@@ -1,4 +1,4 @@
-import type { FC } from 'react';
+import { type FC, useState, useCallback } from 'react';
 import { Modal, Text } from '@lidofinance/lido-ui';
 
 import { vaultTexts } from 'modules/vaults';
@@ -25,29 +25,37 @@ export const WithdrawToVaultModal: FC<WithdrawToVaultModalProps> = ({
   modalData,
   onCloseModal,
 }) => {
+  const [withdrawalType, setWithdrawalType] = useState<'partial' | 'full'>(
+    'partial',
+  );
+
+  const handleCloseModal = useCallback(() => {
+    onCloseModal();
+    setWithdrawalType('partial');
+  }, [onCloseModal]);
+
   if (!modalData) {
     return null;
   }
 
   const { currentModal, pubKey, index, balance } = modalData;
+  const isPartial = withdrawalType === 'partial';
   const availableToPartialWithdraw = bigIntMax(
     0n,
     balance - MIN_ACTIVATION_BALANCE,
   );
-  const isPartial = VALIDATOR_MODALS.partialWithdrawal === currentModal;
-  const isFull = VALIDATOR_MODALS.fullWithdrawal === currentModal;
 
   return (
     <Modal
-      open={isPartial || isFull}
-      onClose={onCloseModal}
+      open={VALIDATOR_MODALS.withdrawalToVault === currentModal}
+      onClose={handleCloseModal}
       windowSize="md"
       title={title}
     >
       <ContentContainer>
         <Text size="xs">{description}</Text>
         <ValidatorInfo pubKey={pubKey} index={index} balance={balance} />
-        <WithdrawalType modalData={modalData} />
+        <WithdrawalType value={withdrawalType} onChange={setWithdrawalType} />
         <WithdrawToVaultModalForm
           isPartial={isPartial}
           availableToPartialWithdraw={availableToPartialWithdraw}

features/validators/shared/hooks/use-validator-menu-actions.ts

@@ -12,7 +12,7 @@ type ValidatorMenuActionItem = {
   modal: ValidatorsModalItem;
 };
 
-const { topUpValidator, partialWithdrawal, withdrawToStVault } =
+const { topUpValidator, withdrawToStVault } =
   vaultTexts.actions.validators.table.menu;
 
 export const useValidatorMenuActions = () => {
@@ -26,15 +26,10 @@ export const useValidatorMenuActions = () => {
         label: topUpValidator,
         modal: VALIDATOR_MODALS.topUpValidator,
       },
-      {
-        hasPermission: isAdmin || hasWithdrawalPermission,
-        label: partialWithdrawal,
-        modal: VALIDATOR_MODALS.partialWithdrawal,
-      },
       {
         hasPermission: isAdmin || hasWithdrawalPermission,
         label: withdrawToStVault,
-        modal: VALIDATOR_MODALS.fullWithdrawal,
+        modal: VALIDATOR_MODALS.withdrawalToVault,
       },
     ],
     [isAdmin, hasDepositorPermission, hasWithdrawalPermission],

modules/vaults/consts/texts.ts

@@ -333,7 +333,6 @@ export const vaultTexts = {
             'Failed to load validator data. Please try again later',
         },
         menu: {
-          partialWithdrawal: 'Partial withdrawal',
           topUpValidator: 'Top up validator',
           withdrawToStVault: 'Withdraw to stVault',
         },