@aztec/prover-client 0.0.0-test.0 → 0.0.1-commit.0208eb9

package/src/bin/get-proof-inputs.ts

@@ -1,59 +0,0 @@
-/* eslint-disable no-console */
-import { AVM_INPUTS_FILENAME } from '@aztec/bb-prover';
-import { jsonParseWithSchema, jsonStringify } from '@aztec/foundation/json-rpc';
-import { createLogger } from '@aztec/foundation/log';
-import { type ProofUri, ProvingJobInputs } from '@aztec/stdlib/interfaces/server';
-import { ProvingRequestType } from '@aztec/stdlib/proofs';
-
-import { mkdirSync, writeFileSync } from 'fs';
-
-import { createProofStoreForUri } from '../proving_broker/index.js';
-
-const logger = createLogger('prover-client:get-proof-inputs');
-
-function printUsage() {
-  console.error('Usage: get-proof-inputs <proof-uri> [out-dir=.]');
-}
-
-async function main() {
-  if (process.argv[2] === '--help') {
-    printUsage();
-    return;
-  }
-
-  const uri = process.argv[2];
-  const outDir = process.argv[3] || '.';
-  if (!uri) {
-    printUsage();
-    throw new Error('Missing proof URI');
-  }
-
-  mkdirSync(outDir, { recursive: true });
-
-  const proofStore = createProofStoreForUri(uri);
-  logger.info(`Processing uri ${uri}`);
-  const input = await proofStore.getProofInput(uri as ProofUri);
-  logger.info(`Found inputs for ${ProvingRequestType[input.type]}`);
-  writeProofInputs(input, outDir);
-
-  console.log((await jsonParseWithSchema(jsonStringify(input), ProvingJobInputs)).inputs);
-}
-
-// This mimics the behavior of bb-prover/src/bb/execute.ts
-function writeProofInputs(input: ProvingJobInputs, outDir: string) {
-  switch (input.type) {
-    case ProvingRequestType.PUBLIC_VM: {
-      writeFileSync(`${outDir}/${AVM_INPUTS_FILENAME}`, input.inputs.serializeWithMessagePack());
-      logger.info(`Wrote AVM inputs to ${AVM_INPUTS_FILENAME}`);
-      break;
-    }
-    default: {
-      throw new Error(`Unimplemented proving request type: ${ProvingRequestType[input.type]}`);
-    }
-  }
-}
-
-main().catch(err => {
-  console.error(err);
-  process.exit(1);
-});

package/src/block_builder/index.ts

@@ -1,6 +0,0 @@
-import type { BlockBuilder, MerkleTreeReadOperations } from '@aztec/stdlib/interfaces/server';
-
-export * from './light.js';
-export interface BlockBuilderFactory {
-  create(db: MerkleTreeReadOperations): BlockBuilder;
-}

package/src/block_builder/light.ts

@@ -1,101 +0,0 @@
-import { SpongeBlob } from '@aztec/blob-lib';
-import { NUMBER_OF_L1_L2_MESSAGES_PER_ROLLUP } from '@aztec/constants';
-import { padArrayEnd } from '@aztec/foundation/collection';
-import { Fr } from '@aztec/foundation/fields';
-import { createLogger } from '@aztec/foundation/log';
-import { L2Block } from '@aztec/stdlib/block';
-import type { BlockBuilder, MerkleTreeWriteOperations } from '@aztec/stdlib/interfaces/server';
-import { MerkleTreeId } from '@aztec/stdlib/trees';
-import { type GlobalVariables, type ProcessedTx, toNumBlobFields } from '@aztec/stdlib/tx';
-import { type TelemetryClient, getTelemetryClient } from '@aztec/telemetry-client';
-
-import {
-  buildBaseRollupHints,
-  buildHeaderAndBodyFromTxs,
-  getTreeSnapshot,
-} from '../orchestrator/block-building-helpers.js';
-
-/**
- * Builds a block and its header from a set of processed tx without running any circuits.
- */
-export class LightweightBlockBuilder implements BlockBuilder {
-  private spongeBlobState?: SpongeBlob;
-  private globalVariables?: GlobalVariables;
-  private l1ToL2Messages?: Fr[];
-
-  private txs: ProcessedTx[] = [];
-
-  private readonly logger = createLogger('prover-client:block_builder');
-
-  constructor(private db: MerkleTreeWriteOperations, private telemetry: TelemetryClient = getTelemetryClient()) {}
-
-  async startNewBlock(globalVariables: GlobalVariables, l1ToL2Messages: Fr[]): Promise<void> {
-    this.logger.debug('Starting new block', { globalVariables: globalVariables.toInspect(), l1ToL2Messages });
-    this.globalVariables = globalVariables;
-    this.l1ToL2Messages = padArrayEnd(l1ToL2Messages, Fr.ZERO, NUMBER_OF_L1_L2_MESSAGES_PER_ROLLUP);
-    this.txs = [];
-    this.spongeBlobState = undefined;
-
-    // Update L1 to L2 tree
-    await this.db.appendLeaves(MerkleTreeId.L1_TO_L2_MESSAGE_TREE, this.l1ToL2Messages!);
-  }
-
-  async addTxs(txs: ProcessedTx[]): Promise<void> {
-    this.spongeBlobState = SpongeBlob.init(toNumBlobFields(txs));
-    for (const tx of txs) {
-      this.logger.debug('Adding new tx to block', { txHash: tx.hash.toString() });
-      this.txs.push(tx);
-      await buildBaseRollupHints(tx, this.globalVariables!, this.db, this.spongeBlobState!);
-    }
-  }
-
-  setBlockCompleted(): Promise<L2Block> {
-    return this.buildBlock();
-  }
-
-  private async buildBlock(): Promise<L2Block> {
-    const { header, body } = await buildHeaderAndBodyFromTxs(
-      this.txs,
-      this.globalVariables!,
-      this.l1ToL2Messages!,
-      this.db,
-    );
-
-    await this.db.updateArchive(header);
-    const newArchive = await getTreeSnapshot(MerkleTreeId.ARCHIVE, this.db);
-
-    const block = new L2Block(newArchive, header, body);
-    this.logger.debug(`Built block ${block.number}`, {
-      globalVariables: this.globalVariables?.toInspect(),
-      archiveRoot: newArchive.root.toString(),
-      blockHash: block.hash.toString(),
-    });
-
-    return block;
-  }
-}
-
-export class LightweightBlockBuilderFactory {
-  constructor(private telemetry: TelemetryClient = getTelemetryClient()) {}
-
-  create(db: MerkleTreeWriteOperations): BlockBuilder {
-    return new LightweightBlockBuilder(db, this.telemetry);
-  }
-}
-
-/**
- * Creates a block builder under the hood with the given txs and messages and creates a block.
- * @param db - A db fork to use for block building.
- */
-export async function buildBlock(
-  txs: ProcessedTx[],
-  globalVariables: GlobalVariables,
-  l1ToL2Messages: Fr[],
-  db: MerkleTreeWriteOperations,
-  telemetry: TelemetryClient = getTelemetryClient(),
-) {
-  const builder = new LightweightBlockBuilder(db, telemetry);
-  await builder.startNewBlock(globalVariables, l1ToL2Messages);
-  await builder.addTxs(txs);
-  return await builder.setBlockCompleted();
-}

package/src/proving_broker/proof_store/gcs_proof_store.ts

@@ -1,72 +0,0 @@
-import {
-  type ProofUri,
-  type ProvingJobId,
-  type ProvingJobInputs,
-  type ProvingJobInputsMap,
-  type ProvingJobResult,
-  type ProvingJobResultsMap,
-  getProvingJobInputClassFor,
-} from '@aztec/stdlib/interfaces/server';
-import { ProvingRequestType } from '@aztec/stdlib/proofs';
-
-import { Storage } from '@google-cloud/storage';
-import { join } from 'path';
-
-import type { ProofStore } from './proof_store.js';
-
-const INPUTS_PATH = 'inputs';
-
-export class GoogleCloudStorageProofStore implements ProofStore {
-  private readonly storage: Storage;
-
-  constructor(private readonly bucketName: string, private readonly path: string) {
-    this.storage = new Storage();
-  }
-
-  public async saveProofInput<T extends ProvingRequestType>(
-    id: ProvingJobId,
-    type: T,
-    inputs: ProvingJobInputsMap[T],
-  ): Promise<ProofUri> {
-    const path = join(this.path, INPUTS_PATH, ProvingRequestType[type], id);
-    const file = this.storage.bucket(this.bucketName).file(path);
-    await file.save(inputs.toBuffer());
-    return file.cloudStorageURI.toString() as ProofUri;
-  }
-
-  saveProofOutput<T extends ProvingRequestType>(
-    _id: ProvingJobId,
-    _type: T,
-    _result: ProvingJobResultsMap[T],
-  ): Promise<ProofUri> {
-    throw new Error('Not implemented');
-  }
-
-  public async getProofInput(uri: ProofUri): Promise<ProvingJobInputs> {
-    try {
-      const url = new URL(uri);
-      const bucket = this.storage.bucket(url.host);
-      const path = url.pathname.replace(/^\/+/, '');
-      const file = bucket.file(path);
-      if (!(await file.exists())) {
-        throw new Error(`File at ${uri} does not exist`);
-      }
-
-      const typeString = path.split('/').at(-2);
-      const type = typeString ? ProvingRequestType[typeString as keyof typeof ProvingRequestType] : undefined;
-      if (type === undefined) {
-        throw new Error(`Unrecognized proof type ${type} in path ${path}`);
-      }
-
-      const contents = await file.download();
-      const inputs = getProvingJobInputClassFor(type).fromBuffer(contents[0]);
-      return { inputs, type } as ProvingJobInputs;
-    } catch (err) {
-      throw new Error(`Error getting proof input at ${uri}: ${err}`);
-    }
-  }
-
-  getProofOutput(_uri: ProofUri): Promise<ProvingJobResult> {
-    throw new Error('Not implemented');
-  }
-}

package/src/proving_broker/proving_agent_instrumentation.ts

@@ -1,21 +0,0 @@
-import type { Timer } from '@aztec/foundation/timer';
-import { type Histogram, Metrics, type TelemetryClient, ValueType } from '@aztec/telemetry-client';
-
-export class ProvingAgentInstrumentation {
-  private idleTime: Histogram;
-
-  constructor(client: TelemetryClient, name = 'ProvingAgent') {
-    const meter = client.getMeter(name);
-
-    this.idleTime = meter.createHistogram(Metrics.PROVING_AGENT_IDLE, {
-      description: 'Records how long an agent was idle',
-      unit: 's',
-      valueType: ValueType.DOUBLE,
-    });
-  }
-
-  recordIdleTime(msOrTimer: Timer | number) {
-    const duration = typeof msOrTimer === 'number' ? msOrTimer : msOrTimer.ms();
-    this.idleTime.record(duration / 1000);
-  }
-}