Fix withdrawals by using withdraw from total balance (#34)

Author: heliuchuan

src/api/modules/aptos/index.ts

@@ -185,7 +185,7 @@ export const withdrawConfidentialBalance = async (
 ) => {
   const decryptionKey = new TwistedEd25519PrivateKey(decryptionKeyHex);
 
-  return confidentialAssets.withdraw({
+  return confidentialAssets.withdrawWithTotalBalance({
     signer: account,
     recipient: receiver,
     tokenAddress,

src/app/dashboard/components/WithdrawForm.tsx

@@ -172,7 +172,7 @@ export default function WithdrawForm({
           setIsSubmitting(false);
           return;
         }
-        const [txReceipt, withdrawErr] = await tryCatch(
+        const [responses, withdrawErr] = await tryCatch(
           withdrawTo(
             parseUnits(String(formData.amount), token.decimals).toString(),
             recipientAddress,
@@ -187,9 +187,15 @@ export default function WithdrawForm({
           setIsSubmitting(false);
           return;
         }
+        if (responses.length === 0) {
+          ErrorHandler.process(new Error('No responses were returned'));
+          enableForm();
+          setIsSubmitting(false);
+          return;
+        }
 
         const [, reloadError] = await tryCatch(
-          reloadBalances(BigInt(txReceipt.version)),
+          reloadBalances(BigInt(responses[responses.length - 1].version)),
         );
         if (reloadError) {
           ErrorHandler.process(reloadError);

src/app/dashboard/context.tsx

@@ -154,7 +154,7 @@ type ConfidentialCoinContextType = {
     opts?: {
       isSyncFirst?: boolean;
     },
-  ) => Promise<CommittedTransactionResponse>;
+  ) => Promise<CommittedTransactionResponse[]>;
   depositTo: (amount: bigint, to: string) => Promise<CommittedTransactionResponse>;
   depositCoinTo: (amount: bigint, to: string) => Promise<CommittedTransactionResponse>;
   // TODO: rotate keys
@@ -212,7 +212,7 @@ const confidentialCoinContext = createContext<ConfidentialCoinContextType>({
   rolloverAccount: async () => ({}) as CommittedTransactionResponse[],
   transfer: async () => ({}) as CommittedTransactionResponse[],
   buildWithdrawToTx: async () => ({}) as SimpleTransaction,
-  withdrawTo: async () => ({}) as CommittedTransactionResponse,
+  withdrawTo: async () => ({}) as CommittedTransactionResponse[],
   depositTo: async () => ({}) as CommittedTransactionResponse,
   depositCoinTo: async () => ({}) as CommittedTransactionResponse,