@aztec/prover-client 5.0.0-private.20260319 → 5.0.0-rc.2

package/dest/prover-client/server-epoch-prover.js

@@ -1,40 +0,0 @@
-/** Encapsulates the proving orchestrator and the broker facade */ export class ServerEpochProver {
-    facade;
-    orchestrator;
-    constructor(facade, orchestrator){
-        this.facade = facade;
-        this.orchestrator = orchestrator;
-    }
-    startNewEpoch(epochNumber, totalNumCheckpoints, finalBlobBatchingChallenges) {
-        this.orchestrator.startNewEpoch(epochNumber, totalNumCheckpoints, finalBlobBatchingChallenges);
-        this.facade.start();
-    }
-    startNewCheckpoint(checkpointIndex, constants, l1ToL2Messages, totalNumBlocks, headerOfLastBlockInPreviousCheckpoint) {
-        return this.orchestrator.startNewCheckpoint(checkpointIndex, constants, l1ToL2Messages, totalNumBlocks, headerOfLastBlockInPreviousCheckpoint);
-    }
-    startChonkVerifierCircuits(txs) {
-        return this.orchestrator.startChonkVerifierCircuits(txs);
-    }
-    setBlockCompleted(blockNumber, expectedBlockHeader) {
-        return this.orchestrator.setBlockCompleted(blockNumber, expectedBlockHeader);
-    }
-    finalizeEpoch() {
-        return this.orchestrator.finalizeEpoch();
-    }
-    cancel() {
-        this.orchestrator.cancel();
-    }
-    getProverId() {
-        return this.orchestrator.getProverId();
-    }
-    async stop() {
-        await this.facade.stop();
-        await this.orchestrator.stop();
-    }
-    startNewBlock(blockNumber, timestamp, totalNumTxs) {
-        return this.orchestrator.startNewBlock(blockNumber, timestamp, totalNumTxs);
-    }
-    addTxs(txs) {
-        return this.orchestrator.addTxs(txs);
-    }
-}

package/src/orchestrator/epoch-proving-state.ts

@@ -1,380 +0,0 @@
-import { BatchedBlob, BatchedBlobAccumulator, type FinalBlobBatchingChallenges } from '@aztec/blob-lib';
-import {
-  type ARCHIVE_HEIGHT,
-  type L1_TO_L2_MSG_SUBTREE_ROOT_SIBLING_PATH_LENGTH,
-  type NESTED_RECURSIVE_PROOF_LENGTH,
-  type NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH,
-  OUT_HASH_TREE_HEIGHT,
-} from '@aztec/constants';
-import { BlockNumber, EpochNumber } from '@aztec/foundation/branded-types';
-import { Fr } from '@aztec/foundation/curves/bn254';
-import type { Tuple } from '@aztec/foundation/serialize';
-import {
-  MerkleTreeCalculator,
-  type TreeNodeLocation,
-  UnbalancedTreeStore,
-  shaMerkleHash,
-} from '@aztec/foundation/trees';
-import type { PublicInputsAndRecursiveProof } from '@aztec/stdlib/interfaces/server';
-import type { Proof } from '@aztec/stdlib/proofs';
-import {
-  CheckpointConstantData,
-  CheckpointMergeRollupPrivateInputs,
-  CheckpointPaddingRollupPrivateInputs,
-  CheckpointRollupPublicInputs,
-  PublicChonkVerifierPublicInputs,
-  RootRollupPrivateInputs,
-  type RootRollupPublicInputs,
-} from '@aztec/stdlib/rollup';
-import { AppendOnlyTreeSnapshot, type MerkleTreeId } from '@aztec/stdlib/trees';
-import type { BlockHeader } from '@aztec/stdlib/tx';
-
-import { toProofData } from './block-building-helpers.js';
-import type { ProofState } from './block-proving-state.js';
-import { CheckpointProvingState } from './checkpoint-proving-state.js';
-
-export type TreeSnapshots = Map<MerkleTreeId, AppendOnlyTreeSnapshot>;
-
-enum PROVING_STATE_LIFECYCLE {
-  PROVING_STATE_CREATED,
-  PROVING_STATE_FULL,
-  PROVING_STATE_RESOLVED,
-  PROVING_STATE_REJECTED,
-}
-
-export type ProvingResult = { status: 'success' } | { status: 'failure'; reason: string };
-
-/**
- * The current state of the proving schedule for an epoch.
- * Contains the raw inputs and intermediate state to generate every constituent proof in the tree.
- * Carries an identifier so we can identify if the proving state is discarded and a new one started.
- * Captures resolve and reject callbacks to provide a promise base interface to the consumer of our proving.
- */
-export class EpochProvingState {
-  private checkpointProofs: UnbalancedTreeStore<
-    ProofState<CheckpointRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>
-  >;
-  private checkpointPaddingProof:
-    | ProofState<CheckpointRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>
-    | undefined;
-  private rootRollupProof: ProofState<RootRollupPublicInputs, typeof NESTED_RECURSIVE_PROOF_LENGTH> | undefined;
-  private checkpoints: (CheckpointProvingState | undefined)[] = [];
-  private startBlobAccumulator: BatchedBlobAccumulator;
-  private endBlobAccumulator: BatchedBlobAccumulator | undefined;
-  private finalBatchedBlob: BatchedBlob | undefined;
-  private provingStateLifecycle = PROVING_STATE_LIFECYCLE.PROVING_STATE_CREATED;
-
-  // Map from tx hash to chonk verifier proof promise. Used when kickstarting chonk verifier proofs before tx processing.
-  public readonly cachedChonkVerifierProofs = new Map<
-    string,
-    Promise<
-      PublicInputsAndRecursiveProof<PublicChonkVerifierPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>
-    >
-  >();
-
-  constructor(
-    public readonly epochNumber: EpochNumber,
-    public readonly totalNumCheckpoints: number,
-    private readonly finalBlobBatchingChallenges: FinalBlobBatchingChallenges,
-    private onCheckpointBlobAccumulatorSet: (checkpoint: CheckpointProvingState) => Promise<void>,
-    private completionCallback: (result: ProvingResult) => void,
-    private rejectionCallback: (reason: string) => void,
-  ) {
-    this.checkpointProofs = new UnbalancedTreeStore(totalNumCheckpoints);
-    this.startBlobAccumulator = BatchedBlobAccumulator.newWithChallenges(finalBlobBatchingChallenges);
-  }
-
-  // Adds a block to the proving state, returns its index
-  // Will update the proving life cycle if this is the last block
-  public startNewCheckpoint(
-    checkpointIndex: number,
-    constants: CheckpointConstantData,
-    totalNumBlocks: number,
-    previousBlockHeader: BlockHeader,
-    lastArchiveSiblingPath: Tuple<Fr, typeof ARCHIVE_HEIGHT>,
-    l1ToL2Messages: Fr[],
-    lastL1ToL2MessageTreeSnapshot: AppendOnlyTreeSnapshot,
-    lastL1ToL2MessageSubtreeRootSiblingPath: Tuple<Fr, typeof L1_TO_L2_MSG_SUBTREE_ROOT_SIBLING_PATH_LENGTH>,
-    newL1ToL2MessageTreeSnapshot: AppendOnlyTreeSnapshot,
-    newL1ToL2MessageSubtreeRootSiblingPath: Tuple<Fr, typeof L1_TO_L2_MSG_SUBTREE_ROOT_SIBLING_PATH_LENGTH>,
-  ): CheckpointProvingState {
-    if (checkpointIndex >= this.totalNumCheckpoints) {
-      throw new Error(
-        `Unable to start a new checkpoint at index ${checkpointIndex}. Expected at most ${this.totalNumCheckpoints} checkpoints.`,
-      );
-    }
-
-    const checkpoint = new CheckpointProvingState(
-      checkpointIndex,
-      constants,
-      totalNumBlocks,
-      this.finalBlobBatchingChallenges,
-      previousBlockHeader,
-      lastArchiveSiblingPath,
-      l1ToL2Messages,
-      lastL1ToL2MessageTreeSnapshot,
-      lastL1ToL2MessageSubtreeRootSiblingPath,
-      newL1ToL2MessageTreeSnapshot,
-      newL1ToL2MessageSubtreeRootSiblingPath,
-      this,
-      this.onCheckpointBlobAccumulatorSet,
-    );
-    this.checkpoints[checkpointIndex] = checkpoint;
-
-    if (this.checkpoints.filter(c => !!c).length === this.totalNumCheckpoints) {
-      this.provingStateLifecycle = PROVING_STATE_LIFECYCLE.PROVING_STATE_FULL;
-    }
-
-    return checkpoint;
-  }
-
-  public getCheckpointProvingState(index: number) {
-    return this.checkpoints[index];
-  }
-
-  public getCheckpointProvingStateByBlockNumber(blockNumber: BlockNumber) {
-    return this.checkpoints.find(
-      c =>
-        c &&
-        Number(blockNumber) >= Number(c.firstBlockNumber) &&
-        Number(blockNumber) < Number(c.firstBlockNumber) + c.totalNumBlocks,
-    );
-  }
-
-  public getBlockProvingStateByBlockNumber(blockNumber: BlockNumber) {
-    return this.getCheckpointProvingStateByBlockNumber(blockNumber)?.getBlockProvingStateByBlockNumber(blockNumber);
-  }
-
-  // Returns true if this proving state is still valid, false otherwise
-  public verifyState() {
-    return (
-      this.provingStateLifecycle === PROVING_STATE_LIFECYCLE.PROVING_STATE_CREATED ||
-      this.provingStateLifecycle === PROVING_STATE_LIFECYCLE.PROVING_STATE_FULL
-    );
-  }
-
-  // Returns true if we are still able to accept checkpoints, false otherwise.
-  public isAcceptingCheckpoints() {
-    return this.checkpoints.filter(c => !!c).length < this.totalNumCheckpoints;
-  }
-
-  public setCheckpointRootRollupProof(
-    checkpointIndex: number,
-    provingOutput: PublicInputsAndRecursiveProof<
-      CheckpointRollupPublicInputs,
-      typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH
-    >,
-  ): TreeNodeLocation {
-    return this.checkpointProofs.setLeaf(checkpointIndex, { provingOutput });
-  }
-
-  public tryStartProvingCheckpointMerge(location: TreeNodeLocation) {
-    if (this.checkpointProofs.getNode(location)?.isProving) {
-      return false;
-    } else {
-      this.checkpointProofs.setNode(location, { isProving: true });
-      return true;
-    }
-  }
-
-  public setCheckpointMergeRollupProof(
-    location: TreeNodeLocation,
-    provingOutput: PublicInputsAndRecursiveProof<
-      CheckpointRollupPublicInputs,
-      typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH
-    >,
-  ) {
-    this.checkpointProofs.setNode(location, { provingOutput });
-  }
-
-  public tryStartProvingRootRollup() {
-    if (this.rootRollupProof?.isProving) {
-      return false;
-    } else {
-      this.rootRollupProof = { isProving: true };
-      return true;
-    }
-  }
-
-  public setRootRollupProof(provingOutput: PublicInputsAndRecursiveProof<RootRollupPublicInputs>) {
-    this.rootRollupProof = { provingOutput };
-  }
-
-  public tryStartProvingPaddingCheckpoint() {
-    if (this.checkpointPaddingProof?.isProving) {
-      return false;
-    } else {
-      this.checkpointPaddingProof = { isProving: true };
-      return true;
-    }
-  }
-
-  public setCheckpointPaddingProof(
-    provingOutput: PublicInputsAndRecursiveProof<
-      CheckpointRollupPublicInputs,
-      typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH
-    >,
-  ) {
-    this.checkpointPaddingProof = { provingOutput };
-  }
-
-  public async accumulateCheckpointOutHashes() {
-    const treeCalculator = await MerkleTreeCalculator.create(OUT_HASH_TREE_HEIGHT, undefined, (left, right) =>
-      Promise.resolve(shaMerkleHash(left, right)),
-    );
-
-    const computeOutHashHint = async (leaves: Fr[]) => {
-      const tree = await treeCalculator.computeTree(leaves.map(l => l.toBuffer()));
-      const nextAvailableLeafIndex = leaves.length;
-      return {
-        treeSnapshot: new AppendOnlyTreeSnapshot(Fr.fromBuffer(tree.root), nextAvailableLeafIndex),
-        siblingPath: tree.getSiblingPath(nextAvailableLeafIndex).map(Fr.fromBuffer) as Tuple<
-          Fr,
-          typeof OUT_HASH_TREE_HEIGHT
-        >,
-      };
-    };
-
-    let hint = this.checkpoints[0]?.getOutHashHint();
-    const outHashes = [];
-    for (let i = 0; i < this.totalNumCheckpoints; i++) {
-      const checkpoint = this.checkpoints[i];
-      if (!checkpoint) {
-        break;
-      }
-
-      // If hints are not set yet, it must be the first checkpoint. Compute the hints with an empty tree.
-      hint ??= await computeOutHashHint([]);
-      checkpoint.setOutHashHint(hint);
-
-      // Get the out hash for this checkpoint.
-      const outHash = checkpoint.accumulateBlockOutHashes();
-      if (!outHash) {
-        break;
-      }
-      outHashes.push(outHash);
-
-      // If this is NOT the last checkpoint, get or create the hint for the next checkpoint.
-      if (i !== this.totalNumCheckpoints - 1) {
-        hint = checkpoint.getOutHashHintForNextCheckpoint() ?? (await computeOutHashHint(outHashes));
-        checkpoint.setOutHashHintForNextCheckpoint(hint);
-      }
-    }
-  }
-
-  public async setBlobAccumulators() {
-    let previousAccumulator = this.startBlobAccumulator;
-    // Accumulate blobs as far as we can for this epoch.
-    for (let i = 0; i < this.totalNumCheckpoints; i++) {
-      const checkpoint = this.checkpoints[i];
-      if (!checkpoint) {
-        break;
-      }
-
-      const endAccumulator =
-        checkpoint.getEndBlobAccumulator() || (await checkpoint.accumulateBlobs(previousAccumulator));
-      if (!endAccumulator) {
-        break;
-      }
-
-      previousAccumulator = endAccumulator;
-
-      // If this is the last checkpoint, set the end blob accumulator.
-      if (i === this.totalNumCheckpoints - 1) {
-        this.endBlobAccumulator = endAccumulator;
-      }
-    }
-  }
-
-  public async finalizeBatchedBlob() {
-    if (!this.endBlobAccumulator) {
-      throw new Error('End blob accumulator not ready.');
-    }
-    this.finalBatchedBlob = await this.endBlobAccumulator.finalize(true /* verifyProof */);
-  }
-
-  public getParentLocation(location: TreeNodeLocation) {
-    return this.checkpointProofs.getParentLocation(location);
-  }
-
-  public getCheckpointMergeRollupInputs(mergeLocation: TreeNodeLocation) {
-    const [left, right] = this.checkpointProofs.getChildren(mergeLocation).map(c => c?.provingOutput);
-    if (!left || !right) {
-      throw new Error('At least one child is not ready for the checkpoint merge rollup.');
-    }
-
-    return new CheckpointMergeRollupPrivateInputs([toProofData(left), toProofData(right)]);
-  }
-
-  public getRootRollupInputs() {
-    const [left, right] = this.#getChildProofsForRoot();
-    if (!left || !right) {
-      throw new Error('At least one child is not ready for the root rollup.');
-    }
-
-    return RootRollupPrivateInputs.from({
-      previousRollups: [toProofData(left), toProofData(right)],
-    });
-  }
-
-  public getPaddingCheckpointInputs() {
-    return new CheckpointPaddingRollupPrivateInputs();
-  }
-
-  public getEpochProofResult(): { proof: Proof; publicInputs: RootRollupPublicInputs; batchedBlobInputs: BatchedBlob } {
-    const provingOutput = this.rootRollupProof?.provingOutput;
-
-    if (!provingOutput || !this.finalBatchedBlob) {
-      throw new Error('Unable to get epoch proof result. Root rollup is not ready.');
-    }
-
-    return {
-      proof: provingOutput.proof.binaryProof,
-      publicInputs: provingOutput.inputs,
-      batchedBlobInputs: this.finalBatchedBlob,
-    };
-  }
-
-  public isReadyForCheckpointMerge(location: TreeNodeLocation) {
-    return !!this.checkpointProofs.getSibling(location)?.provingOutput;
-  }
-
-  // Returns true if we have sufficient inputs to execute the block root rollup
-  public isReadyForRootRollup() {
-    const childProofs = this.#getChildProofsForRoot();
-    return childProofs.every(p => !!p);
-  }
-
-  // Attempts to reject the proving state promise with a reason of 'cancelled'
-  public cancel() {
-    this.reject('Proving cancelled');
-  }
-
-  // Attempts to reject the proving state promise with the given reason
-  // Does nothing if not in a valid state
-  public reject(reason: string) {
-    if (!this.verifyState()) {
-      return;
-    }
-    this.provingStateLifecycle = PROVING_STATE_LIFECYCLE.PROVING_STATE_REJECTED;
-    this.rejectionCallback(reason);
-  }
-
-  // Attempts to resolve the proving state promise with the given result
-  // Does nothing if not in a valid state
-  public resolve(result: ProvingResult) {
-    if (!this.verifyState()) {
-      return;
-    }
-    this.provingStateLifecycle = PROVING_STATE_LIFECYCLE.PROVING_STATE_RESOLVED;
-    this.completionCallback(result);
-  }
-
-  #getChildProofsForRoot() {
-    const rootLocation = { level: 0, index: 0 };
-    // If there's only 1 block, its block root proof will be stored at the root.
-    return this.totalNumCheckpoints === 1
-      ? [this.checkpointProofs.getNode(rootLocation)?.provingOutput, this.checkpointPaddingProof?.provingOutput]
-      : this.checkpointProofs.getChildren(rootLocation).map(c => c?.provingOutput);
-  }
-}

package/src/prover-client/server-epoch-prover.ts

@@ -1,69 +0,0 @@
-import type { BatchedBlob, FinalBlobBatchingChallenges } from '@aztec/blob-lib/types';
-import { BlockNumber, EpochNumber } from '@aztec/foundation/branded-types';
-import type { Fr } from '@aztec/foundation/curves/bn254';
-import type { EthAddress } from '@aztec/stdlib/block';
-import type { EpochProver } from '@aztec/stdlib/interfaces/server';
-import type { Proof } from '@aztec/stdlib/proofs';
-import type { CheckpointConstantData, RootRollupPublicInputs } from '@aztec/stdlib/rollup';
-import type { BlockHeader, ProcessedTx, Tx } from '@aztec/stdlib/tx';
-import type { UInt64 } from '@aztec/stdlib/types';
-
-import type { ProvingOrchestrator } from '../orchestrator/orchestrator.js';
-import type { BrokerCircuitProverFacade } from '../proving_broker/broker_prover_facade.js';
-
-/** Encapsulates the proving orchestrator and the broker facade */
-export class ServerEpochProver implements EpochProver {
-  constructor(
-    private facade: BrokerCircuitProverFacade,
-    private orchestrator: ProvingOrchestrator,
-  ) {}
-
-  startNewEpoch(
-    epochNumber: EpochNumber,
-    totalNumCheckpoints: number,
-    finalBlobBatchingChallenges: FinalBlobBatchingChallenges,
-  ): void {
-    this.orchestrator.startNewEpoch(epochNumber, totalNumCheckpoints, finalBlobBatchingChallenges);
-    this.facade.start();
-  }
-  startNewCheckpoint(
-    checkpointIndex: number,
-    constants: CheckpointConstantData,
-    l1ToL2Messages: Fr[],
-    totalNumBlocks: number,
-    headerOfLastBlockInPreviousCheckpoint: BlockHeader,
-  ): Promise<void> {
-    return this.orchestrator.startNewCheckpoint(
-      checkpointIndex,
-      constants,
-      l1ToL2Messages,
-      totalNumBlocks,
-      headerOfLastBlockInPreviousCheckpoint,
-    );
-  }
-  startChonkVerifierCircuits(txs: Tx[]): Promise<void> {
-    return this.orchestrator.startChonkVerifierCircuits(txs);
-  }
-  setBlockCompleted(blockNumber: BlockNumber, expectedBlockHeader?: BlockHeader): Promise<BlockHeader> {
-    return this.orchestrator.setBlockCompleted(blockNumber, expectedBlockHeader);
-  }
-  finalizeEpoch(): Promise<{ publicInputs: RootRollupPublicInputs; proof: Proof; batchedBlobInputs: BatchedBlob }> {
-    return this.orchestrator.finalizeEpoch();
-  }
-  cancel(): void {
-    this.orchestrator.cancel();
-  }
-  getProverId(): EthAddress {
-    return this.orchestrator.getProverId();
-  }
-  async stop(): Promise<void> {
-    await this.facade.stop();
-    await this.orchestrator.stop();
-  }
-  startNewBlock(blockNumber: BlockNumber, timestamp: UInt64, totalNumTxs: number): Promise<void> {
-    return this.orchestrator.startNewBlock(blockNumber, timestamp, totalNumTxs);
-  }
-  addTxs(txs: ProcessedTx[]): Promise<void> {
-    return this.orchestrator.addTxs(txs);
-  }
-}