@aztec/prover-client 0.0.1-commit.b655e406 → 0.0.1-commit.b8a057fa

package/dest/orchestrator/epoch-proving-state.js

@@ -1,233 +0,0 @@
-import { BatchedBlobAccumulator } from '@aztec/blob-lib';
-import { UnbalancedTreeStore } from '@aztec/foundation/trees';
-import { CheckpointMergeRollupPrivateInputs, CheckpointPaddingRollupPrivateInputs, RootRollupPrivateInputs } from '@aztec/stdlib/rollup';
-import { toProofData } from './block-building-helpers.js';
-import { CheckpointProvingState } from './checkpoint-proving-state.js';
-var PROVING_STATE_LIFECYCLE = /*#__PURE__*/ function(PROVING_STATE_LIFECYCLE) {
-    PROVING_STATE_LIFECYCLE[PROVING_STATE_LIFECYCLE["PROVING_STATE_CREATED"] = 0] = "PROVING_STATE_CREATED";
-    PROVING_STATE_LIFECYCLE[PROVING_STATE_LIFECYCLE["PROVING_STATE_FULL"] = 1] = "PROVING_STATE_FULL";
-    PROVING_STATE_LIFECYCLE[PROVING_STATE_LIFECYCLE["PROVING_STATE_RESOLVED"] = 2] = "PROVING_STATE_RESOLVED";
-    PROVING_STATE_LIFECYCLE[PROVING_STATE_LIFECYCLE["PROVING_STATE_REJECTED"] = 3] = "PROVING_STATE_REJECTED";
-    return PROVING_STATE_LIFECYCLE;
-}(PROVING_STATE_LIFECYCLE || {});
-/**
- * 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 {
-    epochNumber;
-    totalNumCheckpoints;
-    finalBlobBatchingChallenges;
-    onCheckpointBlobAccumulatorSet;
-    completionCallback;
-    rejectionCallback;
-    checkpointProofs;
-    checkpointPaddingProof;
-    rootRollupProof;
-    checkpoints;
-    startBlobAccumulator;
-    endBlobAccumulator;
-    finalBatchedBlob;
-    provingStateLifecycle;
-    // Map from tx hash to chonk verifier proof promise. Used when kickstarting chonk verifier proofs before tx processing.
-    cachedChonkVerifierProofs;
-    constructor(epochNumber, totalNumCheckpoints, finalBlobBatchingChallenges, onCheckpointBlobAccumulatorSet, completionCallback, rejectionCallback){
-        this.epochNumber = epochNumber;
-        this.totalNumCheckpoints = totalNumCheckpoints;
-        this.finalBlobBatchingChallenges = finalBlobBatchingChallenges;
-        this.onCheckpointBlobAccumulatorSet = onCheckpointBlobAccumulatorSet;
-        this.completionCallback = completionCallback;
-        this.rejectionCallback = rejectionCallback;
-        this.checkpoints = [];
-        this.provingStateLifecycle = 0;
-        this.cachedChonkVerifierProofs = new Map();
-        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
-    startNewCheckpoint(checkpointIndex, constants, totalNumBlocks, totalNumBlobFields, previousBlockHeader, lastArchiveSiblingPath, l1ToL2Messages, lastL1ToL2MessageTreeSnapshot, lastL1ToL2MessageSubtreeRootSiblingPath, newL1ToL2MessageTreeSnapshot, newL1ToL2MessageSubtreeRootSiblingPath) {
-        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, totalNumBlobFields, 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 = 1;
-        }
-        return checkpoint;
-    }
-    getCheckpointProvingState(index) {
-        return this.checkpoints[index];
-    }
-    getCheckpointProvingStateByBlockNumber(blockNumber) {
-        return this.checkpoints.find((c)=>c && blockNumber >= c.firstBlockNumber && blockNumber < c.firstBlockNumber + c.totalNumBlocks);
-    }
-    getBlockProvingStateByBlockNumber(blockNumber) {
-        return this.getCheckpointProvingStateByBlockNumber(blockNumber)?.getBlockProvingStateByBlockNumber(blockNumber);
-    }
-    // Returns true if this proving state is still valid, false otherwise
-    verifyState() {
-        return this.provingStateLifecycle === 0 || this.provingStateLifecycle === 1;
-    }
-    // Returns true if we are still able to accept checkpoints, false otherwise.
-    isAcceptingCheckpoints() {
-        return this.checkpoints.filter((c)=>!!c).length < this.totalNumCheckpoints;
-    }
-    setCheckpointRootRollupProof(checkpointIndex, provingOutput) {
-        return this.checkpointProofs.setLeaf(checkpointIndex, {
-            provingOutput
-        });
-    }
-    tryStartProvingCheckpointMerge(location) {
-        if (this.checkpointProofs.getNode(location)?.isProving) {
-            return false;
-        } else {
-            this.checkpointProofs.setNode(location, {
-                isProving: true
-            });
-            return true;
-        }
-    }
-    setCheckpointMergeRollupProof(location, provingOutput) {
-        this.checkpointProofs.setNode(location, {
-            provingOutput
-        });
-    }
-    tryStartProvingRootRollup() {
-        if (this.rootRollupProof?.isProving) {
-            return false;
-        } else {
-            this.rootRollupProof = {
-                isProving: true
-            };
-            return true;
-        }
-    }
-    setRootRollupProof(provingOutput) {
-        this.rootRollupProof = {
-            provingOutput
-        };
-    }
-    tryStartProvingPaddingCheckpoint() {
-        if (this.checkpointPaddingProof?.isProving) {
-            return false;
-        } else {
-            this.checkpointPaddingProof = {
-                isProving: true
-            };
-            return true;
-        }
-    }
-    setCheckpointPaddingProof(provingOutput) {
-        this.checkpointPaddingProof = {
-            provingOutput
-        };
-    }
-    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;
-            }
-        }
-    }
-    async finalizeBatchedBlob() {
-        if (!this.endBlobAccumulator) {
-            throw new Error('End blob accumulator not ready.');
-        }
-        this.finalBatchedBlob = await this.endBlobAccumulator.finalize(true);
-    }
-    getParentLocation(location) {
-        return this.checkpointProofs.getParentLocation(location);
-    }
-    getCheckpointMergeRollupInputs(mergeLocation) {
-        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)
-        ]);
-    }
-    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)
-            ]
-        });
-    }
-    getPaddingCheckpointInputs() {
-        return new CheckpointPaddingRollupPrivateInputs();
-    }
-    getEpochProofResult() {
-        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
-        };
-    }
-    isReadyForCheckpointMerge(location) {
-        return !!this.checkpointProofs.getSibling(location)?.provingOutput;
-    }
-    // Returns true if we have sufficient inputs to execute the block root rollup
-    isReadyForRootRollup() {
-        const childProofs = this.#getChildProofsForRoot();
-        return childProofs.every((p)=>!!p);
-    }
-    // Attempts to reject the proving state promise with a reason of 'cancelled'
-    cancel() {
-        this.reject('Proving cancelled');
-    }
-    // Attempts to reject the proving state promise with the given reason
-    // Does nothing if not in a valid state
-    reject(reason) {
-        if (!this.verifyState()) {
-            return;
-        }
-        this.provingStateLifecycle = 3;
-        this.rejectionCallback(reason);
-    }
-    // Attempts to resolve the proving state promise with the given result
-    // Does nothing if not in a valid state
-    resolve(result) {
-        if (!this.verifyState()) {
-            return;
-        }
-        this.provingStateLifecycle = 2;
-        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/dest/orchestrator/orchestrator.d.ts

@@ -1,113 +0,0 @@
-import { BatchedBlob, FinalBlobBatchingChallenges } from '@aztec/blob-lib';
-import { Fr } from '@aztec/foundation/fields';
-import { EthAddress } from '@aztec/stdlib/block';
-import type { EpochProver, ForkMerkleTreeOperations, ServerCircuitProver } from '@aztec/stdlib/interfaces/server';
-import type { Proof } from '@aztec/stdlib/proofs';
-import { CheckpointConstantData, RootRollupPublicInputs } from '@aztec/stdlib/rollup';
-import type { BlockHeader, ProcessedTx, Tx } from '@aztec/stdlib/tx';
-import type { UInt64 } from '@aztec/stdlib/types';
-import { type TelemetryClient, type Tracer } from '@aztec/telemetry-client';
-import type { BlockProvingState } from './block-proving-state.js';
-/**
- * Implements an event driven proving scheduler to build the recursive proof tree. The idea being:
- * 1. Transactions are provided to the scheduler post simulation.
- * 2. Tree insertions are performed as required to generate transaction specific proofs
- * 3. Those transaction specific proofs are generated in the necessary order accounting for dependencies
- * 4. Once a transaction is proven, it will be incorporated into a merge proof
- * 5. Merge proofs are produced at each level of the tree until the root proof is produced
- *
- * The proving implementation is determined by the provided prover. This could be for example a local prover or a remote prover pool.
- */
-/**
- * The orchestrator, managing the flow of recursive proving operations required to build the rollup proof tree.
- */
-export declare class ProvingOrchestrator implements EpochProver {
-    private dbProvider;
-    private prover;
-    private readonly proverId;
-    private provingState;
-    private pendingProvingJobs;
-    private provingPromise;
-    private metrics;
-    private dbs;
-    constructor(dbProvider: ForkMerkleTreeOperations, prover: ServerCircuitProver, proverId: EthAddress, telemetryClient?: TelemetryClient);
-    get tracer(): Tracer;
-    getProverId(): EthAddress;
-    stop(): Promise<void>;
-    startNewEpoch(epochNumber: number, totalNumCheckpoints: number, finalBlobBatchingChallenges: FinalBlobBatchingChallenges): void;
-    startNewCheckpoint(checkpointIndex: number, constants: CheckpointConstantData, l1ToL2Messages: Fr[], totalNumBlocks: number, totalNumBlobFields: number, headerOfLastBlockInPreviousCheckpoint: BlockHeader): Promise<void>;
-    /**
-     * Starts off a new block
-     * @param blockNumber - The block number
-     * @param timestamp - The timestamp of the block. This is only required for constructing the private inputs for the
-     * block that doesn't have any txs.
-     * @param totalNumTxs - The total number of txs in the block
-     */
-    startNewBlock(blockNumber: number, timestamp: UInt64, totalNumTxs: number): Promise<void>;
-    /**
-     * The interface to add simulated transactions to the scheduler. This can only be called once per block.
-     * @param txs - The transactions to be proven
-     */
-    addTxs(txs: ProcessedTx[]): Promise<void>;
-    /**
-     * Kickstarts chonk verifier circuits for the specified txs. These will be used during epoch proving.
-     * Note that if the chonk verifier circuits are not started this way, they will be started nontheless after processing.
-     */
-    startChonkVerifierCircuits(txs: Tx[]): Promise<void>;
-    /**
-     * Marks the block as completed.
-     * Computes the block header and updates the archive tree.
-     */
-    setBlockCompleted(blockNumber: number, expectedHeader?: BlockHeader): Promise<BlockHeader>;
-    private buildL2BlockHeader;
-    protected verifyBuiltBlockAgainstSyncedState(provingState: BlockProvingState): Promise<void>;
-    /**
-     * Cancel any further proving
-     */
-    cancel(): void;
-    /**
-     * Returns the proof for the current epoch.
-     */
-    finalizeEpoch(): Promise<{
-        publicInputs: RootRollupPublicInputs;
-        proof: Proof;
-        batchedBlobInputs: BatchedBlob;
-    }>;
-    /**
-     * Enqueue a job to be scheduled
-     * @param provingState - The proving state object being operated on
-     * @param jobType - The type of job to be queued
-     * @param job - The actual job, returns a promise notifying of the job's completion
-     */
-    private deferredProving;
-    private updateL1ToL2MessageTree;
-    private prepareBaseRollupInputs;
-    private enqueueBaseRollup;
-    private getOrEnqueueChonkVerifier;
-    private doEnqueueChonkVerifier;
-    private enqueueMergeRollup;
-    private enqueueBlockRootRollup;
-    private enqueueBaseParityCircuit;
-    private checkAndEnqueueRootParityCircuit;
-    private enqueueRootParityCircuit;
-    private enqueueBlockMergeRollup;
-    private enqueueCheckpointRootRollup;
-    private enqueueCheckpointMergeRollup;
-    private enqueueEpochPadding;
-    private enqueueRootRollup;
-    private checkAndEnqueueNextMergeRollup;
-    private checkAndEnqueueBlockRootRollup;
-    private checkAndEnqueueNextBlockMergeRollup;
-    private checkAndEnqueueCheckpointRootRollup;
-    private checkAndEnqueueNextCheckpointMergeRollup;
-    private checkAndEnqueueRootRollup;
-    /**
-     * Executes the VM circuit for a public function, will enqueue the corresponding kernel if the
-     * previous kernel is ready
-     * @param provingState - The proving state being operated on
-     * @param txIndex - The index of the transaction being proven
-     */
-    private enqueueVM;
-    private checkAndEnqueueBaseRollup;
-}
-//# sourceMappingURL=orchestrator.d.ts.map

package/dest/orchestrator/orchestrator.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"orchestrator.d.ts","sourceRoot":"","sources":["../../src/orchestrator/orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,2BAA2B,EAAc,MAAM,iBAAiB,CAAC;AAUvF,OAAO,EAAE,EAAE,EAAE,MAAM,0BAA0B,CAAC;AAQ9C,OAAO,EAAE,UAAU,EAAwB,MAAM,qBAAqB,CAAC;AACvE,OAAO,KAAK,EACV,WAAW,EACX,wBAAwB,EAGxB,mBAAmB,EACpB,MAAM,iCAAiC,CAAC;AACzC,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAML,sBAAsB,EAKtB,sBAAsB,EACvB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AACrE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAEL,KAAK,eAAe,EACpB,KAAK,MAAM,EAIZ,MAAM,yBAAyB,CAAC;AAgBjC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAQlE;;;;;;;;;GASG;AAEH;;GAEG;AACH,qBAAa,mBAAoB,YAAW,WAAW;IASnD,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAV3B,OAAO,CAAC,YAAY,CAA4C;IAChE,OAAO,CAAC,kBAAkB,CAAyB;IAEnD,OAAO,CAAC,cAAc,CAAiD;IACvE,OAAO,CAAC,OAAO,CAA6B;IAC5C,OAAO,CAAC,GAAG,CAAqD;gBAGtD,UAAU,EAAE,wBAAwB,EACpC,MAAM,EAAE,mBAAmB,EAClB,QAAQ,EAAE,UAAU,EACrC,eAAe,GAAE,eAAsC;IAKzD,IAAI,MAAM,IAAI,MAAM,CAEnB;IAEM,WAAW,IAAI,UAAU;IAIzB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAKrB,aAAa,CAClB,WAAW,EAAE,MAAM,EACnB,mBAAmB,EAAE,MAAM,EAC3B,2BAA2B,EAAE,2BAA2B;IAsB7C,kBAAkB,CAC7B,eAAe,EAAE,MAAM,EACvB,SAAS,EAAE,sBAAsB,EACjC,cAAc,EAAE,EAAE,EAAE,EACpB,cAAc,EAAE,MAAM,EACtB,kBAAkB,EAAE,MAAM,EAC1B,qCAAqC,EAAE,WAAW;IA2CpD;;;;;;OAMG;IAIU,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM;IAwDtF;;;OAGG;IAIU,MAAM,CAAC,GAAG,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqFtD;;;OAGG;IAEI,0BAA0B,CAAC,GAAG,EAAE,EAAE,EAAE;IAwB3C;;;OAGG;IAIU,iBAAiB,CAAC,WAAW,EAAE,MAAM,EAAE,cAAc,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;YAgCzF,kBAAkB;cA6BhB,kCAAkC,CAAC,YAAY,EAAE,iBAAiB;IAuDlF;;OAEG;IACI,MAAM;IAQb;;OAEG;IACU,aAAa,IAAI,OAAO,CAAC;QACpC,YAAY,EAAE,sBAAsB,CAAC;QACrC,KAAK,EAAE,KAAK,CAAC;QACb,iBAAiB,EAAE,WAAW,CAAC;KAChC,CAAC;IAsBF;;;;;OAKG;IACH,OAAO,CAAC,eAAe;YAwDT,uBAAuB;YAmCvB,uBAAuB;IAkCrC,OAAO,CAAC,iBAAiB;IAqDzB,OAAO,CAAC,yBAAyB;IA+BjC,OAAO,CAAC,sBAAsB;IAiC9B,OAAO,CAAC,kBAAkB;IA+B1B,OAAO,CAAC,sBAAsB;IAyD9B,OAAO,CAAC,wBAAwB;IAkChC,OAAO,CAAC,gCAAgC;IAUxC,OAAO,CAAC,wBAAwB;IAgChC,OAAO,CAAC,uBAAuB;IA6B/B,OAAO,CAAC,2BAA2B;IA4DnC,OAAO,CAAC,4BAA4B;IA+BpC,OAAO,CAAC,mBAAmB;IAkC3B,OAAO,CAAC,iBAAiB;IA4BzB,OAAO,CAAC,8BAA8B;IAatC,OAAO,CAAC,8BAA8B;IAStC,OAAO,CAAC,mCAAmC;IAa3C,OAAO,CAAC,mCAAmC;IAQ3C,OAAO,CAAC,wCAAwC;IAahD,OAAO,CAAC,yBAAyB;IASjC;;;;;OAKG;IACH,OAAO,CAAC,SAAS;IAoDjB,OAAO,CAAC,yBAAyB;CAWlC"}