fix: update full withdrawal tx modal text

Author: ev-d

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

@@ -82,6 +82,7 @@ export const WithdrawToVaultModalForm: FC<FormProps> = ({
       index,
       pubkey,
       validatorWithdrawalFee,
+      balance,
     },
     disabled:
       !isDappActive ||
@@ -104,8 +105,9 @@ export const WithdrawToVaultModalForm: FC<FormProps> = ({
       index,
       pubkey,
       validatorWithdrawalFee,
+      balance,
     });
-  }, [index, isPartial, pubkey, reset, validatorWithdrawalFee]);
+  }, [index, isPartial, pubkey, reset, validatorWithdrawalFee, balance]);
 
   const onSubmit = useCallback(
     async (formData: WithdrawalFormValidatedValues) => {

features/validators/modals/withdraw-to-vault-modal/hooks/use-withdrawal-to-vault.ts

@@ -34,6 +34,7 @@ export const useWithdrawalToVault = () => {
       index,
       pubkey,
       validatorWithdrawalFee,
+      balance,
     }: WithdrawalFormValidatedValues) => {
       invariant(activeVault, '[useWithdrawalToVault] activeVault is undefined');
       invariant(
@@ -46,9 +47,10 @@ export const useWithdrawalToVault = () => {
       );
 
       const amountInGwei = amount / WEI_PER_GWEI;
+      const withdrawalAmount = amount > 0n ? amount : balance;
 
-      const mainActionLoadingText = loadingText(index, amount);
-      const mainActionCompleteText = mainCompleteText(index, amount);
+      const mainActionLoadingText = loadingText(index, withdrawalAmount);
+      const mainActionCompleteText = mainCompleteText(index, withdrawalAmount);
 
       const prepareTransactions = async () => {
         const calls: TransactionEntry[] = [...prepareReportCalls()];

features/validators/modals/withdraw-to-vault-modal/types.ts

@@ -16,4 +16,5 @@ export type WithdrawalFormFieldValues = {
   index: WithdrawalFormValidatedValues['index'];
   pubkey: WithdrawalFormValidatedValues['pubkey'];
   validatorWithdrawalFee: WithdrawalFormValidatedValues['validatorWithdrawalFee'];
+  balance: WithdrawalFormValidatedValues['balance'];
 };

features/validators/modals/withdraw-to-vault-modal/validation.ts

@@ -34,6 +34,7 @@ export const withdrawalFormSchema = ({
     index: z.number(),
     pubkey: pubkeySchema,
     validatorWithdrawalFee: z.bigint(),
+    balance: z.bigint(),
   });
 };
 

features/validators/shared/components/validator-info/validator-info.tsx

@@ -37,6 +37,7 @@ export const ValidatorInfo: FC<PropsWithChildren<ValidatorInfoProps>> = ({
           <PubKeyWrapper>
             <Text size="xs" strong>
               <Address
+                as="span"
                 address={pubKey}
                 symbols={4}
                 style={{ fontWeight: 700 }}