@aztec/cli-wallet 0.0.1-commit.d431d1c → 0.0.1-commit.d939eb5aa

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (46) hide show
  1. package/dest/cmds/authorize_action.d.ts +2 -2
  2. package/dest/cmds/authorize_action.d.ts.map +1 -1
  3. package/dest/cmds/authorize_action.js +4 -2
  4. package/dest/cmds/check_tx.d.ts +1 -1
  5. package/dest/cmds/check_tx.d.ts.map +1 -1
  6. package/dest/cmds/check_tx.js +41 -10
  7. package/dest/cmds/create_account.d.ts +4 -3
  8. package/dest/cmds/create_account.d.ts.map +1 -1
  9. package/dest/cmds/create_account.js +42 -25
  10. package/dest/cmds/deploy.d.ts +4 -3
  11. package/dest/cmds/deploy.d.ts.map +1 -1
  12. package/dest/cmds/deploy.js +56 -29
  13. package/dest/cmds/deploy_account.d.ts +4 -3
  14. package/dest/cmds/deploy_account.d.ts.map +1 -1
  15. package/dest/cmds/deploy_account.js +37 -20
  16. package/dest/cmds/get_fee_juice_balance.d.ts +5 -0
  17. package/dest/cmds/get_fee_juice_balance.d.ts.map +1 -0
  18. package/dest/cmds/get_fee_juice_balance.js +27 -0
  19. package/dest/cmds/index.d.ts +1 -1
  20. package/dest/cmds/index.d.ts.map +1 -1
  21. package/dest/cmds/index.js +30 -14
  22. package/dest/cmds/send.d.ts +4 -3
  23. package/dest/cmds/send.d.ts.map +1 -1
  24. package/dest/cmds/send.js +32 -16
  25. package/dest/storage/wallet_db.d.ts +6 -1
  26. package/dest/storage/wallet_db.d.ts.map +1 -1
  27. package/dest/storage/wallet_db.js +60 -31
  28. package/dest/utils/options/fees.d.ts +3 -2
  29. package/dest/utils/options/fees.d.ts.map +1 -1
  30. package/dest/utils/options/fees.js +32 -9
  31. package/dest/utils/wallet.d.ts +11 -5
  32. package/dest/utils/wallet.d.ts.map +1 -1
  33. package/dest/utils/wallet.js +73 -49
  34. package/package.json +14 -14
  35. package/src/cmds/authorize_action.ts +1 -1
  36. package/src/cmds/check_tx.ts +47 -12
  37. package/src/cmds/create_account.ts +44 -24
  38. package/src/cmds/deploy.ts +55 -27
  39. package/src/cmds/deploy_account.ts +38 -19
  40. package/src/cmds/get_fee_juice_balance.ts +37 -0
  41. package/src/cmds/index.ts +57 -4
  42. package/src/cmds/send.ts +32 -12
  43. package/src/cmds/simulate.ts +1 -1
  44. package/src/storage/wallet_db.ts +63 -36
  45. package/src/utils/options/fees.ts +39 -9
  46. package/src/utils/wallet.ts +84 -69
@@ -12,6 +12,7 @@ export const Aliases = ['accounts', 'contracts', 'artifacts', 'secrets', 'transa
12
12
  export type AliasType = (typeof Aliases)[number];
13
13
 
14
14
  export class WalletDB {
15
+ #store!: AztecAsyncKVStore;
15
16
  #accounts!: AztecAsyncMap<string, Buffer>;
16
17
  #aliases!: AztecAsyncMap<string, Buffer>;
17
18
  #bridgedFeeJuice!: AztecAsyncMap<string, Buffer>;
@@ -29,6 +30,7 @@ export class WalletDB {
29
30
  }
30
31
 
31
32
  async init(store: AztecAsyncKVStore) {
33
+ this.#store = store;
32
34
  this.#accounts = store.openMap('accounts');
33
35
  this.#aliases = store.openMap('aliases');
34
36
  this.#bridgedFeeJuice = store.openMap('bridgedFeeJuice');
@@ -41,14 +43,17 @@ export class WalletDB {
41
43
  }
42
44
 
43
45
  async pushBridgedFeeJuice(recipient: AztecAddress, secret: Fr, amount: bigint, leafIndex: bigint, log: LogFn) {
44
- let stackPointer = (await this.#bridgedFeeJuice.getAsync(`${recipient.toString()}:stackPointer`))?.readInt8() || 0;
45
- stackPointer++;
46
- await this.#bridgedFeeJuice.set(
47
- `${recipient.toString()}:${stackPointer}`,
48
- Buffer.from(`${amount.toString()}:${secret.toString()}:${leafIndex.toString()}`),
49
- );
50
- await this.#bridgedFeeJuice.set(`${recipient.toString()}:stackPointer`, Buffer.from([stackPointer]));
51
- log(`Pushed ${amount} fee juice for recipient ${recipient.toString()}. Stack pointer ${stackPointer}`);
46
+ await this.#store.transactionAsync(async () => {
47
+ let stackPointer =
48
+ (await this.#bridgedFeeJuice.getAsync(`${recipient.toString()}:stackPointer`))?.readInt8() || 0;
49
+ stackPointer++;
50
+ await this.#bridgedFeeJuice.set(
51
+ `${recipient.toString()}:${stackPointer}`,
52
+ Buffer.from(`${amount.toString()}:${secret.toString()}:${leafIndex.toString()}`),
53
+ );
54
+ await this.#bridgedFeeJuice.set(`${recipient.toString()}:stackPointer`, Buffer.from([stackPointer]));
55
+ log(`Pushed ${amount} fee juice for recipient ${recipient.toString()}. Stack pointer ${stackPointer}`);
56
+ });
52
57
  }
53
58
 
54
59
  async popBridgedFeeJuice(recipient: AztecAddress, log: LogFn) {
@@ -65,6 +70,20 @@ export class WalletDB {
65
70
  return { amount: BigInt(amountStr), secret: secretStr, leafIndex: BigInt(leafIndexStr) };
66
71
  }
67
72
 
73
+ async peekBridgedFeeJuice(recipient: AztecAddress, log: LogFn) {
74
+ const stackPointer =
75
+ (await this.#bridgedFeeJuice.getAsync(`${recipient.toString()}:stackPointer`))?.readInt8() || 0;
76
+ const result = await this.#bridgedFeeJuice.getAsync(`${recipient.toString()}:${stackPointer}`);
77
+ if (!result) {
78
+ throw new Error(
79
+ `No stored fee juice available for recipient ${recipient.toString()}. Please provide claim amount and secret. Stack pointer ${stackPointer}`,
80
+ );
81
+ }
82
+ const [amountStr, secretStr, leafIndexStr] = result.toString().split(':');
83
+ log(`Peeked ${amountStr} fee juice for recipient ${recipient.toString()}. Stack pointer ${stackPointer}`);
84
+ return { amount: BigInt(amountStr), secret: secretStr, leafIndex: BigInt(leafIndexStr) };
85
+ }
86
+
68
87
  async storeAccount(
69
88
  address: AztecAddress,
70
89
  {
@@ -76,19 +95,24 @@ export class WalletDB {
76
95
  }: { type: AccountType; secretKey: Fr; salt: Fr; alias: string | undefined; publicKey: string | undefined },
77
96
  log: LogFn,
78
97
  ) {
79
- if (alias) {
80
- await this.#aliases.set(`accounts:${alias}`, Buffer.from(address.toString()));
81
- }
82
- await this.#accounts.set(`${address.toString()}:type`, Buffer.from(type));
83
- await this.#accounts.set(`${address.toString()}:sk`, secretKey.toBuffer());
84
- await this.#accounts.set(`${address.toString()}:salt`, salt.toBuffer());
98
+ let publicSigningKey: Buffer | undefined;
85
99
  if (type === 'ecdsasecp256r1ssh' && publicKey) {
86
- const publicSigningKey = extractECDSAPublicKeyFromBase64String(publicKey);
87
- await this.storeAccountMetadata(address, 'publicSigningKey', publicSigningKey);
100
+ publicSigningKey = extractECDSAPublicKeyFromBase64String(publicKey);
88
101
  }
89
- await this.#aliases.set('accounts:last', Buffer.from(address.toString()));
90
- log(`Account stored in database with alias${alias ? `es last & ${alias}` : ' last'}`);
91
102
 
103
+ await this.#store.transactionAsync(async () => {
104
+ if (alias) {
105
+ await this.#aliases.set(`accounts:${alias}`, Buffer.from(address.toString()));
106
+ }
107
+ await this.#accounts.set(`${address.toString()}:type`, Buffer.from(type));
108
+ await this.#accounts.set(`${address.toString()}:sk`, secretKey.toBuffer());
109
+ await this.#accounts.set(`${address.toString()}:salt`, salt.toBuffer());
110
+ if (publicSigningKey) {
111
+ await this.#accounts.set(`${address.toString()}:publicSigningKey`, publicSigningKey);
112
+ }
113
+ await this.#aliases.set('accounts:last', Buffer.from(address.toString()));
114
+ });
115
+ log(`Account stored in database with alias${alias ? `es last & ${alias}` : ' last'}`);
92
116
  await this.refreshAliasCache();
93
117
  }
94
118
 
@@ -100,35 +124,38 @@ export class WalletDB {
100
124
  }
101
125
 
102
126
  async storeContract(address: AztecAddress, artifactPath: string, log: LogFn, alias?: string) {
103
- if (alias) {
104
- await this.#aliases.set(`contracts:${alias}`, Buffer.from(address.toString()));
105
- await this.#aliases.set(`artifacts:${alias}`, Buffer.from(artifactPath));
106
- }
107
- await this.#aliases.set(`contracts:last`, Buffer.from(address.toString()));
108
- await this.#aliases.set(`artifacts:last`, Buffer.from(artifactPath));
109
- await this.#aliases.set(`artifacts:${address.toString()}`, Buffer.from(artifactPath));
127
+ await this.#store.transactionAsync(async () => {
128
+ if (alias) {
129
+ await this.#aliases.set(`contracts:${alias}`, Buffer.from(address.toString()));
130
+ await this.#aliases.set(`artifacts:${alias}`, Buffer.from(artifactPath));
131
+ }
132
+ await this.#aliases.set(`contracts:last`, Buffer.from(address.toString()));
133
+ await this.#aliases.set(`artifacts:last`, Buffer.from(artifactPath));
134
+ await this.#aliases.set(`artifacts:${address.toString()}`, Buffer.from(artifactPath));
135
+ });
110
136
  log(`Contract stored in database with alias${alias ? `es last & ${alias}` : ' last'}`);
111
-
112
137
  await this.refreshAliasCache();
113
138
  }
114
139
 
115
140
  async storeAuthwitness(authWit: AuthWitness, log: LogFn, alias?: string) {
116
- if (alias) {
117
- await this.#aliases.set(`authwits:${alias}`, Buffer.from(authWit.toString()));
118
- }
119
- await this.#aliases.set(`authwits:last`, Buffer.from(authWit.toString()));
141
+ await this.#store.transactionAsync(async () => {
142
+ if (alias) {
143
+ await this.#aliases.set(`authwits:${alias}`, Buffer.from(authWit.toString()));
144
+ }
145
+ await this.#aliases.set(`authwits:last`, Buffer.from(authWit.toString()));
146
+ });
120
147
  log(`Authorization witness stored in database with alias${alias ? `es last & ${alias}` : ' last'}`);
121
-
122
148
  await this.refreshAliasCache();
123
149
  }
124
150
 
125
151
  async storeTx({ txHash }: { txHash: TxHash }, log: LogFn, alias?: string) {
126
- if (alias) {
127
- await this.#aliases.set(`transactions:${alias}`, Buffer.from(txHash.toString()));
128
- }
129
- await this.#aliases.set(`transactions:last`, Buffer.from(txHash.toString()));
152
+ await this.#store.transactionAsync(async () => {
153
+ if (alias) {
154
+ await this.#aliases.set(`transactions:${alias}`, Buffer.from(txHash.toString()));
155
+ }
156
+ await this.#aliases.set(`transactions:last`, Buffer.from(txHash.toString()));
157
+ });
130
158
  log(`Transaction hash stored in database with alias${alias ? `es last & ${alias}` : ' last'}`);
131
-
132
159
  await this.refreshAliasCache();
133
160
  }
134
161
 
@@ -5,7 +5,7 @@ import { Fr } from '@aztec/foundation/curves/bn254';
5
5
  import type { LogFn } from '@aztec/foundation/log';
6
6
  import type { FieldsOf } from '@aztec/foundation/types';
7
7
  import { AztecAddress } from '@aztec/stdlib/aztec-address';
8
- import { Gas, GasFees, GasSettings } from '@aztec/stdlib/gas';
8
+ import { Gas, GasFees, GasSettings, ManaUsageEstimate } from '@aztec/stdlib/gas';
9
9
  import type { FeeOptions } from '@aztec/wallet-sdk/base-wallet';
10
10
 
11
11
  import { Option } from 'commander';
@@ -115,6 +115,7 @@ export function parsePaymentMethod(
115
115
  payment: string,
116
116
  log: LogFn,
117
117
  db?: WalletDB,
118
+ estimateOnly?: boolean,
118
119
  ): (wallet: Wallet, from: AztecAddress, gasSettings: GasSettings) => Promise<FeePaymentMethod | undefined> {
119
120
  const parsed = payment.split(',').reduce(
120
121
  (acc, item) => {
@@ -149,7 +150,7 @@ export function parsePaymentMethod(
149
150
  amount: claimAmount,
150
151
  secret: claimSecret,
151
152
  leafIndex: messageLeafIndex,
152
- } = await db.popBridgedFeeJuice(from, log));
153
+ } = estimateOnly ? await db.peekBridgedFeeJuice(from, log) : await db.popBridgedFeeJuice(from, log));
153
154
  } else {
154
155
  ({ claimAmount, claimSecret, messageLeafIndex } = parsed);
155
156
  }
@@ -157,10 +158,10 @@ export function parsePaymentMethod(
157
158
  const { FeeJuicePaymentMethodWithClaim } = await import('@aztec/aztec.js/fee');
158
159
  return new FeeJuicePaymentMethodWithClaim(from, {
159
160
  claimAmount: (typeof claimAmount === 'string'
160
- ? Fr.fromHexString(claimAmount)
161
+ ? Fr.fromString(claimAmount)
161
162
  : new Fr(claimAmount)
162
163
  ).toBigInt(),
163
- claimSecret: Fr.fromHexString(claimSecret),
164
+ claimSecret: typeof claimSecret === 'string' ? Fr.fromString(claimSecret) : claimSecret,
164
165
  messageLeafIndex: BigInt(messageLeafIndex),
165
166
  });
166
167
  } else {
@@ -171,6 +172,9 @@ export function parsePaymentMethod(
171
172
  case 'fpc-public': {
172
173
  const fpc = getFpc();
173
174
  const asset = getAsset();
175
+ log(
176
+ `WARNING: fpc-public is deprecated and will not work on mainnet alpha. Use fee_juice or fpc-sponsored instead.`,
177
+ );
174
178
  log(`Using public fee payment with asset ${asset} via paymaster ${fpc}`);
175
179
  const { PublicFeePaymentMethod } = await import('@aztec/aztec.js/fee');
176
180
  return new PublicFeePaymentMethod(fpc, from, wallet, gasSettings);
@@ -178,6 +182,9 @@ export function parsePaymentMethod(
178
182
  case 'fpc-private': {
179
183
  const fpc = getFpc();
180
184
  const asset = getAsset();
185
+ log(
186
+ `WARNING: fpc-private is deprecated and will not work on mainnet alpha. Use fee_juice or fpc-sponsored instead.`,
187
+ );
181
188
  log(`Using private fee payment with asset ${asset} via paymaster ${fpc}`);
182
189
  const { PrivateFeePaymentMethod } = await import('@aztec/aztec.js/fee');
183
190
  return new PrivateFeePaymentMethod(fpc, from, wallet, gasSettings);
@@ -250,8 +257,9 @@ export class CLIFeeArgs {
250
257
  ) {}
251
258
 
252
259
  async toUserFeeOptions(node: AztecNode, wallet: Wallet, from: AztecAddress): Promise<ParsedFeeOptions> {
253
- const maxFeesPerGas = (await node.getCurrentMinFees()).mul(1 + MIN_FEE_PADDING);
254
- const gasSettings = GasSettings.default({ ...this.gasSettings, maxFeesPerGas });
260
+ const minFees = await this.getMinFees(node);
261
+ const maxFeesPerGas = minFees.mul(1 + MIN_FEE_PADDING);
262
+ const gasSettings = GasSettings.fallback({ ...this.gasSettings, maxFeesPerGas });
255
263
  const paymentMethod = await this.paymentMethod(wallet, from, gasSettings);
256
264
  return {
257
265
  paymentMethod,
@@ -259,10 +267,32 @@ export class CLIFeeArgs {
259
267
  };
260
268
  }
261
269
 
270
+ /**
271
+ * Returns the worst-case min fee across predicted future slots.
272
+ * Falls back to getCurrentMinFees if the node doesn't support getPredictedMinFees.
273
+ */
274
+ private async getMinFees(node: AztecNode): Promise<GasFees> {
275
+ try {
276
+ const predicted = await node.getPredictedMinFees(ManaUsageEstimate.Limit);
277
+ if (predicted.length === 0) {
278
+ return node.getCurrentMinFees();
279
+ }
280
+ return predicted.reduce((worst, fees) => (fees.feePerL2Gas > worst.feePerL2Gas ? fees : worst));
281
+ } catch (err: any) {
282
+ // Fallback for old nodes that don't support getPredictedMinFees.
283
+ // Only fall back on method-not-found errors (JSON-RPC code -32601); rethrow others.
284
+ if (err?.cause?.code === -32601 || err?.message?.includes('Method not found')) {
285
+ return node.getCurrentMinFees();
286
+ }
287
+ throw err;
288
+ }
289
+ }
290
+
262
291
  static parse(args: RawCliFeeArgs, log: LogFn, db?: WalletDB): CLIFeeArgs {
292
+ const estimateOnly = !!args.estimateGasOnly;
263
293
  return new CLIFeeArgs(
264
- !!args.estimateGasOnly,
265
- parsePaymentMethod(args.payment ?? 'method=fee_juice', log, db),
294
+ estimateOnly,
295
+ parsePaymentMethod(args.payment ?? 'method=fee_juice', log, db, estimateOnly),
266
296
  parseGasSettings(args),
267
297
  );
268
298
  }
@@ -288,7 +318,7 @@ function formatGasEstimate(estimate: Pick<GasSettings, 'gasLimits' | 'teardownGa
288
318
  }
289
319
 
290
320
  function getEstimatedCost(estimate: Pick<GasSettings, 'gasLimits' | 'teardownGasLimits'>, maxFeesPerGas: GasFees) {
291
- return GasSettings.default({ ...estimate, maxFeesPerGas })
321
+ return GasSettings.fallback({ ...estimate, maxFeesPerGas })
292
322
  .getFeeLimit()
293
323
  .toBigInt();
294
324
  }
@@ -1,8 +1,9 @@
1
1
  import { EcdsaRAccountContract, EcdsaRSSHAccountContract } from '@aztec/accounts/ecdsa';
2
2
  import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
3
- import { StubAccountContractArtifact, createStubAccount } from '@aztec/accounts/stub';
3
+ import { StubEcdsaAccountContractArtifact, createStubEcdsaAccount } from '@aztec/accounts/stub/ecdsa';
4
+ import { StubSchnorrAccountContractArtifact, createStubSchnorrAccount } from '@aztec/accounts/stub/schnorr';
4
5
  import { getIdentities } from '@aztec/accounts/utils';
5
- import { type Account, type AccountContract, SignerlessAccount } from '@aztec/aztec.js/account';
6
+ import { type Account, type AccountContract, NO_FROM } from '@aztec/aztec.js/account';
6
7
  import {
7
8
  type InteractionFeeOptions,
8
9
  getContractInstanceFromInstantiationParams,
@@ -10,19 +11,21 @@ import {
10
11
  } from '@aztec/aztec.js/contracts';
11
12
  import type { AztecNode } from '@aztec/aztec.js/node';
12
13
  import { AccountManager, type Aliased, type SimulateOptions } from '@aztec/aztec.js/wallet';
14
+ import { TxSimulationResultWithAppOffset } from '@aztec/aztec.js/wallet';
13
15
  import type { DefaultAccountEntrypointOptions } from '@aztec/entrypoints/account';
16
+ import { DefaultEntrypoint } from '@aztec/entrypoints/default';
14
17
  import { Fr } from '@aztec/foundation/curves/bn254';
15
18
  import type { LogFn } from '@aztec/foundation/log';
19
+ import type { NotesFilter } from '@aztec/pxe/client/lazy';
16
20
  import type { PXEConfig } from '@aztec/pxe/config';
17
21
  import type { PXE } from '@aztec/pxe/server';
18
22
  import { createPXE, getPXEConfig } from '@aztec/pxe/server';
19
23
  import { AztecAddress } from '@aztec/stdlib/aztec-address';
20
24
  import { deriveSigningKey } from '@aztec/stdlib/keys';
21
25
  import { NoteDao } from '@aztec/stdlib/note';
22
- import type { NotesFilter } from '@aztec/stdlib/note';
23
- import type { TxProvingResult, TxSimulationResult } from '@aztec/stdlib/tx';
26
+ import type { SimulationOverrides, TxExecutionRequest, TxProvingResult } from '@aztec/stdlib/tx';
24
27
  import { ExecutionPayload, mergeExecutionPayloads } from '@aztec/stdlib/tx';
25
- import { BaseWallet } from '@aztec/wallet-sdk/base-wallet';
28
+ import { BaseWallet, type SimulateViaEntrypointOptions } from '@aztec/wallet-sdk/base-wallet';
26
29
 
27
30
  import type { WalletDB } from '../storage/wallet_db.js';
28
31
  import type { AccountType } from './constants.js';
@@ -55,7 +58,12 @@ export class CLIWallet extends BaseWallet {
55
58
 
56
59
  override async getAccounts(): Promise<Aliased<AztecAddress>[]> {
57
60
  const accounts = (await this.db?.listAliases('accounts')) ?? [];
58
- return Promise.resolve(accounts.map(({ key, value }) => ({ alias: value, item: AztecAddress.fromString(key) })));
61
+ return Promise.resolve(
62
+ accounts.map(({ key, value }) => {
63
+ const alias = key.includes(':') ? key.slice(key.indexOf(':') + 1) : key;
64
+ return { alias, item: AztecAddress.fromString(value) };
65
+ }),
66
+ );
59
67
  }
60
68
 
61
69
  private async createCancellationTxExecutionRequest(
@@ -64,14 +72,19 @@ export class CLIWallet extends BaseWallet {
64
72
  increasedFee: InteractionFeeOptions,
65
73
  ) {
66
74
  const executionPayload = ExecutionPayload.empty();
67
- const feeOptions = await this.completeFeeOptions(from, executionPayload.feePayer, increasedFee.gasSettings);
75
+ const feeOptions = await this.completeFeeOptions({
76
+ from,
77
+ feePayer: executionPayload.feePayer,
78
+ gasSettings: increasedFee.gasSettings,
79
+ });
68
80
  const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
69
81
  const fromAccount = await this.getAccountFromAddress(from);
70
82
  const chainInfo = await this.getChainInfo();
71
83
  const executionOptions: DefaultAccountEntrypointOptions = {
72
84
  txNonce,
73
85
  cancellable: this.cancellableTransactions,
74
- feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions,
86
+ // If from is an address, feeOptions include the way the account contract should handle the fee payment
87
+ feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions!,
75
88
  };
76
89
  return await fromAccount.createTxExecutionRequest(
77
90
  feeExecutionPayload ?? executionPayload,
@@ -87,14 +100,12 @@ export class CLIWallet extends BaseWallet {
87
100
  increasedFee: InteractionFeeOptions,
88
101
  ): Promise<TxProvingResult> {
89
102
  const cancellationTxRequest = await this.createCancellationTxExecutionRequest(from, txNonce, increasedFee);
90
- return await this.pxe.proveTx(cancellationTxRequest);
103
+ return await this.pxe.proveTx(cancellationTxRequest, this.scopesFrom(from));
91
104
  }
92
105
 
93
106
  override async getAccountFromAddress(address: AztecAddress) {
94
107
  let account: Account | undefined;
95
- if (address.equals(AztecAddress.ZERO)) {
96
- account = new SignerlessAccount();
97
- } else if (this.accountCache.has(address.toString())) {
108
+ if (this.accountCache.has(address.toString())) {
98
109
  return this.accountCache.get(address.toString())!;
99
110
  } else {
100
111
  const accountManager = await this.createOrRetrieveAccount(address);
@@ -185,82 +196,86 @@ export class CLIWallet extends BaseWallet {
185
196
  */
186
197
  private async getFakeAccountDataFor(address: AztecAddress) {
187
198
  const originalAccount = await this.getAccountFromAddress(address);
188
- // Account contracts can only be overridden if they have an associated address
189
- // Overwriting SignerlessAccount is not supported, and does not really make sense
190
- // since it has no authorization mechanism.
191
- if (originalAccount instanceof SignerlessAccount) {
192
- throw new Error(`Cannot create fake account data for SignerlessAccount at address: ${address}`);
193
- }
194
- const originalAddress = (originalAccount as Account).getCompleteAddress();
199
+ const originalAddress = originalAccount.getCompleteAddress();
195
200
  const contractInstance = await this.pxe.getContractInstance(originalAddress.address);
196
201
  if (!contractInstance) {
197
202
  throw new Error(`No contract instance found for address: ${originalAddress.address}`);
198
203
  }
199
- const stubAccount = createStubAccount(originalAddress);
200
- const instance = await getContractInstanceFromInstantiationParams(StubAccountContractArtifact, {
201
- salt: Fr.random(),
202
- });
203
- return {
204
- account: stubAccount,
205
- instance,
206
- artifact: StubAccountContractArtifact,
207
- };
204
+ const { type } = await this.db!.retrieveAccount(address);
205
+ const isSchnorr = type === 'schnorr';
206
+ const artifact = isSchnorr ? StubSchnorrAccountContractArtifact : StubEcdsaAccountContractArtifact;
207
+ const stubAccount = isSchnorr ? createStubSchnorrAccount(originalAddress) : createStubEcdsaAccount(originalAddress);
208
+ const instance = await getContractInstanceFromInstantiationParams(artifact, { salt: Fr.random() });
209
+ return { account: stubAccount, instance, artifact };
208
210
  }
209
211
 
210
- override async simulateTx(executionPayload: ExecutionPayload, opts: SimulateOptions): Promise<TxSimulationResult> {
211
- let simulationResults;
212
- const feeOptions = opts.fee?.estimateGas
213
- ? await this.completeFeeOptionsForEstimation(opts.from, executionPayload.feePayer, opts.fee?.gasSettings)
214
- : await this.completeFeeOptions(opts.from, executionPayload.feePayer, opts.fee?.gasSettings);
212
+ override async simulateTx(
213
+ executionPayload: ExecutionPayload,
214
+ opts: SimulateOptions,
215
+ ): Promise<TxSimulationResultWithAppOffset> {
216
+ const simulationResults = await super.simulateTx(executionPayload, opts);
217
+
218
+ if (opts.fee?.estimateGas) {
219
+ const feeOptions = await this.completeFeeOptions({
220
+ from: opts.from,
221
+ feePayer: executionPayload.feePayer,
222
+ gasSettings: opts.fee?.gasSettings,
223
+ });
224
+ const limits = getGasLimits(simulationResults, opts.fee?.estimatedGasPadding);
225
+ printGasEstimates(feeOptions, limits, this.userLog);
226
+ }
227
+ return simulationResults;
228
+ }
229
+
230
+ /**
231
+ * Uses a stub account for kernelless simulation, bypassing real account authorization.
232
+ * Uses DefaultEntrypoint directly for NO_FROM transactions.
233
+ */
234
+ protected override async simulateViaEntrypoint(
235
+ executionPayload: ExecutionPayload,
236
+ opts: SimulateViaEntrypointOptions,
237
+ ): Promise<TxSimulationResultWithAppOffset> {
238
+ const { from, feeOptions, additionalScopes } = opts;
239
+ const scopes = this.scopesFrom(from, additionalScopes);
215
240
  const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
216
- const chainInfo = await this.getChainInfo();
217
- const executionOptions: DefaultAccountEntrypointOptions = {
218
- txNonce: Fr.random(),
219
- cancellable: this.cancellableTransactions,
220
- feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions,
221
- };
222
241
  const finalExecutionPayload = feeExecutionPayload
223
242
  ? mergeExecutionPayloads([feeExecutionPayload, executionPayload])
224
243
  : executionPayload;
244
+ const chainInfo = await this.getChainInfo();
225
245
 
226
- // Kernelless simulations using the multicall entrypoints are not currently supported,
227
- // since we only override proper account contracts.
228
- // TODO: allow disabling kernels even when no overrides are necessary
229
- if (opts.from.equals(AztecAddress.ZERO)) {
230
- const fromAccount = await this.getAccountFromAddress(opts.from);
231
- const txRequest = await fromAccount.createTxExecutionRequest(
232
- finalExecutionPayload,
233
- feeOptions.gasSettings,
234
- chainInfo,
235
- executionOptions,
236
- );
237
- simulationResults = await this.pxe.simulateTx(
238
- txRequest,
239
- true /* simulatePublic */,
240
- opts?.skipTxValidation,
241
- opts?.skipFeeEnforcement ?? true,
242
- );
246
+ let overrides: SimulationOverrides | undefined;
247
+ let txRequest: TxExecutionRequest;
248
+ if (from === NO_FROM) {
249
+ const entrypoint = new DefaultEntrypoint();
250
+ txRequest = await entrypoint.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo);
243
251
  } else {
244
- const { account: fromAccount, instance, artifact } = await this.getFakeAccountDataFor(opts.from);
245
- const txRequest = await fromAccount.createTxExecutionRequest(
252
+ const { account, instance, artifact } = await this.getFakeAccountDataFor(from);
253
+ overrides = {
254
+ contracts: { [from.toString()]: { instance, artifact } },
255
+ };
256
+ const executionOptions: DefaultAccountEntrypointOptions = {
257
+ txNonce: Fr.random(),
258
+ cancellable: this.cancellableTransactions,
259
+ // If from is an address, feeOptions include the way the account contract should handle the fee payment
260
+ feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions!,
261
+ };
262
+ txRequest = await account.createTxExecutionRequest(
246
263
  finalExecutionPayload,
247
264
  feeOptions.gasSettings,
248
265
  chainInfo,
249
266
  executionOptions,
250
267
  );
251
- const contractOverrides = {
252
- [opts.from.toString()]: { instance, artifact },
253
- };
254
- simulationResults = await this.pxe.simulateTx(txRequest, true /* simulatePublic */, true, true, {
255
- contracts: contractOverrides,
256
- });
257
268
  }
258
269
 
259
- if (opts.fee?.estimateGas) {
260
- const limits = getGasLimits(simulationResults, opts.fee?.estimatedGasPadding);
261
- printGasEstimates(feeOptions, limits, this.userLog);
262
- }
263
- return simulationResults;
270
+ const result = await this.pxe.simulateTx(txRequest, {
271
+ simulatePublic: true,
272
+ skipFeeEnforcement: true,
273
+ skipTxValidation: true,
274
+ overrides,
275
+ scopes,
276
+ });
277
+ const appCallOffset = await this.computeAppCallOffset(from, feeOptions);
278
+ return TxSimulationResultWithAppOffset.fromResultAndOffset(result, appCallOffset);
264
279
  }
265
280
 
266
281
  // Exposed because of the `aztec-wallet get-tx` command. It has been decided that it's fine to keep around because