@aztec/epoch-cache 0.0.1-commit.2448fdb → 0.0.1-commit.2606882

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.
@@ -1,6 +1,7 @@
1
1
  import { createEthereumChain } from '@aztec/ethereum/chain';
2
2
  import { makeL1HttpTransport } from '@aztec/ethereum/client';
3
3
  import { NoCommitteeError, RollupContract } from '@aztec/ethereum/contracts';
4
+ import { getFinalizedL1Block } from '@aztec/ethereum/queries';
4
5
  import { EpochNumber, SlotNumber } from '@aztec/foundation/branded-types';
5
6
  import { EthAddress } from '@aztec/foundation/eth-address';
6
7
  import { type Logger, createLogger } from '@aztec/foundation/log';
@@ -10,19 +11,24 @@ import {
10
11
  getEpochAtSlot,
11
12
  getEpochNumberAtTimestamp,
12
13
  getNextL1SlotTimestamp,
14
+ getSlotAtNextL1Block,
13
15
  getSlotAtTimestamp,
14
16
  getSlotRangeForEpoch,
17
+ getStartTimestampForEpoch,
15
18
  getTimestampForSlot,
16
- getTimestampRangeForEpoch,
17
19
  } from '@aztec/stdlib/epoch-helpers';
18
20
 
19
21
  import { createPublicClient, encodeAbiParameters, keccak256 } from 'viem';
20
22
 
21
23
  import { type EpochCacheConfig, getEpochCacheConfigEnvVars } from './config.js';
22
24
 
25
+ /** The proposer pipelines by building one slot ahead. */
26
+ export const PROPOSER_PIPELINING_SLOT_OFFSET = 1;
27
+
28
+ /** Flat return type for compound epoch/slot getters. */
23
29
  export type EpochAndSlot = {
24
- epoch: EpochNumber;
25
30
  slot: SlotNumber;
31
+ epoch: EpochNumber;
26
32
  ts: bigint;
27
33
  };
28
34
 
@@ -36,13 +42,38 @@ export type EpochCommitteeInfo = {
36
42
 
37
43
  export type SlotTag = 'now' | 'next' | SlotNumber;
38
44
 
45
+ /** Minimal L1 block info used for cache provenance. */
46
+ type L1BlockInfo = { number: bigint; hash: `0x${string}`; timestamp: bigint };
47
+
48
+ /** Resolved cache entry with L1 provenance metadata. */
49
+ type CachedEpochEntry = {
50
+ data: EpochCommitteeInfo;
51
+ /** L1 block number at which the committee data was originally queried. */
52
+ lastQueryL1BlockNumber: bigint;
53
+ /** L1 block hash at which the committee data was originally queried. Used to detect reorgs. */
54
+ lastQueryL1BlockHash: `0x${string}`;
55
+ /** Latest L1 block timestamp at the time of the most recent refresh (full fetch or lightweight check). */
56
+ lastRefreshL1Timestamp: bigint;
57
+ /** Whether the epoch's sampling data falls within finalized L1 history. */
58
+ finalized: boolean;
59
+ };
60
+
39
61
  export interface EpochCacheInterface {
40
62
  getCommittee(slot: SlotTag | undefined): Promise<EpochCommitteeInfo>;
63
+ getSlotNow(): SlotNumber;
64
+ getTargetSlot(): SlotNumber;
65
+ getEpochNow(): EpochNumber;
66
+ getTargetEpoch(): EpochNumber;
41
67
  getEpochAndSlotNow(): EpochAndSlot & { nowMs: bigint };
42
- getEpochAndSlotInNextL1Slot(): EpochAndSlot & { now: bigint };
68
+ getEpochAndSlotInNextL1Slot(): EpochAndSlot & { nowSeconds: bigint };
69
+ /** Returns epoch/slot info for the next L1 slot with pipeline offset applied. */
70
+ getTargetEpochAndSlotInNextL1Slot(): EpochAndSlot & { nowSeconds: bigint };
71
+ isEscapeHatchOpen(epoch: EpochNumber): Promise<boolean>;
72
+ isEscapeHatchOpenAtSlot(slot: SlotTag): Promise<boolean>;
43
73
  getProposerIndexEncoding(epoch: EpochNumber, slot: SlotNumber, seed: bigint): `0x${string}`;
44
74
  computeProposerIndex(slot: SlotNumber, epoch: EpochNumber, seed: bigint, size: bigint): bigint;
45
75
  getCurrentAndNextSlot(): { currentSlot: SlotNumber; nextSlot: SlotNumber };
76
+ getTargetAndNextSlot(): { targetSlot: SlotNumber; nextSlot: SlotNumber };
46
77
  getProposerAttesterAddressInSlot(slot: SlotNumber): Promise<EthAddress | undefined>;
47
78
  getRegisteredValidators(): Promise<EthAddress[]>;
48
79
  isInCommittee(slot: SlotTag, validator: EthAddress): Promise<boolean>;
@@ -60,8 +91,11 @@ export interface EpochCacheInterface {
60
91
  * Note: This class is very dependent on the system clock being in sync.
61
92
  */
62
93
  export class EpochCache implements EpochCacheInterface {
63
- // eslint-disable-next-line aztec-custom/no-non-primitive-in-collections
64
- protected cache: Map<EpochNumber, EpochCommitteeInfo> = new Map();
94
+ /**
95
+ * Single map holding both resolved entries and in-flight promises.
96
+ * A `Promise` value means a fetch is in progress; concurrent callers await it.
97
+ */
98
+ protected cache: Map<EpochNumber, CachedEpochEntry | Promise<CachedEpochEntry>> = new Map();
65
99
  private allValidators: Set<string> = new Set();
66
100
  private lastValidatorRefresh = 0;
67
101
  private readonly log: Logger = createLogger('epoch-cache');
@@ -136,13 +170,34 @@ export class EpochCache implements EpochCacheInterface {
136
170
  rollupManaLimit: Number(rollupManaLimit),
137
171
  };
138
172
 
139
- return new EpochCache(rollup, l1RollupConstants, deps.dateProvider);
173
+ return new EpochCache(rollup, l1RollupConstants, deps.dateProvider, {
174
+ cacheSize: 12,
175
+ validatorRefreshIntervalSeconds: 60,
176
+ });
140
177
  }
141
178
 
142
179
  public getL1Constants(): L1RollupConstants {
143
180
  return this.l1constants;
144
181
  }
145
182
 
183
+ public getSlotNow(): SlotNumber {
184
+ return this.getEpochAndSlotNow().slot;
185
+ }
186
+
187
+ public getTargetSlot(): SlotNumber {
188
+ const slotNow = this.getSlotNow();
189
+ const offset = PROPOSER_PIPELINING_SLOT_OFFSET;
190
+ return SlotNumber(slotNow + offset);
191
+ }
192
+
193
+ public getEpochNow(): EpochNumber {
194
+ return this.getEpochAndSlotNow().epoch;
195
+ }
196
+
197
+ public getTargetEpoch(): EpochNumber {
198
+ return getEpochAtSlot(this.getTargetSlot(), this.l1constants);
199
+ }
200
+
146
201
  public getEpochAndSlotNow(): EpochAndSlot & { nowMs: bigint } {
147
202
  const nowMs = BigInt(this.dateProvider.now());
148
203
  const nowSeconds = nowMs / 1000n;
@@ -150,23 +205,29 @@ export class EpochCache implements EpochCacheInterface {
150
205
  }
151
206
 
152
207
  private getEpochAndSlotAtSlot(slot: SlotNumber): EpochAndSlot {
153
- const epoch = getEpochAtSlot(slot, this.l1constants);
154
- const ts = getTimestampRangeForEpoch(epoch, this.l1constants)[0];
155
- return { epoch, ts, slot };
208
+ return this.getEpochAndSlotAtTimestamp(getTimestampForSlot(slot, this.l1constants));
209
+ }
210
+
211
+ public getEpochAndSlotInNextL1Slot(): EpochAndSlot & { nowSeconds: bigint } {
212
+ const nowSeconds = this.dateProvider.nowInSeconds();
213
+ const nextSlotTs = getNextL1SlotTimestamp(nowSeconds, this.l1constants);
214
+ return { ...this.getEpochAndSlotAtTimestamp(nextSlotTs), nowSeconds: BigInt(nowSeconds) };
156
215
  }
157
216
 
158
- public getEpochAndSlotInNextL1Slot(): EpochAndSlot & { now: bigint } {
159
- const now = BigInt(this.dateProvider.nowInSeconds());
160
- const nextSlotTs = getNextL1SlotTimestamp(Number(now), this.l1constants);
161
- return { ...this.getEpochAndSlotAtTimestamp(nextSlotTs), now };
217
+ public getTargetEpochAndSlotInNextL1Slot(): EpochAndSlot & { nowSeconds: bigint } {
218
+ const result = this.getEpochAndSlotInNextL1Slot();
219
+ const offset = PROPOSER_PIPELINING_SLOT_OFFSET;
220
+ const targetSlot = SlotNumber(result.slot + offset);
221
+ return { ...result, slot: targetSlot, epoch: getEpochAtSlot(targetSlot, this.l1constants) };
162
222
  }
163
223
 
164
224
  private getEpochAndSlotAtTimestamp(ts: bigint): EpochAndSlot {
165
225
  const slot = getSlotAtTimestamp(ts, this.l1constants);
226
+ const epoch = getEpochNumberAtTimestamp(ts, this.l1constants);
166
227
  return {
167
- epoch: getEpochNumberAtTimestamp(ts, this.l1constants),
168
- ts: getTimestampForSlot(slot, this.l1constants),
169
228
  slot,
229
+ epoch,
230
+ ts: getTimestampForSlot(slot, this.l1constants),
170
231
  };
171
232
  }
172
233
 
@@ -175,17 +236,8 @@ export class EpochCache implements EpochCacheInterface {
175
236
  return this.getCommittee(startSlot);
176
237
  }
177
238
 
178
- /**
179
- * Returns whether the escape hatch is open for the given epoch.
180
- *
181
- * Uses the already-cached EpochCommitteeInfo when available. If not cached, it will fetch
182
- * the epoch committee info (which includes the escape hatch flag) and return it.
183
- */
239
+ /** Returns whether the escape hatch is open for the given epoch. */
184
240
  public async isEscapeHatchOpen(epoch: EpochNumber): Promise<boolean> {
185
- const cached = this.cache.get(epoch);
186
- if (cached) {
187
- return cached.isEscapeHatchOpen;
188
- }
189
241
  const info = await this.getCommitteeForEpoch(epoch);
190
242
  return info.isEscapeHatchOpen;
191
243
  }
@@ -199,7 +251,7 @@ export class EpochCache implements EpochCacheInterface {
199
251
  public async isEscapeHatchOpenAtSlot(slot: SlotTag = 'now'): Promise<boolean> {
200
252
  const epoch =
201
253
  slot === 'now'
202
- ? this.getEpochAndSlotNow().epoch
254
+ ? this.getEpochNow()
203
255
  : slot === 'next'
204
256
  ? this.getEpochAndSlotInNextL1Slot().epoch
205
257
  : getEpochAtSlot(slot, this.l1constants);
@@ -208,33 +260,52 @@ export class EpochCache implements EpochCacheInterface {
208
260
  }
209
261
 
210
262
  /**
211
- * Get the current validator set
212
- * @param nextSlot - If true, get the validator set for the next slot.
213
- * @returns The current validator set.
263
+ * Get the current validator set.
264
+ *
265
+ * Returns cached data if the entry is finalized or still fresh (queried less than one
266
+ * Ethereum slot ago). Stale non-finalized entries are re-queried, and concurrent callers
267
+ * coalesce on the same in-flight promise so the L1 query happens only once.
214
268
  */
215
269
  public async getCommittee(slot: SlotTag = 'now'): Promise<EpochCommitteeInfo> {
216
270
  const { epoch, ts } = this.getEpochAndTimestamp(slot);
217
271
 
218
- if (this.cache.has(epoch)) {
219
- return this.cache.get(epoch)!;
272
+ const cached = this.cache.get(epoch);
273
+
274
+ // In-flight promise: another caller is already fetching this epoch — just await it.
275
+ if (cached instanceof Promise) {
276
+ return (await cached).data;
220
277
  }
221
278
 
222
- const epochData = await this.computeCommittee({ epoch, ts });
223
- // If the committee size is 0 or undefined, then do not cache
224
- if (!epochData.committee || epochData.committee.length === 0) {
225
- return epochData;
279
+ // Resolved entry: return it if finalized or still fresh.
280
+ if (cached && (cached.finalized || !this.isStale(cached))) {
281
+ return cached.data;
226
282
  }
227
- this.cache.set(epoch, epochData);
228
283
 
229
- const toPurge = Array.from(this.cache.keys())
230
- .sort((a, b) => Number(b - a))
231
- .slice(this.config.cacheSize);
232
- toPurge.forEach(key => this.cache.delete(key));
284
+ // Stale non-finalized entry: do a lightweight refresh first (check block hash + finalized ts).
285
+ // Only fall back to a full re-fetch if the L1 block was reorged.
286
+ if (cached) {
287
+ const promise = this.refreshStaleEntry(cached, epoch, ts);
288
+ this.cache.set(epoch, promise);
289
+ try {
290
+ return (await promise).data;
291
+ } catch (err) {
292
+ this.cache.set(epoch, cached);
293
+ throw err;
294
+ }
295
+ }
233
296
 
234
- return epochData;
297
+ // No entry at all: full fetch.
298
+ const promise = this.fetchAndCache(epoch, ts);
299
+ this.cache.set(epoch, promise);
300
+ try {
301
+ return (await promise).data;
302
+ } catch (err) {
303
+ this.cache.delete(epoch);
304
+ throw err;
305
+ }
235
306
  }
236
307
 
237
- private getEpochAndTimestamp(slot: SlotTag = 'now') {
308
+ private getEpochAndTimestamp(slot: SlotTag = 'now'): { epoch: EpochNumber; ts: bigint } {
238
309
  if (slot === 'now') {
239
310
  return this.getEpochAndSlotNow();
240
311
  } else if (slot === 'next') {
@@ -244,22 +315,140 @@ export class EpochCache implements EpochCacheInterface {
244
315
  }
245
316
  }
246
317
 
247
- private async computeCommittee(when: { epoch: EpochNumber; ts: bigint }): Promise<EpochCommitteeInfo> {
248
- const { ts, epoch } = when;
249
- const [committee, seedBuffer, l1Timestamp, isEscapeHatchOpen] = await Promise.all([
318
+ /** Evicts oldest cache entries (resolved or in-flight) beyond cacheSize. */
319
+ private purgeCache(): void {
320
+ if (this.cache.size <= this.config.cacheSize) {
321
+ return;
322
+ }
323
+ const toPurge = Array.from(this.cache.keys())
324
+ .sort((a, b) => Number(b - a))
325
+ .slice(this.config.cacheSize);
326
+ toPurge.forEach(key => this.cache.delete(key));
327
+ }
328
+
329
+ /** Returns true if a non-finalized cache entry is older than one Ethereum slot. */
330
+ private isStale(entry: CachedEpochEntry): boolean {
331
+ const nowSeconds = BigInt(this.dateProvider.nowInSeconds());
332
+ return nowSeconds - entry.lastRefreshL1Timestamp >= BigInt(this.l1constants.ethereumSlotDuration);
333
+ }
334
+
335
+ /** Whether a cached epoch entry has been marked as finalized. Returns undefined if not cached or still in-flight. */
336
+ public isFinalized(epoch: EpochNumber): boolean | undefined {
337
+ const entry = this.cache.get(epoch);
338
+ if (!entry || entry instanceof Promise) {
339
+ return undefined;
340
+ }
341
+ return entry.finalized;
342
+ }
343
+
344
+ /** Returns the latest L1 timestamp stored in the cached entry. Undefined if not cached or in-flight. */
345
+ public getCachedLastRefreshL1Timestamp(epoch: EpochNumber): bigint | undefined {
346
+ const entry = this.cache.get(epoch);
347
+ if (!entry || entry instanceof Promise) {
348
+ return undefined;
349
+ }
350
+ return entry.lastRefreshL1Timestamp;
351
+ }
352
+
353
+ /** Computes the sampling timestamp for an epoch's committee data. */
354
+ private getSamplingTimestamp(epoch: EpochNumber): bigint {
355
+ const { lagInEpochsForRandao, epochDuration, slotDuration } = this.l1constants;
356
+ const epochStartTs = getStartTimestampForEpoch(epoch, this.l1constants);
357
+ return epochStartTs - BigInt(lagInEpochsForRandao) * BigInt(epochDuration) * BigInt(slotDuration);
358
+ }
359
+
360
+ /**
361
+ * Lightweight refresh for a stale non-finalized entry. Queries only the block hash at
362
+ * the original block number and the finalized block timestamp — avoids the expensive
363
+ * getCommitteeAt and getSampleSeedAt calls on the rollup contract.
364
+ *
365
+ * If the block hash still matches (no L1 reorg), we keep the existing data and just
366
+ * update the provenance timestamp. If the finalized block has caught up, we promote the
367
+ * entry to finalized. If there was a reorg (hash mismatch), we fall back to a full fetch.
368
+ */
369
+ private async refreshStaleEntry(stale: CachedEpochEntry, epoch: EpochNumber, ts: bigint): Promise<CachedEpochEntry> {
370
+ const [blockAtOriginal, l1FinalizedBlock, latestBlock] = await Promise.all([
371
+ this.rollup.client.getBlock({ blockNumber: stale.lastQueryL1BlockNumber, includeTransactions: false }),
372
+ getFinalizedL1Block(this.rollup.client),
373
+ this.rollup.client.getBlock({ includeTransactions: false }),
374
+ ]);
375
+
376
+ if (blockAtOriginal.hash === stale.lastQueryL1BlockHash) {
377
+ // No reorg: the data is still valid. Check if we can now mark it as finalized.
378
+ const samplingTs = this.getSamplingTimestamp(epoch);
379
+ const finalized =
380
+ !!(stale.data.committee && stale.data.committee.length > 0) &&
381
+ l1FinalizedBlock !== undefined &&
382
+ samplingTs <= l1FinalizedBlock.timestamp;
383
+
384
+ const refreshed: CachedEpochEntry = {
385
+ ...stale,
386
+ lastRefreshL1Timestamp: latestBlock.timestamp,
387
+ finalized,
388
+ };
389
+ this.cache.set(epoch, refreshed);
390
+ return refreshed;
391
+ }
392
+
393
+ // Reorg detected: block hash mismatch. Do a full re-fetch.
394
+ // Pass the already-fetched block timestamps to avoid redundant queries.
395
+ this.log.warn(`L1 reorg detected for epoch ${epoch}: block ${stale.lastQueryL1BlockNumber} hash changed`, {
396
+ epoch,
397
+ expectedHash: stale.lastQueryL1BlockHash,
398
+ actualHash: blockAtOriginal.hash,
399
+ });
400
+ return this.fetchAndCache(epoch, ts, { latestBlock, finalizedBlock: l1FinalizedBlock });
401
+ }
402
+
403
+ /**
404
+ * Fetches committee data from L1, determines finalization status, and stores in the cache.
405
+ *
406
+ * Uses `lagInEpochsForRandao` (the binding constraint, always <= lagInEpochsForValidatorSet)
407
+ * and computes the sampling timestamp from the epoch start to match the L1 contract's logic.
408
+ *
409
+ * When called from refreshStaleEntry after a reorg, the latest and finalized blocks are
410
+ * passed in to avoid redundant L1 queries.
411
+ */
412
+ private async fetchAndCache(
413
+ epoch: EpochNumber,
414
+ ts: bigint,
415
+ prefetched?: { latestBlock: L1BlockInfo; finalizedBlock: { timestamp: bigint } | undefined },
416
+ ): Promise<CachedEpochEntry> {
417
+ const [committee, seedBuffer, latestBlock, finalizedBlock, isEscapeHatchOpen] = await Promise.all([
250
418
  this.rollup.getCommitteeAt(ts),
251
419
  this.rollup.getSampleSeedAt(ts),
252
- this.rollup.client.getBlock({ includeTransactions: false }).then(b => b.timestamp),
420
+ prefetched?.latestBlock ?? this.rollup.client.getBlock({ includeTransactions: false }),
421
+ prefetched !== undefined ? prefetched.finalizedBlock : getFinalizedL1Block(this.rollup.client),
253
422
  this.rollup.isEscapeHatchOpen(epoch),
254
423
  ]);
255
- const { lagInEpochsForValidatorSet, epochDuration, slotDuration } = this.l1constants;
256
- const sub = BigInt(lagInEpochsForValidatorSet) * BigInt(epochDuration) * BigInt(slotDuration);
257
- if (ts - sub > l1Timestamp) {
424
+
425
+ const samplingTs = this.getSamplingTimestamp(epoch);
426
+
427
+ if (samplingTs > latestBlock.timestamp) {
258
428
  throw new Error(
259
- `Cannot query committee for future epoch ${epoch} with timestamp ${ts} (current L1 time is ${l1Timestamp}). Check your Ethereum node is synced.`,
429
+ `Cannot query committee for future epoch ${epoch}: ` +
430
+ `sampling timestamp ${samplingTs} is beyond latest L1 block at ${latestBlock.timestamp}. ` +
431
+ `Check your Ethereum node is synced.`,
260
432
  );
261
433
  }
262
- return { committee, seed: seedBuffer.toBigInt(), epoch, isEscapeHatchOpen };
434
+
435
+ // Empty committees are never marked finalized so they always get re-queried after TTL.
436
+ // If L1 has no finalized block yet (devnet startup), entries stay unfinalized.
437
+ const hasCommittee = !!(committee && committee.length > 0);
438
+ const finalized = hasCommittee && finalizedBlock !== undefined && samplingTs <= finalizedBlock.timestamp;
439
+ const data: EpochCommitteeInfo = { committee, seed: seedBuffer.toBigInt(), epoch, isEscapeHatchOpen };
440
+ const entry: CachedEpochEntry = {
441
+ data,
442
+ lastQueryL1BlockNumber: latestBlock.number!,
443
+ lastQueryL1BlockHash: latestBlock.hash!,
444
+ lastRefreshL1Timestamp: latestBlock.timestamp,
445
+ finalized,
446
+ };
447
+
448
+ this.cache.set(epoch, entry);
449
+ this.purgeCache();
450
+
451
+ return entry;
263
452
  }
264
453
 
265
454
  /**
@@ -284,17 +473,31 @@ export class EpochCache implements EpochCacheInterface {
284
473
  return BigInt(keccak256(this.getProposerIndexEncoding(epoch, slot, seed))) % size;
285
474
  }
286
475
 
287
- /** Returns the current and next L2 slot numbers. */
476
+ /** Returns the current and next L2 slot in next eth L1 Slot. */
288
477
  public getCurrentAndNextSlot(): { currentSlot: SlotNumber; nextSlot: SlotNumber } {
289
- const current = this.getEpochAndSlotNow();
478
+ const currentSlot = this.getSlotNow();
290
479
  const next = this.getEpochAndSlotInNextL1Slot();
291
480
 
292
481
  return {
293
- currentSlot: current.slot,
482
+ currentSlot,
294
483
  nextSlot: next.slot,
295
484
  };
296
485
  }
297
486
 
487
+ /** Returns the target and next L2 slot in the next L1 slot. */
488
+ public getTargetAndNextSlot(): { targetSlot: SlotNumber; nextSlot: SlotNumber } {
489
+ const nowSeconds = BigInt(this.dateProvider.nowInSeconds());
490
+ const offset = PROPOSER_PIPELINING_SLOT_OFFSET;
491
+
492
+ const currentSlot = getSlotAtTimestamp(nowSeconds, this.l1constants);
493
+ const targetSlot = SlotNumber(currentSlot + offset);
494
+
495
+ const nextL2SlotOnL1 = getSlotAtNextL1Block(nowSeconds, this.l1constants);
496
+ const nextSlot = SlotNumber(nextL2SlotOnL1 + offset);
497
+
498
+ return { targetSlot, nextSlot };
499
+ }
500
+
298
501
  /**
299
502
  * Get the proposer attester address in the given L2 slot
300
503
  * @returns The proposer attester address. If the committee does not exist, we throw a NoCommitteeError.
@@ -3,7 +3,13 @@ import { EthAddress } from '@aztec/foundation/eth-address';
3
3
  import type { L1RollupConstants } from '@aztec/stdlib/epoch-helpers';
4
4
  import { getEpochAtSlot, getSlotAtTimestamp, getTimestampRangeForEpoch } from '@aztec/stdlib/epoch-helpers';
5
5
 
6
- import type { EpochAndSlot, EpochCacheInterface, EpochCommitteeInfo, SlotTag } from '../epoch_cache.js';
6
+ import {
7
+ type EpochAndSlot,
8
+ type EpochCacheInterface,
9
+ type EpochCommitteeInfo,
10
+ PROPOSER_PIPELINING_SLOT_OFFSET,
11
+ type SlotTag,
12
+ } from '../epoch_cache.js';
7
13
 
8
14
  /** Default L1 constants for testing. */
9
15
  const DEFAULT_L1_CONSTANTS: L1RollupConstants = {
@@ -114,19 +120,52 @@ export class TestEpochCache implements EpochCacheInterface {
114
120
  });
115
121
  }
116
122
 
123
+ getSlotNow(): SlotNumber {
124
+ return this.currentSlot;
125
+ }
126
+
127
+ getTargetSlot(): SlotNumber {
128
+ return SlotNumber(this.currentSlot + PROPOSER_PIPELINING_SLOT_OFFSET);
129
+ }
130
+
131
+ getEpochNow(): EpochNumber {
132
+ return getEpochAtSlot(this.currentSlot, this.l1Constants);
133
+ }
134
+
135
+ getTargetEpoch(): EpochNumber {
136
+ return getEpochAtSlot(this.getTargetSlot(), this.l1Constants);
137
+ }
138
+
117
139
  getEpochAndSlotNow(): EpochAndSlot & { nowMs: bigint } {
118
- const epoch = getEpochAtSlot(this.currentSlot, this.l1Constants);
119
- const ts = getTimestampRangeForEpoch(epoch, this.l1Constants)[0];
120
- return { epoch, slot: this.currentSlot, ts, nowMs: ts * 1000n };
140
+ const epochNow = getEpochAtSlot(this.currentSlot, this.l1Constants);
141
+ const ts = getTimestampRangeForEpoch(epochNow, this.l1Constants)[0];
142
+ return {
143
+ epoch: epochNow,
144
+ slot: this.currentSlot,
145
+ ts,
146
+ nowMs: ts * 1000n,
147
+ };
121
148
  }
122
149
 
123
- getEpochAndSlotInNextL1Slot(): EpochAndSlot & { now: bigint } {
124
- const now = getTimestampRangeForEpoch(getEpochAtSlot(this.currentSlot, this.l1Constants), this.l1Constants)[0];
125
- const nextSlotTs = now + BigInt(this.l1Constants.ethereumSlotDuration);
150
+ getEpochAndSlotInNextL1Slot(): EpochAndSlot & { nowSeconds: bigint } {
151
+ const nowTs = getTimestampRangeForEpoch(getEpochAtSlot(this.currentSlot, this.l1Constants), this.l1Constants)[0];
152
+ const nextSlotTs = nowTs + BigInt(this.l1Constants.ethereumSlotDuration);
126
153
  const nextSlot = getSlotAtTimestamp(nextSlotTs, this.l1Constants);
127
- const epoch = getEpochAtSlot(nextSlot, this.l1Constants);
128
- const ts = getTimestampRangeForEpoch(epoch, this.l1Constants)[0];
129
- return { epoch, slot: nextSlot, ts, now };
154
+ const epochNow = getEpochAtSlot(nextSlot, this.l1Constants);
155
+ const ts = getTimestampRangeForEpoch(epochNow, this.l1Constants)[0];
156
+ return {
157
+ epoch: epochNow,
158
+ slot: nextSlot,
159
+ ts,
160
+ nowSeconds: nowTs,
161
+ };
162
+ }
163
+
164
+ getTargetEpochAndSlotInNextL1Slot(): EpochAndSlot & { nowSeconds: bigint } {
165
+ const result = this.getEpochAndSlotInNextL1Slot();
166
+ const offset = PROPOSER_PIPELINING_SLOT_OFFSET;
167
+ const targetSlot = SlotNumber(result.slot + offset);
168
+ return { ...result, slot: targetSlot, epoch: getEpochAtSlot(targetSlot, this.l1Constants) };
130
169
  }
131
170
 
132
171
  getProposerIndexEncoding(epoch: EpochNumber, slot: SlotNumber, seed: bigint): `0x${string}` {
@@ -142,9 +181,22 @@ export class TestEpochCache implements EpochCacheInterface {
142
181
  }
143
182
 
144
183
  getCurrentAndNextSlot(): { currentSlot: SlotNumber; nextSlot: SlotNumber } {
184
+ const currentSlot = this.getSlotNow();
185
+ const next = this.getEpochAndSlotInNextL1Slot();
186
+
187
+ return {
188
+ currentSlot,
189
+ nextSlot: next.slot,
190
+ };
191
+ }
192
+
193
+ getTargetAndNextSlot(): { targetSlot: SlotNumber; nextSlot: SlotNumber } {
194
+ const targetSlot = this.getTargetSlot();
195
+ const next = this.getTargetEpochAndSlotInNextL1Slot();
196
+
145
197
  return {
146
- currentSlot: this.currentSlot,
147
- nextSlot: SlotNumber(this.currentSlot + 1),
198
+ targetSlot,
199
+ nextSlot: next.slot,
148
200
  };
149
201
  }
150
202
 
@@ -165,6 +217,10 @@ export class TestEpochCache implements EpochCacheInterface {
165
217
  return Promise.resolve(validators.filter(v => committeeSet.has(v.toString())));
166
218
  }
167
219
 
220
+ isEscapeHatchOpen(_epoch: EpochNumber): Promise<boolean> {
221
+ return Promise.resolve(this.escapeHatchOpen);
222
+ }
223
+
168
224
  isEscapeHatchOpenAtSlot(_slot?: SlotTag): Promise<boolean> {
169
225
  return Promise.resolve(this.escapeHatchOpen);
170
226
  }