Smart Wallets with Aave

1
import { LocalAccountSigner } from "@aa-sdk/core";
2
import { createModularAccountAlchemyClient } from "@account-kit/smart-contracts";
3
import { alchemy } from "@account-kit/infra";
4
import { reserve, supply, withdraw } from "@aave/client/actions";
5
import {
6
  currentChain,
7
  getCurrentTokenAddress,
8
  getCurrentAavePoolAddress,
9
  config,
10
} from "./config";
11
12
// Supply assets to Aave
13
async function executeSupply() {
14
  const alchemyApiKey = process.env.ALCHEMY_API_KEY!;
15
  const privateKey = process.env.PRIVATE_KEY! as `0x${string}`;
16
  const policyId = process.env.PAYMASTER_POLICY_ID! as `0x${string}`;
17
18
  // Setup signer
19
  const signer = LocalAccountSigner.privateKeyToAccountSigner(privateKey);
20
21
  // Create smart account client with gas sponsorship
22
  const smartAccountClient = await createModularAccountAlchemyClient({
23
    transport: alchemy({ apiKey: alchemyApiKey }),
24
    policyId, // Enables gas sponsorship
25
    chain: currentChain,
26
    signer,
27
  });
28
29
  const accountAddress = smartAccountClient.account.address;
30
31
  // Get reserve data from Aave
32
  const reserveDataResult = await reserve(client, {
33
    market: evmAddress(getCurrentAavePoolAddress()),
34
    underlyingToken: evmAddress(getCurrentTokenAddress()),
35
    chainId: chainId(currentChain.id),
36
    user: accountAddress,
37
  });
38
39
  const reserveData = reserveDataResult.value;
40
41
  // Get supply transaction data
42
  const supplyResult = await supply(client, {
43
    market: reserveData.market.address,
44
    amount: {
45
      erc20: {
46
        currency: reserveData.underlyingToken.address,
47
        value: config.amounts.deposit,
48
      },
49
    },
50
    sender: evmAddress(accountAddress),
51
    chainId: reserveData.market.chain.chainId,
52
  });
53
54
  let supplyData = supplyResult.value;
55
56
  // Handle transactions (approval + supply or just supply)
57
  if (supplyData.__typename === "ApprovalRequired") {
58
    // Send approval transaction
59
    const approvalTx = await smartAccountClient.sendUserOperation({
60
      account: smartAccountClient.account,
61
      uo: {
62
        target: supplyData.approval.to,
63
        data: supplyData.approval.data,
64
      },
65
    });
66
67
    await smartAccountClient.getUserOperationReceipt(approvalTx.hash);
68
69
    // Send supply transaction
70
    const supplyTx = await smartAccountClient.sendUserOperation({
71
      account: smartAccountClient.account,
72
      uo: {
73
        target: supplyData.originalTransaction.to,
74
        data: supplyData.originalTransaction.data,
75
        value: BigInt(supplyData.originalTransaction.value),
76
      },
77
    });
78
79
    console.log("Supply completed! Hash:", supplyTx.hash);
80
  }
81
}
82
83
// Withdraw assets from Aave
84
async function executeWithdraw() {
85
  // Setup same as supply...
86
  const smartAccountClient = await createModularAccountAlchemyClient({
87
    transport: alchemy({ apiKey: alchemyApiKey }),
88
    policyId,
89
    chain: currentChain,
90
    signer,
91
  });
92
93
  // Get withdraw transaction data
94
  const withdrawResult = await withdraw(client, {
95
    market: reserveData.market.address,
96
    amount: {
97
      erc20: {
98
        currency: reserveData.underlyingToken.address,
99
        value: { exact: config.amounts.withdraw }, // or { max: true } for all
100
      },
101
    },
102
    sender: evmAddress(accountAddress),
103
    chainId: reserveData.market.chain.chainId,
104
  });
105
106
  let withdrawData = withdrawResult.value;
107
108
  if (withdrawData.__typename === "TransactionRequest") {
109
    const withdrawTx = await smartAccountClient.sendUserOperation({
110
      account: smartAccountClient.account,
111
      uo: {
112
        target: withdrawData.to,
113
        data: withdrawData.data,
114
        value: BigInt(withdrawData.value),
115
      },
116
    });
117
118
    console.log("Withdrawal completed! Hash:", withdrawTx.hash);
119
  }
120
}