@aztec/simulator 0.85.0 → 0.86.0

package/src/public/public_tx_simulator/public_tx_context.ts

@@ -9,13 +9,7 @@ import { padArrayEnd } from '@aztec/foundation/collection';
 import { Fr } from '@aztec/foundation/fields';
 import { type Logger, createLogger } from '@aztec/foundation/log';
 import { assertLength } from '@aztec/foundation/serialize';
-import {
-  type AvmCircuitPublicInputs,
-  AvmExecutionHints,
-  AvmTxHint,
-  PublicDataWrite,
-  RevertCode,
-} from '@aztec/stdlib/avm';
+import { type AvmCircuitPublicInputs, PublicDataWrite, RevertCode } from '@aztec/stdlib/avm';
 import type { AztecAddress } from '@aztec/stdlib/aztec-address';
 import type { SimulationError } from '@aztec/stdlib/errors';
 import { computeTransactionFee } from '@aztec/stdlib/fees';
@@ -32,7 +26,6 @@ import { MerkleTreeId } from '@aztec/stdlib/trees';
 import {
   type GlobalVariables,
   PublicCallRequestWithCalldata,
-  type StateReference,
   TreeSnapshots,
   type Tx,
   TxExecutionPhase,
@@ -42,8 +35,7 @@ import {
 import { strict as assert } from 'assert';
 import { inspect } from 'util';
 
-import type { PublicContractsDBInterface } from '../../server.js';
-import { HintingPublicContractsDB, HintingPublicTreesDB } from '../hinting_db_sources.js';
+import type { PublicContractsDBInterface } from '../db_interfaces.js';
 import type { PublicTreesDB } from '../public_db_sources.js';
 import { SideEffectArrayLengths, SideEffectTrace } from '../side_effect_trace.js';
 import { PublicPersistableStateManager } from '../state_manager/state_manager.js';
@@ -66,12 +58,11 @@ export class PublicTxContext {
   private revertCode: RevertCode = RevertCode.OK;
   /* What caused a revert (if one occurred)? */
   public revertReason: SimulationError | undefined;
-
   private constructor(
     public readonly txHash: TxHash,
     public readonly state: PhaseStateManager,
+    private readonly startTreeSnapshots: TreeSnapshots,
     private readonly globalVariables: GlobalVariables,
-    private readonly startStateReference: StateReference,
     private readonly gasSettings: GasSettings,
     private readonly gasUsedByPrivate: Gas,
     private readonly gasAllocatedToPublic: Gas,
@@ -82,7 +73,6 @@ export class PublicTxContext {
     public readonly revertibleAccumulatedDataFromPrivate: PrivateToPublicAccumulatedData,
     public readonly feePayer: AztecAddress,
     private readonly trace: SideEffectTrace,
-    public readonly hints: AvmExecutionHints, // This is public due to enqueued call hinting.
   ) {
     this.log = createLogger(`simulator:public_tx_context`);
   }
@@ -109,22 +99,10 @@ export class PublicTxContext {
 
     const firstNullifier = nonRevertibleAccumulatedDataFromPrivate.nullifiers[0];
 
-    // We wrap the DB to collect AVM hints.
-    const hints = new AvmExecutionHints(await AvmTxHint.fromTx(tx));
-    const hintingContractsDB = new HintingPublicContractsDB(contractsDB, hints);
-    const hintingTreesDB = new HintingPublicTreesDB(treesDB, hints);
-    const startStateReference = await treesDB.getStateReference();
-    hints.startingTreeRoots = new TreeSnapshots(
-      startStateReference.l1ToL2MessageTree,
-      startStateReference.partial.noteHashTree,
-      startStateReference.partial.nullifierTree,
-      startStateReference.partial.publicDataTree,
-    );
-
     // Transaction level state manager that will be forked for revertible phases.
     const txStateManager = PublicPersistableStateManager.create(
-      hintingTreesDB,
-      hintingContractsDB,
+      treesDB,
+      contractsDB,
       trace,
       doMerkleOperations,
       firstNullifier,
@@ -139,8 +117,8 @@ export class PublicTxContext {
     return new PublicTxContext(
       await tx.getTxHash(),
       new PhaseStateManager(txStateManager),
+      await txStateManager.getTreeSnapshots(),
       globalVariables,
-      await treesDB.getStateReference(),
       gasSettings,
       gasUsedByPrivate,
       gasAllocatedToPublic,
@@ -151,7 +129,6 @@ export class PublicTxContext {
       tx.data.forPublic!.revertibleAccumulatedData,
       tx.data.feePayer,
       trace,
-      hints,
     );
   }
 
@@ -324,20 +301,12 @@ export class PublicTxContext {
     assert(this.halted, 'Can only get AvmCircuitPublicInputs after tx execution ends');
     const stateManager = this.state.getActiveStateManager();
 
-    const startTreeSnapshots = new TreeSnapshots(
-      this.startStateReference.l1ToL2MessageTree,
-      this.startStateReference.partial.noteHashTree,
-      this.startStateReference.partial.nullifierTree,
-      this.startStateReference.partial.publicDataTree,
-    );
-
     // FIXME: We are first creating the PIs with the wrong endTreeSnapshots, then patching them.
     // This is because we need to know the lengths of the accumulated data arrays to pad them.
     // We should refactor this to avoid this hack.
     // We should just get the info we need from the trace, and create the rest of the PIs here.
     const avmCircuitPublicInputs = this.trace.toAvmCircuitPublicInputs(
       this.globalVariables,
-      startTreeSnapshots,
       /*startGasUsed=*/ this.gasUsedByPrivate,
       this.gasSettings,
       this.feePayer,
@@ -351,6 +320,7 @@ export class PublicTxContext {
       /*transactionFee=*/ this.getTransactionFeeUnsafe(),
       /*reverted=*/ !this.revertCode.isOK(),
     );
+    avmCircuitPublicInputs.startTreeSnapshots = this.startTreeSnapshots;
 
     const getArrayLengths = (from: PrivateToPublicAccumulatedData) =>
       new PrivateToAvmAccumulatedDataArrayLengths(
@@ -400,18 +370,11 @@ export class PublicTxContext {
     );
     const numNoteHashesToPad =
       MAX_NOTE_HASHES_PER_TX - countAccumulatedItems(avmCircuitPublicInputs.accumulatedData.noteHashes);
-    await stateManager.deprecatedGetTreesForPIGeneration().padTree(MerkleTreeId.NOTE_HASH_TREE, numNoteHashesToPad);
+    await stateManager.padTree(MerkleTreeId.NOTE_HASH_TREE, numNoteHashesToPad);
     const numNullifiersToPad =
       MAX_NULLIFIERS_PER_TX - countAccumulatedItems(avmCircuitPublicInputs.accumulatedData.nullifiers);
-    await stateManager.deprecatedGetTreesForPIGeneration().padTree(MerkleTreeId.NULLIFIER_TREE, numNullifiersToPad);
-
-    const paddedState = await stateManager.deprecatedGetTreesForPIGeneration().getStateReference();
-    avmCircuitPublicInputs.endTreeSnapshots = new TreeSnapshots(
-      paddedState.l1ToL2MessageTree,
-      paddedState.partial.noteHashTree,
-      paddedState.partial.nullifierTree,
-      paddedState.partial.publicDataTree,
-    );
+    await stateManager.padTree(MerkleTreeId.NULLIFIER_TREE, numNullifiersToPad);
+    avmCircuitPublicInputs.endTreeSnapshots = await stateManager.getTreeSnapshots();
 
     return avmCircuitPublicInputs;
   }

package/src/public/public_tx_simulator/public_tx_simulator.ts

@@ -7,11 +7,13 @@ import {
   AvmCircuitPublicInputs,
   AvmExecutionHints,
   type AvmProvingRequest,
+  AvmTxHint,
   type RevertCode,
 } from '@aztec/stdlib/avm';
 import { SimulationError } from '@aztec/stdlib/errors';
 import type { Gas, GasUsed } from '@aztec/stdlib/gas';
 import { ProvingRequestType } from '@aztec/stdlib/proofs';
+import type { MerkleTreeWriteOperations } from '@aztec/stdlib/trees';
 import {
   type GlobalVariables,
   NestedProcessReturnValues,
@@ -22,10 +24,11 @@ import {
 
 import { strict as assert } from 'assert';
 
-import { getPublicFunctionDebugName } from '../../common/debug_fn_name.js';
 import type { AvmFinalizedCallResult } from '../avm/avm_contract_call_result.js';
 import { AvmSimulator } from '../avm/index.js';
-import type { PublicContractsDB, PublicTreesDB } from '../public_db_sources.js';
+import { getPublicFunctionDebugName } from '../debug_fn_name.js';
+import { HintingMerkleWriteOperations, HintingPublicContractsDB } from '../hinting_db_sources.js';
+import { type PublicContractsDB, PublicTreesDB } from '../public_db_sources.js';
 import { NullifierCollisionError } from '../state_manager/nullifiers.js';
 import type { PublicPersistableStateManager } from '../state_manager/state_manager.js';
 import { PublicTxContext } from './public_tx_context.js';
@@ -52,8 +55,8 @@ export class PublicTxSimulator {
   protected log: Logger;
 
   constructor(
-    private treesDB: PublicTreesDB,
-    protected contractsDB: PublicContractsDB,
+    private merkleTree: MerkleTreeWriteOperations,
+    private contractsDB: PublicContractsDB,
     private globalVariables: GlobalVariables,
     private doMerkleOperations: boolean = false,
     private skipFeeEnforcement: boolean = false,
@@ -71,9 +74,15 @@ export class PublicTxSimulator {
       const txHash = await this.computeTxHash(tx);
       this.log.debug(`Simulating ${tx.publicFunctionCalldata.length} public calls for tx ${txHash}`, { txHash });
 
+      // Create hinting DBs.
+      const hints = new AvmExecutionHints(await AvmTxHint.fromTx(tx));
+      const hintingMerkleTree = await HintingMerkleWriteOperations.create(this.merkleTree, hints);
+      const hintingTreesDB = new PublicTreesDB(hintingMerkleTree);
+      const hintingContractsDB = new HintingPublicContractsDB(this.contractsDB, hints);
+
       const context = await PublicTxContext.create(
-        this.treesDB,
-        this.contractsDB,
+        hintingTreesDB,
+        hintingContractsDB,
         tx,
         this.globalVariables,
         this.doMerkleOperations,
@@ -107,7 +116,7 @@ export class PublicTxSimulator {
       await this.payFee(context);
 
       const publicInputs = await context.generateAvmCircuitPublicInputs();
-      const avmProvingRequest = PublicTxSimulator.generateProvingRequest(publicInputs, context.hints);
+      const avmProvingRequest = PublicTxSimulator.generateProvingRequest(publicInputs, hints);
 
       const revertCode = context.getFinalRevertCode();
 
@@ -267,7 +276,7 @@ export class PublicTxSimulator {
     stateManager.traceEnqueuedCall(callRequest.request);
 
     const result = await this.simulateEnqueuedCallInternal(
-      context.state.getActiveStateManager(),
+      stateManager,
       callRequest,
       allocatedGas,
       /*transactionFee=*/ context.getTransactionFee(phase),
@@ -335,7 +344,11 @@ export class PublicTxSimulator {
   protected async insertNonRevertiblesFromPrivate(context: PublicTxContext, tx: Tx) {
     const stateManager = context.state.getActiveStateManager();
     try {
-      await stateManager.writeSiloedNullifiersFromPrivate(context.nonRevertibleAccumulatedDataFromPrivate.nullifiers);
+      for (const siloedNullifier of context.nonRevertibleAccumulatedDataFromPrivate.nullifiers.filter(
+        n => !n.isEmpty(),
+      )) {
+        await stateManager.writeSiloedNullifier(siloedNullifier);
+      }
     } catch (e) {
       if (e instanceof NullifierCollisionError) {
         throw new NullifierCollisionError(
@@ -364,7 +377,9 @@ export class PublicTxSimulator {
     await context.state.fork();
     const stateManager = context.state.getActiveStateManager();
     try {
-      await stateManager.writeSiloedNullifiersFromPrivate(context.revertibleAccumulatedDataFromPrivate.nullifiers);
+      for (const siloedNullifier of context.revertibleAccumulatedDataFromPrivate.nullifiers.filter(n => !n.isEmpty())) {
+        await stateManager.writeSiloedNullifier(siloedNullifier);
+      }
     } catch (e) {
       if (e instanceof NullifierCollisionError) {
         // Instead of throwing, revert the app_logic phase

package/src/public/public_tx_simulator/telemetry_public_tx_simulator.ts

@@ -1,11 +1,12 @@
 import type { Fr } from '@aztec/foundation/fields';
 import type { Gas } from '@aztec/stdlib/gas';
+import type { MerkleTreeWriteOperations } from '@aztec/stdlib/trees';
 import { type GlobalVariables, PublicCallRequestWithCalldata, TxExecutionPhase } from '@aztec/stdlib/tx';
 import { Attributes, type TelemetryClient, type Tracer, getTelemetryClient, trackSpan } from '@aztec/telemetry-client';
 
 import type { AvmFinalizedCallResult } from '../avm/avm_contract_call_result.js';
 import { ExecutorMetrics } from '../executor_metrics.js';
-import type { PublicContractsDB, PublicTreesDB } from '../public_db_sources.js';
+import type { PublicContractsDB } from '../public_db_sources.js';
 import type { PublicPersistableStateManager } from '../state_manager/state_manager.js';
 import { MeasuredPublicTxSimulator } from './measured_public_tx_simulator.js';
 import { PublicTxContext } from './public_tx_context.js';
@@ -18,7 +19,7 @@ export class TelemetryPublicTxSimulator extends MeasuredPublicTxSimulator {
   public readonly tracer: Tracer;
 
   constructor(
-    treesDB: PublicTreesDB,
+    merkleTree: MerkleTreeWriteOperations,
     contractsDB: PublicContractsDB,
     globalVariables: GlobalVariables,
     doMerkleOperations: boolean = false,
@@ -26,7 +27,7 @@ export class TelemetryPublicTxSimulator extends MeasuredPublicTxSimulator {
     telemetryClient: TelemetryClient = getTelemetryClient(),
   ) {
     const metrics = new ExecutorMetrics(telemetryClient, 'PublicTxSimulator');
-    super(treesDB, contractsDB, globalVariables, doMerkleOperations, skipFeeEnforcement, metrics);
+    super(merkleTree, contractsDB, globalVariables, doMerkleOperations, skipFeeEnforcement, metrics);
     this.tracer = metrics.tracer;
   }
 

package/src/public/side_effect_trace.ts

@@ -32,7 +32,7 @@ import {
 } from '@aztec/stdlib/kernel';
 import { PublicLog } from '@aztec/stdlib/logs';
 import { L2ToL1Message, ScopedL2ToL1Message } from '@aztec/stdlib/messaging';
-import type { GlobalVariables, TreeSnapshots } from '@aztec/stdlib/tx';
+import { type GlobalVariables, TreeSnapshots } from '@aztec/stdlib/tx';
 
 import { strict as assert } from 'assert';
 
@@ -277,8 +277,6 @@ export class SideEffectTrace implements PublicSideEffectTraceInterface {
   public toAvmCircuitPublicInputs(
     /** Globals. */
     globalVariables: GlobalVariables,
-    /** Start tree snapshots. */
-    startTreeSnapshots: TreeSnapshots,
     /** Gas used at start of TX. */
     startGasUsed: Gas,
     /** How much gas was available for this public execution. */
@@ -305,7 +303,7 @@ export class SideEffectTrace implements PublicSideEffectTraceInterface {
   ): AvmCircuitPublicInputs {
     return new AvmCircuitPublicInputs(
       globalVariables,
-      startTreeSnapshots,
+      TreeSnapshots.empty(), // will be patched later.
       startGasUsed,
       gasLimits,
       feePayer,

package/src/public/state_manager/state_manager.ts

@@ -11,26 +11,20 @@ import { Fr } from '@aztec/foundation/fields';
 import { jsonStringify } from '@aztec/foundation/json-rpc';
 import { createLogger } from '@aztec/foundation/log';
 import { ProtocolContractAddress } from '@aztec/protocol-contracts';
-import { PublicDataWrite } from '@aztec/stdlib/avm';
 import { AztecAddress } from '@aztec/stdlib/aztec-address';
 import type { ContractClassPublicWithCommitment, ContractInstanceWithAddress } from '@aztec/stdlib/contract';
 import { SerializableContractInstance } from '@aztec/stdlib/contract';
-import {
-  computeNoteHashNonce,
-  computePublicDataTreeLeafSlot,
-  computeUniqueNoteHash,
-  siloNoteHash,
-  siloNullifier,
-} from '@aztec/stdlib/hash';
+import { computeNoteHashNonce, computeUniqueNoteHash, siloNoteHash, siloNullifier } from '@aztec/stdlib/hash';
 import type { PublicCallRequest } from '@aztec/stdlib/kernel';
 import { SharedMutableValues, SharedMutableValuesWithHash } from '@aztec/stdlib/shared-mutable';
 import { MerkleTreeId } from '@aztec/stdlib/trees';
+import type { TreeSnapshots } from '@aztec/stdlib/tx';
 
 import { strict as assert } from 'assert';
 
-import { getPublicFunctionDebugName } from '../../common/debug_fn_name.js';
-import type { PublicContractsDBInterface } from '../../server.js';
 import type { AvmExecutionEnvironment } from '../avm/avm_execution_environment.js';
+import type { PublicContractsDBInterface } from '../db_interfaces.js';
+import { getPublicFunctionDebugName } from '../debug_fn_name.js';
 import type { PublicTreesDB } from '../public_db_sources.js';
 import type { PublicSideEffectTraceInterface } from '../side_effect_trace_interface.js';
 import { NullifierCollisionError, NullifierManager } from './nullifiers.js';
@@ -83,12 +77,6 @@ export class PublicPersistableStateManager {
     );
   }
 
-  // DO NOT USE!
-  // FIXME(fcarreiro): refactor and remove this.
-  public deprecatedGetTreesForPIGeneration() {
-    return this.treesDB;
-  }
-
   /**
    * Create a new state manager forked from this one
    */
@@ -131,14 +119,8 @@ export class PublicPersistableStateManager {
     this.nullifiers.acceptAndMerge(forkedState.nullifiers);
     this.trace.merge(forkedState.trace, reverted);
     if (reverted) {
+      this.log.trace('Reverting forked state...');
       await this.treesDB.revertCheckpoint();
-      if (this.doMerkleOperations) {
-        this.log.trace(
-          `Rolled back nullifier tree to root ${new Fr(
-            (await this.treesDB.getTreeInfo(MerkleTreeId.NULLIFIER_TREE)).root,
-          )}`,
-        );
-      }
     } else {
       this.log.trace('Merging forked state into parent...');
       await this.treesDB.commitCheckpoint();
@@ -153,15 +135,11 @@ export class PublicPersistableStateManager {
    * @param value - the value being written to the slot
    */
   public async writeStorage(contractAddress: AztecAddress, slot: Fr, value: Fr, protocolWrite = false): Promise<void> {
-    const leafSlot = await computePublicDataTreeLeafSlot(contractAddress, slot);
-    this.log.trace(`Storage write (address=${contractAddress}, slot=${slot}): value=${value}, leafSlot=${leafSlot}`);
+    this.log.trace(`Storage write (address=${contractAddress}, slot=${slot}): value=${value}`);
 
     if (this.doMerkleOperations) {
       // write to native merkle trees
-      const publicDataWrite = new PublicDataWrite(leafSlot, value);
-      const result = await this.treesDB.sequentialInsert(MerkleTreeId.PUBLIC_DATA_TREE, [publicDataWrite.toBuffer()]);
-      assert(result !== undefined, 'Public data tree insertion error. You might want to disable doMerkleOperations.');
-      this.log.trace(`Inserted public data tree leaf at leafSlot ${leafSlot}, value: ${value}`);
+      await this.treesDB.storageWrite(contractAddress, slot, value);
     } else {
       // Cache storage writes for later reference/reads
       this.publicStorage.write(contractAddress, slot, value);
@@ -179,8 +157,7 @@ export class PublicPersistableStateManager {
    */
   public async readStorage(contractAddress: AztecAddress, slot: Fr): Promise<Fr> {
     if (this.doMerkleOperations) {
-      const value = await this.treesDB.storageRead(contractAddress, slot);
-      return value;
+      return await this.treesDB.storageRead(contractAddress, slot);
     } else {
       // TODO(fcarreiro): I don't get this. PublicStorage CAN end up reading the tree. Why is it in the "dont do merkle operations" branch?
       const read = await this.publicStorage.read(contractAddress, slot);
@@ -201,8 +178,8 @@ export class PublicPersistableStateManager {
    * @returns true if the note hash exists at the given leaf index, false otherwise
    */
   public async checkNoteHashExists(contractAddress: AztecAddress, noteHash: Fr, leafIndex: Fr): Promise<boolean> {
-    const gotLeafValue = (await this.treesDB.getNoteHash(leafIndex.toBigInt())) ?? Fr.ZERO;
-    const exists = gotLeafValue.equals(noteHash);
+    const gotLeafValue = await this.treesDB.getNoteHash(leafIndex.toBigInt());
+    const exists = gotLeafValue !== undefined && gotLeafValue.equals(noteHash);
     this.log.trace(
       `noteHashes(${contractAddress})@${noteHash} ?? leafIndex: ${leafIndex} | gotLeafValue: ${gotLeafValue}, exists: ${exists}.`,
     );
@@ -236,7 +213,7 @@ export class PublicPersistableStateManager {
   public async writeUniqueNoteHash(uniqueNoteHash: Fr): Promise<void> {
     this.log.trace(`noteHashes += @${uniqueNoteHash}.`);
     if (this.doMerkleOperations) {
-      await this.treesDB.appendLeaves(MerkleTreeId.NOTE_HASH_TREE, [uniqueNoteHash]);
+      await this.treesDB.writeNoteHash(uniqueNoteHash);
     }
     this.trace.traceNewNoteHash(uniqueNoteHash);
   }
@@ -256,7 +233,6 @@ export class PublicPersistableStateManager {
       this.log.trace(`Checked siloed nullifier ${siloedNullifier} (exists=${exists})`);
       return Promise.resolve(exists);
     } else {
-      // TODO: same here, this CAN hit the db.
       const { exists, cacheHit } = await this.nullifiers.checkExists(siloedNullifier);
       this.log.trace(`Checked siloed nullifier ${siloedNullifier} (exists=${exists}), cacheHit=${cacheHit}`);
       return Promise.resolve(exists);
@@ -290,7 +266,7 @@ export class PublicPersistableStateManager {
           `Siloed nullifier ${siloedNullifier} already exists in parent cache or host.`,
         );
       } else {
-        await this.treesDB.sequentialInsert(MerkleTreeId.NULLIFIER_TREE, [siloedNullifier.toBuffer()]);
+        await this.treesDB.writeNullifier(siloedNullifier);
       }
     } else {
       // Cache pending nullifiers for later access
@@ -300,25 +276,15 @@ export class PublicPersistableStateManager {
     this.trace.traceNewNullifier(siloedNullifier);
   }
 
-  public async writeSiloedNullifiersFromPrivate(siloedNullifiers: Fr[]) {
-    for (const siloedNullifier of siloedNullifiers.filter(n => !n.isEmpty())) {
-      await this.writeSiloedNullifier(siloedNullifier);
-    }
-  }
-
   /**
    * Check if an L1 to L2 message exists, trace the check.
    * @param msgHash - the message hash to check existence of
    * @param msgLeafIndex - the message leaf index to use in the check
    * @returns exists - whether the message exists in the L1 to L2 Messages tree
    */
-  public async checkL1ToL2MessageExists(
-    contractAddress: AztecAddress,
-    msgHash: Fr,
-    msgLeafIndex: Fr,
-  ): Promise<boolean> {
-    const valueAtIndex = (await this.treesDB.getL1ToL2LeafValue(msgLeafIndex.toBigInt())) ?? Fr.ZERO;
-    const exists = valueAtIndex.equals(msgHash);
+  public async checkL1ToL2MessageExists(msgHash: Fr, msgLeafIndex: Fr): Promise<boolean> {
+    const valueAtIndex = await this.treesDB.getL1ToL2LeafValue(msgLeafIndex.toBigInt());
+    const exists = valueAtIndex !== undefined && valueAtIndex.equals(msgHash);
     this.log.trace(
       `l1ToL2Messages(@${msgLeafIndex}) ?? exists: ${exists}, expected: ${msgHash}, found: ${valueAtIndex}.`,
     );
@@ -375,7 +341,7 @@ export class PublicPersistableStateManager {
     );
     assert(
       exists == nullifierExistsInTree,
-      'treesDB contains contract instance, but nullifier tree does not contain contract address (or vice versa).... This is a bug!',
+      `Contract instance for address ${contractAddress} in DB: ${exists} != nullifier tree: ${nullifierExistsInTree}. This is a bug!`,
     );
 
     // All that is left is tocheck that the contract updatability information is correct.
@@ -501,6 +467,14 @@ export class PublicPersistableStateManager {
   public async getPublicFunctionDebugName(avmEnvironment: AvmExecutionEnvironment): Promise<string> {
     return await getPublicFunctionDebugName(this.contractsDB, avmEnvironment.address, avmEnvironment.calldata);
   }
+
+  public async padTree(treeId: MerkleTreeId, leavesToInsert: number): Promise<void> {
+    await this.treesDB.padTree(treeId, leavesToInsert);
+  }
+
+  public async getTreeSnapshots(): Promise<TreeSnapshots> {
+    return await this.treesDB.getTreeSnapshots();
+  }
 }
 
 function contractAddressIsCanonical(contractAddress: AztecAddress): boolean {

package/dest/public/avm/fixtures/simple_contract_data_source.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"simple_contract_data_source.d.ts","sourceRoot":"","sources":["../../../../src/public/avm/fixtures/simple_contract_data_source.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,0BAA0B,CAAC;AAEnD,OAAO,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAC5E,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,KAAK,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AAInH;;;;;;GAMG;AACH,qBAAa,wBAAyB,YAAW,kBAAkB;IAC1D,MAAM,yCAA+C;IAG5D,OAAO,CAAC,eAAe,CAA+C;IAEtE,OAAO,CAAC,iBAAiB,CAAuD;IAEhF,OAAO,CAAC,iBAAiB,CAA4C;IAErE,OAAO,CAAC,iBAAiB,CAAkC;IAI3D;;;OAGG;IACG,cAAc,CAClB,gBAAgB,EAAE,gBAAgB,EAClC,aAAa,EAAE,mBAAmB,EAClC,gBAAgB,EAAE,2BAA2B;IAOzC,mBAAmB,CAAC,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,gBAAgB;IAkBjE,cAAc,IAAI,OAAO,CAAC,MAAM,CAAC;IAIjC,gBAAgB,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC;IAIlE,qBAAqB,CAAC,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,EAAE,GAAG,SAAS,CAAC;IAIvD,WAAW,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,2BAA2B,GAAG,SAAS,CAAC;IAIpF,mBAAmB,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;IAI9B,mBAAmB,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAWjF,oBAAoB,CAAC,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,gBAAgB,GAAG,OAAO,CAAC,MAAM,CAAC;IAiB9F,kCAAkC,CAAC,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIhG,gBAAgB,CAAC,aAAa,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAKnE,mBAAmB,CAAC,gBAAgB,EAAE,2BAA2B,GAAG,OAAO,CAAC,IAAI,CAAC;CAIlF"}