feat(withdraw): allow a useMax field in withdraw shortcuts to withdraw the maximum amount on aave (#531)

Author: jh2oman

src/apps/aave/shortcuts.ts

@@ -1,4 +1,10 @@
-import { Address, encodeFunctionData, parseUnits, erc20Abi } from 'viem'
+import {
+  Address,
+  encodeFunctionData,
+  parseUnits,
+  erc20Abi,
+  maxUint256,
+} from 'viem'
 import {
   createShortcut,
   ShortcutsHook,
@@ -141,7 +147,10 @@ const hook: ShortcutsHook = {
           const transactions: Transaction[] = []
 
           // amount in smallest unit
-          const amountToWithdraw = parseUnits(tokens[0].amount, tokenDecimals)
+          // useMax Withdraws entire balance https://docs.aave.com/developers/core-contracts/pool#withdraw
+          const amountToWithdraw = tokens[0].useMax
+            ? maxUint256
+            : parseUnits(tokens[0].amount, tokenDecimals)
 
           const client = getClient(networkId)
 

src/types/shortcuts.ts

@@ -15,6 +15,7 @@ export const tokenAmounts = z
     z.object({
       tokenId: z.string(),
       amount: z.string(),
+      useMax: z.boolean().optional(),
     }),
   )
   .nonempty()