@aztec/ethereum 0.0.1-commit.d6f2b3f94 → 0.0.1-commit.db765a8

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 (93) hide show
  1. package/dest/config.d.ts +3 -1
  2. package/dest/config.d.ts.map +1 -1
  3. package/dest/config.js +10 -2
  4. package/dest/contracts/empire_base.d.ts +3 -1
  5. package/dest/contracts/empire_base.d.ts.map +1 -1
  6. package/dest/contracts/empire_slashing_proposer.d.ts +3 -1
  7. package/dest/contracts/empire_slashing_proposer.d.ts.map +1 -1
  8. package/dest/contracts/empire_slashing_proposer.js +9 -0
  9. package/dest/contracts/fee_asset_price_oracle.d.ts +101 -0
  10. package/dest/contracts/fee_asset_price_oracle.d.ts.map +1 -0
  11. package/dest/contracts/fee_asset_price_oracle.js +651 -0
  12. package/dest/contracts/governance.js +3 -3
  13. package/dest/contracts/governance_proposer.d.ts +3 -1
  14. package/dest/contracts/governance_proposer.d.ts.map +1 -1
  15. package/dest/contracts/governance_proposer.js +9 -0
  16. package/dest/contracts/index.d.ts +2 -1
  17. package/dest/contracts/index.d.ts.map +1 -1
  18. package/dest/contracts/index.js +1 -0
  19. package/dest/contracts/registry.d.ts +3 -1
  20. package/dest/contracts/registry.d.ts.map +1 -1
  21. package/dest/contracts/registry.js +30 -1
  22. package/dest/contracts/rollup.d.ts +5 -5
  23. package/dest/contracts/rollup.d.ts.map +1 -1
  24. package/dest/contracts/rollup.js +12 -2
  25. package/dest/deploy_aztec_l1_contracts.d.ts +2 -3
  26. package/dest/deploy_aztec_l1_contracts.d.ts.map +1 -1
  27. package/dest/deploy_aztec_l1_contracts.js +33 -19
  28. package/dest/deploy_l1_contract.js +3 -3
  29. package/dest/generated/l1-contracts-defaults.d.ts +1 -1
  30. package/dest/generated/l1-contracts-defaults.js +1 -1
  31. package/dest/l1_artifacts.d.ts +1051 -42
  32. package/dest/l1_artifacts.d.ts.map +1 -1
  33. package/dest/l1_tx_utils/config.d.ts +7 -1
  34. package/dest/l1_tx_utils/config.d.ts.map +1 -1
  35. package/dest/l1_tx_utils/config.js +14 -1
  36. package/dest/l1_tx_utils/factory.d.ts +18 -10
  37. package/dest/l1_tx_utils/factory.d.ts.map +1 -1
  38. package/dest/l1_tx_utils/factory.js +17 -7
  39. package/dest/l1_tx_utils/forwarder_l1_tx_utils.d.ts +15 -15
  40. package/dest/l1_tx_utils/forwarder_l1_tx_utils.d.ts.map +1 -1
  41. package/dest/l1_tx_utils/forwarder_l1_tx_utils.js +9 -15
  42. package/dest/l1_tx_utils/index-blobs.d.ts +3 -3
  43. package/dest/l1_tx_utils/index-blobs.d.ts.map +1 -1
  44. package/dest/l1_tx_utils/index-blobs.js +2 -2
  45. package/dest/l1_tx_utils/index.d.ts +2 -1
  46. package/dest/l1_tx_utils/index.d.ts.map +1 -1
  47. package/dest/l1_tx_utils/index.js +1 -0
  48. package/dest/l1_tx_utils/l1_tx_utils.d.ts +18 -7
  49. package/dest/l1_tx_utils/l1_tx_utils.d.ts.map +1 -1
  50. package/dest/l1_tx_utils/l1_tx_utils.js +52 -36
  51. package/dest/l1_tx_utils/tx_delayer.d.ts +56 -0
  52. package/dest/l1_tx_utils/tx_delayer.d.ts.map +1 -0
  53. package/dest/{test → l1_tx_utils}/tx_delayer.js +62 -34
  54. package/dest/test/index.d.ts +1 -3
  55. package/dest/test/index.d.ts.map +1 -1
  56. package/dest/test/index.js +0 -2
  57. package/dest/test/start_anvil.d.ts +9 -3
  58. package/dest/test/start_anvil.d.ts.map +1 -1
  59. package/dest/test/start_anvil.js +128 -29
  60. package/dest/test/upgrade_utils.js +2 -2
  61. package/package.json +5 -7
  62. package/src/config.ts +14 -1
  63. package/src/contracts/empire_base.ts +2 -0
  64. package/src/contracts/empire_slashing_proposer.ts +6 -0
  65. package/src/contracts/fee_asset_price_oracle.ts +280 -0
  66. package/src/contracts/governance.ts +3 -3
  67. package/src/contracts/governance_proposer.ts +6 -0
  68. package/src/contracts/index.ts +1 -0
  69. package/src/contracts/registry.ts +31 -1
  70. package/src/contracts/rollup.ts +20 -6
  71. package/src/deploy_aztec_l1_contracts.ts +56 -29
  72. package/src/deploy_l1_contract.ts +3 -3
  73. package/src/generated/l1-contracts-defaults.ts +1 -1
  74. package/src/l1_tx_utils/config.ts +20 -0
  75. package/src/l1_tx_utils/factory.ts +31 -31
  76. package/src/l1_tx_utils/forwarder_l1_tx_utils.ts +43 -54
  77. package/src/l1_tx_utils/index-blobs.ts +2 -2
  78. package/src/l1_tx_utils/index.ts +1 -0
  79. package/src/l1_tx_utils/l1_tx_utils.ts +54 -26
  80. package/src/{test → l1_tx_utils}/tx_delayer.ts +78 -50
  81. package/src/test/index.ts +0 -2
  82. package/src/test/start_anvil.ts +146 -29
  83. package/src/test/upgrade_utils.ts +2 -2
  84. package/dest/l1_tx_utils/l1_tx_utils_with_blobs.d.ts +0 -26
  85. package/dest/l1_tx_utils/l1_tx_utils_with_blobs.d.ts.map +0 -1
  86. package/dest/l1_tx_utils/l1_tx_utils_with_blobs.js +0 -26
  87. package/dest/test/delayed_tx_utils.d.ts +0 -13
  88. package/dest/test/delayed_tx_utils.d.ts.map +0 -1
  89. package/dest/test/delayed_tx_utils.js +0 -28
  90. package/dest/test/tx_delayer.d.ts +0 -36
  91. package/dest/test/tx_delayer.d.ts.map +0 -1
  92. package/src/l1_tx_utils/l1_tx_utils_with_blobs.ts +0 -77
  93. package/src/test/delayed_tx_utils.ts +0 -52
@@ -1,8 +1,9 @@
1
+ import type { BlobKzgInstance } from '@aztec/blob-lib/types';
1
2
  import { maxBigint } from '@aztec/foundation/bigint';
2
3
  import { merge, pick } from '@aztec/foundation/collection';
3
4
  import { InterruptError, TimeoutError } from '@aztec/foundation/error';
4
5
  import { EthAddress } from '@aztec/foundation/eth-address';
5
- import { type Logger, createLogger } from '@aztec/foundation/log';
6
+ import { type Logger, type LoggerBindings, createLogger } from '@aztec/foundation/log';
6
7
  import { retryUntil } from '@aztec/foundation/retry';
7
8
  import { sleep } from '@aztec/foundation/sleep';
8
9
  import { DateProvider } from '@aztec/foundation/timer';
@@ -13,16 +14,13 @@ import {
13
14
  type Abi,
14
15
  type BlockOverrides,
15
16
  type Hex,
16
- type NonceManager,
17
17
  type PrepareTransactionRequestRequest,
18
18
  type StateOverride,
19
19
  type TransactionReceipt,
20
20
  type TransactionSerializable,
21
- createNonceManager,
22
21
  formatGwei,
23
22
  serializeTransaction,
24
23
  } from 'viem';
25
- import { jsonRpc } from 'viem/nonce';
26
24
 
27
25
  import type { ViemClient } from '../types.js';
28
26
  import { formatViemError } from '../utils.js';
@@ -30,6 +28,7 @@ import { type L1TxUtilsConfig, l1TxUtilsConfigMappings } from './config.js';
30
28
  import { MAX_L1_TX_LIMIT } from './constants.js';
31
29
  import type { IL1TxMetrics, IL1TxStore } from './interfaces.js';
32
30
  import { ReadOnlyL1TxUtils } from './readonly_l1_tx_utils.js';
31
+ import { Delayer, createDelayer, wrapClientWithDelayer } from './tx_delayer.js';
33
32
  import {
34
33
  DroppedTransactionError,
35
34
  type L1BlobInputs,
@@ -45,8 +44,13 @@ import {
45
44
  const MAX_L1_TX_STATES = 32;
46
45
 
47
46
  export class L1TxUtils extends ReadOnlyL1TxUtils {
48
- protected nonceManager: NonceManager;
49
47
  protected txs: L1TxState[] = [];
48
+ /** Last nonce successfully sent to the chain. Used as a lower bound when a fallback RPC node returns a stale count. */
49
+ private lastSentNonce: number | undefined;
50
+ /** Tx delayer for testing. Only set when enableDelayer config is true. */
51
+ public delayer?: Delayer;
52
+ /** KZG instance for blob operations. */
53
+ protected kzg?: BlobKzgInstance;
50
54
 
51
55
  constructor(
52
56
  public override client: ViemClient,
@@ -58,9 +62,25 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
58
62
  debugMaxGasLimit: boolean = false,
59
63
  protected store?: IL1TxStore,
60
64
  protected metrics?: IL1TxMetrics,
65
+ kzg?: BlobKzgInstance,
66
+ delayer?: Delayer,
61
67
  ) {
62
68
  super(client, logger, dateProvider, config, debugMaxGasLimit);
63
- this.nonceManager = createNonceManager({ source: jsonRpc() });
69
+ this.kzg = kzg;
70
+
71
+ // Set up delayer: use provided one or create new
72
+ if (config?.enableDelayer && config?.ethereumSlotDuration) {
73
+ this.delayer =
74
+ delayer ?? this.createDelayer({ ethereumSlotDuration: config.ethereumSlotDuration }, logger.getBindings());
75
+ this.client = wrapClientWithDelayer(this.client, this.delayer);
76
+ if (config.txDelayerMaxInclusionTimeIntoSlot !== undefined) {
77
+ this.delayer.setMaxInclusionTimeIntoSlot(config.txDelayerMaxInclusionTimeIntoSlot);
78
+ }
79
+ } else if (delayer) {
80
+ // Delayer provided but enableDelayer not set — just store it without wrapping
81
+ logger.warn('Delayer provided but enableDelayer config is not set; delayer will not be used');
82
+ this.delayer = delayer;
83
+ }
64
84
  }
65
85
 
66
86
  public get state() {
@@ -87,6 +107,11 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
87
107
  this.metrics?.recordMinedTx(l1TxState, new Date(l1Timestamp));
88
108
  } else if (newState === TxUtilsState.NOT_MINED) {
89
109
  this.metrics?.recordDroppedTx(l1TxState);
110
+ // The tx was dropped: the chain nonce reverted to l1TxState.nonce, so our lower bound is
111
+ // no longer valid. Clear it so the next send fetches the real nonce from the chain.
112
+ if (this.lastSentNonce === l1TxState.nonce) {
113
+ this.lastSentNonce = undefined;
114
+ }
90
115
  }
91
116
 
92
117
  // Update state in the store
@@ -221,15 +246,6 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
221
246
  throw new InterruptError(`Transaction sending is interrupted`);
222
247
  }
223
248
 
224
- const nonce = await this.nonceManager.consume({
225
- client: this.client,
226
- address: account,
227
- chainId: this.client.chain.id,
228
- });
229
-
230
- const baseState = { request, gasLimit, blobInputs, gasPrice, nonce };
231
- const txData = this.makeTxData(baseState, { isCancelTx: false });
232
-
233
249
  const now = new Date(await this.getL1Timestamp());
234
250
  if (gasConfig.txTimeoutAt && now > gasConfig.txTimeoutAt) {
235
251
  throw new TimeoutError(
@@ -237,9 +253,20 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
237
253
  );
238
254
  }
239
255
 
256
+ const chainNonce = await this.client.getTransactionCount({ address: account, blockTag: 'pending' });
257
+ // If a fallback RPC node returns a stale count (lower than what we last sent), use our
258
+ // local lower bound to avoid sending a duplicate of an already-pending transaction.
259
+ const nonce =
260
+ this.lastSentNonce !== undefined && chainNonce <= this.lastSentNonce ? this.lastSentNonce + 1 : chainNonce;
261
+
262
+ const baseState = { request, gasLimit, blobInputs, gasPrice, nonce };
263
+ const txData = this.makeTxData(baseState, { isCancelTx: false });
264
+
240
265
  // Send the new tx
241
266
  const signedRequest = await this.prepareSignedTransaction(txData);
242
267
  const txHash = await this.client.sendRawTransaction({ serializedTransaction: signedRequest });
268
+ // Update after tx is sent successfully
269
+ this.lastSentNonce = nonce;
243
270
 
244
271
  // Create the new state for monitoring
245
272
  const l1TxState: L1TxState = {
@@ -423,7 +450,6 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
423
450
  { nonce, account, pendingNonce, timePassed },
424
451
  );
425
452
  await this.updateState(state, TxUtilsState.NOT_MINED);
426
- this.nonceManager.reset({ address: account, chainId: this.client.chain.id });
427
453
  throw new DroppedTransactionError(nonce, account);
428
454
  }
429
455
 
@@ -515,12 +541,7 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
515
541
 
516
542
  // Oh no, the transaction has timed out!
517
543
  if (isCancelTx || !gasConfig.cancelTxOnTimeout) {
518
- // If this was already a cancellation tx, or we are configured to not cancel txs, we just mark it as NOT_MINED
519
- // and reset the nonce manager, so the next tx that comes along can reuse the nonce if/when this tx gets dropped.
520
- // This is the nastiest scenario for us, since the new tx could acquire the next nonce, but then this tx is dropped,
521
- // and the new tx would never get mined. Eventually, the new tx would also drop.
522
544
  await this.updateState(state, TxUtilsState.NOT_MINED);
523
- this.nonceManager.reset({ address: account, chainId: this.client.chain.id });
524
545
  } else {
525
546
  // Otherwise we fire the cancellation without awaiting to avoid blocking the caller,
526
547
  // and monitor it in the background so we can speed it up as needed.
@@ -659,7 +680,6 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
659
680
  { nonce, account },
660
681
  );
661
682
  await this.updateState(state, TxUtilsState.NOT_MINED);
662
- this.nonceManager.reset({ address: account, chainId: this.client.chain.id });
663
683
  return;
664
684
  }
665
685
 
@@ -671,7 +691,6 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
671
691
  { nonce, account, currentNonce },
672
692
  );
673
693
  await this.updateState(state, TxUtilsState.NOT_MINED);
674
- this.nonceManager.reset({ address: account, chainId: this.client.chain.id });
675
694
  return;
676
695
  }
677
696
 
@@ -731,8 +750,17 @@ export class L1TxUtils extends ReadOnlyL1TxUtils {
731
750
  return Number(timestamp) * 1000;
732
751
  }
733
752
 
734
- /** Makes empty blob inputs for the cancellation tx. To be overridden in L1TxUtilsWithBlobs. */
735
- protected makeEmptyBlobInputs(_maxFeePerBlobGas: bigint): Required<L1BlobInputs> {
736
- throw new Error('Cannot make empty blob inputs for cancellation');
753
+ /** Makes empty blob inputs for the cancellation tx. */
754
+ protected makeEmptyBlobInputs(maxFeePerBlobGas: bigint): Required<L1BlobInputs> {
755
+ if (!this.kzg) {
756
+ throw new Error('Cannot make empty blob inputs for cancellation without kzg');
757
+ }
758
+ const blobData = new Uint8Array(131072).fill(0);
759
+ return { blobs: [blobData], kzg: this.kzg, maxFeePerBlobGas };
760
+ }
761
+
762
+ /** Creates a new delayer instance. */
763
+ protected createDelayer(opts: { ethereumSlotDuration: bigint | number }, bindings: LoggerBindings): Delayer {
764
+ return createDelayer(this.dateProvider, opts, bindings);
737
765
  }
738
766
  }
@@ -1,5 +1,5 @@
1
1
  import { omit } from '@aztec/foundation/collection';
2
- import { type Logger, createLogger } from '@aztec/foundation/log';
2
+ import { type Logger, type LoggerBindings, createLogger } from '@aztec/foundation/log';
3
3
  import { retryUntil } from '@aztec/foundation/retry';
4
4
  import type { DateProvider } from '@aztec/foundation/timer';
5
5
 
@@ -9,14 +9,16 @@ import {
9
9
  type Hex,
10
10
  type PublicClient,
11
11
  type TransactionSerializableEIP4844,
12
+ type TransactionSerialized,
12
13
  keccak256,
13
14
  parseTransaction,
14
15
  publicActions,
16
+ recoverTransactionAddress,
15
17
  serializeTransaction,
16
18
  walletActions,
17
19
  } from 'viem';
18
20
 
19
- import { type ViemClient, isExtendedClient } from '../types.js';
21
+ import type { ExtendedViemWalletClient, ViemClient } from '../types.js';
20
22
 
21
23
  const MAX_WAIT_TIME_SECONDS = 180;
22
24
 
@@ -73,82 +75,98 @@ export function waitUntilL1Timestamp<T extends Client>(
73
75
  );
74
76
  }
75
77
 
76
- export interface Delayer {
77
- /** Returns the hashes of all effectively sent txs. */
78
- getSentTxHashes(): Hex[];
79
- /** Returns the raw hex for all cancelled txs. */
80
- getCancelledTxs(): Hex[];
81
- /** Delays the next tx to be sent so it lands on the given L1 block number. */
82
- pauseNextTxUntilBlock(l1BlockNumber: number | bigint | undefined): void;
83
- /** Delays the next tx to be sent so it lands on the given timestamp. */
84
- pauseNextTxUntilTimestamp(l1Timestamp: number | bigint | undefined): void;
85
- /** Delays the next tx to be sent indefinitely. */
86
- cancelNextTx(): void;
87
- /**
88
- * Sets max inclusion time into slot. If more than this many seconds have passed
89
- * since the last L1 block was mined, then any tx will not be mined in the current
90
- * L1 slot but will be deferred for the next one.
91
- */
92
- setMaxInclusionTimeIntoSlot(seconds: number | bigint | undefined): void;
93
- }
78
+ /** Manages tx delaying for testing, intercepting sendRawTransaction calls to delay or cancel them. */
79
+ export class Delayer {
80
+ private logger: Logger;
81
+
82
+ public maxInclusionTimeIntoSlot: number | undefined = undefined;
83
+ public ethereumSlotDuration: bigint;
84
+ public nextWait: { l1Timestamp: bigint } | { l1BlockNumber: bigint } | { indefinitely: true } | undefined = undefined;
85
+ public sentTxHashes: Hex[] = [];
86
+ public cancelledTxs: Hex[] = [];
94
87
 
95
- class DelayerImpl implements Delayer {
96
- private logger = createLogger('ethereum:tx_delayer');
97
88
  constructor(
98
89
  public dateProvider: DateProvider,
99
90
  opts: { ethereumSlotDuration: bigint | number },
91
+ bindings: LoggerBindings,
100
92
  ) {
101
93
  this.ethereumSlotDuration = BigInt(opts.ethereumSlotDuration);
94
+ this.logger = createLogger('ethereum:tx_delayer', bindings);
102
95
  }
103
96
 
104
- public maxInclusionTimeIntoSlot: number | undefined = undefined;
105
- public ethereumSlotDuration: bigint;
106
- public nextWait: { l1Timestamp: bigint } | { l1BlockNumber: bigint } | { indefinitely: true } | undefined = undefined;
107
- public sentTxHashes: Hex[] = [];
108
- public cancelledTxs: Hex[] = [];
97
+ /** Returns the logger instance used by this delayer. */
98
+ getLogger(): Logger {
99
+ return this.logger;
100
+ }
109
101
 
102
+ /** Returns the hashes of all effectively sent txs. */
110
103
  getSentTxHashes() {
111
104
  return this.sentTxHashes;
112
105
  }
113
106
 
107
+ /** Returns the raw hex for all cancelled txs. */
114
108
  getCancelledTxs(): Hex[] {
115
109
  return this.cancelledTxs;
116
110
  }
117
111
 
112
+ /** Delays the next tx to be sent so it lands on the given L1 block number. */
118
113
  pauseNextTxUntilBlock(l1BlockNumber: number | bigint) {
119
114
  this.nextWait = { l1BlockNumber: BigInt(l1BlockNumber) };
120
115
  }
121
116
 
117
+ /** Delays the next tx to be sent so it lands on the given timestamp. */
122
118
  pauseNextTxUntilTimestamp(l1Timestamp: number | bigint) {
123
119
  this.nextWait = { l1Timestamp: BigInt(l1Timestamp) };
124
120
  }
125
121
 
122
+ /** Delays the next tx to be sent indefinitely. */
126
123
  cancelNextTx() {
127
124
  this.nextWait = { indefinitely: true };
128
125
  }
129
126
 
127
+ /**
128
+ * Sets max inclusion time into slot. If more than this many seconds have passed
129
+ * since the last L1 block was mined, then any tx will not be mined in the current
130
+ * L1 slot but will be deferred for the next one.
131
+ */
130
132
  setMaxInclusionTimeIntoSlot(seconds: number | undefined) {
131
133
  this.maxInclusionTimeIntoSlot = seconds;
132
134
  }
133
135
  }
134
136
 
135
137
  /**
136
- * Returns a new client (without modifying the one passed in) with an injected tx delayer.
137
- * The delayer can be used to hold off the next tx to be sent until a given block number.
138
- * TODO(#10824): This doesn't play along well with blob txs for some reason.
138
+ * Creates a new Delayer instance. Exposed so callers can create a single shared delayer
139
+ * and pass it to multiple `wrapClientWithDelayer` calls.
139
140
  */
140
- export function withDelayer<T extends ViemClient>(
141
- client: T,
141
+ export function createDelayer(
142
142
  dateProvider: DateProvider,
143
143
  opts: { ethereumSlotDuration: bigint | number },
144
- ): { client: T; delayer: Delayer } {
145
- if (!isExtendedClient(client)) {
146
- throw new Error('withDelayer has to be instantiated with a wallet viem client.');
144
+ bindings: LoggerBindings,
145
+ ): Delayer {
146
+ return new Delayer(dateProvider, opts, bindings);
147
+ }
148
+
149
+ /** Tries to recover the sender address from a serialized signed transaction. */
150
+ async function tryRecoverSender(serializedTransaction: Hex): Promise<string | undefined> {
151
+ try {
152
+ return await recoverTransactionAddress({
153
+ serializedTransaction: serializedTransaction as TransactionSerialized,
154
+ });
155
+ } catch {
156
+ return undefined;
147
157
  }
148
- const logger = createLogger('ethereum:tx_delayer');
149
- const delayer = new DelayerImpl(dateProvider, opts);
158
+ }
150
159
 
151
- const extended = client
160
+ /**
161
+ * Wraps a viem client with tx delaying logic. Returns the wrapped client.
162
+ * The delayer intercepts sendRawTransaction calls and delays them based on the delayer's state.
163
+ */
164
+ export function wrapClientWithDelayer<T extends ViemClient>(client: T, delayer: Delayer): T {
165
+ const logger = delayer.getLogger();
166
+
167
+ // Cast to ExtendedViemWalletClient for the extend chain since it has sendRawTransaction.
168
+ // The sendRawTransaction override is applied to all clients regardless of type.
169
+ const withRawTx = (client as unknown as ExtendedViemWalletClient)
152
170
  // Tweak sendRawTransaction so it uses the delay defined in the delayer.
153
171
  // Note that this will only work with local accounts (ie accounts for which we have the private key).
154
172
  // Transactions signed by the node will not be delayed since they use sendTransaction directly,
@@ -160,6 +178,7 @@ export function withDelayer<T extends ViemClient>(
160
178
 
161
179
  const { serializedTransaction } = args[0];
162
180
  const publicClient = client as unknown as PublicClient;
181
+ const sender = await tryRecoverSender(serializedTransaction);
163
182
 
164
183
  if (delayer.nextWait !== undefined) {
165
184
  // Check if we have been instructed to delay the next tx.
@@ -171,7 +190,7 @@ export function withDelayer<T extends ViemClient>(
171
190
 
172
191
  // Cancel tx outright if instructed
173
192
  if ('indefinitely' in waitUntil && waitUntil.indefinitely) {
174
- logger.info(`Cancelling tx ${txHash}`);
193
+ logger.info(`Cancelling tx ${txHash}`, { sender });
175
194
  delayer.cancelledTxs.push(serializedTransaction);
176
195
  return Promise.resolve(txHash);
177
196
  }
@@ -185,6 +204,7 @@ export function withDelayer<T extends ViemClient>(
185
204
  : undefined;
186
205
 
187
206
  logger.info(`Delaying tx ${txHash} until ${inspect(waitUntil)}`, {
207
+ sender,
188
208
  argsLen: args.length,
189
209
  ...omit(parseTransaction(serializedTransaction), 'data', 'sidecars'),
190
210
  });
@@ -196,6 +216,7 @@ export function withDelayer<T extends ViemClient>(
196
216
 
197
217
  txHash = computeTxHash(serializedTransaction);
198
218
  const logData = {
219
+ sender,
199
220
  ...omit(parseTransaction(serializedTransaction), 'data', 'sidecars'),
200
221
  lastBlockTimestamp,
201
222
  now,
@@ -225,28 +246,35 @@ export function withDelayer<T extends ViemClient>(
225
246
  computedTxHash: txHash,
226
247
  });
227
248
  }
228
- logger.info(`Sent previously delayed tx ${clientTxHash}`);
249
+ logger.info(`Sent previously delayed tx ${clientTxHash}`, { sender });
229
250
  delayer.sentTxHashes.push(clientTxHash);
230
251
  })
231
252
  .catch(err => logger.error(`Error sending tx after delay`, err));
232
253
  return Promise.resolve(txHash!);
233
254
  } else {
234
255
  const txHash = await client.sendRawTransaction(...args);
235
- logger.debug(`Sent tx immediately ${txHash}`);
256
+ logger.debug(`Sent tx immediately ${txHash}`, { sender });
236
257
  delayer.sentTxHashes.push(txHash);
237
258
  return txHash;
238
259
  }
239
260
  },
240
- }))
241
- // Re-extend with sendTransaction so it uses the modified sendRawTransaction.
242
- .extend(client => ({ sendTransaction: walletActions(client).sendTransaction }))
243
- // And with the actions that depend on the modified sendTransaction
244
- .extend(client => ({
245
- writeContract: walletActions(client).writeContract,
246
- deployContract: walletActions(client).deployContract,
247
- })) as T;
261
+ }));
262
+
263
+ // Only re-bind wallet actions (sendTransaction, writeContract, deployContract) for wallet clients.
264
+ // This is needed for tests that use wallet actions directly rather than sendRawTransaction.
265
+ const isWalletClient = 'account' in client && client.account !== undefined;
266
+ const extended = isWalletClient
267
+ ? withRawTx
268
+ // Re-extend with sendTransaction so it uses the modified sendRawTransaction.
269
+ .extend(client => ({ sendTransaction: walletActions(client).sendTransaction }))
270
+ // And with the actions that depend on the modified sendTransaction
271
+ .extend(client => ({
272
+ writeContract: walletActions(client).writeContract,
273
+ deployContract: walletActions(client).deployContract,
274
+ }))
275
+ : withRawTx;
248
276
 
249
- return { client: extended, delayer };
277
+ return extended as T;
250
278
  }
251
279
 
252
280
  /**
package/src/test/index.ts CHANGED
@@ -1,8 +1,6 @@
1
- export * from './delayed_tx_utils.js';
2
1
  export * from './eth_cheat_codes.js';
3
2
  export * from './eth_cheat_codes_with_state.js';
4
3
  export * from './start_anvil.js';
5
- export * from './tx_delayer.js';
6
4
  export * from './upgrade_utils.js';
7
5
  export * from './chain_monitor.js';
8
6
  export * from './rollup_cheat_codes.js';
@@ -2,9 +2,17 @@ import { createLogger } from '@aztec/foundation/log';
2
2
  import { makeBackoff, retry } from '@aztec/foundation/retry';
3
3
  import { fileURLToPath } from '@aztec/foundation/url';
4
4
 
5
- import { type Anvil, createAnvil } from '@viem/anvil';
5
+ import { type ChildProcess, spawn } from 'child_process';
6
6
  import { dirname, resolve } from 'path';
7
7
 
8
+ /** Minimal interface matching the @viem/anvil Anvil shape used by callers. */
9
+ export interface Anvil {
10
+ readonly port: number;
11
+ readonly host: string;
12
+ readonly status: 'listening' | 'idle';
13
+ stop(): Promise<void>;
14
+ }
15
+
8
16
  /**
9
17
  * Ensures there's a running Anvil instance and returns the RPC URL.
10
18
  */
@@ -16,7 +24,7 @@ export async function startAnvil(
16
24
  captureMethodCalls?: boolean;
17
25
  accounts?: number;
18
26
  chainId?: number;
19
- /** The hardfork to use - note: @viem/anvil types are out of date but 'cancun' and 'latest' work */
27
+ /** The hardfork to use (e.g. 'cancun', 'latest'). */
20
28
  hardfork?: string;
21
29
  } = {},
22
30
  ): Promise<{ anvil: Anvil; methodCalls?: string[]; rpcUrl: string; stop: () => Promise<void> }> {
@@ -24,48 +32,157 @@ export async function startAnvil(
24
32
  const logger = opts.log ? createLogger('ethereum:anvil') : undefined;
25
33
  const methodCalls = opts.captureMethodCalls ? ([] as string[]) : undefined;
26
34
 
27
- let port: number | undefined;
35
+ let detectedPort: number | undefined;
28
36
 
29
- // Start anvil.
30
- // We go via a wrapper script to ensure if the parent dies, anvil dies.
31
37
  const anvil = await retry(
32
38
  async () => {
33
- const anvil = createAnvil({
34
- anvilBinary,
35
- host: '127.0.0.1',
36
- port: opts.port ?? (process.env.ANVIL_PORT ? parseInt(process.env.ANVIL_PORT) : 8545),
37
- blockTime: opts.l1BlockTime,
38
- stopTimeout: 1000,
39
- accounts: opts.accounts ?? 20,
40
- gasLimit: 45_000_000n,
41
- chainId: opts.chainId ?? 31337,
39
+ const port = opts.port ?? (process.env.ANVIL_PORT ? parseInt(process.env.ANVIL_PORT) : 8545);
40
+ const args: string[] = [
41
+ '--host',
42
+ '127.0.0.1',
43
+ '--port',
44
+ String(port),
45
+ '--accounts',
46
+ String(opts.accounts ?? 20),
47
+ '--gas-limit',
48
+ String(45_000_000),
49
+ '--chain-id',
50
+ String(opts.chainId ?? 31337),
51
+ ];
52
+ if (opts.l1BlockTime !== undefined) {
53
+ args.push('--block-time', String(opts.l1BlockTime));
54
+ }
55
+ if (opts.hardfork !== undefined) {
56
+ args.push('--hardfork', opts.hardfork);
57
+ }
58
+
59
+ const child = spawn(anvilBinary, args, {
60
+ stdio: ['ignore', 'pipe', 'pipe'],
61
+ env: { ...process.env, RAYON_NUM_THREADS: '1' },
42
62
  });
43
63
 
44
- // Listen to the anvil output to get the port.
45
- const removeHandler = anvil.on('message', (message: string) => {
46
- logger?.debug(message.trim());
64
+ // Wait for "Listening on" or an early exit.
65
+ await new Promise<void>((resolve, reject) => {
66
+ let stderr = '';
67
+
68
+ const onStdout = (data: Buffer) => {
69
+ const text = data.toString();
70
+ logger?.debug(text.trim());
71
+ methodCalls?.push(...(text.match(/eth_[^\s]+/g) || []));
47
72
 
48
- methodCalls?.push(...(message.match(/eth_[^\s]+/g) || []));
49
- if (port === undefined && message.includes('Listening on')) {
50
- port = parseInt(message.match(/Listening on ([^:]+):(\d+)/)![2]);
51
- }
73
+ if (detectedPort === undefined && text.includes('Listening on')) {
74
+ const match = text.match(/Listening on ([^:]+):(\d+)/);
75
+ if (match) {
76
+ detectedPort = parseInt(match[2]);
77
+ }
78
+ }
79
+ if (detectedPort !== undefined) {
80
+ child.stdout?.removeListener('data', onStdout);
81
+ child.stderr?.removeListener('data', onStderr);
82
+ child.removeListener('close', onClose);
83
+ resolve();
84
+ }
85
+ };
86
+
87
+ const onStderr = (data: Buffer) => {
88
+ stderr += data.toString();
89
+ logger?.debug(data.toString().trim());
90
+ };
91
+
92
+ const onClose = (code: number | null) => {
93
+ child.stdout?.removeListener('data', onStdout);
94
+ child.stderr?.removeListener('data', onStderr);
95
+ reject(new Error(`Anvil exited with code ${code} before listening. stderr: ${stderr}`));
96
+ };
97
+
98
+ child.stdout?.on('data', onStdout);
99
+ child.stderr?.on('data', onStderr);
100
+ child.once('close', onClose);
52
101
  });
53
- await anvil.start();
54
- if (!logger && !opts.captureMethodCalls) {
55
- removeHandler();
102
+
103
+ // Continue piping for logging / method-call capture after startup.
104
+ if (logger || opts.captureMethodCalls) {
105
+ child.stdout?.on('data', (data: Buffer) => {
106
+ const text = data.toString();
107
+ logger?.debug(text.trim());
108
+ methodCalls?.push(...(text.match(/eth_[^\s]+/g) || []));
109
+ });
110
+ child.stderr?.on('data', (data: Buffer) => {
111
+ logger?.debug(data.toString().trim());
112
+ });
113
+ } else {
114
+ // Consume streams so the child process doesn't block on full pipe buffers.
115
+ child.stdout?.resume();
116
+ child.stderr?.resume();
56
117
  }
57
118
 
58
- return anvil;
119
+ return child;
59
120
  },
60
121
  'Start anvil',
61
122
  makeBackoff([5, 5, 5]),
62
123
  );
63
124
 
64
- if (!port) {
125
+ if (!detectedPort) {
65
126
  throw new Error('Failed to start anvil');
66
127
  }
67
128
 
68
- // Monkeypatch the anvil instance to include the actually assigned port
69
- // Object.defineProperty(anvil, 'port', { value: port, writable: false });
70
- return { anvil, methodCalls, stop: () => anvil.stop(), rpcUrl: `http://127.0.0.1:${port}` };
129
+ const port = detectedPort;
130
+ let status: 'listening' | 'idle' = 'listening';
131
+
132
+ anvil.once('close', () => {
133
+ status = 'idle';
134
+ });
135
+
136
+ const stop = async () => {
137
+ if (status === 'idle') {
138
+ return;
139
+ }
140
+ await killChild(anvil);
141
+ };
142
+
143
+ const anvilObj: Anvil = {
144
+ port,
145
+ host: '127.0.0.1',
146
+ get status() {
147
+ return status;
148
+ },
149
+ stop,
150
+ };
151
+
152
+ return { anvil: anvilObj, methodCalls, stop, rpcUrl: `http://127.0.0.1:${port}` };
153
+ }
154
+
155
+ /** Send SIGTERM, wait up to 5 s, then SIGKILL. All timers are always cleared. */
156
+ function killChild(child: ChildProcess): Promise<void> {
157
+ return new Promise<void>(resolve => {
158
+ if (child.exitCode !== null || child.killed) {
159
+ child.stdout?.destroy();
160
+ child.stderr?.destroy();
161
+ resolve();
162
+ return;
163
+ }
164
+
165
+ let killTimer: NodeJS.Timeout | undefined;
166
+
167
+ const onClose = () => {
168
+ if (killTimer !== undefined) {
169
+ clearTimeout(killTimer);
170
+ }
171
+ // Destroy stdio streams so their PipeWrap handles don't keep the event loop alive.
172
+ child.stdout?.destroy();
173
+ child.stderr?.destroy();
174
+ resolve();
175
+ };
176
+
177
+ child.once('close', onClose);
178
+ child.kill('SIGTERM');
179
+
180
+ killTimer = setTimeout(() => {
181
+ killTimer = undefined;
182
+ child.kill('SIGKILL');
183
+ }, 5000);
184
+
185
+ // Ensure the timer does not prevent Node from exiting.
186
+ killTimer.unref();
187
+ });
71
188
  }
@@ -7,7 +7,7 @@ import { type GetContractReturnType, type PrivateKeyAccount, getContract } from
7
7
 
8
8
  import { extractProposalIdFromLogs } from '../contracts/governance.js';
9
9
  import type { L1ContractAddresses } from '../l1_contract_addresses.js';
10
- import { createL1TxUtilsFromViemWallet } from '../l1_tx_utils/index.js';
10
+ import { createL1TxUtils } from '../l1_tx_utils/index.js';
11
11
  import type { ExtendedViemWalletClient, ViemPublicClient } from '../types.js';
12
12
  import { EthCheatCodes } from './eth_cheat_codes.js';
13
13
 
@@ -22,7 +22,7 @@ export async function executeGovernanceProposal(
22
22
  ) {
23
23
  const proposal = await governance.read.getProposal([proposalId]);
24
24
 
25
- const l1TxUtils = createL1TxUtilsFromViemWallet(l1Client);
25
+ const l1TxUtils = createL1TxUtils(l1Client);
26
26
 
27
27
  const waitL1Block = async () => {
28
28
  await l1TxUtils.sendAndMonitorTransaction({