@aztec/prover-node 0.0.1-commit.b655e406 → 0.0.1-commit.c31f2472

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 (53) hide show
  1. package/dest/actions/download-epoch-proving-job.d.ts +4 -4
  2. package/dest/actions/index.d.ts +1 -1
  3. package/dest/actions/rerun-epoch-proving-job.d.ts +3 -2
  4. package/dest/actions/rerun-epoch-proving-job.d.ts.map +1 -1
  5. package/dest/actions/rerun-epoch-proving-job.js +5 -3
  6. package/dest/actions/upload-epoch-proof-failure.d.ts +1 -1
  7. package/dest/bin/run-failed-epoch.d.ts +1 -1
  8. package/dest/config.d.ts +5 -4
  9. package/dest/config.d.ts.map +1 -1
  10. package/dest/config.js +4 -3
  11. package/dest/factory.d.ts +2 -4
  12. package/dest/factory.d.ts.map +1 -1
  13. package/dest/factory.js +21 -16
  14. package/dest/index.d.ts +2 -1
  15. package/dest/index.d.ts.map +1 -1
  16. package/dest/index.js +1 -0
  17. package/dest/job/epoch-proving-job-data.d.ts +8 -6
  18. package/dest/job/epoch-proving-job-data.d.ts.map +1 -1
  19. package/dest/job/epoch-proving-job-data.js +25 -18
  20. package/dest/job/epoch-proving-job.d.ts +7 -13
  21. package/dest/job/epoch-proving-job.d.ts.map +1 -1
  22. package/dest/job/epoch-proving-job.js +486 -99
  23. package/dest/metrics.d.ts +4 -3
  24. package/dest/metrics.d.ts.map +1 -1
  25. package/dest/metrics.js +30 -98
  26. package/dest/monitors/epoch-monitor.d.ts +3 -2
  27. package/dest/monitors/epoch-monitor.d.ts.map +1 -1
  28. package/dest/monitors/epoch-monitor.js +3 -11
  29. package/dest/monitors/index.d.ts +1 -1
  30. package/dest/prover-node-publisher.d.ts +12 -9
  31. package/dest/prover-node-publisher.d.ts.map +1 -1
  32. package/dest/prover-node-publisher.js +47 -40
  33. package/dest/prover-node.d.ts +9 -8
  34. package/dest/prover-node.d.ts.map +1 -1
  35. package/dest/prover-node.js +432 -51
  36. package/dest/prover-publisher-factory.d.ts +7 -3
  37. package/dest/prover-publisher-factory.d.ts.map +1 -1
  38. package/dest/prover-publisher-factory.js +4 -2
  39. package/dest/test/index.d.ts +1 -1
  40. package/dest/test/index.d.ts.map +1 -1
  41. package/package.json +26 -25
  42. package/src/actions/rerun-epoch-proving-job.ts +5 -3
  43. package/src/bin/run-failed-epoch.ts +1 -1
  44. package/src/config.ts +6 -4
  45. package/src/factory.ts +31 -18
  46. package/src/index.ts +1 -0
  47. package/src/job/epoch-proving-job-data.ts +31 -25
  48. package/src/job/epoch-proving-job.ts +114 -99
  49. package/src/metrics.ts +31 -82
  50. package/src/monitors/epoch-monitor.ts +5 -11
  51. package/src/prover-node-publisher.ts +68 -55
  52. package/src/prover-node.ts +49 -43
  53. package/src/prover-publisher-factory.ts +14 -6
@@ -1,11 +1,13 @@
1
- import type { BatchedBlob } from '@aztec/blob-lib';
1
+ import { BatchedBlob, getEthBlobEvaluationInputs } from '@aztec/blob-lib';
2
2
  import { AZTEC_MAX_EPOCH_DURATION } from '@aztec/constants';
3
- import type { L1TxUtils, RollupContract, ViemCommitteeAttestation } from '@aztec/ethereum';
3
+ import type { RollupContract, ViemCommitteeAttestation } from '@aztec/ethereum/contracts';
4
+ import type { L1TxUtils } from '@aztec/ethereum/l1-tx-utils';
4
5
  import { makeTuple } from '@aztec/foundation/array';
6
+ import { CheckpointNumber, EpochNumber } from '@aztec/foundation/branded-types';
5
7
  import { areArraysEqual } from '@aztec/foundation/collection';
8
+ import { Fr } from '@aztec/foundation/curves/bn254';
6
9
  import { EthAddress } from '@aztec/foundation/eth-address';
7
- import { Fr } from '@aztec/foundation/fields';
8
- import { createLogger } from '@aztec/foundation/log';
10
+ import { type Logger, type LoggerBindings, createLogger } from '@aztec/foundation/log';
9
11
  import type { Tuple } from '@aztec/foundation/serialize';
10
12
  import { Timer } from '@aztec/foundation/timer';
11
13
  import { RollupAbi } from '@aztec/l1-artifacts';
@@ -37,7 +39,7 @@ export class ProverNodePublisher {
37
39
  private interrupted = false;
38
40
  private metrics: ProverNodePublisherMetrics;
39
41
 
40
- protected log = createLogger('prover-node:l1-tx-publisher');
42
+ protected log: Logger;
41
43
 
42
44
  protected rollupContract: RollupContract;
43
45
 
@@ -50,10 +52,12 @@ export class ProverNodePublisher {
50
52
  l1TxUtils: L1TxUtils;
51
53
  telemetry?: TelemetryClient;
52
54
  },
55
+ bindings?: LoggerBindings,
53
56
  ) {
54
57
  const telemetry = deps.telemetry ?? getTelemetryClient();
55
58
 
56
59
  this.metrics = new ProverNodePublisherMetrics(telemetry, 'ProverNode');
60
+ this.log = createLogger('prover-node:l1-tx-publisher', bindings);
57
61
 
58
62
  this.rollupContract = deps.rollupContract;
59
63
  this.l1TxUtils = deps.l1TxUtils;
@@ -85,16 +89,16 @@ export class ProverNodePublisher {
85
89
  }
86
90
 
87
91
  public async submitEpochProof(args: {
88
- epochNumber: number;
89
- fromBlock: number;
90
- toBlock: number;
92
+ epochNumber: EpochNumber;
93
+ fromCheckpoint: CheckpointNumber;
94
+ toCheckpoint: CheckpointNumber;
91
95
  publicInputs: RootRollupPublicInputs;
92
96
  proof: Proof;
93
97
  batchedBlobInputs: BatchedBlob;
94
98
  attestations: ViemCommitteeAttestation[];
95
99
  }): Promise<boolean> {
96
- const { epochNumber, fromBlock, toBlock } = args;
97
- const ctx = { epochNumber, fromBlock, toBlock };
100
+ const { epochNumber, fromCheckpoint, toCheckpoint } = args;
101
+ const ctx = { epochNumber, fromCheckpoint, toCheckpoint };
98
102
 
99
103
  if (!this.interrupted) {
100
104
  const timer = new Timer();
@@ -103,6 +107,7 @@ export class ProverNodePublisher {
103
107
 
104
108
  const txReceipt = await this.sendSubmitEpochProofTx(args);
105
109
  if (!txReceipt) {
110
+ this.log.error(`Failed to mine submitEpochProof tx`, undefined, ctx);
106
111
  return false;
107
112
  }
108
113
 
@@ -135,47 +140,51 @@ export class ProverNodePublisher {
135
140
  }
136
141
 
137
142
  this.metrics.recordFailedTx();
138
- this.log.error(`Rollup.submitEpochProof tx status failed ${txReceipt.transactionHash}`, undefined, ctx);
143
+ this.log.error(`Rollup submitEpochProof tx reverted ${txReceipt.transactionHash}`, undefined, ctx);
139
144
  }
140
145
 
141
- this.log.verbose('L2 block data syncing interrupted', ctx);
146
+ this.log.verbose('Checkpoint data syncing interrupted', ctx);
142
147
  return false;
143
148
  }
144
149
 
145
150
  private async validateEpochProofSubmission(args: {
146
- fromBlock: number;
147
- toBlock: number;
151
+ fromCheckpoint: CheckpointNumber;
152
+ toCheckpoint: CheckpointNumber;
148
153
  publicInputs: RootRollupPublicInputs;
149
154
  proof: Proof;
150
155
  batchedBlobInputs: BatchedBlob;
151
156
  attestations: ViemCommitteeAttestation[];
152
157
  }) {
153
- const { fromBlock, toBlock, publicInputs, batchedBlobInputs } = args;
158
+ const { fromCheckpoint, toCheckpoint, publicInputs, batchedBlobInputs } = args;
154
159
 
155
- // Check that the block numbers match the expected epoch to be proven
156
- const { pendingBlockNumber: pending, provenBlockNumber: proven } = await this.rollupContract.getTips();
157
- // Don't publish if proven is beyond our toBlock, pointless to do so
158
- if (proven > BigInt(toBlock)) {
159
- throw new Error(`Cannot submit epoch proof for ${fromBlock}-${toBlock} as proven block is ${proven}`);
160
+ // Check that the checkpoint numbers match the expected epoch to be proven
161
+ const { pending, proven } = await this.rollupContract.getTips();
162
+ // Don't publish if proven is beyond our toCheckpoint, pointless to do so
163
+ if (proven > toCheckpoint) {
164
+ throw new Error(
165
+ `Cannot submit epoch proof for ${fromCheckpoint}-${toCheckpoint} as proven checkpoint is ${proven}`,
166
+ );
160
167
  }
161
- // toBlock can't be greater than pending
162
- if (toBlock > pending) {
163
- throw new Error(`Cannot submit epoch proof for ${fromBlock}-${toBlock} as pending block is ${pending}`);
168
+ // toCheckpoint can't be greater than pending
169
+ if (toCheckpoint > pending) {
170
+ throw new Error(
171
+ `Cannot submit epoch proof for ${fromCheckpoint}-${toCheckpoint} as pending checkpoint is ${pending}`,
172
+ );
164
173
  }
165
174
 
166
- // Check the archive for the immediate block before the epoch
167
- const blockLog = await this.rollupContract.getBlock(BigInt(fromBlock - 1));
168
- if (publicInputs.previousArchiveRoot.toString() !== blockLog.archive) {
175
+ // Check the archive for the immediate checkpoint before the epoch
176
+ const checkpointLog = await this.rollupContract.getCheckpoint(CheckpointNumber(fromCheckpoint - 1));
177
+ if (!publicInputs.previousArchiveRoot.equals(checkpointLog.archive)) {
169
178
  throw new Error(
170
- `Previous archive root mismatch: ${publicInputs.previousArchiveRoot.toString()} !== ${blockLog.archive}`,
179
+ `Previous archive root mismatch: ${publicInputs.previousArchiveRoot.toString()} !== ${checkpointLog.archive.toString()}`,
171
180
  );
172
181
  }
173
182
 
174
- // Check the archive for the last block in the epoch
175
- const endBlockLog = await this.rollupContract.getBlock(BigInt(toBlock));
176
- if (publicInputs.endArchiveRoot.toString() !== endBlockLog.archive) {
183
+ // Check the archive for the last checkpoint in the epoch
184
+ const endCheckpointLog = await this.rollupContract.getCheckpoint(toCheckpoint);
185
+ if (!publicInputs.endArchiveRoot.equals(endCheckpointLog.archive)) {
177
186
  throw new Error(
178
- `End archive root mismatch: ${publicInputs.endArchiveRoot.toString()} !== ${endBlockLog.archive}`,
187
+ `End archive root mismatch: ${publicInputs.endArchiveRoot.toString()} !== ${endCheckpointLog.archive.toString()}`,
179
188
  );
180
189
  }
181
190
 
@@ -193,7 +202,7 @@ export class ProverNodePublisher {
193
202
  );
194
203
  const argsPublicInputs = [...publicInputs.toFields()];
195
204
 
196
- if (!areArraysEqual(rollupPublicInputs.map(Fr.fromHexString), argsPublicInputs, (a, b) => a.equals(b))) {
205
+ if (!areArraysEqual(rollupPublicInputs, argsPublicInputs, (a, b) => a.equals(b))) {
197
206
  const fmt = (inputs: Fr[] | readonly string[]) => inputs.map(x => x.toString()).join(', ');
198
207
  throw new Error(
199
208
  `Root rollup public inputs mismatch:\nRollup: ${fmt(rollupPublicInputs)}\nComputed:${fmt(argsPublicInputs)}`,
@@ -202,8 +211,8 @@ export class ProverNodePublisher {
202
211
  }
203
212
 
204
213
  private async sendSubmitEpochProofTx(args: {
205
- fromBlock: number;
206
- toBlock: number;
214
+ fromCheckpoint: CheckpointNumber;
215
+ toCheckpoint: CheckpointNumber;
207
216
  publicInputs: RootRollupPublicInputs;
208
217
  proof: Proof;
209
218
  batchedBlobInputs: BatchedBlob;
@@ -213,8 +222,8 @@ export class ProverNodePublisher {
213
222
 
214
223
  this.log.info(`Submitting epoch proof to L1 rollup contract`, {
215
224
  proofSize: args.proof.withoutPublicInputs().length,
216
- fromBlock: args.fromBlock,
217
- toBlock: args.toBlock,
225
+ fromCheckpoint: args.fromCheckpoint,
226
+ toCheckpoint: args.toCheckpoint,
218
227
  });
219
228
  const data = encodeFunctionData({
220
229
  abi: RollupAbi,
@@ -223,39 +232,43 @@ export class ProverNodePublisher {
223
232
  });
224
233
  try {
225
234
  const { receipt } = await this.l1TxUtils.sendAndMonitorTransaction({ to: this.rollupContract.address, data });
235
+ if (receipt.status !== 'success') {
236
+ const errorMsg = await this.l1TxUtils.tryGetErrorFromRevertedTx(
237
+ data,
238
+ {
239
+ args: [...txArgs],
240
+ functionName: 'submitEpochRootProof',
241
+ abi: RollupAbi,
242
+ address: this.rollupContract.address,
243
+ },
244
+ /*blobInputs*/ undefined,
245
+ /*stateOverride*/ [],
246
+ );
247
+ this.log.error(`Rollup submit epoch proof tx reverted with ${errorMsg ?? 'unknown error'}`);
248
+ return undefined;
249
+ }
226
250
  return receipt;
227
251
  } catch (err) {
228
252
  this.log.error(`Rollup submit epoch proof failed`, err);
229
- const errorMsg = await this.l1TxUtils.tryGetErrorFromRevertedTx(
230
- data,
231
- {
232
- args: [...txArgs],
233
- functionName: 'submitEpochRootProof',
234
- abi: RollupAbi,
235
- address: this.rollupContract.address,
236
- },
237
- /*blobInputs*/ undefined,
238
- /*stateOverride*/ [],
239
- );
240
- this.log.error(`Rollup submit epoch proof tx reverted. ${errorMsg}`);
241
253
  return undefined;
242
254
  }
243
255
  }
244
256
 
245
257
  private getEpochProofPublicInputsArgs(args: {
246
- fromBlock: number;
247
- toBlock: number;
258
+ fromCheckpoint: CheckpointNumber;
259
+ toCheckpoint: CheckpointNumber;
248
260
  publicInputs: RootRollupPublicInputs;
249
261
  batchedBlobInputs: BatchedBlob;
250
262
  attestations: ViemCommitteeAttestation[];
251
263
  }) {
252
264
  // Returns arguments for EpochProofLib.sol -> getEpochProofPublicInputs()
253
265
  return [
254
- BigInt(args.fromBlock) /*_start*/,
255
- BigInt(args.toBlock) /*_end*/,
266
+ BigInt(args.fromCheckpoint) /*_start*/,
267
+ BigInt(args.toCheckpoint) /*_end*/,
256
268
  {
257
269
  previousArchive: args.publicInputs.previousArchiveRoot.toString(),
258
270
  endArchive: args.publicInputs.endArchiveRoot.toString(),
271
+ outHash: args.publicInputs.outHash.toString(),
259
272
  proverId: EthAddress.fromField(args.publicInputs.constants.proverId).toString(),
260
273
  } /*_args*/,
261
274
  makeTuple(AZTEC_MAX_EPOCH_DURATION * 2, i =>
@@ -263,13 +276,13 @@ export class ProverNodePublisher {
263
276
  ? args.publicInputs.fees[i / 2].recipient.toField().toString()
264
277
  : args.publicInputs.fees[(i - 1) / 2].value.toString(),
265
278
  ) /*_fees*/,
266
- args.batchedBlobInputs.getEthBlobEvaluationInputs() /*_blobPublicInputs*/,
279
+ getEthBlobEvaluationInputs(args.batchedBlobInputs) /*_blobPublicInputs*/,
267
280
  ] as const;
268
281
  }
269
282
 
270
283
  private getSubmitEpochProofArgs(args: {
271
- fromBlock: number;
272
- toBlock: number;
284
+ fromCheckpoint: CheckpointNumber;
285
+ toCheckpoint: CheckpointNumber;
273
286
  publicInputs: RootRollupPublicInputs;
274
287
  proof: Proof;
275
288
  batchedBlobInputs: BatchedBlob;
@@ -1,14 +1,16 @@
1
1
  import type { Archiver } from '@aztec/archiver';
2
- import type { RollupContract } from '@aztec/ethereum';
2
+ import type { RollupContract } from '@aztec/ethereum/contracts';
3
+ import { BlockNumber, CheckpointNumber, EpochNumber } from '@aztec/foundation/branded-types';
3
4
  import { assertRequired, compact, pick, sum } from '@aztec/foundation/collection';
5
+ import type { Fr } from '@aztec/foundation/curves/bn254';
4
6
  import { memoize } from '@aztec/foundation/decorators';
5
- import type { Fr } from '@aztec/foundation/fields';
6
7
  import { createLogger } from '@aztec/foundation/log';
7
8
  import { DateProvider } from '@aztec/foundation/timer';
8
9
  import type { DataStoreConfig } from '@aztec/kv-store/config';
9
10
  import type { P2PClient } from '@aztec/p2p';
10
11
  import { PublicProcessorFactory } from '@aztec/simulator/server';
11
- import type { L2Block, L2BlockSource } from '@aztec/stdlib/block';
12
+ import type { L2BlockSource } from '@aztec/stdlib/block';
13
+ import type { Checkpoint } from '@aztec/stdlib/checkpoint';
12
14
  import type { ChainConfig } from '@aztec/stdlib/config';
13
15
  import type { ContractDataSource } from '@aztec/stdlib/contract';
14
16
  import { getProofSubmissionDeadlineTimestamp } from '@aztec/stdlib/epoch-helpers';
@@ -114,7 +116,7 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
114
116
  * @param epochNumber - The epoch number that was just completed.
115
117
  * @returns false if there is an error, true otherwise
116
118
  */
117
- async handleEpochReadyToProve(epochNumber: bigint): Promise<boolean> {
119
+ async handleEpochReadyToProve(epochNumber: EpochNumber): Promise<boolean> {
118
120
  try {
119
121
  this.log.debug(`Running jobs as ${epochNumber} is ready to prove`, {
120
122
  jobs: Array.from(this.jobs.values()).map(job => `${job.getEpochNumber()}:${job.getId()}`),
@@ -184,8 +186,8 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
184
186
  /**
185
187
  * Starts a proving process and returns immediately.
186
188
  */
187
- public async startProof(epochNumber: number | bigint) {
188
- const job = await this.createProvingJob(BigInt(epochNumber), { skipEpochCheck: true });
189
+ public async startProof(epochNumber: EpochNumber) {
190
+ const job = await this.createProvingJob(epochNumber, { skipEpochCheck: true });
189
191
  void this.runJob(job);
190
192
  }
191
193
 
@@ -238,21 +240,20 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
238
240
  /**
239
241
  * Returns an array of jobs being processed.
240
242
  */
241
- public getJobs(): Promise<{ uuid: string; status: EpochProvingJobState; epochNumber: number }[]> {
243
+ public getJobs(): Promise<{ uuid: string; status: EpochProvingJobState; epochNumber: EpochNumber }[]> {
242
244
  return Promise.resolve(
243
245
  Array.from(this.jobs.entries()).map(([uuid, job]) => ({
244
246
  uuid,
245
247
  status: job.getState(),
246
- epochNumber: Number(job.getEpochNumber()),
248
+ epochNumber: job.getEpochNumber(),
247
249
  })),
248
250
  );
249
251
  }
250
252
 
251
253
  protected async getActiveJobsForEpoch(
252
- epochBigInt: bigint,
254
+ epochNumber: EpochNumber,
253
255
  ): Promise<{ uuid: string; status: EpochProvingJobState }[]> {
254
256
  const jobs = await this.getJobs();
255
- const epochNumber = Number(epochBigInt);
256
257
  return jobs.filter(job => job.epochNumber === epochNumber && !EpochProvingJobTerminalState.includes(job.status));
257
258
  }
258
259
 
@@ -263,18 +264,21 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
263
264
  }
264
265
  }
265
266
 
266
- @trackSpan('ProverNode.createProvingJob', epochNumber => ({ [Attributes.EPOCH_NUMBER]: Number(epochNumber) }))
267
- private async createProvingJob(epochNumber: bigint, opts: { skipEpochCheck?: boolean } = {}) {
267
+ @trackSpan('ProverNode.createProvingJob', epochNumber => ({ [Attributes.EPOCH_NUMBER]: epochNumber }))
268
+ private async createProvingJob(epochNumber: EpochNumber, opts: { skipEpochCheck?: boolean } = {}) {
268
269
  this.checkMaximumPendingJobs();
269
270
 
270
271
  this.publisher = await this.publisherFactory.create();
271
272
 
272
273
  // Gather all data for this epoch
273
274
  const epochData = await this.gatherEpochData(epochNumber);
274
-
275
- const fromBlock = epochData.blocks[0].number;
276
- const toBlock = epochData.blocks.at(-1)!.number;
277
- this.log.verbose(`Creating proving job for epoch ${epochNumber} for block range ${fromBlock} to ${toBlock}`);
275
+ const fromCheckpoint = epochData.checkpoints[0].number;
276
+ const toCheckpoint = epochData.checkpoints.at(-1)!.number;
277
+ const fromBlock = epochData.checkpoints[0].blocks[0].number;
278
+ const toBlock = epochData.checkpoints.at(-1)!.blocks.at(-1)!.number;
279
+ this.log.verbose(
280
+ `Creating proving job for epoch ${epochNumber} for checkpoint range ${fromCheckpoint} to ${toCheckpoint} and block range ${fromBlock} to ${toBlock}`,
281
+ );
278
282
 
279
283
  // Fast forward world state to right before the target block and get a fork
280
284
  await this.worldState.syncImmediate(toBlock);
@@ -284,12 +288,12 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
284
288
  this.contractDataSource,
285
289
  this.dateProvider,
286
290
  this.telemetryClient,
291
+ this.log.getBindings(),
287
292
  );
288
293
 
289
294
  // Set deadline for this job to run. It will abort if it takes too long.
290
295
  const deadlineTs = getProofSubmissionDeadlineTimestamp(epochNumber, await this.getL1Constants());
291
296
  const deadline = new Date(Number(deadlineTs) * 1000);
292
-
293
297
  const job = this.doCreateEpochProvingJob(epochData, deadline, publicProcessorFactory, this.publisher, opts);
294
298
  this.jobs.set(job.getId(), job);
295
299
  return job;
@@ -300,30 +304,32 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
300
304
  return this.l2BlockSource.getL1Constants();
301
305
  }
302
306
 
303
- @trackSpan('ProverNode.gatherEpochData', epochNumber => ({ [Attributes.EPOCH_NUMBER]: Number(epochNumber) }))
304
- private async gatherEpochData(epochNumber: bigint): Promise<EpochProvingJobData> {
305
- const blocks = await this.gatherBlocks(epochNumber);
306
- const txArray = await this.gatherTxs(epochNumber, blocks);
307
+ @trackSpan('ProverNode.gatherEpochData', epochNumber => ({ [Attributes.EPOCH_NUMBER]: epochNumber }))
308
+ private async gatherEpochData(epochNumber: EpochNumber): Promise<EpochProvingJobData> {
309
+ const checkpoints = await this.gatherCheckpoints(epochNumber);
310
+ const txArray = await this.gatherTxs(epochNumber, checkpoints);
307
311
  const txs = new Map<string, Tx>(txArray.map(tx => [tx.getTxHash().toString(), tx]));
308
- const l1ToL2Messages = await this.gatherMessages(epochNumber, blocks);
309
- const previousBlockHeader = await this.gatherPreviousBlockHeader(epochNumber, blocks[0]);
310
- const [lastBlock] = await this.l2BlockSource.getPublishedBlocks(blocks.at(-1)!.number, 1);
311
- const attestations = lastBlock?.attestations ?? [];
312
+ const l1ToL2Messages = await this.gatherMessages(epochNumber, checkpoints);
313
+ const [firstBlock] = checkpoints[0].blocks;
314
+ const previousBlockHeader = await this.gatherPreviousBlockHeader(epochNumber, firstBlock.number - 1);
315
+ const [lastPublishedCheckpoint] = await this.l2BlockSource.getCheckpoints(checkpoints.at(-1)!.number, 1);
316
+ const attestations = lastPublishedCheckpoint?.attestations ?? [];
312
317
 
313
- return { blocks, txs, l1ToL2Messages, epochNumber, previousBlockHeader, attestations };
318
+ return { checkpoints, txs, l1ToL2Messages, epochNumber, previousBlockHeader, attestations };
314
319
  }
315
320
 
316
- private async gatherBlocks(epochNumber: bigint) {
317
- const blocks = await this.l2BlockSource.getBlocksForEpoch(epochNumber);
318
- if (blocks.length === 0) {
321
+ private async gatherCheckpoints(epochNumber: EpochNumber) {
322
+ const checkpoints = await this.l2BlockSource.getCheckpointsForEpoch(epochNumber);
323
+ if (checkpoints.length === 0) {
319
324
  throw new EmptyEpochError(epochNumber);
320
325
  }
321
- return blocks;
326
+ return checkpoints;
322
327
  }
323
328
 
324
- private async gatherTxs(epochNumber: bigint, blocks: L2Block[]) {
329
+ private async gatherTxs(epochNumber: EpochNumber, checkpoints: Checkpoint[]) {
325
330
  const deadline = new Date(this.dateProvider.now() + this.config.txGatheringTimeoutMs);
326
331
  const txProvider = this.p2pClient.getTxProvider();
332
+ const blocks = checkpoints.flatMap(checkpoint => checkpoint.blocks);
327
333
  const txsByBlock = await Promise.all(blocks.map(block => txProvider.getTxsForBlock(block, { deadline })));
328
334
  const txs = txsByBlock.map(({ txs }) => txs).flat();
329
335
  const missingTxs = txsByBlock.map(({ missingTxs }) => missingTxs).flat();
@@ -336,25 +342,24 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
336
342
  throw new Error(`Txs not found for epoch ${epochNumber}: ${missingTxs.map(hash => hash.toString()).join(', ')}`);
337
343
  }
338
344
 
339
- private async gatherMessages(epochNumber: bigint, blocks: L2Block[]) {
340
- const messages = await Promise.all(blocks.map(b => this.l1ToL2MessageSource.getL1ToL2Messages(b.number)));
345
+ private async gatherMessages(epochNumber: EpochNumber, checkpoints: Checkpoint[]) {
346
+ const messages = await Promise.all(checkpoints.map(c => this.l1ToL2MessageSource.getL1ToL2Messages(c.number)));
341
347
  const messageCount = sum(messages.map(m => m.length));
342
348
  this.log.verbose(`Gathered all ${messageCount} messages for epoch ${epochNumber}`, { epochNumber });
343
- const messagesByBlock: Record<number, Fr[]> = {};
344
- for (let i = 0; i < blocks.length; i++) {
345
- messagesByBlock[blocks[i].number] = messages[i];
349
+ const messagesByCheckpoint: Record<CheckpointNumber, Fr[]> = {};
350
+ for (let i = 0; i < checkpoints.length; i++) {
351
+ messagesByCheckpoint[checkpoints[i].number] = messages[i];
346
352
  }
347
- return messagesByBlock;
353
+ return messagesByCheckpoint;
348
354
  }
349
355
 
350
- private async gatherPreviousBlockHeader(epochNumber: bigint, initialBlock: L2Block) {
351
- const previousBlockNumber = initialBlock.number - 1;
356
+ private async gatherPreviousBlockHeader(epochNumber: EpochNumber, previousBlockNumber: number) {
352
357
  const header = await (previousBlockNumber === 0
353
358
  ? this.worldState.getCommitted().getInitialHeader()
354
- : this.l2BlockSource.getBlockHeader(previousBlockNumber));
359
+ : this.l2BlockSource.getBlockHeader(BlockNumber(previousBlockNumber)));
355
360
 
356
361
  if (!header) {
357
- throw new Error(`Previous block header ${initialBlock.number} not found for proving epoch ${epochNumber}`);
362
+ throw new Error(`Previous block header ${previousBlockNumber} not found for proving epoch ${epochNumber}`);
358
363
  }
359
364
 
360
365
  this.log.verbose(`Gathered previous block header ${header.getBlockNumber()} for epoch ${epochNumber}`);
@@ -380,6 +385,7 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
380
385
  this.jobMetrics,
381
386
  deadline,
382
387
  { parallelBlockLimit, skipSubmitProof: proverNodeDisableProofPublish, ...opts },
388
+ this.log.getBindings(),
383
389
  );
384
390
  }
385
391
 
@@ -391,7 +397,7 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
391
397
  private validateConfig() {
392
398
  if (
393
399
  this.config.proverNodeFailedEpochStore &&
394
- (!this.config.dataDirectory || !this.config.l1ChainId || !this.config.rollupVersion)
400
+ (!this.config.dataDirectory || !this.config.l1ChainId || this.config.rollupVersion === undefined)
395
401
  ) {
396
402
  this.log.warn(
397
403
  `Invalid prover-node config (missing dataDirectory, l1ChainId, or rollupVersion)`,
@@ -405,7 +411,7 @@ export class ProverNode implements EpochMonitorHandler, ProverNodeApi, Traceable
405
411
  }
406
412
 
407
413
  class EmptyEpochError extends Error {
408
- constructor(epochNumber: bigint) {
414
+ constructor(epochNumber: EpochNumber) {
409
415
  super(`No blocks found for epoch ${epochNumber}`);
410
416
  this.name = 'EmptyEpochError';
411
417
  }
@@ -1,4 +1,7 @@
1
- import type { L1TxUtils, PublisherManager, RollupContract } from '@aztec/ethereum';
1
+ import type { RollupContract } from '@aztec/ethereum/contracts';
2
+ import type { L1TxUtils } from '@aztec/ethereum/l1-tx-utils';
3
+ import type { PublisherManager } from '@aztec/ethereum/publisher-manager';
4
+ import type { LoggerBindings } from '@aztec/foundation/log';
2
5
  import type { PublisherConfig, TxSenderConfig } from '@aztec/sequencer-client';
3
6
  import type { TelemetryClient } from '@aztec/telemetry-client';
4
7
 
@@ -12,6 +15,7 @@ export class ProverPublisherFactory {
12
15
  publisherManager: PublisherManager<L1TxUtils>;
13
16
  telemetry?: TelemetryClient;
14
17
  },
18
+ private bindings?: LoggerBindings,
15
19
  ) {}
16
20
 
17
21
  public async start() {
@@ -28,10 +32,14 @@ export class ProverPublisherFactory {
28
32
  */
29
33
  public async create(): Promise<ProverNodePublisher> {
30
34
  const l1Publisher = await this.deps.publisherManager.getAvailablePublisher();
31
- return new ProverNodePublisher(this.config, {
32
- rollupContract: this.deps.rollupContract,
33
- l1TxUtils: l1Publisher,
34
- telemetry: this.deps.telemetry,
35
- });
35
+ return new ProverNodePublisher(
36
+ this.config,
37
+ {
38
+ rollupContract: this.deps.rollupContract,
39
+ l1TxUtils: l1Publisher,
40
+ telemetry: this.deps.telemetry,
41
+ },
42
+ this.bindings,
43
+ );
36
44
  }
37
45
  }