feat: sweep-all

Kodylow · Nov 10, 2023 · ebb214c · ebb214c
1 parent 9ce8ee6
commit ebb214c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/apps/gateway-ui/src/GatewayApi.ts b/apps/gateway-ui/src/GatewayApi.ts
@@ -117,7 +117,7 @@ export class GatewayApi {
 
   requestWithdrawal = async (
     federationId: string,
-    amountSat: number,
+    amountSat: number | 'all',
     address: string
   ): Promise<string> => {
     try {

diff --git a/apps/gateway-ui/src/components/WithdrawCard.tsx b/apps/gateway-ui/src/components/WithdrawCard.tsx
@@ -56,9 +56,16 @@ export const WithdrawCard = React.memo(function WithdrawCard({
     setModalState(true);
   }, [amount, address, t]);
 
+  const sweepAll = useCallback(() => {
+    setAmount(balanceMsat / 1000);
+    createWithdrawal();
+  }, [balanceMsat, createWithdrawal]);
+
   const startWithdrawal = useCallback(() => {
+    const withdrawalAmount = amount === balanceMsat / 1000 ? 'all' : amount;
+
     gateway
-      .requestWithdrawal(federationId, amount, address)
+      .requestWithdrawal(federationId, withdrawalAmount, address)
       .then((txId) => {
         alert(`${t('withdraw-card.your-transaction')} ${txId}`);
         setAddress('');
@@ -69,7 +76,7 @@ export const WithdrawCard = React.memo(function WithdrawCard({
         console.error(error);
         setError(`${t('withdraw-card.error-request')}`);
       });
-  }, [gateway, federationId, amount, address, t]);
+  }, [gateway, federationId, amount, address, balanceMsat, t]);
 
   return (
     <>
@@ -161,6 +168,15 @@ export const WithdrawCard = React.memo(function WithdrawCard({
           >
             {t('withdraw-card.card-header')}
           </Button>
+          <Button
+            borderRadius='8px'
+            maxW='145px'
+            isDisabled={!address}
+            fontSize='sm'
+            onClick={sweepAll}
+          >
+            {t('withdraw-card.sweep-all')}
+          </Button>
         </Stack>
       </GatewayCard>