@aztec/archiver 0.0.1-commit.d431d1c → 0.0.1-commit.d58ff9d0

package/dest/modules/data_source_base.js

@@ -1,291 +1,322 @@
-import { BlockNumber, CheckpointNumber } from '@aztec/foundation/branded-types';
-import { isDefined } from '@aztec/foundation/types';
-import { CheckpointedL2Block, CommitteeAttestation } from '@aztec/stdlib/block';
+import { INITIAL_L2_BLOCK_NUM } from '@aztec/constants';
+import { BlockNumber, CheckpointNumber, IndexWithinCheckpoint } from '@aztec/foundation/branded-types';
+import { Body, L2Block } from '@aztec/stdlib/block';
 import { Checkpoint, PublishedCheckpoint } from '@aztec/stdlib/checkpoint';
-import { getSlotRangeForEpoch } from '@aztec/stdlib/epoch-helpers';
+import { getEpochAtSlot, getEpochNumberAtTimestamp, getLastL1SlotTimestampForL2Slot, getProofSubmissionDeadlineEpoch, getSlotRangeForEpoch } from '@aztec/stdlib/epoch-helpers';
+import { AppendOnlyTreeSnapshot } from '@aztec/stdlib/trees';
 /**
- * Abstract base class implementing ArchiverDataSource using a KVArchiverDataStore.
- * Provides implementations for all store-delegating methods and declares abstract methods
- * for L1-dependent functionality that subclasses must implement.
+ * Abstract base class implementing ArchiverDataSource using a bundle of archiver substores.
+ * Provides implementations for all read-side methods and declares abstract methods for
+ * L1-dependent functionality that subclasses must implement.
  */ export class ArchiverDataSourceBase {
-    store;
+    stores;
     l1Constants;
-    constructor(store, l1Constants){
-        this.store = store;
+    /** The injected genesis block header. */ initialHeader;
+    /** Precomputed hash of the initial header, exposed via {@link getGenesisBlockHash}. */ initialBlockHash;
+    /** Archive root after block 0 was appended — read from L1 (`Rollup.getGenesisArchiveTreeRoot`). */ genesisArchiveRoot;
+    /** Memoized synthetic genesis block — callers rely on referential identity for caching. */ genesisBlock;
+    /** Memoized synthetic genesis block data — kept consistent with {@link genesisBlock}. */ genesisBlockData;
+    constructor(stores, l1Constants, initialHeader, initialBlockHash, genesisArchiveRoot){
+        this.stores = stores;
         this.l1Constants = l1Constants;
+        this.initialHeader = initialHeader;
+        this.initialBlockHash = initialBlockHash;
+        this.genesisArchiveRoot = genesisArchiveRoot;
+        const genesisArchive = new AppendOnlyTreeSnapshot(genesisArchiveRoot, 1);
+        this.genesisBlock = new L2Block(genesisArchive, initialHeader, Body.empty(), CheckpointNumber.ZERO, IndexWithinCheckpoint(0));
+        this.genesisBlockData = {
+            header: initialHeader,
+            archive: genesisArchive,
+            blockHash: initialBlockHash,
+            checkpointNumber: CheckpointNumber.ZERO,
+            indexWithinCheckpoint: IndexWithinCheckpoint(0)
+        };
     }
-    getCheckpointNumber() {
-        return this.store.getSynchedCheckpointNumber();
-    }
-    getSynchedCheckpointNumber() {
-        return this.store.getSynchedCheckpointNumber();
+    /** Returns the precomputed hash of the genesis block header. */ getGenesisBlockHash() {
+        return this.initialBlockHash;
     }
-    getProvenCheckpointNumber() {
-        return this.store.getProvenCheckpointNumber();
+    /** Returns the synthetic genesis L2Block (memoized — same instance across calls). */ getGenesisBlock() {
+        return this.genesisBlock;
     }
-    getBlockNumber() {
-        return this.store.getLatestBlockNumber();
+    /** Returns genesis block data (memoized — same instance across calls). */ getGenesisBlockData() {
+        return this.genesisBlockData;
     }
-    getProvenBlockNumber() {
-        return this.store.getProvenBlockNumber();
+    /**
+   * Type guard distinguishing the genesis sentinel from a {@link ResolvedBlockQuery}.
+   * `resolveBlockQuery` already rewrites every genesis-matching shape to the sentinel,
+   * so callers only need this single sync check.
+   */ isGenesisBlockQuery(query) {
+        return 'genesis' in query;
     }
-    async getBlockHeader(number) {
-        const blockNumber = number === 'latest' ? await this.store.getLatestBlockNumber() : number;
-        if (blockNumber === 0) {
-            return undefined;
+    async isPruneDueAtSlot(slot) {
+        if (!this.l1Constants) {
+            throw new Error('isPruneDueAtSlot requires l1Constants');
+        }
+        const tips = await this.getL2Tips();
+        const proven = tips.proven.checkpoint.number;
+        const pending = tips.checkpointed.checkpoint.number;
+        if (pending === proven) {
+            return false;
+        }
+        const oldestUnproven = await this.getCheckpointData({
+            number: CheckpointNumber(Number(proven) + 1)
+        });
+        if (!oldestUnproven) {
+            return false;
         }
-        const headers = await this.store.getBlockHeaders(blockNumber, 1);
-        return headers.length === 0 ? undefined : headers[0];
+        const slotTs = getLastL1SlotTimestampForL2Slot(slot, this.l1Constants);
+        const slotEpoch = getEpochNumberAtTimestamp(slotTs, this.l1Constants);
+        const oldestUnprovenEpoch = getEpochAtSlot(oldestUnproven.header.slotNumber, this.l1Constants);
+        const deadlineEpoch = getProofSubmissionDeadlineEpoch(oldestUnprovenEpoch, this.l1Constants);
+        return slotEpoch >= deadlineEpoch;
     }
-    getCheckpointedBlock(number) {
-        return this.store.getCheckpointedBlock(number);
+    getCheckpointNumber() {
+        return this.stores.blocks.getLatestCheckpointNumber();
     }
-    getCheckpointedBlockNumber() {
-        return this.store.getCheckpointedL2BlockNumber();
+    getProvenCheckpointNumber() {
+        return this.stores.blocks.getProvenCheckpointNumber();
     }
-    async getCheckpointHeader(number) {
-        if (number === 'latest') {
-            number = await this.store.getSynchedCheckpointNumber();
+    async getBlockNumber(query) {
+        if (!query) {
+            return this.stores.blocks.getLatestL2BlockNumber();
         }
-        if (number === 0) {
+        const resolved = await this.resolveBlockQuery(query);
+        if (resolved === undefined) {
             return undefined;
         }
-        const checkpoint = await this.store.getCheckpointData(number);
-        if (!checkpoint) {
-            return undefined;
+        if (this.isGenesisBlockQuery(resolved)) {
+            return BlockNumber.ZERO;
+        }
+        return this.stores.blocks.getBlockNumber(resolved);
+    }
+    /**
+   * Resolves a {@link CheckpointQuery} to a concrete `CheckpointNumber`, or undefined when the
+   * query refers to a position that has no checkpoint yet (e.g. `{ slot }` not found).
+   */ resolveCheckpointQuery(query) {
+        if ('number' in query) {
+            return Promise.resolve(query.number);
+        }
+        if ('slot' in query) {
+            return this.stores.blocks.getCheckpointNumberBySlot(query.slot);
+        }
+        // tag variant
+        switch(query.tag){
+            case 'checkpointed':
+                return this.stores.blocks.getLatestCheckpointNumber();
+            case 'proven':
+                return this.stores.blocks.getProvenCheckpointNumber();
+            case 'finalized':
+                return this.stores.blocks.getFinalizedCheckpointNumber();
         }
-        return checkpoint.header;
     }
-    async getLastBlockNumberInCheckpoint(checkpointNumber) {
-        const checkpointData = await this.store.getCheckpointData(checkpointNumber);
-        if (!checkpointData) {
+    async getCheckpoint(query) {
+        const number = await this.resolveCheckpointQuery(query);
+        if (number === undefined || number === 0) {
             return undefined;
         }
-        return BlockNumber(checkpointData.startBlock + checkpointData.numBlocks - 1);
-    }
-    async getCheckpointedBlocks(from, limit, proven) {
-        const blocks = await this.store.getCheckpointedBlocks(from, limit);
-        if (proven === true) {
-            const provenBlockNumber = await this.store.getProvenBlockNumber();
-            return blocks.filter((b)=>b.block.number <= provenBlockNumber);
+        const data = await this.stores.blocks.getCheckpointData(number);
+        if (!data) {
+            return undefined;
         }
-        return blocks;
+        return this.getPublishedCheckpointFromCheckpointData(data);
     }
-    getBlockHeaderByHash(blockHash) {
-        return this.store.getBlockHeaderByHash(blockHash);
+    async getCheckpoints(query) {
+        const checkpoints = await this.getCheckpointsData(query);
+        return Promise.all(checkpoints.map((ch)=>this.getPublishedCheckpointFromCheckpointData(ch)));
     }
-    getBlockHeaderByArchive(archive) {
-        return this.store.getBlockHeaderByArchive(archive);
+    async getCheckpointData(query) {
+        const number = await this.resolveCheckpointQuery(query);
+        if (number === undefined || number === 0) {
+            return undefined;
+        }
+        return this.stores.blocks.getCheckpointData(number);
     }
-    async getL2BlockNew(number) {
-        // If the number provided is -ve, then return the latest block.
-        if (number < 0) {
-            number = await this.store.getLatestBlockNumber();
+    async getCheckpointsData(query) {
+        if ('fromSlot' in query) {
+            return this.stores.blocks.getCheckpointsBySlot(query.fromSlot, query.limit, query.reverse ?? false);
         }
-        if (number === 0) {
-            return undefined;
+        if ('from' in query) {
+            return this.stores.blocks.getRangeOfCheckpoints(query.from, query.limit);
         }
-        const publishedBlock = await this.store.getBlock(number);
-        return publishedBlock;
+        const numbers = await this.getCheckpointNumbersForEpoch(query.epoch);
+        return numbers.length > 0 ? this.stores.blocks.getRangeOfCheckpoints(numbers[0], numbers.length) : [];
     }
-    getTxEffect(txHash) {
-        return this.store.getTxEffect(txHash);
+    getProposedCheckpointData(query) {
+        if (!query || 'tag' in query) {
+            return this.stores.blocks.getLastProposedCheckpoint();
+        }
+        if ('number' in query) {
+            return this.stores.blocks.getProposedCheckpointByNumber(query.number);
+        }
+        return this.stores.blocks.getProposedCheckpointBySlot(query.slot);
     }
-    getSettledTxReceipt(txHash) {
-        return this.store.getSettledTxReceipt(txHash);
+    getTxEffect(txHash) {
+        return this.stores.blocks.getTxEffect(txHash);
     }
     isPendingChainInvalid() {
         return this.getPendingChainValidationStatus().then((status)=>!status.valid);
     }
     async getPendingChainValidationStatus() {
-        return await this.store.getPendingChainValidationStatus() ?? {
+        return await this.stores.blocks.getPendingChainValidationStatus() ?? {
             valid: true
         };
     }
-    async getL2BlocksNew(from, limit, proven) {
-        const blocks = await this.store.getBlocks(from, limit);
-        if (proven === true) {
-            const provenBlockNumber = await this.store.getProvenBlockNumber();
-            return blocks.filter((b)=>b.number <= provenBlockNumber);
-        }
-        return blocks;
-    }
-    getPrivateLogsByTags(tags, page) {
-        return this.store.getPrivateLogsByTags(tags, page);
-    }
-    getPublicLogsByTagsFromContract(contractAddress, tags, page) {
-        return this.store.getPublicLogsByTagsFromContract(contractAddress, tags, page);
-    }
-    getPublicLogs(filter) {
-        return this.store.getPublicLogs(filter);
+    getPrivateLogsByTags(query) {
+        return this.stores.logs.getPrivateLogsByTags(query);
     }
-    getContractClassLogs(filter) {
-        return this.store.getContractClassLogs(filter);
+    getPublicLogsByTags(query) {
+        return this.stores.logs.getPublicLogsByTags(query);
     }
     getContractClass(id) {
-        return this.store.getContractClass(id);
+        return this.stores.contractClasses.getContractClass(id);
     }
     getBytecodeCommitment(id) {
-        return this.store.getBytecodeCommitment(id);
-    }
-    async getContract(address, maybeTimestamp) {
-        let timestamp;
-        if (maybeTimestamp === undefined) {
-            const latestBlockHeader = await this.getBlockHeader('latest');
-            // If we get undefined block header, it means that the archiver has not yet synced any block so we default to 0.
-            timestamp = latestBlockHeader ? latestBlockHeader.globalVariables.timestamp : 0n;
-        } else {
-            timestamp = maybeTimestamp;
-        }
-        return this.store.getContractInstance(address, timestamp);
+        return this.stores.contractClasses.getBytecodeCommitment(id);
+    }
+    getContract(address, timestamp) {
+        return this.stores.contractInstances.getContractInstance(address, timestamp);
     }
     getContractClassIds() {
-        return this.store.getContractClassIds();
+        return this.stores.contractClasses.getContractClassIds();
     }
-    getDebugFunctionName(address, selector) {
-        return this.store.getDebugFunctionName(address, selector);
+    /** Looks up a public function name given a selector. */ getDebugFunctionName(_address, selector) {
+        return Promise.resolve(this.stores.functionNames.get(selector));
     }
-    registerContractFunctionSignatures(signatures) {
-        return this.store.registerContractFunctionSignatures(signatures);
+    /** Register public function signatures so they can be looked up by selector. */ registerContractFunctionSignatures(signatures) {
+        return this.stores.functionNames.register(signatures);
     }
     getL1ToL2Messages(checkpointNumber) {
-        return this.store.getL1ToL2Messages(checkpointNumber);
+        return this.stores.messages.getL1ToL2Messages(checkpointNumber);
     }
     getL1ToL2MessageIndex(l1ToL2Message) {
-        return this.store.getL1ToL2MessageIndex(l1ToL2Message);
-    }
-    async getPublishedCheckpoints(checkpointNumber, limit) {
-        const checkpoints = await this.store.getRangeOfCheckpoints(checkpointNumber, limit);
-        const blocks = (await Promise.all(checkpoints.map((ch)=>this.store.getBlocksForCheckpoint(ch.checkpointNumber)))).filter(isDefined);
-        const fullCheckpoints = [];
-        for(let i = 0; i < checkpoints.length; i++){
-            const blocksForCheckpoint = blocks[i];
-            const checkpoint = checkpoints[i];
-            const fullCheckpoint = new Checkpoint(checkpoint.archive, checkpoint.header, blocksForCheckpoint, checkpoint.checkpointNumber);
-            const publishedCheckpoint = new PublishedCheckpoint(fullCheckpoint, checkpoint.l1, checkpoint.attestations.map((x)=>CommitteeAttestation.fromBuffer(x)));
-            fullCheckpoints.push(publishedCheckpoint);
+        return this.stores.messages.getL1ToL2MessageIndex(l1ToL2Message);
+    }
+    async getPublishedCheckpointFromCheckpointData(checkpoint) {
+        const blocksForCheckpoint = await this.stores.blocks.getBlocksForCheckpoint(checkpoint.checkpointNumber);
+        if (!blocksForCheckpoint) {
+            throw new Error(`Blocks for checkpoint ${checkpoint.checkpointNumber} not found`);
         }
-        return fullCheckpoints;
+        const fullCheckpoint = new Checkpoint(checkpoint.archive, checkpoint.header, blocksForCheckpoint, checkpoint.checkpointNumber, checkpoint.feeAssetPriceModifier);
+        return new PublishedCheckpoint(fullCheckpoint, checkpoint.l1, checkpoint.attestations);
     }
     getBlocksForSlot(slotNumber) {
-        return this.store.getBlocksForSlot(slotNumber);
+        return this.stores.blocks.getBlocksForSlot(slotNumber);
     }
-    async getBlocksForEpoch(epochNumber) {
+    /** Returns just the checkpoint numbers for all checkpoints whose slot falls within the given epoch. */ getCheckpointNumbersForEpoch(epochNumber) {
         if (!this.l1Constants) {
             throw new Error('L1 constants not set');
         }
         const [start, end] = getSlotRangeForEpoch(epochNumber, this.l1Constants);
-        const blocks = [];
-        // Walk the list of checkpoints backwards and filter by slots matching the requested epoch.
-        // We'll typically ask for checkpoints for a very recent epoch, so we shouldn't need an index here.
-        let checkpoint = await this.store.getCheckpointData(await this.store.getSynchedCheckpointNumber());
-        const slot = (b)=>b.header.slotNumber;
-        while(checkpoint && slot(checkpoint) >= start){
-            if (slot(checkpoint) <= end) {
-                // push the blocks on backwards
-                const endBlock = checkpoint.startBlock + checkpoint.numBlocks - 1;
-                for(let i = endBlock; i >= checkpoint.startBlock; i--){
-                    const block = await this.getBlock(BlockNumber(i));
-                    if (block) {
-                        blocks.push(block);
-                    }
-                }
-            }
-            checkpoint = await this.store.getCheckpointData(CheckpointNumber(checkpoint.checkpointNumber - 1));
-        }
-        return blocks.reverse();
+        return this.stores.blocks.getCheckpointNumbersForSlotRange(start, end);
     }
-    async getBlockHeadersForEpoch(epochNumber) {
-        if (!this.l1Constants) {
-            throw new Error('L1 constants not set');
+    async getBlock(query) {
+        const resolved = await this.resolveBlockQuery(query);
+        if (resolved === undefined) {
+            return undefined;
         }
-        const [start, end] = getSlotRangeForEpoch(epochNumber, this.l1Constants);
-        const blocks = [];
-        // Walk the list of checkpoints backwards and filter by slots matching the requested epoch.
-        // We'll typically ask for checkpoints for a very recent epoch, so we shouldn't need an index here.
-        let checkpoint = await this.store.getCheckpointData(await this.store.getSynchedCheckpointNumber());
-        const slot = (b)=>b.header.slotNumber;
-        while(checkpoint && slot(checkpoint) >= start){
-            if (slot(checkpoint) <= end) {
-                // push the blocks on backwards
-                const endBlock = checkpoint.startBlock + checkpoint.numBlocks - 1;
-                for(let i = endBlock; i >= checkpoint.startBlock; i--){
-                    const block = await this.getBlockHeader(BlockNumber(i));
-                    if (block) {
-                        blocks.push(block);
-                    }
-                }
-            }
-            checkpoint = await this.store.getCheckpointData(CheckpointNumber(checkpoint.checkpointNumber - 1));
+        if (this.isGenesisBlockQuery(resolved)) {
+            return this.getGenesisBlock();
         }
-        return blocks.reverse();
-    }
-    async getCheckpointsForEpoch(epochNumber) {
-        if (!this.l1Constants) {
-            throw new Error('L1 constants not set');
+        return this.stores.blocks.getBlock(resolved);
+    }
+    /**
+   * Range queries iterate physical blocks only; the genesis block is NOT prepended.
+   * `L2BlockStream` consumers (`world-state.handleL2Blocks`, etc.) emit `blocks-added` events for
+   * real blocks and would be surprised by a synthetic block 0. Use {@link getBlock} or
+   * {@link getBlockData} for genesis-aware single-block lookups.
+   */ async getBlocks(query) {
+        const resolved = await this.resolveBlocksQuery(query);
+        return resolved ? this.stores.blocks.getBlocks(resolved) : [];
+    }
+    async getBlockData(query) {
+        const resolved = await this.resolveBlockQuery(query);
+        if (resolved === undefined) {
+            return undefined;
         }
-        const [start, end] = getSlotRangeForEpoch(epochNumber, this.l1Constants);
-        const checkpoints = [];
-        // Walk the list of checkpoints backwards and filter by slots matching the requested epoch.
-        // We'll typically ask for checkpoints for a very recent epoch, so we shouldn't need an index here.
-        let checkpointData = await this.store.getCheckpointData(await this.store.getSynchedCheckpointNumber());
-        const slot = (b)=>b.header.slotNumber;
-        while(checkpointData && slot(checkpointData) >= start){
-            if (slot(checkpointData) <= end) {
-                // push the checkpoints on backwards
-                const [checkpoint] = await this.getPublishedCheckpoints(checkpointData.checkpointNumber, 1);
-                checkpoints.push(checkpoint.checkpoint);
-            }
-            checkpointData = await this.store.getCheckpointData(CheckpointNumber(checkpointData.checkpointNumber - 1));
+        if (this.isGenesisBlockQuery(resolved)) {
+            return this.getGenesisBlockData();
         }
-        return checkpoints.reverse();
-    }
-    async getPublishedBlocks(from, limit, proven) {
-        const checkpoints = await this.store.getRangeOfCheckpoints(CheckpointNumber(from), limit);
-        const provenCheckpointNumber = await this.store.getProvenCheckpointNumber();
-        const blocks = (await Promise.all(checkpoints.map((ch)=>this.store.getBlocksForCheckpoint(ch.checkpointNumber)))).filter(isDefined);
-        const publishedBlocks = [];
-        for(let i = 0; i < checkpoints.length; i++){
-            const blockForCheckpoint = blocks[i][0];
-            const checkpoint = checkpoints[i];
-            if (checkpoint.checkpointNumber > provenCheckpointNumber && proven === true) {
-                continue;
-            }
-            const publishedBlock = new CheckpointedL2Block(checkpoint.checkpointNumber, blockForCheckpoint, checkpoint.l1, checkpoint.attestations.map((x)=>CommitteeAttestation.fromBuffer(x)));
-            publishedBlocks.push(publishedBlock);
+        return this.stores.blocks.getBlockData(resolved);
+    }
+    /** See {@link getBlocks} — range queries do not prepend the genesis block. */ async getBlocksData(query) {
+        const resolved = await this.resolveBlocksQuery(query);
+        return resolved ? this.stores.blocks.getBlocksData(resolved) : [];
+    }
+    /**
+   * Resolves a {@link BlockQuery} to either the genesis sentinel or a {@link ResolvedBlockQuery}
+   * understood by BlockStore. Detects every shape that points at block 0 — `{number:0}`,
+   * `{hash}` matching the initial header, `{archive}` matching the post-genesis archive root,
+   * and `{tag}` resolving to 0 — and rewrites them to the sentinel so callers branch once.
+   */ async resolveBlockQuery(query) {
+        if ('number' in query) {
+            return query.number === BlockNumber.ZERO ? {
+                genesis: true
+            } : query;
         }
-        return publishedBlocks;
-    }
-    async getBlock(number) {
-        // If the number provided is -ve, then return the latest block.
-        if (number < 0) {
-            number = await this.store.getLatestBlockNumber();
+        if ('hash' in query) {
+            return query.hash.equals(this.initialBlockHash) ? {
+                genesis: true
+            } : query;
         }
-        if (number === 0) {
-            return undefined;
+        if ('archive' in query) {
+            return query.archive.equals(this.genesisArchiveRoot) ? {
+                genesis: true
+            } : query;
         }
-        return this.store.getBlock(number);
-    }
-    async getBlocks(from, limit, proven) {
-        const blocks = await this.store.getBlocks(from, limit);
-        if (proven === true) {
-            const provenBlockNumber = await this.store.getProvenBlockNumber();
-            return blocks.filter((b)=>b.number <= provenBlockNumber);
+        const number = await this.resolveBlockTag(query.tag);
+        if (number === BlockNumber.ZERO) {
+            return {
+                genesis: true
+            };
         }
-        return blocks;
-    }
-    getPublishedBlockByHash(blockHash) {
-        return this.store.getCheckpointedBlockByHash(blockHash);
-    }
-    getPublishedBlockByArchive(archive) {
-        return this.store.getCheckpointedBlockByArchive(archive);
+        return {
+            number
+        };
     }
-    async getL2BlockNewByHash(blockHash) {
-        const checkpointedBlock = await this.store.getCheckpointedBlockByHash(blockHash);
-        return checkpointedBlock?.block;
+    /** Maps a {@link BlockTag} to the matching block number for the current chain state. */ resolveBlockTag(tag) {
+        switch(tag){
+            case 'latest':
+            case 'proposed':
+                return this.stores.blocks.getLatestL2BlockNumber();
+            case 'checkpointed':
+                return this.stores.blocks.getCheckpointedL2BlockNumber();
+            case 'proven':
+                return this.stores.blocks.getProvenBlockNumber();
+            case 'finalized':
+                return this.stores.blocks.getFinalizedL2BlockNumber();
+        }
     }
-    async getL2BlockNewByArchive(archive) {
-        const checkpointedBlock = await this.store.getCheckpointedBlockByArchive(archive);
-        return checkpointedBlock?.block;
+    /**
+   * Converts an epoch-based BlocksQuery to a from/limit query using l1Constants.
+   * Returns undefined when the epoch has no checkpoints, so callers can return [] without
+   * entering BlockStore. Reads only the two endpoint checkpoints rather than the whole epoch.
+   */ async resolveBlocksQuery(query) {
+        if (!('epoch' in query)) {
+            if (query.from < INITIAL_L2_BLOCK_NUM) {
+                throw new Error(`getBlocks/getBlocksData: 'from' must be >= ${INITIAL_L2_BLOCK_NUM}, got ${query.from}. ` + `Use getBlock({number:0})/getBlockData({number:0}) for genesis-aware single-block lookups.`);
+            }
+            return query;
+        }
+        const checkpointNumbers = await this.getCheckpointNumbersForEpoch(query.epoch);
+        if (checkpointNumbers.length === 0) {
+            return undefined;
+        }
+        const firstNumber = checkpointNumbers[0];
+        const lastNumber = checkpointNumbers[checkpointNumbers.length - 1];
+        const first = await this.stores.blocks.getCheckpointData(firstNumber);
+        if (!first) {
+            return undefined;
+        }
+        const last = firstNumber === lastNumber ? first : await this.stores.blocks.getCheckpointData(lastNumber);
+        if (!last) {
+            return undefined;
+        }
+        const from = BlockNumber(first.startBlock);
+        const limit = last.startBlock + last.blockCount - first.startBlock;
+        return {
+            from,
+            limit,
+            onlyCheckpointed: true
+        };
     }
 }