@aztec/wallet-sdk 0.0.1-commit.0b941701 → 0.0.1-commit.0dc957cde
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.
- package/dest/base-wallet/base_wallet.d.ts +85 -35
- package/dest/base-wallet/base_wallet.d.ts.map +1 -1
- package/dest/base-wallet/base_wallet.js +238 -69
- package/dest/base-wallet/index.d.ts +3 -2
- package/dest/base-wallet/index.d.ts.map +1 -1
- package/dest/base-wallet/index.js +1 -0
- package/dest/base-wallet/utils.d.ts +50 -0
- package/dest/base-wallet/utils.d.ts.map +1 -0
- package/dest/base-wallet/utils.js +133 -0
- package/dest/crypto.d.ts +39 -1
- package/dest/crypto.d.ts.map +1 -1
- package/dest/crypto.js +88 -0
- package/dest/extension/provider/extension_wallet.d.ts +4 -6
- package/dest/extension/provider/extension_wallet.d.ts.map +1 -1
- package/dest/extension/provider/extension_wallet.js +9 -2
- package/dest/extension/provider/index.d.ts +2 -2
- package/dest/extension/provider/index.d.ts.map +1 -1
- package/dest/iframe/handlers/iframe_connection_handler.d.ts +118 -0
- package/dest/iframe/handlers/iframe_connection_handler.d.ts.map +1 -0
- package/dest/iframe/handlers/iframe_connection_handler.js +228 -0
- package/dest/iframe/handlers/index.d.ts +2 -0
- package/dest/iframe/handlers/index.d.ts.map +1 -0
- package/dest/iframe/handlers/index.js +1 -0
- package/dest/iframe/provider/iframe_discovery.d.ts +25 -0
- package/dest/iframe/provider/iframe_discovery.d.ts.map +1 -0
- package/dest/iframe/provider/iframe_discovery.js +167 -0
- package/dest/iframe/provider/iframe_provider.d.ts +65 -0
- package/dest/iframe/provider/iframe_provider.d.ts.map +1 -0
- package/dest/iframe/provider/iframe_provider.js +257 -0
- package/dest/iframe/provider/iframe_wallet.d.ts +68 -0
- package/dest/iframe/provider/iframe_wallet.d.ts.map +1 -0
- package/dest/iframe/provider/iframe_wallet.js +200 -0
- package/dest/iframe/provider/index.d.ts +4 -0
- package/dest/iframe/provider/index.d.ts.map +1 -0
- package/dest/iframe/provider/index.js +3 -0
- package/dest/manager/types.d.ts +6 -5
- package/dest/manager/types.d.ts.map +1 -1
- package/dest/manager/wallet_manager.d.ts +1 -1
- package/dest/manager/wallet_manager.d.ts.map +1 -1
- package/dest/manager/wallet_manager.js +48 -18
- package/dest/types.d.ts +14 -2
- package/dest/types.d.ts.map +1 -1
- package/dest/types.js +4 -0
- package/package.json +19 -9
- package/src/base-wallet/base_wallet.ts +330 -122
- package/src/base-wallet/index.ts +7 -1
- package/src/base-wallet/utils.ts +240 -0
- package/src/crypto.ts +104 -0
- package/src/extension/provider/extension_wallet.ts +13 -10
- package/src/extension/provider/index.ts +1 -1
- package/src/iframe/handlers/iframe_connection_handler.ts +328 -0
- package/src/iframe/handlers/index.ts +7 -0
- package/src/iframe/provider/iframe_discovery.ts +185 -0
- package/src/iframe/provider/iframe_provider.ts +331 -0
- package/src/iframe/provider/iframe_wallet.ts +229 -0
- package/src/iframe/provider/index.ts +3 -0
- package/src/manager/types.ts +5 -4
- package/src/manager/wallet_manager.ts +55 -23
- package/src/types.ts +13 -0
|
@@ -1,16 +1,21 @@
|
|
|
1
|
-
import {
|
|
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 {
|
|
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';
|
|
11
|
+
import { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
8
12
|
import { computePartialAddress, getContractClassFromArtifact } from '@aztec/stdlib/contract';
|
|
9
13
|
import { SimulationError } from '@aztec/stdlib/errors';
|
|
10
|
-
import { Gas, GasSettings } from '@aztec/stdlib/gas';
|
|
11
|
-
import {
|
|
14
|
+
import { Gas, GasFees, GasSettings, ManaUsageEstimate } from '@aztec/stdlib/gas';
|
|
15
|
+
import { computeSiloedPrivateInitializationNullifier, computeSiloedPublicInitializationNullifier } from '@aztec/stdlib/hash';
|
|
12
16
|
import { mergeExecutionPayloads } from '@aztec/stdlib/tx';
|
|
13
17
|
import { inspect } from 'util';
|
|
18
|
+
import { buildMergedSimulationResult, extractOptimizablePublicStaticCalls, simulateViaNode } from './utils.js';
|
|
14
19
|
/**
|
|
15
20
|
* A base class for Wallet implementations
|
|
16
21
|
*/ export class BaseWallet {
|
|
@@ -19,14 +24,27 @@ import { inspect } from 'util';
|
|
|
19
24
|
log;
|
|
20
25
|
minFeePadding;
|
|
21
26
|
cancellableTransactions;
|
|
27
|
+
// A wallet is instantiated for a particular chain, so chain info never changes during its lifetime.
|
|
28
|
+
// We cache it here because getChainInfo is called frequently (every tx simulation, send, auth wit, etc.).
|
|
29
|
+
nodeInfoPromise;
|
|
22
30
|
// Protected because we want to force wallets to instantiate their own PXE.
|
|
23
|
-
constructor(pxe, aztecNode){
|
|
31
|
+
constructor(pxe, aztecNode, log = createLogger('wallet-sdk:base_wallet')){
|
|
24
32
|
this.pxe = pxe;
|
|
25
33
|
this.aztecNode = aztecNode;
|
|
26
|
-
this.log =
|
|
34
|
+
this.log = log;
|
|
27
35
|
this.minFeePadding = 0.5;
|
|
28
36
|
this.cancellableTransactions = false;
|
|
29
37
|
}
|
|
38
|
+
scopesFrom(from, additionalScopes = []) {
|
|
39
|
+
const allScopes = from === NO_FROM ? additionalScopes : [
|
|
40
|
+
from,
|
|
41
|
+
...additionalScopes
|
|
42
|
+
];
|
|
43
|
+
const scopeSet = new Set(allScopes.map((address)=>address.toString()));
|
|
44
|
+
return [
|
|
45
|
+
...scopeSet
|
|
46
|
+
].map(AztecAddress.fromString);
|
|
47
|
+
}
|
|
30
48
|
/**
|
|
31
49
|
* Returns the list of aliased contacts associated with the wallet.
|
|
32
50
|
* This base implementation directly returns PXE's senders, but note that in general contacts are a superset of senders.
|
|
@@ -41,7 +59,10 @@ import { inspect } from 'util';
|
|
|
41
59
|
}));
|
|
42
60
|
}
|
|
43
61
|
async getChainInfo() {
|
|
44
|
-
|
|
62
|
+
if (!this.nodeInfoPromise) {
|
|
63
|
+
this.nodeInfoPromise = this.aztecNode.getNodeInfo();
|
|
64
|
+
}
|
|
65
|
+
const { l1ChainId, rollupVersion } = await this.nodeInfoPromise;
|
|
45
66
|
return {
|
|
46
67
|
chainId: new Fr(l1ChainId),
|
|
47
68
|
version: new Fr(rollupVersion)
|
|
@@ -49,24 +70,45 @@ import { inspect } from 'util';
|
|
|
49
70
|
}
|
|
50
71
|
async createTxExecutionRequestFromPayloadAndFee(executionPayload, from, feeOptions) {
|
|
51
72
|
const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
|
|
52
|
-
const executionOptions = {
|
|
53
|
-
txNonce: Fr.random(),
|
|
54
|
-
cancellable: this.cancellableTransactions,
|
|
55
|
-
feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions
|
|
56
|
-
};
|
|
57
73
|
const finalExecutionPayload = feeExecutionPayload ? mergeExecutionPayloads([
|
|
58
74
|
feeExecutionPayload,
|
|
59
75
|
executionPayload
|
|
60
76
|
]) : executionPayload;
|
|
61
|
-
const fromAccount = await this.getAccountFromAddress(from);
|
|
62
77
|
const chainInfo = await this.getChainInfo();
|
|
63
|
-
|
|
78
|
+
if (from === NO_FROM) {
|
|
79
|
+
const entrypoint = new DefaultEntrypoint();
|
|
80
|
+
return entrypoint.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo);
|
|
81
|
+
} else {
|
|
82
|
+
const fromAccount = await this.getAccountFromAddress(from);
|
|
83
|
+
const executionOptions = {
|
|
84
|
+
txNonce: Fr.random(),
|
|
85
|
+
cancellable: this.cancellableTransactions,
|
|
86
|
+
// If from is an address, feeOptions include the way the account contract should handle the fee payment
|
|
87
|
+
feePaymentMethodOptions: feeOptions.accountFeePaymentMethodOptions
|
|
88
|
+
};
|
|
89
|
+
return fromAccount.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo, executionOptions);
|
|
90
|
+
}
|
|
64
91
|
}
|
|
65
92
|
async createAuthWit(from, messageHashOrIntent) {
|
|
66
93
|
const account = await this.getAccountFromAddress(from);
|
|
67
94
|
const chainInfo = await this.getChainInfo();
|
|
68
95
|
return account.createAuthWit(messageHashOrIntent, chainInfo);
|
|
69
96
|
}
|
|
97
|
+
/**
|
|
98
|
+
* Request capabilities from the wallet.
|
|
99
|
+
*
|
|
100
|
+
* This method is wallet-implementation-dependent and must be provided by classes extending BaseWallet.
|
|
101
|
+
* Embedded wallets typically don't support capability-based authorization (no user authorization flow),
|
|
102
|
+
* while external wallets (browser extensions, hardware wallets) implement this to reduce authorization
|
|
103
|
+
* friction by allowing apps to request permissions upfront.
|
|
104
|
+
*
|
|
105
|
+
* TODO: Consider making it abstract so implementing it is a conscious decision. Leaving it as-is
|
|
106
|
+
* while the feature stabilizes.
|
|
107
|
+
*
|
|
108
|
+
* @param _manifest - Application capability manifest declaring what operations the app needs
|
|
109
|
+
*/ requestCapabilities(_manifest) {
|
|
110
|
+
throw new Error('Not implemented');
|
|
111
|
+
}
|
|
70
112
|
async batch(methods) {
|
|
71
113
|
const results = [];
|
|
72
114
|
for (const method of methods){
|
|
@@ -88,26 +130,33 @@ import { inspect } from 'util';
|
|
|
88
130
|
}
|
|
89
131
|
/**
|
|
90
132
|
* Completes partial user-provided fee options with wallet defaults.
|
|
91
|
-
* @param
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
*/ async completeFeeOptions(from, feePayer, gasSettings) {
|
|
96
|
-
const maxFeesPerGas = gasSettings?.maxFeesPerGas ?? (await this.aztecNode.getCurrentMinFees()).mul(1 + this.minFeePadding);
|
|
133
|
+
* @param config - Fee completion config.
|
|
134
|
+
*/ async completeFeeOptions(config) {
|
|
135
|
+
const { from, feePayer, gasSettings, forEstimation, congestionEstimate } = config;
|
|
136
|
+
const maxFeesPerGas = gasSettings?.maxFeesPerGas ?? (await this.getMinFees(congestionEstimate)).mul(1 + this.minFeePadding);
|
|
97
137
|
let accountFeePaymentMethodOptions;
|
|
98
|
-
//
|
|
99
|
-
//
|
|
100
|
-
if (
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
138
|
+
// If from is an address, we need to determine the appropriate fee payment method options for the
|
|
139
|
+
// account contract entrypoint to use
|
|
140
|
+
if (from !== NO_FROM) {
|
|
141
|
+
if (!feePayer) {
|
|
142
|
+
// The transaction does not include a fee payment method, so we set the flag
|
|
143
|
+
// for the account to use its fee juice balance
|
|
144
|
+
accountFeePaymentMethodOptions = AccountFeePaymentMethodOptions.PREEXISTING_FEE_JUICE;
|
|
145
|
+
} else {
|
|
146
|
+
// The transaction includes fee payment method, so we check if we are the fee payer for it
|
|
147
|
+
// (this can only happen if the embedded payment method is FeeJuiceWithClaim)
|
|
148
|
+
accountFeePaymentMethodOptions = from.equals(feePayer) ? AccountFeePaymentMethodOptions.FEE_JUICE_WITH_CLAIM : AccountFeePaymentMethodOptions.EXTERNAL;
|
|
149
|
+
}
|
|
106
150
|
}
|
|
107
|
-
const
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
151
|
+
const gasSettingsOverrides = {
|
|
152
|
+
gasLimits: gasSettings?.gasLimits ? Gas.from(gasSettings.gasLimits) : undefined,
|
|
153
|
+
teardownGasLimits: gasSettings?.teardownGasLimits ? Gas.from(gasSettings.teardownGasLimits) : undefined,
|
|
154
|
+
maxFeesPerGas,
|
|
155
|
+
maxPriorityFeesPerGas: gasSettings?.maxPriorityFeesPerGas ?? GasFees.empty()
|
|
156
|
+
};
|
|
157
|
+
// When estimating gas (simulation), use high limits so the simulation doesn't run out of gas.
|
|
158
|
+
// When sending for real, use protocol max limits that the network will actually accept.
|
|
159
|
+
const fullGasSettings = forEstimation ? GasSettings.forEstimation(gasSettingsOverrides) : GasSettings.fallback(gasSettingsOverrides);
|
|
111
160
|
this.log.debug(`Using L2 gas settings`, fullGasSettings);
|
|
112
161
|
return {
|
|
113
162
|
gasSettings: fullGasSettings,
|
|
@@ -116,22 +165,24 @@ import { inspect } from 'util';
|
|
|
116
165
|
};
|
|
117
166
|
}
|
|
118
167
|
/**
|
|
119
|
-
*
|
|
120
|
-
*
|
|
121
|
-
* to
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
168
|
+
* Returns the worst-case min fee across predicted future slots.
|
|
169
|
+
* Falls back to getCurrentMinFees if the node doesn't support getPredictedMinFees.
|
|
170
|
+
* @param estimate - The mana usage estimate to use for fee prediction. Defaults to Limit for conservative estimation.
|
|
171
|
+
*/ async getMinFees(estimate = ManaUsageEstimate.Limit) {
|
|
172
|
+
try {
|
|
173
|
+
const predicted = await this.aztecNode.getPredictedMinFees(estimate);
|
|
174
|
+
if (predicted.length === 0) {
|
|
175
|
+
return this.aztecNode.getCurrentMinFees();
|
|
176
|
+
}
|
|
177
|
+
return predicted.reduce((worst, fees)=>fees.feePerL2Gas > worst.feePerL2Gas ? fees : worst);
|
|
178
|
+
} catch (err) {
|
|
179
|
+
// Fallback for old nodes that don't support getPredictedMinFees.
|
|
180
|
+
// Only fall back on method-not-found errors (JSON-RPC code -32601); rethrow others.
|
|
181
|
+
if (err?.cause?.code === -32601 || err?.message?.includes('Method not found')) {
|
|
182
|
+
return this.aztecNode.getCurrentMinFees();
|
|
183
|
+
}
|
|
184
|
+
throw err;
|
|
185
|
+
}
|
|
135
186
|
}
|
|
136
187
|
registerSender(address, _alias = '') {
|
|
137
188
|
return this.pxe.registerSender(address);
|
|
@@ -168,20 +219,99 @@ import { inspect } from 'util';
|
|
|
168
219
|
}
|
|
169
220
|
return instance;
|
|
170
221
|
}
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
222
|
+
/**
|
|
223
|
+
* Simulates calls through the standard PXE path (account entrypoint).
|
|
224
|
+
* @param executionPayload - The execution payload to simulate.
|
|
225
|
+
* @param opts - Simulation options.
|
|
226
|
+
*/ async simulateViaEntrypoint(executionPayload, opts) {
|
|
227
|
+
const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, opts.feeOptions);
|
|
228
|
+
const result = await this.pxe.simulateTx(txRequest, {
|
|
229
|
+
simulatePublic: true,
|
|
230
|
+
skipTxValidation: opts.skipTxValidation,
|
|
231
|
+
skipFeeEnforcement: opts.skipFeeEnforcement,
|
|
232
|
+
scopes: this.scopesFrom(opts.from, opts.additionalScopes)
|
|
233
|
+
});
|
|
234
|
+
const appCallOffset = await this.computeAppCallOffset(opts.from, opts.feeOptions);
|
|
235
|
+
return TxSimulationResultWithAppOffset.fromResultAndOffset(result, appCallOffset);
|
|
236
|
+
}
|
|
237
|
+
/**
|
|
238
|
+
* Computes the index where the app's calls begin in the flattened array of calls (0 = entrypoint/root, 1..N = fee
|
|
239
|
+
* calls, N+1 = app).
|
|
240
|
+
* @param from - The sender address, or NO_FROM for the default entrypoint.
|
|
241
|
+
* @param feeOptions - Fee options containing the wallet fee payment method.
|
|
242
|
+
*/ async computeAppCallOffset(from, feeOptions) {
|
|
243
|
+
if (from === NO_FROM) {
|
|
244
|
+
return 0;
|
|
245
|
+
}
|
|
246
|
+
const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
|
|
247
|
+
return (feeExecutionPayload?.calls.length ?? 0) + 1; // +1 for entrypoint
|
|
248
|
+
}
|
|
249
|
+
/**
|
|
250
|
+
* Simulates a transaction, optimizing leading public static calls by running them directly
|
|
251
|
+
* on the node while sending the remaining calls through the standard PXE path.
|
|
252
|
+
* Return values from both paths are merged back in original call order.
|
|
253
|
+
* @param executionPayload - The execution payload to simulate.
|
|
254
|
+
* @param opts - Simulation options (from address, fee settings, etc.).
|
|
255
|
+
* @returns The merged simulation result.
|
|
256
|
+
*/ async simulateTx(executionPayload, opts) {
|
|
257
|
+
const feeOptions = await this.completeFeeOptions({
|
|
258
|
+
from: opts.from,
|
|
259
|
+
feePayer: executionPayload.feePayer,
|
|
260
|
+
gasSettings: opts.fee?.gasSettings,
|
|
261
|
+
forEstimation: true,
|
|
262
|
+
congestionEstimate: opts.fee?.congestionEstimate
|
|
263
|
+
});
|
|
264
|
+
const { optimizableCalls, remainingCalls } = extractOptimizablePublicStaticCalls(executionPayload);
|
|
265
|
+
const remainingPayload = {
|
|
266
|
+
...executionPayload,
|
|
267
|
+
calls: remainingCalls
|
|
268
|
+
};
|
|
269
|
+
const chainInfo = await this.getChainInfo();
|
|
270
|
+
let blockHeader;
|
|
271
|
+
// PXE might not be synced yet, so we pull the latest header from the node
|
|
272
|
+
// To keep things consistent, we'll always try with PXE first
|
|
273
|
+
try {
|
|
274
|
+
blockHeader = await this.pxe.getSyncedBlockHeader();
|
|
275
|
+
} catch {
|
|
276
|
+
blockHeader = await this.aztecNode.getBlockHeader();
|
|
277
|
+
}
|
|
278
|
+
const simulationOrigin = opts.from === NO_FROM ? AztecAddress.ZERO : opts.from;
|
|
279
|
+
const [optimizedResults, normalResult] = await Promise.all([
|
|
280
|
+
optimizableCalls.length > 0 ? simulateViaNode(this.aztecNode, optimizableCalls, simulationOrigin, chainInfo, feeOptions.gasSettings, blockHeader, opts.skipFeeEnforcement ?? true, this.getContractName.bind(this)) : Promise.resolve([]),
|
|
281
|
+
remainingCalls.length > 0 ? this.simulateViaEntrypoint(remainingPayload, {
|
|
282
|
+
from: opts.from,
|
|
283
|
+
feeOptions,
|
|
284
|
+
additionalScopes: opts.additionalScopes,
|
|
285
|
+
skipTxValidation: opts.skipTxValidation,
|
|
286
|
+
skipFeeEnforcement: opts.skipFeeEnforcement ?? true
|
|
287
|
+
}) : Promise.resolve(null)
|
|
288
|
+
]);
|
|
289
|
+
return buildMergedSimulationResult(optimizedResults, normalResult);
|
|
175
290
|
}
|
|
176
291
|
async profileTx(executionPayload, opts) {
|
|
177
|
-
const feeOptions = await this.completeFeeOptions(
|
|
292
|
+
const feeOptions = await this.completeFeeOptions({
|
|
293
|
+
from: opts.from,
|
|
294
|
+
feePayer: executionPayload.feePayer,
|
|
295
|
+
gasSettings: opts.fee?.gasSettings,
|
|
296
|
+
congestionEstimate: opts.fee?.congestionEstimate
|
|
297
|
+
});
|
|
178
298
|
const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, feeOptions);
|
|
179
|
-
return this.pxe.profileTx(txRequest,
|
|
299
|
+
return this.pxe.profileTx(txRequest, {
|
|
300
|
+
profileMode: opts.profileMode,
|
|
301
|
+
skipProofGeneration: opts.skipProofGeneration ?? true,
|
|
302
|
+
scopes: this.scopesFrom(opts.from, opts.additionalScopes)
|
|
303
|
+
});
|
|
180
304
|
}
|
|
181
305
|
async sendTx(executionPayload, opts) {
|
|
182
|
-
const feeOptions = await this.completeFeeOptions(
|
|
306
|
+
const feeOptions = await this.completeFeeOptions({
|
|
307
|
+
from: opts.from,
|
|
308
|
+
feePayer: executionPayload.feePayer,
|
|
309
|
+
gasSettings: opts.fee?.gasSettings,
|
|
310
|
+
congestionEstimate: opts.fee?.congestionEstimate
|
|
311
|
+
});
|
|
183
312
|
const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, feeOptions);
|
|
184
|
-
const provenTx = await this.pxe.proveTx(txRequest);
|
|
313
|
+
const provenTx = await this.pxe.proveTx(txRequest, this.scopesFrom(opts.from, opts.additionalScopes));
|
|
314
|
+
const offchainOutput = extractOffchainOutput(provenTx.getOffchainEffects(), provenTx.publicInputs.constants.anchorBlockHeader.globalVariables.timestamp);
|
|
185
315
|
const tx = await provenTx.toTx();
|
|
186
316
|
const txHash = tx.getTxHash();
|
|
187
317
|
if (await this.aztecNode.getTxEffect(txHash)) {
|
|
@@ -194,11 +324,33 @@ import { inspect } from 'util';
|
|
|
194
324
|
this.log.info(`Sent transaction ${txHash}`);
|
|
195
325
|
// If wait is NO_WAIT, return txHash immediately
|
|
196
326
|
if (opts.wait === NO_WAIT) {
|
|
197
|
-
return
|
|
327
|
+
return {
|
|
328
|
+
txHash,
|
|
329
|
+
...offchainOutput
|
|
330
|
+
};
|
|
198
331
|
}
|
|
199
332
|
// Otherwise, wait for the full receipt (default behavior on wait: undefined)
|
|
200
333
|
const waitOpts = typeof opts.wait === 'object' ? opts.wait : undefined;
|
|
201
|
-
|
|
334
|
+
const receipt = await waitForTx(this.aztecNode, txHash, waitOpts);
|
|
335
|
+
// Display debug logs from public execution if present (served in test mode only)
|
|
336
|
+
if (receipt.debugLogs?.length) {
|
|
337
|
+
await displayDebugLogs(receipt.debugLogs, this.getContractName.bind(this));
|
|
338
|
+
}
|
|
339
|
+
return {
|
|
340
|
+
receipt,
|
|
341
|
+
...offchainOutput
|
|
342
|
+
};
|
|
343
|
+
}
|
|
344
|
+
/**
|
|
345
|
+
* Resolves a contract address to a human-readable name via PXE, if available.
|
|
346
|
+
* @param address - The contract address to resolve.
|
|
347
|
+
*/ async getContractName(address) {
|
|
348
|
+
const instance = await this.pxe.getContractInstance(address);
|
|
349
|
+
if (!instance) {
|
|
350
|
+
return undefined;
|
|
351
|
+
}
|
|
352
|
+
const artifact = await this.pxe.getContractArtifact(instance.currentContractClassId);
|
|
353
|
+
return artifact?.name;
|
|
202
354
|
}
|
|
203
355
|
contextualizeError(err, ...context) {
|
|
204
356
|
let contextStr = '';
|
|
@@ -213,8 +365,11 @@ import { inspect } from 'util';
|
|
|
213
365
|
}
|
|
214
366
|
return err;
|
|
215
367
|
}
|
|
216
|
-
|
|
217
|
-
return this.pxe.
|
|
368
|
+
executeUtility(call, opts) {
|
|
369
|
+
return this.pxe.executeUtility(call, {
|
|
370
|
+
authwits: opts.authWitnesses,
|
|
371
|
+
scopes: opts.scopes
|
|
372
|
+
});
|
|
218
373
|
}
|
|
219
374
|
async getPrivateEvents(eventDef, eventFilter) {
|
|
220
375
|
const pxeEvents = await this.pxe.getPrivateEvents(eventDef.eventSelector, eventFilter);
|
|
@@ -232,20 +387,34 @@ import { inspect } from 'util';
|
|
|
232
387
|
});
|
|
233
388
|
return decodedEvents;
|
|
234
389
|
}
|
|
235
|
-
|
|
390
|
+
/**
|
|
391
|
+
* Returns metadata about a contract, including whether it has been initialized, published, and updated.
|
|
392
|
+
* @param address - The contract address to query.
|
|
393
|
+
*/ async getContractMetadata(address) {
|
|
236
394
|
const instance = await this.pxe.getContractInstance(address);
|
|
237
|
-
const
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
395
|
+
const publiclyRegisteredContractPromise = this.aztecNode.getContract(address);
|
|
396
|
+
let initializationStatus;
|
|
397
|
+
if (instance) {
|
|
398
|
+
// We have the instance, so we can compute the private initialization nullifier (which includes init_hash and is
|
|
399
|
+
// emitted by both private and public initializers) and get a definitive INITIALIZED/UNINITIALIZED answer.
|
|
400
|
+
const initNullifier = await computeSiloedPrivateInitializationNullifier(address, instance.initializationHash);
|
|
401
|
+
const witness = await this.aztecNode.getNullifierMembershipWitness('latest', initNullifier);
|
|
402
|
+
initializationStatus = witness ? ContractInitializationStatus.INITIALIZED : ContractInitializationStatus.UNINITIALIZED;
|
|
403
|
+
} else {
|
|
404
|
+
// Without the instance we lack the init_hash needed for the private nullifier. We fall back to checking the
|
|
405
|
+
// public initialization nullifier (computed from address alone). Not all contracts emit it (only those with
|
|
406
|
+
// public functions that require initialization checks), so its absence doesn't mean the contract is
|
|
407
|
+
// uninitialized.
|
|
408
|
+
const publicNullifier = await computeSiloedPublicInitializationNullifier(address);
|
|
409
|
+
const witness = await this.aztecNode.getNullifierMembershipWitness('latest', publicNullifier);
|
|
410
|
+
initializationStatus = witness ? ContractInitializationStatus.INITIALIZED : ContractInitializationStatus.UNKNOWN;
|
|
411
|
+
}
|
|
412
|
+
const publiclyRegisteredContract = await publiclyRegisteredContractPromise;
|
|
243
413
|
const isContractUpdated = publiclyRegisteredContract && !publiclyRegisteredContract.currentContractClassId.equals(publiclyRegisteredContract.originalContractClassId);
|
|
244
414
|
return {
|
|
245
415
|
instance: instance ?? undefined,
|
|
246
|
-
|
|
416
|
+
initializationStatus,
|
|
247
417
|
isContractPublished: !!publiclyRegisteredContract,
|
|
248
|
-
isContractClassPubliclyRegistered: !!publiclyRegisteredContractClass,
|
|
249
418
|
isContractUpdated: !!isContractUpdated,
|
|
250
419
|
updatedContractClassId: isContractUpdated ? publiclyRegisteredContract.currentContractClassId : undefined
|
|
251
420
|
};
|
|
@@ -1,2 +1,3 @@
|
|
|
1
|
-
export { BaseWallet, type FeeOptions } from './base_wallet.js';
|
|
2
|
-
|
|
1
|
+
export { BaseWallet, type CompleteFeeOptionsConfig, type FeeOptions, type SimulateViaEntrypointOptions, } from './base_wallet.js';
|
|
2
|
+
export { simulateViaNode, buildMergedSimulationResult, extractOptimizablePublicStaticCalls } from './utils.js';
|
|
3
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9iYXNlLXdhbGxldC9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQ0wsVUFBVSxFQUNWLEtBQUssd0JBQXdCLEVBQzdCLEtBQUssVUFBVSxFQUNmLEtBQUssNEJBQTRCLEdBQ2xDLE1BQU0sa0JBQWtCLENBQUM7QUFDMUIsT0FBTyxFQUFFLGVBQWUsRUFBRSwyQkFBMkIsRUFBRSxtQ0FBbUMsRUFBRSxNQUFNLFlBQVksQ0FBQyJ9
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/base-wallet/index.ts"],"names":[],"mappings":"AAAA,OAAO,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/base-wallet/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,EACV,KAAK,wBAAwB,EAC7B,KAAK,UAAU,EACf,KAAK,4BAA4B,GAClC,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,eAAe,EAAE,2BAA2B,EAAE,mCAAmC,EAAE,MAAM,YAAY,CAAC"}
|
|
@@ -0,0 +1,50 @@
|
|
|
1
|
+
import type { AztecNode } from '@aztec/aztec.js/node';
|
|
2
|
+
import { TxSimulationResultWithAppOffset } from '@aztec/aztec.js/wallet';
|
|
3
|
+
import type { ChainInfo } from '@aztec/entrypoints/interfaces';
|
|
4
|
+
import type { ContractNameResolver } from '@aztec/pxe/client/lazy';
|
|
5
|
+
import { type FunctionCall } from '@aztec/stdlib/abi';
|
|
6
|
+
import type { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
7
|
+
import type { GasSettings } from '@aztec/stdlib/gas';
|
|
8
|
+
import { type BlockHeader, type ExecutionPayload, TxSimulationResult } from '@aztec/stdlib/tx';
|
|
9
|
+
/**
|
|
10
|
+
* Splits an execution payload into a leading prefix of public static calls
|
|
11
|
+
* (eligible for direct node simulation) and the remaining calls.
|
|
12
|
+
*
|
|
13
|
+
* Only a leading run of public static calls is eligible for optimization.
|
|
14
|
+
* Any non-public-static call may enqueue public state mutations
|
|
15
|
+
* (e.g. private calls can enqueue public calls), so all calls that follow
|
|
16
|
+
* must go through the normal simulation path to see the correct state.
|
|
17
|
+
*
|
|
18
|
+
*/
|
|
19
|
+
export declare function extractOptimizablePublicStaticCalls(payload: ExecutionPayload): {
|
|
20
|
+
/** Leading public static calls eligible for direct node simulation. */
|
|
21
|
+
optimizableCalls: FunctionCall[];
|
|
22
|
+
/** All remaining calls. */
|
|
23
|
+
remainingCalls: FunctionCall[];
|
|
24
|
+
};
|
|
25
|
+
/**
|
|
26
|
+
* Simulates public static calls by splitting them into batches of MAX_ENQUEUED_CALLS_PER_CALL
|
|
27
|
+
* and sending each batch directly to the node.
|
|
28
|
+
*
|
|
29
|
+
* @param node - The Aztec node to simulate on.
|
|
30
|
+
* @param publicStaticCalls - Array of public static function calls to optimize.
|
|
31
|
+
* @param from - The account address making the calls.
|
|
32
|
+
* @param chainInfo - Chain information (chainId and version).
|
|
33
|
+
* @param gasSettings - Gas settings for the transaction.
|
|
34
|
+
* @param blockHeader - Block header to use as anchor.
|
|
35
|
+
* @param skipFeeEnforcement - Whether to skip fee enforcement during simulation.
|
|
36
|
+
* @returns Array of TxSimulationResult, one per batch.
|
|
37
|
+
*/
|
|
38
|
+
export declare function simulateViaNode(node: AztecNode, publicStaticCalls: FunctionCall[], from: AztecAddress, chainInfo: ChainInfo, gasSettings: GasSettings, blockHeader: BlockHeader, skipFeeEnforcement: boolean | undefined, getContractName: ContractNameResolver): Promise<TxSimulationResult[]>;
|
|
39
|
+
/**
|
|
40
|
+
* Merges simulation results from the optimized (public static) and normal paths.
|
|
41
|
+
* Since optimized calls are always a leading prefix, return values are simply
|
|
42
|
+
* concatenated: optimized first, then normal.
|
|
43
|
+
* Stats are taken from the normal result only (the optimized path doesn't produce them).
|
|
44
|
+
*
|
|
45
|
+
* @param optimizedResults - Results from optimized public static call batches.
|
|
46
|
+
* @param normalResult - Result from normal simulation (null if all calls were optimized).
|
|
47
|
+
* @returns A single TxSimulationResult with return values in original call order.
|
|
48
|
+
*/
|
|
49
|
+
export declare function buildMergedSimulationResult(optimizedResults: TxSimulationResult[], normalResult: TxSimulationResultWithAppOffset | null): TxSimulationResultWithAppOffset;
|
|
50
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidXRpbHMuZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9iYXNlLXdhbGxldC91dGlscy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEtBQUssRUFBRSxTQUFTLEVBQUUsTUFBTSxzQkFBc0IsQ0FBQztBQUN0RCxPQUFPLEVBQUUsK0JBQStCLEVBQUUsTUFBTSx3QkFBd0IsQ0FBQztBQUV6RSxPQUFPLEtBQUssRUFBRSxTQUFTLEVBQUUsTUFBTSwrQkFBK0IsQ0FBQztBQUkvRCxPQUFPLEtBQUssRUFBRSxvQkFBb0IsRUFBRSxNQUFNLHdCQUF3QixDQUFDO0FBR25FLE9BQU8sRUFBRSxLQUFLLFlBQVksRUFBb0IsTUFBTSxtQkFBbUIsQ0FBQztBQUN4RSxPQUFPLEtBQUssRUFBRSxZQUFZLEVBQUUsTUFBTSw2QkFBNkIsQ0FBQztBQUNoRSxPQUFPLEtBQUssRUFBRSxXQUFXLEVBQUUsTUFBTSxtQkFBbUIsQ0FBQztBQVFyRCxPQUFPLEVBQ0wsS0FBSyxXQUFXLEVBQ2hCLEtBQUssZ0JBQWdCLEVBT3JCLGtCQUFrQixFQUNuQixNQUFNLGtCQUFrQixDQUFDO0FBRTFCOzs7Ozs7Ozs7R0FTRztBQUNILHdCQUFnQixtQ0FBbUMsQ0FBQyxPQUFPLEVBQUUsZ0JBQWdCLEdBQUc7SUFDOUUsdUVBQXVFO0lBQ3ZFLGdCQUFnQixFQUFFLFlBQVksRUFBRSxDQUFDO0lBQ2pDLDJCQUEyQjtJQUMzQixjQUFjLEVBQUUsWUFBWSxFQUFFLENBQUM7Q0FDaEMsQ0FPQTtBQXVHRDs7Ozs7Ozs7Ozs7O0dBWUc7QUFDSCx3QkFBc0IsZUFBZSxDQUNuQyxJQUFJLEVBQUUsU0FBUyxFQUNmLGlCQUFpQixFQUFFLFlBQVksRUFBRSxFQUNqQyxJQUFJLEVBQUUsWUFBWSxFQUNsQixTQUFTLEVBQUUsU0FBUyxFQUNwQixXQUFXLEVBQUUsV0FBVyxFQUN4QixXQUFXLEVBQUUsV0FBVyxFQUN4QixrQkFBa0IscUJBQWdCLEVBQ2xDLGVBQWUsRUFBRSxvQkFBb0IsR0FDcEMsT0FBTyxDQUFDLGtCQUFrQixFQUFFLENBQUMsQ0F3Qi9CO0FBRUQ7Ozs7Ozs7OztHQVNHO0FBQ0gsd0JBQWdCLDJCQUEyQixDQUN6QyxnQkFBZ0IsRUFBRSxrQkFBa0IsRUFBRSxFQUN0QyxZQUFZLEVBQUUsK0JBQStCLEdBQUcsSUFBSSxHQUNuRCwrQkFBK0IsQ0FxQmpDIn0=
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/base-wallet/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,+BAA+B,EAAE,MAAM,wBAAwB,CAAC;AAEzE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAI/D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAGnE,OAAO,EAAE,KAAK,YAAY,EAAoB,MAAM,mBAAmB,CAAC;AACxE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAQrD,OAAO,EACL,KAAK,WAAW,EAChB,KAAK,gBAAgB,EAOrB,kBAAkB,EACnB,MAAM,kBAAkB,CAAC;AAE1B;;;;;;;;;GASG;AACH,wBAAgB,mCAAmC,CAAC,OAAO,EAAE,gBAAgB,GAAG;IAC9E,uEAAuE;IACvE,gBAAgB,EAAE,YAAY,EAAE,CAAC;IACjC,2BAA2B;IAC3B,cAAc,EAAE,YAAY,EAAE,CAAC;CAChC,CAOA;AAuGD;;;;;;;;;;;;GAYG;AACH,wBAAsB,eAAe,CACnC,IAAI,EAAE,SAAS,EACf,iBAAiB,EAAE,YAAY,EAAE,EACjC,IAAI,EAAE,YAAY,EAClB,SAAS,EAAE,SAAS,EACpB,WAAW,EAAE,WAAW,EACxB,WAAW,EAAE,WAAW,EACxB,kBAAkB,qBAAgB,EAClC,eAAe,EAAE,oBAAoB,GACpC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAwB/B;AAED;;;;;;;;;GASG;AACH,wBAAgB,2BAA2B,CACzC,gBAAgB,EAAE,kBAAkB,EAAE,EACtC,YAAY,EAAE,+BAA+B,GAAG,IAAI,GACnD,+BAA+B,CAqBjC"}
|
|
@@ -0,0 +1,133 @@
|
|
|
1
|
+
import { TxSimulationResultWithAppOffset } from '@aztec/aztec.js/wallet';
|
|
2
|
+
import { MAX_ENQUEUED_CALLS_PER_CALL } from '@aztec/constants';
|
|
3
|
+
import { makeTuple } from '@aztec/foundation/array';
|
|
4
|
+
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
5
|
+
import { displayDebugLogs } from '@aztec/pxe/client/lazy';
|
|
6
|
+
import { generateSimulatedProvingResult } from '@aztec/pxe/simulator';
|
|
7
|
+
import { ClaimedLengthArray, CountedPublicCallRequest, PrivateCircuitPublicInputs, PublicCallRequest } from '@aztec/stdlib/kernel';
|
|
8
|
+
import { ChonkProof } from '@aztec/stdlib/proofs';
|
|
9
|
+
import { HashedValues, PrivateCallExecutionResult, PrivateExecutionResult, Tx, TxContext, TxSimulationResult } from '@aztec/stdlib/tx';
|
|
10
|
+
/**
|
|
11
|
+
* Splits an execution payload into a leading prefix of public static calls
|
|
12
|
+
* (eligible for direct node simulation) and the remaining calls.
|
|
13
|
+
*
|
|
14
|
+
* Only a leading run of public static calls is eligible for optimization.
|
|
15
|
+
* Any non-public-static call may enqueue public state mutations
|
|
16
|
+
* (e.g. private calls can enqueue public calls), so all calls that follow
|
|
17
|
+
* must go through the normal simulation path to see the correct state.
|
|
18
|
+
*
|
|
19
|
+
*/ export function extractOptimizablePublicStaticCalls(payload) {
|
|
20
|
+
const splitIndex = payload.calls.findIndex((call)=>!call.isPublicStatic());
|
|
21
|
+
const boundary = splitIndex === -1 ? payload.calls.length : splitIndex;
|
|
22
|
+
return {
|
|
23
|
+
optimizableCalls: payload.calls.slice(0, boundary),
|
|
24
|
+
remainingCalls: payload.calls.slice(boundary)
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
/**
|
|
28
|
+
* Simulates a batch of public static calls by bypassing account entrypoint and private execution,
|
|
29
|
+
* directly constructing a minimal Tx and calling node.simulatePublicCalls.
|
|
30
|
+
*
|
|
31
|
+
* @param node - The Aztec node to simulate on.
|
|
32
|
+
* @param publicStaticCalls - Array of public static function calls (max MAX_ENQUEUED_CALLS_PER_CALL).
|
|
33
|
+
* @param from - The account address making the calls.
|
|
34
|
+
* @param chainInfo - Chain information (chainId and version).
|
|
35
|
+
* @param gasSettings - Gas settings for the transaction.
|
|
36
|
+
* @param blockHeader - Block header to use as anchor.
|
|
37
|
+
* @param skipFeeEnforcement - Whether to skip fee enforcement during simulation.
|
|
38
|
+
* @returns TxSimulationResult with public return values.
|
|
39
|
+
*/ async function simulateBatchViaNode(node, publicStaticCalls, from, chainInfo, gasSettings, blockHeader, skipFeeEnforcement, getContractName) {
|
|
40
|
+
const txContext = new TxContext(chainInfo.chainId, chainInfo.version, gasSettings);
|
|
41
|
+
const publicFunctionCalldata = [];
|
|
42
|
+
for (const call of publicStaticCalls){
|
|
43
|
+
const calldata = await HashedValues.fromCalldata([
|
|
44
|
+
call.selector.toField(),
|
|
45
|
+
...call.args
|
|
46
|
+
]);
|
|
47
|
+
publicFunctionCalldata.push(calldata);
|
|
48
|
+
}
|
|
49
|
+
const publicCallRequests = makeTuple(MAX_ENQUEUED_CALLS_PER_CALL, (i)=>{
|
|
50
|
+
const call = publicStaticCalls[i];
|
|
51
|
+
if (!call) {
|
|
52
|
+
return CountedPublicCallRequest.empty();
|
|
53
|
+
}
|
|
54
|
+
const publicCallRequest = new PublicCallRequest(from, call.to, call.isStatic, publicFunctionCalldata[i].hash);
|
|
55
|
+
// Counter starts at 1 (minRevertibleSideEffectCounter) so all calls are revertible
|
|
56
|
+
return new CountedPublicCallRequest(publicCallRequest, i + 1);
|
|
57
|
+
});
|
|
58
|
+
const publicCallRequestsArray = new ClaimedLengthArray(publicCallRequests, publicStaticCalls.length);
|
|
59
|
+
const publicInputs = PrivateCircuitPublicInputs.from({
|
|
60
|
+
...PrivateCircuitPublicInputs.empty(),
|
|
61
|
+
anchorBlockHeader: blockHeader,
|
|
62
|
+
txContext: txContext,
|
|
63
|
+
publicCallRequests: publicCallRequestsArray,
|
|
64
|
+
startSideEffectCounter: new Fr(0),
|
|
65
|
+
endSideEffectCounter: new Fr(publicStaticCalls.length + 1)
|
|
66
|
+
});
|
|
67
|
+
// Minimal entrypoint structure — no real private execution, just public call requests
|
|
68
|
+
const emptyEntrypoint = new PrivateCallExecutionResult(Buffer.alloc(0), Buffer.alloc(0), new Map(), publicInputs, [], new Map(), [], [], [], [], []);
|
|
69
|
+
const privateResult = new PrivateExecutionResult(emptyEntrypoint, Fr.random(), publicFunctionCalldata);
|
|
70
|
+
const provingResult = await generateSimulatedProvingResult(privateResult, (_contractAddress, _functionSelector)=>Promise.resolve(''), node, 1);
|
|
71
|
+
provingResult.publicInputs.feePayer = from;
|
|
72
|
+
const tx = await Tx.create({
|
|
73
|
+
data: provingResult.publicInputs,
|
|
74
|
+
chonkProof: ChonkProof.empty(),
|
|
75
|
+
contractClassLogFields: [],
|
|
76
|
+
publicFunctionCalldata: publicFunctionCalldata
|
|
77
|
+
});
|
|
78
|
+
const publicOutput = await node.simulatePublicCalls(tx, skipFeeEnforcement);
|
|
79
|
+
if (publicOutput.revertReason) {
|
|
80
|
+
throw publicOutput.revertReason;
|
|
81
|
+
}
|
|
82
|
+
// Display debug logs from the public simulation.
|
|
83
|
+
await displayDebugLogs(publicOutput.debugLogs, getContractName);
|
|
84
|
+
return new TxSimulationResult(privateResult, provingResult.publicInputs, publicOutput, undefined);
|
|
85
|
+
}
|
|
86
|
+
/**
|
|
87
|
+
* Simulates public static calls by splitting them into batches of MAX_ENQUEUED_CALLS_PER_CALL
|
|
88
|
+
* and sending each batch directly to the node.
|
|
89
|
+
*
|
|
90
|
+
* @param node - The Aztec node to simulate on.
|
|
91
|
+
* @param publicStaticCalls - Array of public static function calls to optimize.
|
|
92
|
+
* @param from - The account address making the calls.
|
|
93
|
+
* @param chainInfo - Chain information (chainId and version).
|
|
94
|
+
* @param gasSettings - Gas settings for the transaction.
|
|
95
|
+
* @param blockHeader - Block header to use as anchor.
|
|
96
|
+
* @param skipFeeEnforcement - Whether to skip fee enforcement during simulation.
|
|
97
|
+
* @returns Array of TxSimulationResult, one per batch.
|
|
98
|
+
*/ export async function simulateViaNode(node, publicStaticCalls, from, chainInfo, gasSettings, blockHeader, skipFeeEnforcement = true, getContractName) {
|
|
99
|
+
const batches = [];
|
|
100
|
+
for(let i = 0; i < publicStaticCalls.length; i += MAX_ENQUEUED_CALLS_PER_CALL){
|
|
101
|
+
batches.push(publicStaticCalls.slice(i, i + MAX_ENQUEUED_CALLS_PER_CALL));
|
|
102
|
+
}
|
|
103
|
+
const results = [];
|
|
104
|
+
for (const batch of batches){
|
|
105
|
+
const result = await simulateBatchViaNode(node, batch, from, chainInfo, gasSettings, blockHeader, skipFeeEnforcement, getContractName);
|
|
106
|
+
results.push(result);
|
|
107
|
+
}
|
|
108
|
+
return results;
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* Merges simulation results from the optimized (public static) and normal paths.
|
|
112
|
+
* Since optimized calls are always a leading prefix, return values are simply
|
|
113
|
+
* concatenated: optimized first, then normal.
|
|
114
|
+
* Stats are taken from the normal result only (the optimized path doesn't produce them).
|
|
115
|
+
*
|
|
116
|
+
* @param optimizedResults - Results from optimized public static call batches.
|
|
117
|
+
* @param normalResult - Result from normal simulation (null if all calls were optimized).
|
|
118
|
+
* @returns A single TxSimulationResult with return values in original call order.
|
|
119
|
+
*/ export function buildMergedSimulationResult(optimizedResults, normalResult) {
|
|
120
|
+
const optimizedReturnValues = optimizedResults.flatMap((r)=>r.publicOutput?.publicReturnValues ?? []);
|
|
121
|
+
const normalReturnValues = normalResult?.publicOutput?.publicReturnValues ?? [];
|
|
122
|
+
const allReturnValues = [
|
|
123
|
+
...optimizedReturnValues,
|
|
124
|
+
...normalReturnValues
|
|
125
|
+
];
|
|
126
|
+
const baseResult = normalResult ?? optimizedResults[0];
|
|
127
|
+
const mergedPublicOutput = baseResult.publicOutput ? {
|
|
128
|
+
...baseResult.publicOutput,
|
|
129
|
+
publicReturnValues: allReturnValues
|
|
130
|
+
} : undefined;
|
|
131
|
+
const merged = new TxSimulationResult(baseResult.privateExecutionResult, baseResult.publicInputs, mergedPublicOutput, normalResult?.stats);
|
|
132
|
+
return TxSimulationResultWithAppOffset.fromResultAndOffset(merged, normalResult?.appCallOffset ?? 0);
|
|
133
|
+
}
|