WIP

yarn-project/aztec-node/src/aztec-node/http_rpc_server.ts

@@ -5,8 +5,8 @@ import {
   L2Block,
   LogId,
   NullifierMembershipWitness,
-  ProcessOutput,
   PublicDataWitness,
+  PublicSimulationOutput,
   SiblingPath,
   Tx,
   TxEffect,
@@ -42,7 +42,14 @@ export function createAztecNodeRpcServer(node: AztecNode) {
       PublicDataWitness,
       SiblingPath,
     },
-    { ProcessOutput, Tx, TxReceipt, EncryptedL2BlockL2Logs, UnencryptedL2BlockL2Logs, NullifierMembershipWitness },
+    {
+      PublicSimulationOutput,
+      Tx,
+      TxReceipt,
+      EncryptedL2BlockL2Logs,
+      UnencryptedL2BlockL2Logs,
+      NullifierMembershipWitness,
+    },
     // disable methods not part of the AztecNode interface
     ['start', 'stop'],
   );

yarn-project/aztec-node/src/aztec-node/server.ts

@@ -13,10 +13,10 @@ import {
   LogType,
   MerkleTreeId,
   NullifierMembershipWitness,
-  ProcessOutput,
   type ProverClient,
   type ProverConfig,
   PublicDataWitness,
+  PublicSimulationOutput,
   type SequencerConfig,
   type SiblingPath,
   type Tx,
@@ -634,7 +634,7 @@ export class AztecNodeService implements AztecNode {
    * Simulates the public part of a transaction with the current state.
    * @param tx - The transaction to simulate.
    **/
-  public async simulatePublicCalls(tx: Tx): Promise<ProcessOutput> {
+  public async simulatePublicCalls(tx: Tx): Promise<PublicSimulationOutput> {
     this.log.info(`Simulating tx ${tx.getTxHash()}`);
     const blockNumber = (await this.blockSource.getBlockNumber()) + 1;
 
@@ -674,7 +674,7 @@ export class AztecNodeService implements AztecNode {
     }
     this.log.debug(`Simulated tx ${tx.getTxHash()} succeeds`);
     const [processedTx] = processedTxs;
-    return new ProcessOutput(
+    return new PublicSimulationOutput(
       processedTx.encryptedLogs,
       processedTx.unencryptedLogs,
       processedTx.revertReason,

yarn-project/circuit-types/src/aztec_node/rpc/aztec_node_client.ts

@@ -11,7 +11,7 @@ import { L2Block } from '../../l2_block.js';
 import { EncryptedL2BlockL2Logs, ExtendedUnencryptedL2Log, LogId, UnencryptedL2BlockL2Logs } from '../../logs/index.js';
 import { PublicDataWitness } from '../../public_data_witness.js';
 import { SiblingPath } from '../../sibling_path/index.js';
-import { ProcessOutput, Tx, TxHash, TxReceipt } from '../../tx/index.js';
+import { PublicSimulationOutput, Tx, TxHash, TxReceipt } from '../../tx/index.js';
 import { TxEffect } from '../../tx_effect.js';
 
 /**
@@ -38,7 +38,14 @@ export function createAztecNodeClient(url: string, fetch = defaultFetch): AztecN
       PublicDataWitness,
       SiblingPath,
     },
-    { ProcessOutput, Tx, TxReceipt, EncryptedL2BlockL2Logs, UnencryptedL2BlockL2Logs, NullifierMembershipWitness },
+    {
+      PublicSimulationOutput,
+      Tx,
+      TxReceipt,
+      EncryptedL2BlockL2Logs,
+      UnencryptedL2BlockL2Logs,
+      NullifierMembershipWitness,
+    },
     false,
     'node',
     fetch,

yarn-project/circuit-types/src/interfaces/aztec-node.ts

@@ -22,7 +22,7 @@ import {
 import { type MerkleTreeId } from '../merkle_tree_id.js';
 import { type PublicDataWitness } from '../public_data_witness.js';
 import { type SiblingPath } from '../sibling_path/index.js';
-import { type ProcessOutput, type Tx, type TxHash, type TxReceipt } from '../tx/index.js';
+import { type PublicSimulationOutput, type Tx, type TxHash, type TxReceipt } from '../tx/index.js';
 import { type TxEffect } from '../tx_effect.js';
 import { type SequencerConfig } from './configs.js';
 import { type L2BlockNumber } from './l2_block_number.js';
@@ -283,7 +283,7 @@ export interface AztecNode {
    * This currently just checks that the transaction execution succeeds.
    * @param tx - The transaction to simulate.
    **/
-  simulatePublicCalls(tx: Tx): Promise<ProcessOutput>;
+  simulatePublicCalls(tx: Tx): Promise<PublicSimulationOutput>;
 
   /**
    * Updates the configuration of this node.

yarn-project/circuit-types/src/mocks.ts

@@ -27,7 +27,7 @@ import { type ContractInstanceWithAddress, SerializableContractInstance } from '
 import { EncryptedL2Log } from './logs/encrypted_l2_log.js';
 import { EncryptedFunctionL2Logs, EncryptedTxL2Logs, Note, UnencryptedTxL2Logs } from './logs/index.js';
 import { ExtendedNote } from './notes/index.js';
-import { ProcessOutput, type ProcessReturnValues, SimulatedTx, Tx, TxHash } from './tx/index.js';
+import { type ProcessReturnValues, PublicSimulationOutput, SimulatedTx, Tx, TxHash } from './tx/index.js';
 
 /**
  * Testing utility to create empty logs composed from a single empty log.
@@ -129,7 +129,7 @@ export const mockTxForRollup = (seed = 1, { hasLogs = false }: { hasLogs?: boole
 export const mockSimulatedTx = (seed = 1, hasLogs = true) => {
   const tx = mockTx(seed, { hasLogs });
   const dec: ProcessReturnValues = [new Fr(1n), new Fr(2n), new Fr(3n), new Fr(4n)];
-  const output = new ProcessOutput(
+  const output = new PublicSimulationOutput(
     tx.encryptedLogs,
     tx.unencryptedLogs,
     undefined,

yarn-project/circuit-types/src/tx/index.ts

@@ -3,4 +3,5 @@ export * from './simulated_tx.js';
 export * from './tx_hash.js';
 export * from './tx_receipt.js';
 export * from './processed_tx.js';
+export * from './public_simulation_output.js';
 export * from './tx_validator.js';

yarn-project/circuit-types/src/tx/public_simulation_output.ts

@@ -0,0 +1,48 @@
+import { CombinedAccumulatedData, CombinedConstantData, Fr, Gas } from '@aztec/circuits.js';
+import { mapValues } from '@aztec/foundation/collection';
+
+import { EncryptedTxL2Logs, UnencryptedTxL2Logs } from '../logs/tx_l2_logs.js';
+import { type SimulationError } from '../simulation_error.js';
+import { type PublicKernelType } from './processed_tx.js';
+
+/** Return values of simulating a circuit. */
+export type ProcessReturnValues = Fr[] | undefined;
+
+/**
+ * Outputs of processing the public component of a transaction.
+ */
+export class PublicSimulationOutput {
+  constructor(
+    public encryptedLogs: EncryptedTxL2Logs,
+    public unencryptedLogs: UnencryptedTxL2Logs,
+    public revertReason: SimulationError | undefined,
+    public constants: CombinedConstantData,
+    public end: CombinedAccumulatedData,
+    public publicReturnValues: ProcessReturnValues,
+    public gasUsed: Partial<Record<PublicKernelType, Gas>>,
+  ) {}
+
+  toJSON() {
+    return {
+      encryptedLogs: this.encryptedLogs.toJSON(),
+      unencryptedLogs: this.unencryptedLogs.toJSON(),
+      revertReason: this.revertReason,
+      constants: this.constants.toBuffer().toString('hex'),
+      end: this.end.toBuffer().toString('hex'),
+      publicReturnValues: this.publicReturnValues?.map(fr => fr.toString()),
+      gasUsed: mapValues(this.gasUsed, gas => gas?.toJSON()),
+    };
+  }
+
+  static fromJSON(json: any): PublicSimulationOutput {
+    return new PublicSimulationOutput(
+      EncryptedTxL2Logs.fromJSON(json.encryptedLogs),
+      UnencryptedTxL2Logs.fromJSON(json.unencryptedLogs),
+      json.revertReason,
+      CombinedConstantData.fromBuffer(Buffer.from(json.constants, 'hex')),
+      CombinedAccumulatedData.fromBuffer(Buffer.from(json.end, 'hex')),
+      json.publicReturnValues?.map(Fr.fromString),
+      mapValues(json.gasUsed, gas => (gas ? Gas.fromJSON(gas) : undefined)),
+    );
+  }
+}

yarn-project/circuit-types/src/tx/simulated_tx.ts

@@ -1,58 +1,21 @@
-import { CombinedAccumulatedData, CombinedConstantData, Fr, Gas } from '@aztec/circuits.js';
-import { mapValues } from '@aztec/foundation/collection';
+import { Fr, Gas } from '@aztec/circuits.js';
 
-import { EncryptedTxL2Logs, UnencryptedTxL2Logs } from '../logs/index.js';
-import { type SimulationError } from '../simulation_error.js';
 import { PublicKernelType } from './processed_tx.js';
+import { type ProcessReturnValues, PublicSimulationOutput } from './public_simulation_output.js';
 import { Tx } from './tx.js';
 
-/** Return values of simulating a circuit. */
-export type ProcessReturnValues = Fr[] | undefined;
-
-export class ProcessOutput {
-  constructor(
-    public encryptedLogs: EncryptedTxL2Logs,
-    public unencryptedLogs: UnencryptedTxL2Logs,
-    public revertReason: SimulationError | undefined,
-    public constants: CombinedConstantData,
-    public end: CombinedAccumulatedData,
-    public publicReturnValues: ProcessReturnValues,
-    public gasUsed: Partial<Record<PublicKernelType, Gas>>,
-  ) {}
-
-  toJSON() {
-    return {
-      encryptedLogs: this.encryptedLogs.toJSON(),
-      unencryptedLogs: this.unencryptedLogs.toJSON(),
-      revertReason: this.revertReason,
-      constants: this.constants.toBuffer().toString('hex'),
-      end: this.end.toBuffer().toString('hex'),
-      publicReturnValues: this.publicReturnValues?.map(fr => fr.toString()),
-      gasUsed: mapValues(this.gasUsed, gas => gas?.toJSON()),
-    };
-  }
-
-  static fromJSON(json: any): ProcessOutput {
-    return new ProcessOutput(
-      EncryptedTxL2Logs.fromJSON(json.encryptedLogs),
-      UnencryptedTxL2Logs.fromJSON(json.unencryptedLogs),
-      json.revertReason,
-      CombinedConstantData.fromBuffer(Buffer.from(json.constants, 'hex')),
-      CombinedAccumulatedData.fromBuffer(Buffer.from(json.end, 'hex')),
-      json.publicReturnValues?.map(Fr.fromString),
-      mapValues(json.gasUsed, gas => (gas ? Gas.fromJSON(gas) : undefined)),
-    );
-  }
-}
-
 // REFACTOR: Review what we need to expose to the user when running a simulation.
 // Eg tx already has encrypted and unencrypted logs, but those cover only the ones
 // emitted during private. We need the ones from ProcessOutput to include the public
 // ones as well. However, those would only be present if the user chooses to simulate
 // the public side of things. This also points at this class needing to be split into
 // two: one with just private simulation, and one that also includes public simulation.
 export class SimulatedTx {
-  constructor(public tx: Tx, public privateReturnValues?: ProcessReturnValues, public publicOutput?: ProcessOutput) {}
+  constructor(
+    public tx: Tx,
+    public privateReturnValues?: ProcessReturnValues,
+    public publicOutput?: PublicSimulationOutput,
+  ) {}
 
   /**
    * Returns suggested total and teardown gas limits for the simulated tx.
@@ -96,7 +59,7 @@ export class SimulatedTx {
    */
   public static fromJSON(obj: any) {
     const tx = Tx.fromJSON(obj.tx);
-    const publicOutput = obj.publicOutput ? ProcessOutput.fromJSON(obj.publicOutput) : undefined;
+    const publicOutput = obj.publicOutput ? PublicSimulationOutput.fromJSON(obj.publicOutput) : undefined;
     const privateReturnValues = obj.privateReturnValues?.map(Fr.fromString);
 
     return new SimulatedTx(tx, privateReturnValues, publicOutput);