@aztec/wallet-sdk 0.0.1-commit.6d3c34e → 0.0.1-commit.7cf39cb55

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 (77) hide show
  1. package/README.md +218 -355
  2. package/dest/base-wallet/base_wallet.d.ts +54 -12
  3. package/dest/base-wallet/base_wallet.d.ts.map +1 -1
  4. package/dest/base-wallet/base_wallet.js +120 -25
  5. package/dest/base-wallet/index.d.ts +2 -1
  6. package/dest/base-wallet/index.d.ts.map +1 -1
  7. package/dest/base-wallet/index.js +1 -0
  8. package/dest/base-wallet/utils.d.ts +48 -0
  9. package/dest/base-wallet/utils.d.ts.map +1 -0
  10. package/dest/base-wallet/utils.js +128 -0
  11. package/dest/crypto.d.ts +73 -27
  12. package/dest/crypto.d.ts.map +1 -1
  13. package/dest/crypto.js +219 -41
  14. package/dest/emoji_alphabet.d.ts +35 -0
  15. package/dest/emoji_alphabet.d.ts.map +1 -0
  16. package/dest/emoji_alphabet.js +299 -0
  17. package/dest/extension/handlers/background_connection_handler.d.ts +158 -0
  18. package/dest/extension/handlers/background_connection_handler.d.ts.map +1 -0
  19. package/dest/extension/handlers/background_connection_handler.js +258 -0
  20. package/dest/extension/handlers/content_script_connection_handler.d.ts +56 -0
  21. package/dest/extension/handlers/content_script_connection_handler.d.ts.map +1 -0
  22. package/dest/extension/handlers/content_script_connection_handler.js +174 -0
  23. package/dest/extension/handlers/index.d.ts +12 -0
  24. package/dest/extension/handlers/index.d.ts.map +1 -0
  25. package/dest/extension/handlers/index.js +10 -0
  26. package/dest/extension/handlers/internal_message_types.d.ts +63 -0
  27. package/dest/extension/handlers/internal_message_types.d.ts.map +1 -0
  28. package/dest/extension/handlers/internal_message_types.js +22 -0
  29. package/dest/extension/provider/extension_provider.d.ts +107 -0
  30. package/dest/extension/provider/extension_provider.d.ts.map +1 -0
  31. package/dest/extension/provider/extension_provider.js +160 -0
  32. package/dest/extension/provider/extension_wallet.d.ts +131 -0
  33. package/dest/extension/provider/extension_wallet.d.ts.map +1 -0
  34. package/dest/extension/provider/extension_wallet.js +271 -0
  35. package/dest/extension/provider/index.d.ts +3 -0
  36. package/dest/extension/provider/index.d.ts.map +1 -0
  37. package/dest/{providers/extension → extension/provider}/index.js +0 -1
  38. package/dest/manager/index.d.ts +2 -7
  39. package/dest/manager/index.d.ts.map +1 -1
  40. package/dest/manager/index.js +0 -4
  41. package/dest/manager/types.d.ts +108 -5
  42. package/dest/manager/types.d.ts.map +1 -1
  43. package/dest/manager/types.js +17 -1
  44. package/dest/manager/wallet_manager.d.ts +50 -7
  45. package/dest/manager/wallet_manager.d.ts.map +1 -1
  46. package/dest/manager/wallet_manager.js +178 -29
  47. package/dest/types.d.ts +55 -15
  48. package/dest/types.d.ts.map +1 -1
  49. package/dest/types.js +10 -2
  50. package/package.json +18 -11
  51. package/src/base-wallet/base_wallet.ts +164 -45
  52. package/src/base-wallet/index.ts +1 -0
  53. package/src/base-wallet/utils.ts +230 -0
  54. package/src/crypto.ts +263 -47
  55. package/src/emoji_alphabet.ts +317 -0
  56. package/src/extension/handlers/background_connection_handler.ts +423 -0
  57. package/src/extension/handlers/content_script_connection_handler.ts +246 -0
  58. package/src/extension/handlers/index.ts +25 -0
  59. package/src/extension/handlers/internal_message_types.ts +69 -0
  60. package/src/extension/provider/extension_provider.ts +233 -0
  61. package/src/extension/provider/extension_wallet.ts +321 -0
  62. package/src/extension/provider/index.ts +7 -0
  63. package/src/manager/index.ts +3 -9
  64. package/src/manager/types.ts +112 -4
  65. package/src/manager/wallet_manager.ts +204 -31
  66. package/src/types.ts +57 -14
  67. package/dest/providers/extension/extension_provider.d.ts +0 -17
  68. package/dest/providers/extension/extension_provider.d.ts.map +0 -1
  69. package/dest/providers/extension/extension_provider.js +0 -56
  70. package/dest/providers/extension/extension_wallet.d.ts +0 -95
  71. package/dest/providers/extension/extension_wallet.d.ts.map +0 -1
  72. package/dest/providers/extension/extension_wallet.js +0 -225
  73. package/dest/providers/extension/index.d.ts +0 -5
  74. package/dest/providers/extension/index.d.ts.map +0 -1
  75. package/src/providers/extension/extension_provider.ts +0 -72
  76. package/src/providers/extension/extension_wallet.ts +0 -275
  77. package/src/providers/extension/index.ts +0 -11
@@ -1 +1 @@
1
- {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAEzD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAErD;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,uCAAuC;IACvC,EAAE,EAAE,MAAM,CAAC;IACX,iCAAiC;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,+BAA+B;IAC/B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,qBAAqB;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,gEAAgE;IAChE,SAAS,EAAE,iBAAiB,CAAC;CAC9B;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,+CAA+C;IAC/C,SAAS,EAAE,MAAM,CAAC;IAClB,gCAAgC;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,+BAA+B;IAC/B,IAAI,EAAE,OAAO,EAAE,CAAC;IAChB,wBAAwB;IACxB,SAAS,EAAE,SAAS,CAAC;IACrB,wCAAwC;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,4CAA4C;IAC5C,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,sCAAsC;IACtC,SAAS,EAAE,MAAM,CAAC;IAClB,kCAAkC;IAClC,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,6BAA6B;IAC7B,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,uCAAuC;IACvC,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,iCAAiC;IACjC,IAAI,EAAE,wBAAwB,CAAC;IAC/B,iBAAiB;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,iEAAiE;IACjE,SAAS,EAAE,SAAS,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,0CAA0C;IAC1C,IAAI,EAAE,iCAAiC,CAAC;IACxC,gDAAgD;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,yBAAyB;IACzB,UAAU,EAAE,UAAU,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,kCAAkC;IAClC,IAAI,EAAE,sBAAsB,CAAC;IAC7B,uBAAuB;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,qBAAqB;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,wDAAwD;IACxD,SAAS,EAAE,iBAAiB,CAAC;CAC9B"}
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAEzD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAErD;;;GAGG;AACH,oBAAY,iBAAiB;IAC3B,kDAAkD;IAClD,SAAS,2BAA2B;IACpC,uCAAuC;IACvC,kBAAkB,oCAAoC;IACtD,sEAAsE;IACtE,UAAU,4BAA4B;IACtC,oDAAoD;IACpD,oBAAoB,sCAAsC;IAC1D,qDAAqD;IACrD,qBAAqB,uCAAuC;CAC7D;AAED;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB,uCAAuC;IACvC,EAAE,EAAE,MAAM,CAAC;IACX,iCAAiC;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,+BAA+B;IAC/B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,qBAAqB;IACrB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,mBAAoB,SAAQ,UAAU;IACrD,gEAAgE;IAChE,SAAS,EAAE,iBAAiB,CAAC;IAC7B;;;;OAIG;IACH,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,+CAA+C;IAC/C,SAAS,EAAE,MAAM,CAAC;IAClB,gCAAgC;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,+BAA+B;IAC/B,IAAI,EAAE,OAAO,EAAE,CAAC;IAChB,wBAAwB;IACxB,SAAS,EAAE,SAAS,CAAC;IACrB,wCAAwC;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,4CAA4C;IAC5C,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,sCAAsC;IACtC,SAAS,EAAE,MAAM,CAAC;IAClB,kCAAkC;IAClC,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,6BAA6B;IAC7B,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,uCAAuC;IACvC,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,iCAAiC;IACjC,IAAI,EAAE,iBAAiB,CAAC,SAAS,CAAC;IAClC,iBAAiB;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,wCAAwC;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,iEAAiE;IACjE,SAAS,EAAE,SAAS,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,0CAA0C;IAC1C,IAAI,EAAE,iBAAiB,CAAC,kBAAkB,CAAC;IAC3C,gDAAgD;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,+BAA+B;IAC/B,UAAU,EAAE,UAAU,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,mBAAmB;IACnB,IAAI,EAAE,iBAAiB,CAAC,oBAAoB,CAAC;IAC7C,gDAAgD;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,wDAAwD;IACxD,SAAS,EAAE,iBAAiB,CAAC;CAC9B;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,mBAAmB;IACnB,IAAI,EAAE,iBAAiB,CAAC,qBAAqB,CAAC;IAC9C,gDAAgD;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,0DAA0D;IAC1D,SAAS,EAAE,iBAAiB,CAAC;CAC9B"}
package/dest/types.js CHANGED
@@ -1,3 +1,11 @@
1
1
  /**
2
- * Connection request to establish secure channel
3
- */ export { };
2
+ * Message types for wallet SDK communication.
3
+ * All types are prefixed with 'aztec-wallet-' for namespacing.
4
+ */ export var WalletMessageType = /*#__PURE__*/ function(WalletMessageType) {
5
+ /** Discovery request to find installed wallets */ WalletMessageType["DISCOVERY"] = "aztec-wallet-discovery";
6
+ /** Discovery response from a wallet */ WalletMessageType["DISCOVERY_RESPONSE"] = "aztec-wallet-discovery-response";
7
+ /** Disconnect message (unencrypted control message, bidirectional) */ WalletMessageType["DISCONNECT"] = "aztec-wallet-disconnect";
8
+ /** Key exchange request sent over MessageChannel */ WalletMessageType["KEY_EXCHANGE_REQUEST"] = "aztec-wallet-key-exchange-request";
9
+ /** Key exchange response sent over MessageChannel */ WalletMessageType["KEY_EXCHANGE_RESPONSE"] = "aztec-wallet-key-exchange-response";
10
+ return WalletMessageType;
11
+ }({});
package/package.json CHANGED
@@ -1,19 +1,26 @@
1
1
  {
2
2
  "name": "@aztec/wallet-sdk",
3
3
  "homepage": "https://github.com/AztecProtocol/aztec-packages/tree/master/yarn-project/wallet-sdk",
4
- "version": "0.0.1-commit.6d3c34e",
4
+ "version": "0.0.1-commit.7cf39cb55",
5
5
  "type": "module",
6
6
  "exports": {
7
7
  "./base-wallet": "./dest/base-wallet/index.js",
8
- "./providers/extension": "./dest/providers/extension/index.js",
8
+ "./extension/handlers": "./dest/extension/handlers/index.js",
9
+ "./extension/provider": "./dest/extension/provider/index.js",
9
10
  "./crypto": "./dest/crypto.js",
10
11
  "./types": "./dest/types.js",
11
12
  "./manager": "./dest/manager/index.js"
12
13
  },
13
14
  "typedocOptions": {
14
15
  "entryPoints": [
15
- "./src/index.ts"
16
+ "./src/base-wallet/index.ts",
17
+ "./src/extension/handlers/index.ts",
18
+ "./src/extension/provider/index.ts",
19
+ "./src/crypto.ts",
20
+ "./src/types.ts",
21
+ "./src/manager/index.ts"
16
22
  ],
23
+ "name": "Wallet SDK",
17
24
  "tsconfig": "./tsconfig.json"
18
25
  },
19
26
  "scripts": {
@@ -64,19 +71,19 @@
64
71
  ]
65
72
  },
66
73
  "dependencies": {
67
- "@aztec/aztec.js": "0.0.1-commit.6d3c34e",
68
- "@aztec/constants": "0.0.1-commit.6d3c34e",
69
- "@aztec/entrypoints": "0.0.1-commit.6d3c34e",
70
- "@aztec/foundation": "0.0.1-commit.6d3c34e",
71
- "@aztec/pxe": "0.0.1-commit.6d3c34e",
72
- "@aztec/stdlib": "0.0.1-commit.6d3c34e"
74
+ "@aztec/aztec.js": "0.0.1-commit.7cf39cb55",
75
+ "@aztec/constants": "0.0.1-commit.7cf39cb55",
76
+ "@aztec/entrypoints": "0.0.1-commit.7cf39cb55",
77
+ "@aztec/foundation": "0.0.1-commit.7cf39cb55",
78
+ "@aztec/pxe": "0.0.1-commit.7cf39cb55",
79
+ "@aztec/stdlib": "0.0.1-commit.7cf39cb55"
73
80
  },
74
81
  "devDependencies": {
75
- "@aztec/noir-contracts.js": "0.0.1-commit.6d3c34e",
82
+ "@aztec/noir-contracts.js": "0.0.1-commit.7cf39cb55",
76
83
  "@jest/globals": "^30.0.0",
77
84
  "@types/jest": "^30.0.0",
78
85
  "@types/node": "^22.15.17",
79
- "@typescript/native-preview": "7.0.0-dev.20251126.1",
86
+ "@typescript/native-preview": "7.0.0-dev.20260113.1",
80
87
  "jest": "^30.0.0",
81
88
  "jest-mock-extended": "^4.0.0",
82
89
  "resolve-typescript-plugin": "^2.0.1",
@@ -1,17 +1,21 @@
1
1
  import type { Account } from '@aztec/aztec.js/account';
2
2
  import type { CallIntent, IntentInnerHash } from '@aztec/aztec.js/authorization';
3
+ import { type InteractionWaitOptions, NO_WAIT, type SendReturn } from '@aztec/aztec.js/contracts';
3
4
  import type { FeePaymentMethod } from '@aztec/aztec.js/fee';
5
+ import { waitForTx } from '@aztec/aztec.js/node';
4
6
  import type {
5
7
  Aliased,
8
+ AppCapabilities,
6
9
  BatchResults,
7
- BatchableMethods,
8
10
  BatchedMethod,
9
11
  PrivateEvent,
10
12
  PrivateEventFilter,
11
13
  ProfileOptions,
12
14
  SendOptions,
13
15
  SimulateOptions,
16
+ SimulateUtilityOptions,
14
17
  Wallet,
18
+ WalletCapabilities,
15
19
  } from '@aztec/aztec.js/wallet';
16
20
  import {
17
21
  GAS_ESTIMATION_DA_GAS_LIMIT,
@@ -34,27 +38,27 @@ import {
34
38
  import type { AuthWitness } from '@aztec/stdlib/auth-witness';
35
39
  import type { AztecAddress } from '@aztec/stdlib/aztec-address';
36
40
  import {
37
- type ContractClassMetadata,
38
41
  type ContractInstanceWithAddress,
39
- type ContractMetadata,
40
42
  computePartialAddress,
41
43
  getContractClassFromArtifact,
42
44
  } from '@aztec/stdlib/contract';
43
45
  import { SimulationError } from '@aztec/stdlib/errors';
44
46
  import { Gas, GasSettings } from '@aztec/stdlib/gas';
47
+ import { siloNullifier } from '@aztec/stdlib/hash';
45
48
  import type { AztecNode } from '@aztec/stdlib/interfaces/client';
46
- import type {
47
- TxExecutionRequest,
48
- TxHash,
49
- TxProfileResult,
50
- TxReceipt,
49
+ import {
50
+ BlockHeader,
51
+ type TxExecutionRequest,
52
+ type TxProfileResult,
51
53
  TxSimulationResult,
52
- UtilitySimulationResult,
54
+ type UtilitySimulationResult,
53
55
  } from '@aztec/stdlib/tx';
54
56
  import { ExecutionPayload, mergeExecutionPayloads } from '@aztec/stdlib/tx';
55
57
 
56
58
  import { inspect } from 'util';
57
59
 
60
+ import { buildMergedSimulationResult, extractOptimizablePublicStaticCalls, simulateViaNode } from './utils.js';
61
+
58
62
  /**
59
63
  * Options to configure fee payment for a transaction
60
64
  */
@@ -74,8 +78,6 @@ export type FeeOptions = {
74
78
  * A base class for Wallet implementations
75
79
  */
76
80
  export abstract class BaseWallet implements Wallet {
77
- protected log = createLogger('wallet-sdk:base_wallet');
78
-
79
81
  protected minFeePadding = 0.5;
80
82
  protected cancellableTransactions = false;
81
83
 
@@ -83,8 +85,15 @@ export abstract class BaseWallet implements Wallet {
83
85
  protected constructor(
84
86
  protected readonly pxe: PXE,
85
87
  protected readonly aztecNode: AztecNode,
88
+ protected log = createLogger('wallet-sdk:base_wallet'),
86
89
  ) {}
87
90
 
91
+ // When `from` is the zero address (e.g. when deploying a new account contract), we return an
92
+ // empty scope list which acts as deny-all: no notes are visible and no keys are accessible.
93
+ protected scopesFor(from: AztecAddress): AztecAddress[] {
94
+ return from.isZero() ? [] : [from];
95
+ }
96
+
88
97
  protected abstract getAccountFromAddress(address: AztecAddress): Promise<Account>;
89
98
 
90
99
  abstract getAccounts(): Promise<Aliased<AztecAddress>[]>;
@@ -121,20 +130,42 @@ export abstract class BaseWallet implements Wallet {
121
130
  ? mergeExecutionPayloads([feeExecutionPayload, executionPayload])
122
131
  : executionPayload;
123
132
  const fromAccount = await this.getAccountFromAddress(from);
124
- return fromAccount.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, executionOptions);
133
+ const chainInfo = await this.getChainInfo();
134
+ return fromAccount.createTxExecutionRequest(
135
+ finalExecutionPayload,
136
+ feeOptions.gasSettings,
137
+ chainInfo,
138
+ executionOptions,
139
+ );
125
140
  }
126
141
 
127
142
  public async createAuthWit(
128
143
  from: AztecAddress,
129
- messageHashOrIntent: Fr | IntentInnerHash | CallIntent,
144
+ messageHashOrIntent: IntentInnerHash | CallIntent,
130
145
  ): Promise<AuthWitness> {
131
146
  const account = await this.getAccountFromAddress(from);
132
- return account.createAuthWit(messageHashOrIntent);
147
+ const chainInfo = await this.getChainInfo();
148
+ return account.createAuthWit(messageHashOrIntent, chainInfo);
149
+ }
150
+
151
+ /**
152
+ * Request capabilities from the wallet.
153
+ *
154
+ * This method is wallet-implementation-dependent and must be provided by classes extending BaseWallet.
155
+ * Embedded wallets typically don't support capability-based authorization (no user authorization flow),
156
+ * while external wallets (browser extensions, hardware wallets) implement this to reduce authorization
157
+ * friction by allowing apps to request permissions upfront.
158
+ *
159
+ * TODO: Consider making it abstract so implementing it is a conscious decision. Leaving it as-is
160
+ * while the feature stabilizes.
161
+ *
162
+ * @param _manifest - Application capability manifest declaring what operations the app needs
163
+ */
164
+ public requestCapabilities(_manifest: AppCapabilities): Promise<WalletCapabilities> {
165
+ throw new Error('Not implemented');
133
166
  }
134
167
 
135
- public async batch<const T extends readonly BatchedMethod<keyof BatchableMethods>[]>(
136
- methods: T,
137
- ): Promise<BatchResults<T>> {
168
+ public async batch<const T extends readonly BatchedMethod[]>(methods: T): Promise<BatchResults<T>> {
138
169
  const results: any[] = [];
139
170
  for (const method of methods) {
140
171
  const { name, args } = method;
@@ -227,7 +258,7 @@ export abstract class BaseWallet implements Wallet {
227
258
  artifact?: ContractArtifact,
228
259
  secretKey?: Fr,
229
260
  ): Promise<ContractInstanceWithAddress> {
230
- const { contractInstance: existingInstance } = await this.pxe.getContractMetadata(instance.address);
261
+ const existingInstance = await this.pxe.getContractInstance(instance.address);
231
262
 
232
263
  if (existingInstance) {
233
264
  // Instance already registered in the wallet
@@ -244,13 +275,12 @@ export abstract class BaseWallet implements Wallet {
244
275
  // Instance not registered yet
245
276
  if (!artifact) {
246
277
  // Try to get the artifact from the wallet's contract class storage
247
- const classMetadata = await this.pxe.getContractClassMetadata(instance.currentContractClassId, true);
248
- if (!classMetadata.artifact) {
278
+ artifact = await this.pxe.getContractArtifact(instance.currentContractClassId);
279
+ if (!artifact) {
249
280
  throw new Error(
250
281
  `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()}`,
251
282
  );
252
283
  }
253
- artifact = classMetadata.artifact;
254
284
  }
255
285
  await this.pxe.registerContract({ artifact, instance });
256
286
  }
@@ -261,29 +291,96 @@ export abstract class BaseWallet implements Wallet {
261
291
  return instance;
262
292
  }
263
293
 
294
+ /**
295
+ * Simulates calls through the standard PXE path (account entrypoint).
296
+ * @param executionPayload - The execution payload to simulate.
297
+ * @param from - The sender address.
298
+ * @param feeOptions - Fee options for the transaction.
299
+ * @param skipTxValidation - Whether to skip tx validation.
300
+ * @param skipFeeEnforcement - Whether to skip fee enforcement.
301
+ * @param scopes - The scopes to use for the simulation.
302
+ */
303
+ protected async simulateViaEntrypoint(
304
+ executionPayload: ExecutionPayload,
305
+ from: AztecAddress,
306
+ feeOptions: FeeOptions,
307
+ skipTxValidation?: boolean,
308
+ skipFeeEnforcement?: boolean,
309
+ scopes?: AztecAddress[],
310
+ ) {
311
+ const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, from, feeOptions);
312
+ return this.pxe.simulateTx(txRequest, { simulatePublic: true, skipTxValidation, skipFeeEnforcement, scopes });
313
+ }
314
+
315
+ /**
316
+ * Simulates a transaction, optimizing leading public static calls by running them directly
317
+ * on the node while sending the remaining calls through the standard PXE path.
318
+ * Return values from both paths are merged back in original call order.
319
+ * @param executionPayload - The execution payload to simulate.
320
+ * @param opts - Simulation options (from address, fee settings, etc.).
321
+ * @returns The merged simulation result.
322
+ */
264
323
  async simulateTx(executionPayload: ExecutionPayload, opts: SimulateOptions): Promise<TxSimulationResult> {
265
324
  const feeOptions = opts.fee?.estimateGas
266
325
  ? await this.completeFeeOptionsForEstimation(opts.from, executionPayload.feePayer, opts.fee?.gasSettings)
267
326
  : await this.completeFeeOptions(opts.from, executionPayload.feePayer, opts.fee?.gasSettings);
268
- const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, feeOptions);
269
- return this.pxe.simulateTx(
270
- txRequest,
271
- true /* simulatePublic */,
272
- opts?.skipTxValidation,
273
- opts?.skipFeeEnforcement ?? true,
274
- );
327
+ const { optimizableCalls, remainingCalls } = extractOptimizablePublicStaticCalls(executionPayload);
328
+ const remainingPayload = { ...executionPayload, calls: remainingCalls };
329
+
330
+ const chainInfo = await this.getChainInfo();
331
+ let blockHeader: BlockHeader;
332
+ // PXE might not be synced yet, so we pull the latest header from the node
333
+ // To keep things consistent, we'll always try with PXE first
334
+ try {
335
+ blockHeader = await this.pxe.getSyncedBlockHeader();
336
+ } catch {
337
+ blockHeader = (await this.aztecNode.getBlockHeader())!;
338
+ }
339
+
340
+ const [optimizedResults, normalResult] = await Promise.all([
341
+ optimizableCalls.length > 0
342
+ ? simulateViaNode(
343
+ this.aztecNode,
344
+ optimizableCalls,
345
+ opts.from,
346
+ chainInfo,
347
+ feeOptions.gasSettings,
348
+ blockHeader,
349
+ opts.skipFeeEnforcement ?? true,
350
+ )
351
+ : Promise.resolve([]),
352
+ remainingCalls.length > 0
353
+ ? this.simulateViaEntrypoint(
354
+ remainingPayload,
355
+ opts.from,
356
+ feeOptions,
357
+ opts.skipTxValidation,
358
+ opts.skipFeeEnforcement ?? true,
359
+ this.scopesFor(opts.from),
360
+ )
361
+ : Promise.resolve(null),
362
+ ]);
363
+
364
+ return buildMergedSimulationResult(optimizedResults, normalResult);
275
365
  }
276
366
 
277
367
  async profileTx(executionPayload: ExecutionPayload, opts: ProfileOptions): Promise<TxProfileResult> {
278
368
  const feeOptions = await this.completeFeeOptions(opts.from, executionPayload.feePayer, opts.fee?.gasSettings);
279
369
  const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, feeOptions);
280
- return this.pxe.profileTx(txRequest, opts.profileMode, opts.skipProofGeneration ?? true);
370
+ return this.pxe.profileTx(txRequest, {
371
+ profileMode: opts.profileMode,
372
+ skipProofGeneration: opts.skipProofGeneration ?? true,
373
+ scopes: this.scopesFor(opts.from),
374
+ });
281
375
  }
282
376
 
283
- async sendTx(executionPayload: ExecutionPayload, opts: SendOptions): Promise<TxHash> {
377
+ public async sendTx<W extends InteractionWaitOptions = undefined>(
378
+ executionPayload: ExecutionPayload,
379
+ opts: SendOptions<W>,
380
+ ): Promise<SendReturn<W>> {
284
381
  const feeOptions = await this.completeFeeOptions(opts.from, executionPayload.feePayer, opts.fee?.gasSettings);
285
382
  const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(executionPayload, opts.from, feeOptions);
286
- const provenTx = await this.pxe.proveTx(txRequest);
383
+ const provenTx = await this.pxe.proveTx(txRequest, this.scopesFor(opts.from));
287
384
  const tx = await provenTx.toTx();
288
385
  const txHash = tx.getTxHash();
289
386
  if (await this.aztecNode.getTxEffect(txHash)) {
@@ -294,7 +391,15 @@ export abstract class BaseWallet implements Wallet {
294
391
  throw this.contextualizeError(err, inspect(tx));
295
392
  });
296
393
  this.log.info(`Sent transaction ${txHash}`);
297
- return txHash;
394
+
395
+ // If wait is NO_WAIT, return txHash immediately
396
+ if (opts.wait === NO_WAIT) {
397
+ return txHash as SendReturn<W>;
398
+ }
399
+
400
+ // Otherwise, wait for the full receipt (default behavior on wait: undefined)
401
+ const waitOpts = typeof opts.wait === 'object' ? opts.wait : undefined;
402
+ return (await waitForTx(this.aztecNode, txHash, waitOpts)) as SendReturn<W>;
298
403
  }
299
404
 
300
405
  protected contextualizeError(err: Error, ...context: string[]): Error {
@@ -311,19 +416,8 @@ export abstract class BaseWallet implements Wallet {
311
416
  return err;
312
417
  }
313
418
 
314
- simulateUtility(call: FunctionCall, authwits?: AuthWitness[]): Promise<UtilitySimulationResult> {
315
- return this.pxe.simulateUtility(call, authwits);
316
- }
317
-
318
- getContractClassMetadata(id: Fr, includeArtifact: boolean = false): Promise<ContractClassMetadata> {
319
- return this.pxe.getContractClassMetadata(id, includeArtifact);
320
- }
321
- getContractMetadata(address: AztecAddress): Promise<ContractMetadata> {
322
- return this.pxe.getContractMetadata(address);
323
- }
324
-
325
- getTxReceipt(txHash: TxHash): Promise<TxReceipt> {
326
- return this.aztecNode.getTxReceipt(txHash);
419
+ simulateUtility(call: FunctionCall, opts: SimulateUtilityOptions): Promise<UtilitySimulationResult> {
420
+ return this.pxe.simulateUtility(call, { authwits: opts.authWitnesses, scopes: [opts.scope] });
327
421
  }
328
422
 
329
423
  async getPrivateEvents<T>(
@@ -345,4 +439,29 @@ export abstract class BaseWallet implements Wallet {
345
439
 
346
440
  return decodedEvents;
347
441
  }
442
+
443
+ async getContractMetadata(address: AztecAddress) {
444
+ const instance = await this.pxe.getContractInstance(address);
445
+ const initNullifier = await siloNullifier(address, address.toField());
446
+ const publiclyRegisteredContract = await this.aztecNode.getContract(address);
447
+ const initNullifierMembershipWitness = await this.aztecNode.getNullifierMembershipWitness('latest', initNullifier);
448
+ const isContractUpdated =
449
+ publiclyRegisteredContract &&
450
+ !publiclyRegisteredContract.currentContractClassId.equals(publiclyRegisteredContract.originalContractClassId);
451
+ return {
452
+ instance: instance ?? undefined,
453
+ isContractInitialized: !!initNullifierMembershipWitness,
454
+ isContractPublished: !!publiclyRegisteredContract,
455
+ isContractUpdated: !!isContractUpdated,
456
+ updatedContractClassId: isContractUpdated ? publiclyRegisteredContract.currentContractClassId : undefined,
457
+ };
458
+ }
459
+
460
+ async getContractClassMetadata(id: Fr) {
461
+ const publiclyRegisteredContractClass = await this.aztecNode.getContractClass(id);
462
+ return {
463
+ isArtifactRegistered: !!(await this.pxe.getContractArtifact(id)),
464
+ isContractClassPubliclyRegistered: !!publiclyRegisteredContractClass,
465
+ };
466
+ }
348
467
  }
@@ -1 +1,2 @@
1
1
  export { BaseWallet, type FeeOptions } from './base_wallet.js';
2
+ export { simulateViaNode, buildMergedSimulationResult, extractOptimizablePublicStaticCalls } from './utils.js';
@@ -0,0 +1,230 @@
1
+ import type { AztecNode } from '@aztec/aztec.js/node';
2
+ import { MAX_ENQUEUED_CALLS_PER_CALL } from '@aztec/constants';
3
+ import type { ChainInfo } from '@aztec/entrypoints/interfaces';
4
+ import { makeTuple } from '@aztec/foundation/array';
5
+ import { Fr } from '@aztec/foundation/curves/bn254';
6
+ import type { Tuple } from '@aztec/foundation/serialize';
7
+ import { generateSimulatedProvingResult } from '@aztec/pxe/simulator';
8
+ import { type FunctionCall, FunctionSelector } from '@aztec/stdlib/abi';
9
+ import type { AztecAddress } from '@aztec/stdlib/aztec-address';
10
+ import type { GasSettings } from '@aztec/stdlib/gas';
11
+ import {
12
+ ClaimedLengthArray,
13
+ CountedPublicCallRequest,
14
+ PrivateCircuitPublicInputs,
15
+ PublicCallRequest,
16
+ } from '@aztec/stdlib/kernel';
17
+ import { ChonkProof } from '@aztec/stdlib/proofs';
18
+ import {
19
+ type BlockHeader,
20
+ type ExecutionPayload,
21
+ HashedValues,
22
+ PrivateCallExecutionResult,
23
+ PrivateExecutionResult,
24
+ PublicSimulationOutput,
25
+ Tx,
26
+ TxContext,
27
+ TxSimulationResult,
28
+ } from '@aztec/stdlib/tx';
29
+
30
+ /**
31
+ * Splits an execution payload into a leading prefix of public static calls
32
+ * (eligible for direct node simulation) and the remaining calls.
33
+ *
34
+ * Only a leading run of public static calls is eligible for optimization.
35
+ * Any non-public-static call may enqueue public state mutations
36
+ * (e.g. private calls can enqueue public calls), so all calls that follow
37
+ * must go through the normal simulation path to see the correct state.
38
+ *
39
+ */
40
+ export function extractOptimizablePublicStaticCalls(payload: ExecutionPayload): {
41
+ /** Leading public static calls eligible for direct node simulation. */
42
+ optimizableCalls: FunctionCall[];
43
+ /** All remaining calls. */
44
+ remainingCalls: FunctionCall[];
45
+ } {
46
+ const splitIndex = payload.calls.findIndex(call => !call.isPublicStatic());
47
+ const boundary = splitIndex === -1 ? payload.calls.length : splitIndex;
48
+ return {
49
+ optimizableCalls: payload.calls.slice(0, boundary),
50
+ remainingCalls: payload.calls.slice(boundary),
51
+ };
52
+ }
53
+
54
+ /**
55
+ * Simulates a batch of public static calls by bypassing account entrypoint and private execution,
56
+ * directly constructing a minimal Tx and calling node.simulatePublicCalls.
57
+ *
58
+ * @param node - The Aztec node to simulate on.
59
+ * @param publicStaticCalls - Array of public static function calls (max MAX_ENQUEUED_CALLS_PER_CALL).
60
+ * @param from - The account address making the calls.
61
+ * @param chainInfo - Chain information (chainId and version).
62
+ * @param gasSettings - Gas settings for the transaction.
63
+ * @param blockHeader - Block header to use as anchor.
64
+ * @param skipFeeEnforcement - Whether to skip fee enforcement during simulation.
65
+ * @returns TxSimulationResult with public return values.
66
+ */
67
+ async function simulateBatchViaNode(
68
+ node: AztecNode,
69
+ publicStaticCalls: FunctionCall[],
70
+ from: AztecAddress,
71
+ chainInfo: ChainInfo,
72
+ gasSettings: GasSettings,
73
+ blockHeader: BlockHeader,
74
+ skipFeeEnforcement: boolean,
75
+ ): Promise<TxSimulationResult> {
76
+ const txContext = new TxContext(chainInfo.chainId, chainInfo.version, gasSettings);
77
+
78
+ const publicFunctionCalldata: HashedValues[] = [];
79
+ for (const call of publicStaticCalls) {
80
+ const calldata = await HashedValues.fromCalldata([call.selector.toField(), ...call.args]);
81
+ publicFunctionCalldata.push(calldata);
82
+ }
83
+
84
+ const publicCallRequests = makeTuple(MAX_ENQUEUED_CALLS_PER_CALL, i => {
85
+ const call = publicStaticCalls[i];
86
+ if (!call) {
87
+ return CountedPublicCallRequest.empty();
88
+ }
89
+ const publicCallRequest = new PublicCallRequest(from, call.to, call.isStatic, publicFunctionCalldata[i]!.hash);
90
+ // Counter starts at 1 (minRevertibleSideEffectCounter) so all calls are revertible
91
+ return new CountedPublicCallRequest(publicCallRequest, i + 1);
92
+ });
93
+
94
+ const publicCallRequestsArray: ClaimedLengthArray<CountedPublicCallRequest, typeof MAX_ENQUEUED_CALLS_PER_CALL> =
95
+ new ClaimedLengthArray(
96
+ publicCallRequests as Tuple<CountedPublicCallRequest, typeof MAX_ENQUEUED_CALLS_PER_CALL>,
97
+ publicStaticCalls.length,
98
+ );
99
+
100
+ const publicInputs = PrivateCircuitPublicInputs.from({
101
+ ...PrivateCircuitPublicInputs.empty(),
102
+ anchorBlockHeader: blockHeader,
103
+ txContext: txContext,
104
+ publicCallRequests: publicCallRequestsArray,
105
+ startSideEffectCounter: new Fr(0),
106
+ endSideEffectCounter: new Fr(publicStaticCalls.length + 1),
107
+ });
108
+
109
+ // Minimal entrypoint structure — no real private execution, just public call requests
110
+ const emptyEntrypoint = new PrivateCallExecutionResult(
111
+ Buffer.alloc(0),
112
+ Buffer.alloc(0),
113
+ new Map(),
114
+ publicInputs,
115
+ [],
116
+ new Map(),
117
+ [],
118
+ [],
119
+ [],
120
+ [],
121
+ [],
122
+ );
123
+
124
+ const privateResult = new PrivateExecutionResult(emptyEntrypoint, Fr.random(), publicFunctionCalldata);
125
+
126
+ const provingResult = await generateSimulatedProvingResult(
127
+ privateResult,
128
+ (_contractAddress: AztecAddress, _functionSelector: FunctionSelector) => Promise.resolve(''),
129
+ node,
130
+ 1, // minRevertibleSideEffectCounter
131
+ );
132
+
133
+ provingResult.publicInputs.feePayer = from;
134
+
135
+ const tx = await Tx.create({
136
+ data: provingResult.publicInputs,
137
+ chonkProof: ChonkProof.empty(),
138
+ contractClassLogFields: [],
139
+ publicFunctionCalldata: publicFunctionCalldata,
140
+ });
141
+
142
+ const publicOutput = await node.simulatePublicCalls(tx, skipFeeEnforcement);
143
+
144
+ if (publicOutput.revertReason) {
145
+ throw publicOutput.revertReason;
146
+ }
147
+
148
+ return new TxSimulationResult(privateResult, provingResult.publicInputs, publicOutput, undefined);
149
+ }
150
+
151
+ /**
152
+ * Simulates public static calls by splitting them into batches of MAX_ENQUEUED_CALLS_PER_CALL
153
+ * and sending each batch directly to the node.
154
+ *
155
+ * @param node - The Aztec node to simulate on.
156
+ * @param publicStaticCalls - Array of public static function calls to optimize.
157
+ * @param from - The account address making the calls.
158
+ * @param chainInfo - Chain information (chainId and version).
159
+ * @param gasSettings - Gas settings for the transaction.
160
+ * @param blockHeader - Block header to use as anchor.
161
+ * @param skipFeeEnforcement - Whether to skip fee enforcement during simulation.
162
+ * @returns Array of TxSimulationResult, one per batch.
163
+ */
164
+ export async function simulateViaNode(
165
+ node: AztecNode,
166
+ publicStaticCalls: FunctionCall[],
167
+ from: AztecAddress,
168
+ chainInfo: ChainInfo,
169
+ gasSettings: GasSettings,
170
+ blockHeader: BlockHeader,
171
+ skipFeeEnforcement: boolean = true,
172
+ ): Promise<TxSimulationResult[]> {
173
+ const batches: FunctionCall[][] = [];
174
+
175
+ for (let i = 0; i < publicStaticCalls.length; i += MAX_ENQUEUED_CALLS_PER_CALL) {
176
+ batches.push(publicStaticCalls.slice(i, i + MAX_ENQUEUED_CALLS_PER_CALL));
177
+ }
178
+
179
+ const results: TxSimulationResult[] = [];
180
+
181
+ for (const batch of batches) {
182
+ const result = await simulateBatchViaNode(
183
+ node,
184
+ batch,
185
+ from,
186
+ chainInfo,
187
+ gasSettings,
188
+ blockHeader,
189
+ skipFeeEnforcement,
190
+ );
191
+ results.push(result);
192
+ }
193
+
194
+ return results;
195
+ }
196
+
197
+ /**
198
+ * Merges simulation results from the optimized (public static) and normal paths.
199
+ * Since optimized calls are always a leading prefix, return values are simply
200
+ * concatenated: optimized first, then normal.
201
+ * Stats are taken from the normal result only (the optimized path doesn't produce them).
202
+ *
203
+ * @param optimizedResults - Results from optimized public static call batches.
204
+ * @param normalResult - Result from normal simulation (null if all calls were optimized).
205
+ * @returns A single TxSimulationResult with return values in original call order.
206
+ */
207
+ export function buildMergedSimulationResult(
208
+ optimizedResults: TxSimulationResult[],
209
+ normalResult: TxSimulationResult | null,
210
+ ): TxSimulationResult {
211
+ const optimizedReturnValues = optimizedResults.flatMap(r => r.publicOutput?.publicReturnValues ?? []);
212
+ const normalReturnValues = normalResult?.publicOutput?.publicReturnValues ?? [];
213
+ const allReturnValues = [...optimizedReturnValues, ...normalReturnValues];
214
+
215
+ const baseResult = normalResult ?? optimizedResults[0];
216
+
217
+ const mergedPublicOutput: PublicSimulationOutput | undefined = baseResult.publicOutput
218
+ ? {
219
+ ...baseResult.publicOutput,
220
+ publicReturnValues: allReturnValues,
221
+ }
222
+ : undefined;
223
+
224
+ return new TxSimulationResult(
225
+ baseResult.privateExecutionResult,
226
+ baseResult.publicInputs,
227
+ mergedPublicOutput,
228
+ normalResult?.stats,
229
+ );
230
+ }