@aztec/wallet-sdk 0.0.1-commit.a072138 → 0.0.1-commit.a5db02d

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 (76) hide show
  1. package/README.md +125 -0
  2. package/dest/base-wallet/base_wallet.d.ts +99 -37
  3. package/dest/base-wallet/base_wallet.d.ts.map +1 -1
  4. package/dest/base-wallet/base_wallet.js +324 -100
  5. package/dest/base-wallet/get_gas_limits.d.ts +36 -0
  6. package/dest/base-wallet/get_gas_limits.d.ts.map +1 -0
  7. package/dest/base-wallet/get_gas_limits.js +55 -0
  8. package/dest/base-wallet/index.d.ts +4 -2
  9. package/dest/base-wallet/index.d.ts.map +1 -1
  10. package/dest/base-wallet/index.js +2 -0
  11. package/dest/base-wallet/utils.d.ts +52 -0
  12. package/dest/base-wallet/utils.d.ts.map +1 -0
  13. package/dest/base-wallet/utils.js +137 -0
  14. package/dest/crypto.d.ts +39 -1
  15. package/dest/crypto.d.ts.map +1 -1
  16. package/dest/crypto.js +88 -0
  17. package/dest/extension/handlers/background_connection_handler.d.ts +12 -2
  18. package/dest/extension/handlers/background_connection_handler.d.ts.map +1 -1
  19. package/dest/extension/handlers/background_connection_handler.js +44 -8
  20. package/dest/extension/handlers/content_script_connection_handler.d.ts +2 -1
  21. package/dest/extension/handlers/content_script_connection_handler.d.ts.map +1 -1
  22. package/dest/extension/handlers/content_script_connection_handler.js +19 -0
  23. package/dest/extension/handlers/internal_message_types.d.ts +3 -1
  24. package/dest/extension/handlers/internal_message_types.d.ts.map +1 -1
  25. package/dest/extension/handlers/internal_message_types.js +3 -1
  26. package/dest/extension/provider/extension_wallet.d.ts +27 -6
  27. package/dest/extension/provider/extension_wallet.d.ts.map +1 -1
  28. package/dest/extension/provider/extension_wallet.js +89 -11
  29. package/dest/extension/provider/index.d.ts +2 -2
  30. package/dest/extension/provider/index.d.ts.map +1 -1
  31. package/dest/iframe/handlers/iframe_connection_handler.d.ts +122 -0
  32. package/dest/iframe/handlers/iframe_connection_handler.d.ts.map +1 -0
  33. package/dest/iframe/handlers/iframe_connection_handler.js +239 -0
  34. package/dest/iframe/handlers/index.d.ts +2 -0
  35. package/dest/iframe/handlers/index.d.ts.map +1 -0
  36. package/dest/iframe/handlers/index.js +1 -0
  37. package/dest/iframe/provider/iframe_discovery.d.ts +25 -0
  38. package/dest/iframe/provider/iframe_discovery.d.ts.map +1 -0
  39. package/dest/iframe/provider/iframe_discovery.js +167 -0
  40. package/dest/iframe/provider/iframe_provider.d.ts +65 -0
  41. package/dest/iframe/provider/iframe_provider.d.ts.map +1 -0
  42. package/dest/iframe/provider/iframe_provider.js +257 -0
  43. package/dest/iframe/provider/iframe_wallet.d.ts +85 -0
  44. package/dest/iframe/provider/iframe_wallet.d.ts.map +1 -0
  45. package/dest/iframe/provider/iframe_wallet.js +269 -0
  46. package/dest/iframe/provider/index.d.ts +4 -0
  47. package/dest/iframe/provider/index.d.ts.map +1 -0
  48. package/dest/iframe/provider/index.js +3 -0
  49. package/dest/manager/types.d.ts +6 -5
  50. package/dest/manager/types.d.ts.map +1 -1
  51. package/dest/manager/wallet_manager.d.ts +1 -1
  52. package/dest/manager/wallet_manager.d.ts.map +1 -1
  53. package/dest/manager/wallet_manager.js +48 -18
  54. package/dest/types.d.ts +64 -2
  55. package/dest/types.d.ts.map +1 -1
  56. package/dest/types.js +29 -0
  57. package/package.json +19 -9
  58. package/src/base-wallet/base_wallet.ts +421 -157
  59. package/src/base-wallet/get_gas_limits.ts +88 -0
  60. package/src/base-wallet/index.ts +8 -1
  61. package/src/base-wallet/utils.ts +248 -0
  62. package/src/crypto.ts +104 -0
  63. package/src/extension/handlers/background_connection_handler.ts +42 -9
  64. package/src/extension/handlers/content_script_connection_handler.ts +18 -0
  65. package/src/extension/handlers/internal_message_types.ts +2 -0
  66. package/src/extension/provider/extension_wallet.ts +106 -17
  67. package/src/extension/provider/index.ts +1 -1
  68. package/src/iframe/handlers/iframe_connection_handler.ts +341 -0
  69. package/src/iframe/handlers/index.ts +7 -0
  70. package/src/iframe/provider/iframe_discovery.ts +185 -0
  71. package/src/iframe/provider/iframe_provider.ts +331 -0
  72. package/src/iframe/provider/iframe_wallet.ts +323 -0
  73. package/src/iframe/provider/index.ts +3 -0
  74. package/src/manager/types.ts +5 -4
  75. package/src/manager/wallet_manager.ts +55 -23
  76. package/src/types.ts +72 -0
@@ -1,16 +1,23 @@
1
- import { NO_WAIT } from '@aztec/aztec.js/contracts';
1
+ import { NO_FROM } from '@aztec/aztec.js/account';
2
+ import { NO_WAIT, extractOffchainOutput } from '@aztec/aztec.js/contracts';
2
3
  import { waitForTx } from '@aztec/aztec.js/node';
3
- import { GAS_ESTIMATION_DA_GAS_LIMIT, GAS_ESTIMATION_L2_GAS_LIMIT, GAS_ESTIMATION_TEARDOWN_DA_GAS_LIMIT, GAS_ESTIMATION_TEARDOWN_L2_GAS_LIMIT } from '@aztec/constants';
4
+ import { ContractInitializationStatus, TxSimulationResultWithAppOffset } from '@aztec/aztec.js/wallet';
4
5
  import { AccountFeePaymentMethodOptions } from '@aztec/entrypoints/account';
6
+ import { DefaultEntrypoint } from '@aztec/entrypoints/default';
5
7
  import { Fr } from '@aztec/foundation/curves/bn254';
6
8
  import { createLogger } from '@aztec/foundation/log';
9
+ import { displayDebugLogs } from '@aztec/pxe/client/lazy';
7
10
  import { decodeFromAbi } from '@aztec/stdlib/abi';
8
- import { computePartialAddress, getContractClassFromArtifact } from '@aztec/stdlib/contract';
11
+ import { AztecAddress } from '@aztec/stdlib/aztec-address';
12
+ import { computePartialAddress } from '@aztec/stdlib/contract';
9
13
  import { SimulationError } from '@aztec/stdlib/errors';
10
- import { Gas, GasSettings } from '@aztec/stdlib/gas';
11
- import { siloNullifier } from '@aztec/stdlib/hash';
14
+ import { Gas, GasFees, GasSettings, ManaUsageEstimate } from '@aztec/stdlib/gas';
15
+ import { computeSiloedPrivateInitializationNullifier, computeSiloedPublicInitializationNullifier } from '@aztec/stdlib/hash';
16
+ import { deriveKeys, deriveKeysFromMasterSecretKeys } from '@aztec/stdlib/keys';
12
17
  import { mergeExecutionPayloads } from '@aztec/stdlib/tx';
13
18
  import { inspect } from 'util';
19
+ import { assertGasLimitsWithinNetworkLimits } from './get_gas_limits.js';
20
+ import { buildMergedSimulationResult, extractOptimizablePublicStaticCalls, simulateViaNode } from './utils.js';
14
21
  /**
15
22
  * A base class for Wallet implementations
16
23
  */ export class BaseWallet {
@@ -19,14 +26,39 @@ import { inspect } from 'util';
19
26
  log;
20
27
  minFeePadding;
21
28
  cancellableTransactions;
29
+ // Poll interval (in seconds) injected into sendTx waits when the caller does not specify one. Left undefined on
30
+ // production wallets so the DefaultWaitOpts 1s cadence stands; test wallets talking to in-process nodes lower it.
31
+ defaultWaitInterval;
32
+ // A wallet is instantiated for a particular chain, so chain info never changes during its lifetime.
33
+ // We cache it here because getChainInfo is called frequently (every tx simulation, send, auth wit, etc.).
34
+ nodeInfoPromise;
22
35
  // Protected because we want to force wallets to instantiate their own PXE.
23
- constructor(pxe, aztecNode){
36
+ constructor(pxe, aztecNode, log = createLogger('wallet-sdk:base_wallet')){
24
37
  this.pxe = pxe;
25
38
  this.aztecNode = aztecNode;
26
- this.log = createLogger('wallet-sdk:base_wallet');
39
+ this.log = log;
27
40
  this.minFeePadding = 0.5;
28
41
  this.cancellableTransactions = false;
29
42
  }
43
+ scopesFrom(from, additionalScopes = []) {
44
+ const allScopes = from === NO_FROM ? additionalScopes : [
45
+ from,
46
+ ...additionalScopes
47
+ ];
48
+ const scopeSet = new Set(allScopes.map((address)=>address.toString()));
49
+ return [
50
+ ...scopeSet
51
+ ].map(AztecAddress.fromStringUnsafe);
52
+ }
53
+ /**
54
+ * Picks the sender address PXE should tag private messages with. Returns `undefined` when there is no signing
55
+ * account (`from === NO_FROM`) and no explicit override; in that case any private log emitted by the tx using
56
+ * the wallet-supplied default sender will fail the "Sender for tags is not set" assertion.
57
+ * @param from - Tx sender, or `NO_FROM`.
58
+ * @param sendMessagesAs - Explicit override.
59
+ */ senderForTagsFrom(from, sendMessagesAs) {
60
+ return sendMessagesAs ?? (from === NO_FROM ? undefined : from);
61
+ }
30
62
  /**
31
63
  * Returns the list of aliased contacts associated with the wallet.
32
64
  * This base implementation directly returns PXE's senders, but note that in general contacts are a superset of senders.
@@ -34,33 +66,64 @@ import { inspect } from 'util';
34
66
  * - Contacts: more general concept akin to a phone's contact list.
35
67
  * @returns The aliased collection of AztecAddresses that form this wallet's address book
36
68
  */ async getAddressBook() {
37
- const senders = await this.pxe.getSenders();
38
- return senders.map((sender)=>({
39
- item: sender,
69
+ const sources = await this.pxe.getTaggingSecretSources({
70
+ kind: 'address-derived'
71
+ });
72
+ return sources.map((source)=>({
73
+ item: source.sender,
40
74
  alias: ''
41
75
  }));
42
76
  }
77
+ /**
78
+ * Fetches and caches the node info for the wallet's lifetime, since a wallet talks to a single network and
79
+ * node info never changes. A rejected fetch clears the cache so the next call retries instead of replaying
80
+ * the cached rejection forever — important because the gas-limit fill-in and validation (run on every send)
81
+ * depend on it.
82
+ */ getNodeInfo() {
83
+ if (!this.nodeInfoPromise) {
84
+ this.nodeInfoPromise = this.aztecNode.getNodeInfo().catch((err)=>{
85
+ this.nodeInfoPromise = undefined;
86
+ throw err;
87
+ });
88
+ }
89
+ return this.nodeInfoPromise;
90
+ }
43
91
  async getChainInfo() {
44
- const { l1ChainId, rollupVersion } = await this.aztecNode.getNodeInfo();
92
+ const { l1ChainId, rollupVersion } = await this.getNodeInfo();
45
93
  return {
46
94
  chainId: new Fr(l1ChainId),
47
95
  version: new Fr(rollupVersion)
48
96
  };
49
97
  }
98
+ /**
99
+ * Returns the maximum gas limits a single transaction may declare on this wallet's network (the
100
+ * node-advertised `txsLimits.gas`). Internal helper used to fill in default gas limits when sending a
101
+ * transaction without explicit limits, and to validate caller-provided limits before sending. Backed by
102
+ * the cached node info, since a wallet talks to a single network.
103
+ */ async getMaxTxGasLimits() {
104
+ const { txsLimits } = await this.getNodeInfo();
105
+ return new Gas(txsLimits.gas.daGas, txsLimits.gas.l2Gas);
106
+ }
50
107
  async createTxExecutionRequestFromPayloadAndFee(executionPayload, from, feeOptions) {
51
108
  const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
52
- const executionOptions = {
53
- txNonce: Fr.random(),
54
- cancellable: this.cancellableTransactions,
55
- feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions
56
- };
57
109
  const finalExecutionPayload = feeExecutionPayload ? mergeExecutionPayloads([
58
110
  feeExecutionPayload,
59
111
  executionPayload
60
112
  ]) : executionPayload;
61
- const fromAccount = await this.getAccountFromAddress(from);
62
113
  const chainInfo = await this.getChainInfo();
63
- return fromAccount.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo, executionOptions);
114
+ if (from === NO_FROM) {
115
+ const entrypoint = new DefaultEntrypoint();
116
+ return entrypoint.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo);
117
+ } else {
118
+ const fromAccount = await this.getAccountFromAddress(from);
119
+ const executionOptions = {
120
+ txNonce: Fr.random(),
121
+ cancellable: this.cancellableTransactions,
122
+ // If from is an address, feeOptions include the way the account contract should handle the fee payment
123
+ feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions
124
+ };
125
+ return fromAccount.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo, executionOptions);
126
+ }
64
127
  }
65
128
  async createAuthWit(from, messageHashOrIntent) {
66
129
  const account = await this.getAccountFromAddress(from);
@@ -103,26 +166,50 @@ import { inspect } from 'util';
103
166
  }
104
167
  /**
105
168
  * Completes partial user-provided fee options with wallet defaults.
106
- * @param from - The address where the transaction is being sent from
107
- * @param feePayer - The address paying for fees (if any fee payment method is embedded in the execution payload)
108
- * @param gasSettings - User-provided partial gas settings
109
- * @returns - Complete fee options that can be used to create a transaction execution request
110
- */ async completeFeeOptions(from, feePayer, gasSettings) {
111
- const maxFeesPerGas = gasSettings?.maxFeesPerGas ?? (await this.aztecNode.getCurrentMinFees()).mul(1 + this.minFeePadding);
169
+ * @param config - Fee completion config.
170
+ */ async completeFeeOptions(config) {
171
+ const { from, feePayer, gasSettings, forEstimation, congestionEstimate } = config;
172
+ const maxFeesPerGas = gasSettings?.maxFeesPerGas ?? (await this.getMinFees(congestionEstimate)).mul(1 + this.minFeePadding);
112
173
  let accountFeePaymentMethodOptions;
113
- // The transaction does not include a fee payment method, so we set the flag
114
- // for the account to use its fee juice balance
115
- if (!feePayer) {
116
- accountFeePaymentMethodOptions = AccountFeePaymentMethodOptions.PREEXISTING_FEE_JUICE;
174
+ // If from is an address, we need to determine the appropriate fee payment method options for the
175
+ // account contract entrypoint to use
176
+ if (from !== NO_FROM) {
177
+ if (!feePayer) {
178
+ // The transaction does not include a fee payment method, so we set the flag
179
+ // for the account to use its fee juice balance
180
+ accountFeePaymentMethodOptions = AccountFeePaymentMethodOptions.PREEXISTING_FEE_JUICE;
181
+ } else {
182
+ // The transaction includes fee payment method, so we check if we are the fee payer for it
183
+ // (this can only happen if the embedded payment method is FeeJuiceWithClaim)
184
+ accountFeePaymentMethodOptions = from.equals(feePayer) ? AccountFeePaymentMethodOptions.FEE_JUICE_WITH_CLAIM : AccountFeePaymentMethodOptions.EXTERNAL;
185
+ }
186
+ }
187
+ const gasSettingsOverrides = {
188
+ gasLimits: gasSettings?.gasLimits ? Gas.from(gasSettings.gasLimits) : undefined,
189
+ teardownGasLimits: gasSettings?.teardownGasLimits ? Gas.from(gasSettings.teardownGasLimits) : undefined,
190
+ maxFeesPerGas,
191
+ maxPriorityFeesPerGas: gasSettings?.maxPriorityFeesPerGas ?? GasFees.empty()
192
+ };
193
+ // When estimating gas (simulation), use high limits so the simulation doesn't run out of gas.
194
+ // When sending for real without explicit limits, declare the most a single tx may use on this network
195
+ // (the node's per-tx admission limit), so the proposer does not skip the tx for over-declaring gas.
196
+ let fullGasSettings;
197
+ if (forEstimation) {
198
+ // Estimation deliberately uses very high internal limits and skips tx validation, so we do not
199
+ // validate against the network admission limit here.
200
+ fullGasSettings = GasSettings.forEstimation(gasSettingsOverrides);
117
201
  } else {
118
- // The transaction includes fee payment method, so we check if we are the fee payer for it
119
- // (this can only happen if the embedded payment method is FeeJuiceWithClaim)
120
- accountFeePaymentMethodOptions = from.equals(feePayer) ? AccountFeePaymentMethodOptions.FEE_JUICE_WITH_CLAIM : AccountFeePaymentMethodOptions.EXTERNAL;
202
+ const maxTxGasLimits = await this.getMaxTxGasLimits();
203
+ // If the caller declared explicit gas limits, reject them up front when they exceed the network's
204
+ // per-tx admission limit (mirroring the node's GasLimitsValidator). Otherwise fill in the limit.
205
+ if (gasSettingsOverrides.gasLimits) {
206
+ assertGasLimitsWithinNetworkLimits(gasSettingsOverrides.gasLimits, maxTxGasLimits);
207
+ }
208
+ fullGasSettings = GasSettings.fallback({
209
+ ...gasSettingsOverrides,
210
+ gasLimits: gasSettingsOverrides.gasLimits ?? maxTxGasLimits
211
+ });
121
212
  }
122
- const fullGasSettings = GasSettings.default({
123
- ...gasSettings,
124
- maxFeesPerGas
125
- });
126
213
  this.log.debug(`Using L2 gas settings`, fullGasSettings);
127
214
  return {
128
215
  gasSettings: fullGasSettings,
@@ -131,75 +218,161 @@ import { inspect } from 'util';
131
218
  };
132
219
  }
133
220
  /**
134
- * Completes partial user-provided fee options with unreasonably high gas limits
135
- * for gas estimation. Uses the same logic as completeFeeOptions but sets high limits
136
- * to avoid running out of gas during estimation.
137
- * @param from - The address where the transaction is being sent from
138
- * @param feePayer - The address paying for fees (if any fee payment method is embedded in the execution payload)
139
- * @param gasSettings - User-provided partial gas settings
140
- */ async completeFeeOptionsForEstimation(from, feePayer, gasSettings) {
141
- const defaultFeeOptions = await this.completeFeeOptions(from, feePayer, gasSettings);
142
- const { gasSettings: { maxFeesPerGas, maxPriorityFeesPerGas } } = defaultFeeOptions;
143
- // Use unrealistically high gas limits for estimation to avoid running out of gas.
144
- // They will be tuned down after the simulation.
145
- const gasSettingsForEstimation = new GasSettings(new Gas(GAS_ESTIMATION_DA_GAS_LIMIT, GAS_ESTIMATION_L2_GAS_LIMIT), new Gas(GAS_ESTIMATION_TEARDOWN_DA_GAS_LIMIT, GAS_ESTIMATION_TEARDOWN_L2_GAS_LIMIT), maxFeesPerGas, maxPriorityFeesPerGas);
146
- return {
147
- ...defaultFeeOptions,
148
- gasSettings: gasSettingsForEstimation
149
- };
221
+ * Returns the worst-case min fee across predicted future slots.
222
+ * Falls back to getCurrentMinFees if the node doesn't support getPredictedMinFees.
223
+ * @param estimate - The mana usage estimate to use for fee prediction. Defaults to Limit for conservative estimation.
224
+ */ async getMinFees(estimate = ManaUsageEstimate.Limit) {
225
+ try {
226
+ const predicted = await this.aztecNode.getPredictedMinFees(estimate);
227
+ if (predicted.length === 0) {
228
+ return this.aztecNode.getCurrentMinFees();
229
+ }
230
+ return predicted.reduce((worst, fees)=>fees.feePerL2Gas > worst.feePerL2Gas ? fees : worst);
231
+ } catch (err) {
232
+ // Fallback for old nodes that don't support getPredictedMinFees.
233
+ // Only fall back on method-not-found errors (JSON-RPC code -32601); rethrow others.
234
+ if (err?.cause?.code === -32601 || err?.message?.includes('Method not found')) {
235
+ return this.aztecNode.getCurrentMinFees();
236
+ }
237
+ throw err;
238
+ }
150
239
  }
151
- registerSender(address, _alias = '') {
152
- return this.pxe.registerSender(address);
240
+ async registerSender(address, _alias = '') {
241
+ await this.pxe.registerTaggingSecretSource({
242
+ kind: 'address-derived',
243
+ sender: address
244
+ });
245
+ return address;
153
246
  }
154
- async registerContract(instance, artifact, secretKey) {
155
- const existingInstance = await this.pxe.getContractInstance(instance.address);
156
- if (existingInstance) {
157
- // Instance already registered in the wallet
158
- if (artifact) {
159
- const thisContractClass = await getContractClassFromArtifact(artifact);
160
- if (!thisContractClass.id.equals(existingInstance.currentContractClassId)) {
161
- // wallet holds an outdated version of this contract
162
- await this.pxe.updateContract(instance.address, artifact);
163
- instance.currentContractClassId = thisContractClass.id;
164
- }
165
- }
166
- // If no artifact provided, we just use the existing registration
167
- } else {
168
- // Instance not registered yet
169
- if (!artifact) {
170
- // Try to get the artifact from the wallet's contract class storage
171
- artifact = await this.pxe.getContractArtifact(instance.currentContractClassId);
172
- if (!artifact) {
173
- throw new Error(`Cannot register contract at ${instance.address.toString()}: artifact is required but not provided, and wallet does not have the artifact for contract class ${instance.currentContractClassId.toString()}`);
174
- }
247
+ async registerContract(instance, artifact, secretKeyOrKeys) {
248
+ // Classes and instances are registered independently: register the artifact (if provided) then the instance.
249
+ // Neither call validates that the artifact matches the class the instance runs, a missing artifact only surfaces
250
+ // when the contract is later simulated.
251
+ if (artifact) {
252
+ await this.pxe.registerContractClass(artifact);
253
+ }
254
+ const contractAddress = await this.pxe.registerContract(instance);
255
+ if (secretKeyOrKeys) {
256
+ // PXE never receives the account seed (from which the message-signing/fallback secret keys could be re-derived):
257
+ // the wallet derives the keys here. Of these, PXE only reads and stores the four privacy secret keys and the
258
+ // message-signing and fallback *public* keys — it never touches the message-signing or fallback secret keys.
259
+ //
260
+ // Since PXE recomputes the address from those keys, we assert it matches the instance's address: a mismatch means
261
+ // the provided keys don't correspond to this account.
262
+ const derivedKeys = secretKeyOrKeys instanceof Fr ? await deriveKeys(secretKeyOrKeys) : await deriveKeysFromMasterSecretKeys(secretKeyOrKeys);
263
+ const { address } = await this.pxe.registerAccount(derivedKeys, await computePartialAddress(instance));
264
+ if (!address.equals(contractAddress)) {
265
+ throw new Error(`Registered account address ${address.toString()} does not match contract instance address ${contractAddress.toString()}: the provided keys do not correspond to this account.`);
175
266
  }
176
- await this.pxe.registerContract({
177
- artifact,
178
- instance
179
- });
180
267
  }
181
- if (secretKey) {
182
- await this.pxe.registerAccount(secretKey, await computePartialAddress(instance));
268
+ }
269
+ registerContractClass(artifact) {
270
+ return this.pxe.registerContractClass(artifact);
271
+ }
272
+ /**
273
+ * Simulates calls through the standard PXE path (account entrypoint).
274
+ * @param executionPayload - The execution payload to simulate.
275
+ * @param opts - Simulation options.
276
+ */ async simulateViaEntrypoint(executionPayload, opts) {
277
+ const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, opts.feeOptions);
278
+ const result = await this.pxe.simulateTx(txRequest, {
279
+ simulatePublic: true,
280
+ skipTxValidation: opts.skipTxValidation,
281
+ skipFeeEnforcement: opts.skipFeeEnforcement,
282
+ scopes: this.scopesFrom(opts.from, opts.additionalScopes),
283
+ senderForTags: this.senderForTagsFrom(opts.from, opts.sendMessagesAs),
284
+ overrides: opts.overrides
285
+ });
286
+ const appCallOffset = await this.computeAppCallOffset(opts.from, opts.feeOptions);
287
+ return TxSimulationResultWithAppOffset.fromResultAndOffset(result, appCallOffset);
288
+ }
289
+ /**
290
+ * Computes the index where the app's calls begin in the flattened array of calls (0 = entrypoint/root, 1..N = fee
291
+ * calls, N+1 = app).
292
+ * @param from - The sender address, or NO_FROM for the default entrypoint.
293
+ * @param feeOptions - Fee options containing the wallet fee payment method.
294
+ */ async computeAppCallOffset(from, feeOptions) {
295
+ if (from === NO_FROM) {
296
+ return 0;
183
297
  }
184
- return instance;
298
+ const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
299
+ return (feeExecutionPayload?.calls.length ?? 0) + 1; // +1 for entrypoint
185
300
  }
186
- async simulateTx(executionPayload, opts) {
187
- const feeOptions = opts.fee?.estimateGas ? await this.completeFeeOptionsForEstimation(opts.from, executionPayload.feePayer, opts.fee?.gasSettings) : await this.completeFeeOptions(opts.from, executionPayload.feePayer, opts.fee?.gasSettings);
188
- const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, feeOptions);
189
- return this.pxe.simulateTx(txRequest, true, opts?.skipTxValidation, opts?.skipFeeEnforcement ?? true);
301
+ /**
302
+ * Simulates a transaction, optimizing leading public static calls by running them directly
303
+ * on the node while sending the remaining calls through the standard PXE path.
304
+ * Return values from both paths are merged back in original call order.
305
+ * @param executionPayload - The execution payload to simulate.
306
+ * @param opts - Simulation options (from address, fee settings, etc.).
307
+ * @returns The merged simulation result.
308
+ */ async simulateTx(executionPayload, opts) {
309
+ const feeOptions = await this.completeFeeOptions({
310
+ from: opts.from,
311
+ feePayer: executionPayload.feePayer,
312
+ gasSettings: opts.fee?.gasSettings,
313
+ forEstimation: true,
314
+ congestionEstimate: opts.fee?.congestionEstimate
315
+ });
316
+ const { optimizableCalls, remainingCalls } = extractOptimizablePublicStaticCalls(executionPayload);
317
+ const remainingPayload = {
318
+ ...executionPayload,
319
+ calls: remainingCalls
320
+ };
321
+ const chainInfo = await this.getChainInfo();
322
+ let blockHeader;
323
+ // PXE might not be synced yet, so we pull the latest header from the node
324
+ // To keep things consistent, we'll always try with PXE first
325
+ try {
326
+ blockHeader = await this.pxe.getSyncedBlockHeader();
327
+ } catch {
328
+ blockHeader = (await this.aztecNode.getBlockData('latest')).header;
329
+ }
330
+ const simulationOrigin = opts.from === NO_FROM ? AztecAddress.ZERO : opts.from;
331
+ const [optimizedResults, normalResult] = await Promise.all([
332
+ optimizableCalls.length > 0 ? simulateViaNode(this.aztecNode, optimizableCalls, simulationOrigin, chainInfo, feeOptions.gasSettings, blockHeader, opts.skipFeeEnforcement ?? true, this.getContractName.bind(this), opts.overrides) : Promise.resolve([]),
333
+ remainingCalls.length > 0 ? this.simulateViaEntrypoint(remainingPayload, {
334
+ from: opts.from,
335
+ feeOptions,
336
+ additionalScopes: opts.additionalScopes,
337
+ skipTxValidation: opts.skipTxValidation,
338
+ skipFeeEnforcement: opts.skipFeeEnforcement ?? true,
339
+ sendMessagesAs: opts.sendMessagesAs,
340
+ overrides: opts.overrides
341
+ }) : Promise.resolve(null)
342
+ ]);
343
+ return buildMergedSimulationResult(optimizedResults, normalResult);
190
344
  }
191
345
  async profileTx(executionPayload, opts) {
192
- const feeOptions = await this.completeFeeOptions(opts.from, executionPayload.feePayer, opts.fee?.gasSettings);
346
+ const feeOptions = await this.completeFeeOptions({
347
+ from: opts.from,
348
+ feePayer: executionPayload.feePayer,
349
+ gasSettings: opts.fee?.gasSettings,
350
+ congestionEstimate: opts.fee?.congestionEstimate
351
+ });
193
352
  const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, feeOptions);
194
- return this.pxe.profileTx(txRequest, opts.profileMode, opts.skipProofGeneration ?? true);
353
+ return this.pxe.profileTx(txRequest, {
354
+ profileMode: opts.profileMode,
355
+ skipProofGeneration: opts.skipProofGeneration ?? true,
356
+ scopes: this.scopesFrom(opts.from, opts.additionalScopes),
357
+ senderForTags: this.senderForTagsFrom(opts.from, opts.sendMessagesAs)
358
+ });
195
359
  }
196
360
  async sendTx(executionPayload, opts) {
197
- const feeOptions = await this.completeFeeOptions(opts.from, executionPayload.feePayer, opts.fee?.gasSettings);
361
+ const feeOptions = await this.completeFeeOptions({
362
+ from: opts.from,
363
+ feePayer: executionPayload.feePayer,
364
+ gasSettings: opts.fee?.gasSettings,
365
+ congestionEstimate: opts.fee?.congestionEstimate
366
+ });
198
367
  const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, feeOptions);
199
- const provenTx = await this.pxe.proveTx(txRequest);
368
+ const provenTx = await this.pxe.proveTx(txRequest, {
369
+ scopes: this.scopesFrom(opts.from, opts.additionalScopes),
370
+ senderForTags: this.senderForTagsFrom(opts.from, opts.sendMessagesAs)
371
+ });
372
+ const offchainOutput = extractOffchainOutput(provenTx.getOffchainEffects(), provenTx.publicInputs.constants.anchorBlockHeader.globalVariables.timestamp);
200
373
  const tx = await provenTx.toTx();
201
374
  const txHash = tx.getTxHash();
202
- if (await this.aztecNode.getTxEffect(txHash)) {
375
+ if ((await this.aztecNode.getTxReceipt(txHash)).isMined()) {
203
376
  throw new Error(`A settled tx with equal hash ${txHash.toString()} exists.`);
204
377
  }
205
378
  this.log.debug(`Sending transaction ${txHash}`);
@@ -209,11 +382,41 @@ import { inspect } from 'util';
209
382
  this.log.info(`Sent transaction ${txHash}`);
210
383
  // If wait is NO_WAIT, return txHash immediately
211
384
  if (opts.wait === NO_WAIT) {
212
- return txHash;
385
+ return {
386
+ txHash,
387
+ ...offchainOutput
388
+ };
213
389
  }
214
390
  // Otherwise, wait for the full receipt (default behavior on wait: undefined)
215
- const waitOpts = typeof opts.wait === 'object' ? opts.wait : undefined;
216
- return await waitForTx(this.aztecNode, txHash, waitOpts);
391
+ const callerWaitOpts = typeof opts.wait === 'object' ? opts.wait : undefined;
392
+ const waitOpts = this.defaultWaitInterval !== undefined && callerWaitOpts?.interval === undefined ? {
393
+ ...callerWaitOpts,
394
+ interval: this.defaultWaitInterval
395
+ } : callerWaitOpts;
396
+ const receipt = await waitForTx(this.aztecNode, txHash, waitOpts);
397
+ // Display debug logs from public execution if present (served in test mode only)
398
+ if (receipt.isMined() && receipt.debugLogs?.length) {
399
+ await displayDebugLogs(receipt.debugLogs, this.getContractName.bind(this));
400
+ }
401
+ return {
402
+ receipt,
403
+ ...offchainOutput
404
+ };
405
+ }
406
+ /**
407
+ * Resolves a contract address to a human-readable name via PXE, if available.
408
+ * @param address - The contract address to resolve.
409
+ */ async getContractName(address) {
410
+ const instance = await this.pxe.getContractInstance(address);
411
+ if (!instance) {
412
+ return undefined;
413
+ }
414
+ // Contract names are class-stable (an upgrade preserves the contract name), so the original class artifact is a
415
+ // sufficient source for the display name without resolving the current class against the node.
416
+ // TODO: if a contract were to be upgraded and its original artifact never registered, then this would fail and we'd
417
+ // want to fallback to the current class.
418
+ const artifact = await this.pxe.getContractArtifact(instance.originalContractClassId);
419
+ return artifact?.name;
217
420
  }
218
421
  contextualizeError(err, ...context) {
219
422
  let contextStr = '';
@@ -228,8 +431,11 @@ import { inspect } from 'util';
228
431
  }
229
432
  return err;
230
433
  }
231
- simulateUtility(call, authwits) {
232
- return this.pxe.simulateUtility(call, authwits);
434
+ executeUtility(call, opts) {
435
+ return this.pxe.executeUtility(call, {
436
+ authwits: opts.authWitnesses,
437
+ scopes: opts.scopes
438
+ });
233
439
  }
234
440
  async getPrivateEvents(eventDef, eventFilter) {
235
441
  const pxeEvents = await this.pxe.getPrivateEvents(eventDef.eventSelector, eventFilter);
@@ -247,15 +453,33 @@ import { inspect } from 'util';
247
453
  });
248
454
  return decodedEvents;
249
455
  }
250
- async getContractMetadata(address) {
456
+ /**
457
+ * Returns metadata about a contract, including whether it has been initialized, published, and updated.
458
+ * @param address - The contract address to query.
459
+ */ async getContractMetadata(address) {
251
460
  const instance = await this.pxe.getContractInstance(address);
252
- const initNullifier = await siloNullifier(address, address.toField());
253
- const publiclyRegisteredContract = await this.aztecNode.getContract(address);
254
- const initNullifierMembershipWitness = await this.aztecNode.getNullifierMembershipWitness('latest', initNullifier);
461
+ const publiclyRegisteredContractPromise = this.aztecNode.getContract(address);
462
+ let initializationStatus;
463
+ if (instance) {
464
+ // We have the instance, so we can compute the private initialization nullifier (which includes init_hash and is
465
+ // emitted by both private and public initializers) and get a definitive INITIALIZED/UNINITIALIZED answer.
466
+ const initNullifier = await computeSiloedPrivateInitializationNullifier(address, instance.initializationHash);
467
+ const witness = await this.aztecNode.getNullifierMembershipWitness('latest', initNullifier);
468
+ initializationStatus = witness ? ContractInitializationStatus.INITIALIZED : ContractInitializationStatus.UNINITIALIZED;
469
+ } else {
470
+ // Without the instance we lack the init_hash needed for the private nullifier. We fall back to checking the
471
+ // public initialization nullifier (computed from address alone). Not all contracts emit it (only those with
472
+ // public functions that require initialization checks), so its absence doesn't mean the contract is
473
+ // uninitialized.
474
+ const publicNullifier = await computeSiloedPublicInitializationNullifier(address);
475
+ const witness = await this.aztecNode.getNullifierMembershipWitness('latest', publicNullifier);
476
+ initializationStatus = witness ? ContractInitializationStatus.INITIALIZED : ContractInitializationStatus.UNKNOWN;
477
+ }
478
+ const publiclyRegisteredContract = await publiclyRegisteredContractPromise;
255
479
  const isContractUpdated = publiclyRegisteredContract && !publiclyRegisteredContract.currentContractClassId.equals(publiclyRegisteredContract.originalContractClassId);
256
480
  return {
257
481
  instance: instance ?? undefined,
258
- isContractInitialized: !!initNullifierMembershipWitness,
482
+ initializationStatus,
259
483
  isContractPublished: !!publiclyRegisteredContract,
260
484
  isContractUpdated: !!isContractUpdated,
261
485
  updatedContractClassId: isContractUpdated ? publiclyRegisteredContract.currentContractClassId : undefined
@@ -0,0 +1,36 @@
1
+ import { Gas, type GasUsed } from '@aztec/stdlib/gas';
2
+ /**
3
+ * Returns suggested total and teardown gas limits for a simulated tx, clamped to the network's per-tx
4
+ * admission limits.
5
+ *
6
+ * The network only admits transactions that declare up to `maxTxGasLimits` per dimension (the
7
+ * node-advertised `txsLimits.gas`). Wallets pass the value read from their own node info, but since node info
8
+ * is remote input it is defensively clamped here to the per-tx protocol maxima so a value above them is never
9
+ * honored. If the simulated usage already exceeds the resulting admission limits the tx can never be included,
10
+ * so this throws a descriptive error instead of returning a limit the node would reject. Otherwise it pads the
11
+ * usage and clamps each dimension to the admission limit.
12
+ * @param gasUsed - The gas actually consumed during simulation.
13
+ * @param maxTxGasLimits - The maximum gas a single tx may declare on this network (the node-advertised `txsLimits.gas`).
14
+ * @param pad - Fraction to pad the suggested gas limits by (as a decimal, e.g. 0.1 for 10%). The effective
15
+ * padding shrinks to zero as usage approaches the network limit, since the network will not admit a higher
16
+ * declared limit regardless of the buffer.
17
+ */
18
+ export declare function getGasLimits(gasUsed: GasUsed, maxTxGasLimits: Gas, pad?: number): {
19
+ /**
20
+ * Gas limit for the tx, excluding teardown gas
21
+ */
22
+ gasLimits: Gas;
23
+ /**
24
+ * Gas limit for the teardown phase
25
+ */
26
+ teardownGasLimits: Gas;
27
+ };
28
+ /**
29
+ * Validates that caller-declared gas limits do not exceed the network's per-tx admission limits, throwing a
30
+ * descriptive error per dimension when they do. The node's inbound validation checks declared
31
+ * `gasSettings.gasLimits`, so we mirror that here to surface the rejection locally before the tx is sent.
32
+ * @param gasLimits - The gas limits the transaction will declare.
33
+ * @param maxTxGasLimits - The maximum gas a single tx may declare on this network (the node-advertised `txsLimits.gas`).
34
+ */
35
+ export declare function assertGasLimitsWithinNetworkLimits(gasLimits: Gas, maxTxGasLimits: Gas): void;
36
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZ2V0X2dhc19saW1pdHMuZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9iYXNlLXdhbGxldC9nZXRfZ2FzX2xpbWl0cy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFDQSxPQUFPLEVBQUUsR0FBRyxFQUFFLEtBQUssT0FBTyxFQUFFLE1BQU0sbUJBQW1CLENBQUM7QUFFdEQ7Ozs7Ozs7Ozs7Ozs7OztHQWVHO0FBQ0gsd0JBQWdCLFlBQVksQ0FDMUIsT0FBTyxFQUFFLE9BQU8sRUFDaEIsY0FBYyxFQUFFLEdBQUcsRUFDbkIsR0FBRyxTQUFNLEdBQ1I7SUFDRDs7T0FFRztJQUNILFNBQVMsRUFBRSxHQUFHLENBQUM7SUFDZjs7T0FFRztJQUNILGlCQUFpQixFQUFFLEdBQUcsQ0FBQztDQUN4QixDQTZCQTtBQVFEOzs7Ozs7R0FNRztBQUNILHdCQUFnQixrQ0FBa0MsQ0FBQyxTQUFTLEVBQUUsR0FBRyxFQUFFLGNBQWMsRUFBRSxHQUFHLEdBQUcsSUFBSSxDQVc1RiJ9
@@ -0,0 +1 @@
1
+ {"version":3,"file":"get_gas_limits.d.ts","sourceRoot":"","sources":["../../src/base-wallet/get_gas_limits.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,GAAG,EAAE,KAAK,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEtD;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,YAAY,CAC1B,OAAO,EAAE,OAAO,EAChB,cAAc,EAAE,GAAG,EACnB,GAAG,SAAM,GACR;IACD;;OAEG;IACH,SAAS,EAAE,GAAG,CAAC;IACf;;OAEG;IACH,iBAAiB,EAAE,GAAG,CAAC;CACxB,CA6BA;AAQD;;;;;;GAMG;AACH,wBAAgB,kCAAkC,CAAC,SAAS,EAAE,GAAG,EAAE,cAAc,EAAE,GAAG,GAAG,IAAI,CAW5F"}
@@ -0,0 +1,55 @@
1
+ import { MAX_PROCESSABLE_L2_GAS, MAX_TX_DA_GAS } from '@aztec/constants';
2
+ import { Gas } from '@aztec/stdlib/gas';
3
+ /**
4
+ * Returns suggested total and teardown gas limits for a simulated tx, clamped to the network's per-tx
5
+ * admission limits.
6
+ *
7
+ * The network only admits transactions that declare up to `maxTxGasLimits` per dimension (the
8
+ * node-advertised `txsLimits.gas`). Wallets pass the value read from their own node info, but since node info
9
+ * is remote input it is defensively clamped here to the per-tx protocol maxima so a value above them is never
10
+ * honored. If the simulated usage already exceeds the resulting admission limits the tx can never be included,
11
+ * so this throws a descriptive error instead of returning a limit the node would reject. Otherwise it pads the
12
+ * usage and clamps each dimension to the admission limit.
13
+ * @param gasUsed - The gas actually consumed during simulation.
14
+ * @param maxTxGasLimits - The maximum gas a single tx may declare on this network (the node-advertised `txsLimits.gas`).
15
+ * @param pad - Fraction to pad the suggested gas limits by (as a decimal, e.g. 0.1 for 10%). The effective
16
+ * padding shrinks to zero as usage approaches the network limit, since the network will not admit a higher
17
+ * declared limit regardless of the buffer.
18
+ */ export function getGasLimits(gasUsed, maxTxGasLimits, pad = 0.1) {
19
+ const { totalGas, teardownGas } = gasUsed;
20
+ // `maxTxGasLimits` is the node-advertised admission limit. Node info is remote input, so we defensively
21
+ // clamp to the per-tx protocol maxima so a value above them can never be honored.
22
+ const maxLimits = new Gas(Math.min(maxTxGasLimits.daGas, MAX_TX_DA_GAS), Math.min(maxTxGasLimits.l2Gas, MAX_PROCESSABLE_L2_GAS));
23
+ // The simulated usage must fit within the admission limits, otherwise the tx can never be included.
24
+ if (totalGas.daGas > maxLimits.daGas) {
25
+ throw new Error(`Transaction consumes ${totalGas.daGas} DA gas but the network only admits transactions declaring up to ${maxLimits.daGas} DA gas`);
26
+ }
27
+ if (totalGas.l2Gas > maxLimits.l2Gas) {
28
+ throw new Error(`Transaction consumes ${totalGas.l2Gas} L2 gas but the network only admits transactions declaring up to ${maxLimits.l2Gas} L2 gas`);
29
+ }
30
+ // Pad the limits by the buffer, then cap each dimension at the admission limit so the buffer cannot push a
31
+ // declared limit past what inbound validation accepts. Teardown is part of the total, so clamping it to the
32
+ // admission limit is safe.
33
+ return {
34
+ gasLimits: padGas(totalGas, pad, maxLimits),
35
+ teardownGasLimits: padGas(teardownGas, pad, maxLimits)
36
+ };
37
+ }
38
+ /** Pads each gas dimension, capping it at the network admission limit. */ function padGas(gas, pad, cap) {
39
+ const padded = gas.mul(1 + pad);
40
+ return new Gas(Math.min(padded.daGas, cap.daGas), Math.min(padded.l2Gas, cap.l2Gas));
41
+ }
42
+ /**
43
+ * Validates that caller-declared gas limits do not exceed the network's per-tx admission limits, throwing a
44
+ * descriptive error per dimension when they do. The node's inbound validation checks declared
45
+ * `gasSettings.gasLimits`, so we mirror that here to surface the rejection locally before the tx is sent.
46
+ * @param gasLimits - The gas limits the transaction will declare.
47
+ * @param maxTxGasLimits - The maximum gas a single tx may declare on this network (the node-advertised `txsLimits.gas`).
48
+ */ export function assertGasLimitsWithinNetworkLimits(gasLimits, maxTxGasLimits) {
49
+ if (gasLimits.daGas > maxTxGasLimits.daGas) {
50
+ throw new Error(`Declared DA gas limit (${gasLimits.daGas}) exceeds the maximum this network allows per tx (${maxTxGasLimits.daGas})`);
51
+ }
52
+ if (gasLimits.l2Gas > maxTxGasLimits.l2Gas) {
53
+ throw new Error(`Declared L2 gas limit (${gasLimits.l2Gas}) exceeds the maximum this network allows per tx (${maxTxGasLimits.l2Gas})`);
54
+ }
55
+ }
@@ -1,2 +1,4 @@
1
- export { BaseWallet, type FeeOptions } from './base_wallet.js';
2
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9iYXNlLXdhbGxldC9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQUUsVUFBVSxFQUFFLEtBQUssVUFBVSxFQUFFLE1BQU0sa0JBQWtCLENBQUMifQ==
1
+ export { BaseWallet, type CompleteFeeOptionsConfig, type FeeOptions, type SimulateViaEntrypointOptions, } from './base_wallet.js';
2
+ export { simulateViaNode, buildMergedSimulationResult, extractOptimizablePublicStaticCalls } from './utils.js';
3
+ export { getGasLimits, assertGasLimitsWithinNetworkLimits } from './get_gas_limits.js';
4
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9iYXNlLXdhbGxldC9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQ0wsVUFBVSxFQUNWLEtBQUssd0JBQXdCLEVBQzdCLEtBQUssVUFBVSxFQUNmLEtBQUssNEJBQTRCLEdBQ2xDLE1BQU0sa0JBQWtCLENBQUM7QUFDMUIsT0FBTyxFQUFFLGVBQWUsRUFBRSwyQkFBMkIsRUFBRSxtQ0FBbUMsRUFBRSxNQUFNLFlBQVksQ0FBQztBQUMvRyxPQUFPLEVBQUUsWUFBWSxFQUFFLGtDQUFrQyxFQUFFLE1BQU0scUJBQXFCLENBQUMifQ==