@aztec/pxe 0.0.1-commit.b655e406 → 0.0.1-commit.c7c42ec

package/dest/pxe.js

@@ -1,70 +1,77 @@
-import { Fr } from '@aztec/foundation/fields';
+import { Fr } from '@aztec/foundation/curves/bn254';
 import { createLogger } from '@aztec/foundation/log';
 import { SerialQueue } from '@aztec/foundation/queue';
 import { Timer } from '@aztec/foundation/timer';
 import { KeyStore } from '@aztec/key-store';
 import { L2TipsKVStore } from '@aztec/kv-store/stores';
 import { protocolContractNames } from '@aztec/protocol-contracts';
-import { FunctionSelector, FunctionType, decodeFromAbi, decodeFunctionSignature, encodeArguments } from '@aztec/stdlib/abi';
+import { FunctionType, decodeFunctionSignature } from '@aztec/stdlib/abi';
 import { computeContractAddressFromInstance, getContractClassFromArtifact } from '@aztec/stdlib/contract';
 import { SimulationError } from '@aztec/stdlib/errors';
-import { siloNullifier } from '@aztec/stdlib/hash';
-import { UniqueNote } from '@aztec/stdlib/note';
+import { computeProtocolNullifier, siloNullifier } from '@aztec/stdlib/hash';
 import { PrivateSimulationResult, TxProfileResult, TxProvingResult, TxSimulationResult } from '@aztec/stdlib/tx';
 import { inspect } from 'util';
+import { BlockSynchronizer } from './block_synchronizer/index.js';
 import { ContractFunctionSimulator, generateSimulatedProvingResult } from './contract_function_simulator/contract_function_simulator.js';
 import { readCurrentClassId } from './contract_function_simulator/oracle/private_execution.js';
-import { ProxiedContractDataProviderFactory } from './contract_function_simulator/proxied_contract_data_source.js';
-import { ProxiedNodeFactory } from './contract_function_simulator/proxied_node.js';
-import { PXEOracleInterface } from './contract_function_simulator/pxe_oracle_interface.js';
+import { ProxiedContractStoreFactory } from './contract_function_simulator/proxied_contract_data_source.js';
+import { PXEDebugUtils } from './debug/pxe_debug_utils.js';
 import { enrichPublicSimulationError, enrichSimulationError } from './error_enriching.js';
+import { PrivateEventFilterValidator } from './events/private_event_filter_validator.js';
 import { PrivateKernelExecutionProver } from './private_kernel/private_kernel_execution_prover.js';
 import { PrivateKernelOracleImpl } from './private_kernel/private_kernel_oracle_impl.js';
-import { AddressDataProvider } from './storage/address_data_provider/address_data_provider.js';
-import { CapsuleDataProvider } from './storage/capsule_data_provider/capsule_data_provider.js';
-import { ContractDataProvider } from './storage/contract_data_provider/contract_data_provider.js';
-import { NoteDataProvider } from './storage/note_data_provider/note_data_provider.js';
-import { PrivateEventDataProvider } from './storage/private_event_data_provider/private_event_data_provider.js';
-import { SyncDataProvider } from './storage/sync_data_provider/sync_data_provider.js';
-import { TaggingDataProvider } from './storage/tagging_data_provider/tagging_data_provider.js';
-import { Synchronizer } from './synchronizer/index.js';
+import { AddressStore } from './storage/address_store/address_store.js';
+import { AnchorBlockStore } from './storage/anchor_block_store/anchor_block_store.js';
+import { CapsuleStore } from './storage/capsule_store/capsule_store.js';
+import { ContractStore } from './storage/contract_store/contract_store.js';
+import { NoteStore } from './storage/note_store/note_store.js';
+import { PrivateEventStore } from './storage/private_event_store/private_event_store.js';
+import { RecipientTaggingStore } from './storage/tagging_store/recipient_tagging_store.js';
+import { SenderAddressBookStore } from './storage/tagging_store/sender_address_book_store.js';
+import { SenderTaggingStore } from './storage/tagging_store/sender_tagging_store.js';
 /**
  * Private eXecution Environment (PXE) is a library used by wallets to simulate private phase of transactions and to
  * manage private state of users.
  */ export class PXE {
     node;
-    synchronizer;
+    blockStateSynchronizer;
     keyStore;
-    contractDataProvider;
-    noteDataProvider;
-    capsuleDataProvider;
-    syncDataProvider;
-    taggingDataProvider;
-    addressDataProvider;
-    privateEventDataProvider;
+    contractStore;
+    noteStore;
+    capsuleStore;
+    anchorBlockStore;
+    senderTaggingStore;
+    senderAddressBookStore;
+    recipientTaggingStore;
+    addressStore;
+    privateEventStore;
     simulator;
     proverEnabled;
     proofCreator;
     protocolContractsProvider;
     log;
     jobQueue;
-    constructor(node, synchronizer, keyStore, contractDataProvider, noteDataProvider, capsuleDataProvider, syncDataProvider, taggingDataProvider, addressDataProvider, privateEventDataProvider, simulator, proverEnabled, proofCreator, protocolContractsProvider, log, jobQueue){
+    debug;
+    constructor(node, blockStateSynchronizer, keyStore, contractStore, noteStore, capsuleStore, anchorBlockStore, senderTaggingStore, senderAddressBookStore, recipientTaggingStore, addressStore, privateEventStore, simulator, proverEnabled, proofCreator, protocolContractsProvider, log, jobQueue, debug){
         this.node = node;
-        this.synchronizer = synchronizer;
+        this.blockStateSynchronizer = blockStateSynchronizer;
         this.keyStore = keyStore;
-        this.contractDataProvider = contractDataProvider;
-        this.noteDataProvider = noteDataProvider;
-        this.capsuleDataProvider = capsuleDataProvider;
-        this.syncDataProvider = syncDataProvider;
-        this.taggingDataProvider = taggingDataProvider;
-        this.addressDataProvider = addressDataProvider;
-        this.privateEventDataProvider = privateEventDataProvider;
+        this.contractStore = contractStore;
+        this.noteStore = noteStore;
+        this.capsuleStore = capsuleStore;
+        this.anchorBlockStore = anchorBlockStore;
+        this.senderTaggingStore = senderTaggingStore;
+        this.senderAddressBookStore = senderAddressBookStore;
+        this.recipientTaggingStore = recipientTaggingStore;
+        this.addressStore = addressStore;
+        this.privateEventStore = privateEventStore;
         this.simulator = simulator;
         this.proverEnabled = proverEnabled;
         this.proofCreator = proofCreator;
         this.protocolContractsProvider = protocolContractsProvider;
         this.log = log;
         this.jobQueue = jobQueue;
+        this.debug = debug;
     }
     /**
    * Creates an instance of a PXE by instantiating all the necessary data providers and services.
@@ -75,18 +82,22 @@ import { Synchronizer } from './synchronizer/index.js';
    */ static async create(node, store, proofCreator, simulator, protocolContractsProvider, config, loggerOrSuffix) {
         const log = !loggerOrSuffix || typeof loggerOrSuffix === 'string' ? createLogger(loggerOrSuffix ? `pxe:service:${loggerOrSuffix}` : `pxe:service`) : loggerOrSuffix;
         const proverEnabled = !!config.proverEnabled;
-        const addressDataProvider = new AddressDataProvider(store);
-        const privateEventDataProvider = new PrivateEventDataProvider(store);
-        const contractDataProvider = new ContractDataProvider(store);
-        const noteDataProvider = await NoteDataProvider.create(store);
-        const syncDataProvider = new SyncDataProvider(store);
-        const taggingDataProvider = new TaggingDataProvider(store);
-        const capsuleDataProvider = new CapsuleDataProvider(store);
+        const addressStore = new AddressStore(store);
+        const privateEventStore = new PrivateEventStore(store);
+        const contractStore = new ContractStore(store);
+        const noteStore = await NoteStore.create(store);
+        const anchorBlockStore = new AnchorBlockStore(store);
+        const senderTaggingStore = new SenderTaggingStore(store);
+        const senderAddressBookStore = new SenderAddressBookStore(store);
+        const recipientTaggingStore = new RecipientTaggingStore(store);
+        const capsuleStore = new CapsuleStore(store);
         const keyStore = new KeyStore(store);
         const tipsStore = new L2TipsKVStore(store, 'pxe');
-        const synchronizer = new Synchronizer(node, syncDataProvider, noteDataProvider, taggingDataProvider, tipsStore, config, loggerOrSuffix);
+        const synchronizer = new BlockSynchronizer(node, anchorBlockStore, noteStore, tipsStore, config, loggerOrSuffix);
+        const debugUtils = new PXEDebugUtils(contractStore, noteStore);
         const jobQueue = new SerialQueue();
-        const pxe = new PXE(node, synchronizer, keyStore, contractDataProvider, noteDataProvider, capsuleDataProvider, syncDataProvider, taggingDataProvider, addressDataProvider, privateEventDataProvider, simulator, proverEnabled, proofCreator, protocolContractsProvider, log, jobQueue);
+        const pxe = new PXE(node, synchronizer, keyStore, contractStore, noteStore, capsuleStore, anchorBlockStore, senderTaggingStore, senderAddressBookStore, recipientTaggingStore, addressStore, privateEventStore, simulator, proverEnabled, proofCreator, protocolContractsProvider, log, jobQueue, debugUtils);
+        debugUtils.setPXE(pxe);
         pxe.jobQueue.start();
         await pxe.#registerProtocolContracts();
         const info = await node.getNodeInfo();
@@ -95,8 +106,8 @@ import { Synchronizer } from './synchronizer/index.js';
     }
     // Internal methods
     #getSimulatorForTx(overrides) {
-        const pxeOracleInterface = new PXEOracleInterface(ProxiedNodeFactory.create(this.node), this.keyStore, ProxiedContractDataProviderFactory.create(this.contractDataProvider, overrides?.contracts), this.noteDataProvider, this.capsuleDataProvider, this.syncDataProvider, this.taggingDataProvider, this.addressDataProvider, this.privateEventDataProvider, this.log);
-        return new ContractFunctionSimulator(pxeOracleInterface, this.simulator);
+        const proxyContractStore = ProxiedContractStoreFactory.create(this.contractStore, overrides?.contracts);
+        return new ContractFunctionSimulator(proxyContractStore, this.noteStore, this.keyStore, this.addressStore, this.node, this.anchorBlockStore, this.senderTaggingStore, this.recipientTaggingStore, this.senderAddressBookStore, this.capsuleStore, this.privateEventStore, this.simulator);
     }
     #contextualizeError(err, ...context) {
         let contextStr = '';
@@ -127,8 +138,8 @@ import { Synchronizer } from './synchronizer/index.js';
         const registered = {};
         for (const name of protocolContractNames){
             const { address, contractClass, instance, artifact } = await this.protocolContractsProvider.getProtocolContractArtifact(name);
-            await this.contractDataProvider.addContractArtifact(contractClass.id, artifact);
-            await this.contractDataProvider.addContractInstance(instance);
+            await this.contractStore.addContractArtifact(contractClass.id, artifact);
+            await this.contractStore.addContractInstance(instance);
             registered[name] = address.toString();
         }
         this.log.verbose(`Registered protocol contracts in pxe`, registered);
@@ -143,39 +154,20 @@ import { Synchronizer } from './synchronizer/index.js';
         const initNullifier = await siloNullifier(address, address.toField());
         return !!await this.node.getNullifierMembershipWitness('latest', initNullifier);
     }
-    async #getFunctionCall(functionName, args, to) {
-        const contract = await this.contractDataProvider.getContract(to);
-        if (!contract) {
-            throw new Error(`Unknown contract ${to}: add it to PXE by calling server.addContracts(...).\nSee docs for context: https://docs.aztec.network/developers/resources/debugging/aztecnr-errors#unknown-contract-0x0-add-it-to-pxe-by-calling-serveraddcontracts`);
-        }
-        const functionDao = contract.functions.find((f)=>f.name === functionName);
-        if (!functionDao) {
-            throw new Error(`Unknown function ${functionName} in contract ${contract.name}.`);
-        }
-        return {
-            name: functionDao.name,
-            args: encodeArguments(functionDao, args),
-            selector: await FunctionSelector.fromNameAndParameters(functionDao.name, functionDao.parameters),
-            type: functionDao.functionType,
-            to,
-            hideMsgSender: false,
-            isStatic: functionDao.isStatic,
-            returnTypes: functionDao.returnTypes
-        };
-    }
     // Executes the entrypoint private function, as well as all nested private
     // functions that might arise.
     async #executePrivate(contractFunctionSimulator, txRequest, scopes) {
         const { origin: contractAddress, functionSelector } = txRequest;
         try {
-            const result = await contractFunctionSimulator.run(txRequest, contractAddress, functionSelector, undefined, // The sender for tags is set by contracts, typically by an account
+            const anchorBlockHeader = await this.anchorBlockStore.getBlockHeader();
+            const result = await contractFunctionSimulator.run(txRequest, contractAddress, functionSelector, undefined, anchorBlockHeader, // The sender for tags is set by contracts, typically by an account
             // contract entrypoint
             undefined, scopes);
             this.log.debug(`Private simulation completed for ${contractAddress.toString()}:${functionSelector}`);
             return result;
         } catch (err) {
             if (err instanceof SimulationError) {
-                await enrichSimulationError(err, this.contractDataProvider, this.log);
+                await enrichSimulationError(err, this.contractStore, this.log);
             }
             throw err;
         }
@@ -190,10 +182,11 @@ import { Synchronizer } from './synchronizer/index.js';
    * @returns The simulation result containing the outputs of the utility function.
    */ async #simulateUtility(contractFunctionSimulator, call, authWitnesses, scopes) {
         try {
-            return contractFunctionSimulator.runUtility(call, authWitnesses ?? [], scopes);
+            const anchorBlockHeader = await this.anchorBlockStore.getBlockHeader();
+            return contractFunctionSimulator.runUtility(call, authWitnesses ?? [], anchorBlockHeader, scopes);
         } catch (err) {
             if (err instanceof SimulationError) {
-                await enrichSimulationError(err, this.contractDataProvider, this.log);
+                await enrichSimulationError(err, this.contractStore, this.log);
             }
             throw err;
         }
@@ -215,7 +208,7 @@ import { Synchronizer } from './synchronizer/index.js';
         } catch (err) {
             if (err instanceof SimulationError) {
                 try {
-                    await enrichPublicSimulationError(err, this.contractDataProvider, this.log);
+                    await enrichPublicSimulationError(err, this.contractStore, this.log);
                 } catch (enrichErr) {
                     this.log.error(`Failed to enrich public simulation error: ${enrichErr}`);
                 }
@@ -235,14 +228,14 @@ import { Synchronizer } from './synchronizer/index.js';
    * @returns An object that contains the output of the kernel execution, including the ChonkProof if proving is enabled.
    */ async #prove(txExecutionRequest, proofCreator, privateExecutionResult, config) {
         const simulationAnchorBlock = privateExecutionResult.getSimulationAnchorBlockNumber();
-        const kernelOracle = new PrivateKernelOracleImpl(this.contractDataProvider, this.keyStore, this.node, simulationAnchorBlock);
+        const kernelOracle = new PrivateKernelOracleImpl(this.contractStore, this.keyStore, this.node, simulationAnchorBlock);
         const kernelTraceProver = new PrivateKernelExecutionProver(kernelOracle, proofCreator, !this.proverEnabled);
         this.log.debug(`Executing kernel trace prover (${JSON.stringify(config)})...`);
         return await kernelTraceProver.proveWithKernels(txExecutionRequest.toTxRequest(), privateExecutionResult, config);
     }
     // Public API
     getContractInstance(address) {
-        return this.contractDataProvider.getContractInstance(address);
+        return this.contractStore.getContractInstance(address);
     }
     /**
    * Returns the contract class metadata given a contract class id.
@@ -258,7 +251,7 @@ import { Synchronizer } from './synchronizer/index.js';
    * during a public deployment. We probably want a nicer and more general API for this, but it'll have to
    * do for the time being.
    */ async getContractClassMetadata(id, includeArtifact = false) {
-        const artifact = await this.contractDataProvider.getContractArtifact(id);
+        const artifact = await this.contractStore.getContractArtifact(id);
         if (!artifact) {
             this.log.warn(`No artifact found for contract class ${id.toString()} when looking for its metadata`);
         }
@@ -281,7 +274,7 @@ import { Synchronizer } from './synchronizer/index.js';
    */ async getContractMetadata(address) {
         let instance;
         try {
-            instance = await this.contractDataProvider.getContractInstance(address);
+            instance = await this.contractStore.getContractInstance(address);
         } catch  {
             this.log.warn(`No instance found for contract ${address.toString()} when looking for its metadata`);
         }
@@ -310,46 +303,48 @@ import { Synchronizer } from './synchronizer/index.js';
             this.log.info(`Registered account ${accountCompleteAddress.address.toString()}`);
             this.log.debug(`Registered account\n ${accountCompleteAddress.toReadableString()}`);
         }
-        await this.addressDataProvider.addCompleteAddress(accountCompleteAddress);
-        await this.noteDataProvider.addScope(accountCompleteAddress.address);
+        await this.addressStore.addCompleteAddress(accountCompleteAddress);
+        await this.noteStore.addScope(accountCompleteAddress.address);
         return accountCompleteAddress;
     }
     /**
-   * Registers a user contact in PXE.
+   * Registers a sender in this PXE.
    *
-   * Once a new contact is registered, the PXE will be able to receive notes tagged from this contact.
-   * Will do nothing if the account is already registered.
+   * After registering a new sender, the PXE will sync private logs that are tagged with this sender's address.
+   * Will do nothing if the address is already registered.
    *
-   * @param address - Address of the user to add to the address book
-   * @returns The address address of the account.
-   */ async registerSender(address) {
+   * @param sender - Address of the sender to register.
+   * @returns The address of the sender.
+   * TODO: It's strange that we return the address here and I (benesjan) think we should drop the return value.
+   */ async registerSender(sender) {
         const accounts = await this.keyStore.getAccounts();
-        if (accounts.includes(address)) {
-            this.log.info(`Sender:\n "${address.toString()}"\n already registered.`);
-            return address;
+        if (accounts.includes(sender)) {
+            this.log.info(`Sender:\n "${sender.toString()}"\n already registered.`);
+            return sender;
         }
-        const wasAdded = await this.taggingDataProvider.addSenderAddress(address);
+        const wasAdded = await this.senderAddressBookStore.addSender(sender);
         if (wasAdded) {
-            this.log.info(`Added sender:\n ${address.toString()}`);
+            this.log.info(`Added sender:\n ${sender.toString()}`);
         } else {
-            this.log.info(`Sender:\n "${address.toString()}"\n already registered.`);
+            this.log.info(`Sender:\n "${sender.toString()}"\n already registered.`);
         }
-        return address;
+        return sender;
     }
     /**
-   * Retrieves the addresses stored as senders on this PXE.
-   * @returns An array of the senders on this PXE.
+   * Retrieves senders registered in this PXE.
+   * @returns Senders registered in this PXE.
    */ getSenders() {
-        return this.taggingDataProvider.getSenderAddresses();
+        return this.senderAddressBookStore.getSenders();
     }
     /**
-   * Removes a sender in the address book.
-   */ async removeSender(address) {
-        const wasRemoved = await this.taggingDataProvider.removeSenderAddress(address);
+   * Removes a sender registered in this PXE.
+   * @param sender - The address of the sender to remove.
+   */ async removeSender(sender) {
+        const wasRemoved = await this.senderAddressBookStore.removeSender(sender);
         if (wasRemoved) {
-            this.log.info(`Removed sender:\n ${address.toString()}`);
+            this.log.info(`Removed sender:\n ${sender.toString()}`);
         } else {
-            this.log.info(`Sender:\n "${address.toString()}"\n not in address book.`);
+            this.log.info(`Sender:\n "${sender.toString()}"\n not registered in PXE.`);
         }
     }
     /**
@@ -357,7 +352,7 @@ import { Synchronizer } from './synchronizer/index.js';
    * @returns An array of the accounts registered on this PXE.
    */ async getRegisteredAccounts() {
         // Get complete addresses of both the recipients and the accounts
-        const completeAddresses = await this.addressDataProvider.getCompleteAddresses();
+        const completeAddresses = await this.addressStore.getCompleteAddresses();
         // Filter out the addresses not corresponding to accounts
         const accounts = await this.keyStore.getAccounts();
         return completeAddresses.filter((completeAddress)=>accounts.find((address)=>address.equals(completeAddress.address)));
@@ -368,7 +363,7 @@ import { Synchronizer } from './synchronizer/index.js';
    * @param artifact - The build artifact for the contract class.
    */ async registerContractClass(artifact) {
         const { id: contractClassId } = await getContractClassFromArtifact(artifact);
-        await this.contractDataProvider.addContractArtifact(contractClassId, artifact);
+        await this.contractStore.addContractArtifact(contractClassId, artifact);
         this.log.info(`Added contract class ${artifact.name} with id ${contractClassId}`);
     }
     /**
@@ -392,17 +387,17 @@ import { Synchronizer } from './synchronizer/index.js';
             if (!computedAddress.equals(instance.address)) {
                 throw new Error('Added a contract in which the address does not match the contract instance.');
             }
-            await this.contractDataProvider.addContractArtifact(contractClass.id, artifact);
+            await this.contractStore.addContractArtifact(contractClass.id, artifact);
             const publicFunctionSignatures = artifact.functions.filter((fn)=>fn.functionType === FunctionType.PUBLIC).map((fn)=>decodeFunctionSignature(fn.name, fn.parameters));
             await this.node.registerContractFunctionSignatures(publicFunctionSignatures);
         } else {
             // Otherwise, make sure there is an artifact already registered for that class id
-            artifact = await this.contractDataProvider.getContractArtifact(instance.currentContractClassId);
+            artifact = await this.contractStore.getContractArtifact(instance.currentContractClassId);
             if (!artifact) {
                 throw new Error(`Artifact not found when registering an instance. Contract class: ${instance.currentContractClassId}.`);
             }
         }
-        await this.contractDataProvider.addContractInstance(instance);
+        await this.contractStore.addContractInstance(instance);
         this.log.info(`Added contract ${artifact.name} at ${instance.address.toString()} with class ${instance.currentContractClassId}`);
     }
     /**
@@ -417,22 +412,22 @@ import { Synchronizer } from './synchronizer/index.js';
         // We disable concurrently updating contracts to avoid concurrently syncing with the node, or changing a contract's
         // class while we're simulating it.
         return this.#putInJobQueue(async ()=>{
-            const currentInstance = await this.contractDataProvider.getContractInstance(contractAddress);
+            const currentInstance = await this.contractStore.getContractInstance(contractAddress);
             if (!currentInstance) {
                 throw new Error(`Instance not found when updating a contract. Contract address: ${contractAddress}.`);
             }
             const contractClass = await getContractClassFromArtifact(artifact);
-            await this.synchronizer.sync();
-            const header = await this.syncDataProvider.getBlockHeader();
+            await this.blockStateSynchronizer.sync();
+            const header = await this.anchorBlockStore.getBlockHeader();
             const currentClassId = await readCurrentClassId(contractAddress, currentInstance, this.node, header.globalVariables.blockNumber, header.globalVariables.timestamp);
             if (!contractClass.id.equals(currentClassId)) {
                 throw new Error('Could not update contract to a class different from the current one.');
             }
-            await this.contractDataProvider.addContractArtifact(contractClass.id, artifact);
+            await this.contractStore.addContractArtifact(contractClass.id, artifact);
             const publicFunctionSignatures = artifact.functions.filter((fn)=>fn.functionType === FunctionType.PUBLIC).map((fn)=>decodeFunctionSignature(fn.name, fn.parameters));
             await this.node.registerContractFunctionSignatures(publicFunctionSignatures);
             currentInstance.currentContractClassId = contractClass.id;
-            await this.contractDataProvider.addContractInstance(currentInstance);
+            await this.contractStore.addContractInstance(currentInstance);
             this.log.info(`Updated contract ${artifact.name} at ${contractAddress.toString()} to class ${contractClass.id}`);
         });
     }
@@ -440,33 +435,7 @@ import { Synchronizer } from './synchronizer/index.js';
    * Retrieves the addresses of contracts added to this PXE.
    * @returns An array of contracts addresses registered on this PXE.
    */ getContracts() {
-        return this.contractDataProvider.getContractsAddresses();
-    }
-    /**
-   * A debugging utility to get notes based on the provided filter.
-   *
-   * Note that this should not be used in production code because the structure of notes is considered to be
-   * an implementation detail of contracts. This is only meant to be used for debugging purposes. If you need to obtain
-   * note-related information in production code, please implement a custom utility function on your contract and call
-   * that function instead (e.g. `get_balance(owner: AztecAddress) -> u128` utility function on a Token contract).
-   *
-   * @param filter - The filter to apply to the notes.
-   * @returns The requested notes.
-   */ async getNotes(filter) {
-        // We need to manually trigger private state sync to have a guarantee that all the notes are available.
-        await this.simulateUtility('sync_private_state', [], filter.contractAddress);
-        const noteDaos = await this.noteDataProvider.getNotes(filter);
-        const uniqueNotes = noteDaos.map(async (dao)=>{
-            const completeAddresses = await this.addressDataProvider.getCompleteAddresses();
-            const completeAddressIndex = completeAddresses.findIndex((completeAddress)=>completeAddress.address.equals(dao.recipient));
-            const completeAddress = completeAddresses[completeAddressIndex];
-            if (completeAddress === undefined) {
-                throw new Error(`Cannot find complete address for recipient ${dao.recipient.toString()}`);
-            }
-            const recipient = completeAddress.address;
-            return new UniqueNote(dao.note, recipient, dao.contractAddress, dao.storageSlot, dao.txHash, dao.noteNonce);
-        });
-        return Promise.all(uniqueNotes);
+        return this.contractStore.getContractsAddresses();
     }
     /**
    * Proves the private portion of a simulated transaction, ready to send to the network
@@ -484,7 +453,7 @@ import { Synchronizer } from './synchronizer/index.js';
             const totalTimer = new Timer();
             try {
                 const syncTimer = new Timer();
-                await this.synchronizer.sync();
+                await this.blockStateSynchronizer.sync();
                 const syncTime = syncTimer.ms();
                 const contractFunctionSimulator = this.#getSimulatorForTx();
                 privateExecutionResult = await this.#executePrivate(contractFunctionSimulator, txRequest);
@@ -513,14 +482,21 @@ import { Synchronizer } from './synchronizer/index.js';
                     timings,
                     nodeRPCCalls: contractFunctionSimulator?.getStats().nodeRPCCalls
                 });
+                // While not strictly necessary to store tagging cache contents in the DB since we sync tagging indexes from
+                // chain before sending new logs, the sync can only see logs already included in blocks. If we send another
+                // transaction before this one is included in a block from this PXE, and that transaction contains a log with
+                // a tag derived from the same secret, we would reuse the tag and the transactions would be linked. Hence
+                // storing the tags here prevents linkage of txs sent from the same PXE.
                 const preTagsUsedInTheTx = privateExecutionResult.entrypoint.preTags;
                 if (preTagsUsedInTheTx.length > 0) {
-                    await this.taggingDataProvider.setLastUsedIndexesAsSender(preTagsUsedInTheTx);
-                    this.log.debug(`Stored used pre tags as sender for the tx`, {
+                    // TODO(benesjan): The following is an expensive operation. Figure out a way to avoid it.
+                    const txHash = (await txProvingResult.toTx()).txHash;
+                    await this.senderTaggingStore.storePendingIndexes(preTagsUsedInTheTx, txHash);
+                    this.log.debug(`Stored used pre-tags as sender for the tx`, {
                         preTagsUsedInTheTx
                     });
                 } else {
-                    this.log.debug(`No pre tags used in the tx`);
+                    this.log.debug(`No pre-tags used in the tx`);
                 }
                 return txProvingResult;
             } catch (err) {
@@ -551,7 +527,7 @@ import { Synchronizer } from './synchronizer/index.js';
                 };
                 this.log.info(`Profiling transaction execution request to ${txRequest.functionSelector} at ${txRequest.origin}`, txInfo);
                 const syncTimer = new Timer();
-                await this.synchronizer.sync();
+                await this.blockStateSynchronizer.sync();
                 const syncTime = syncTimer.ms();
                 const contractFunctionSimulator = this.#getSimulatorForTx();
                 const privateExecutionResult = await this.#executePrivate(contractFunctionSimulator, txRequest);
@@ -627,7 +603,7 @@ import { Synchronizer } from './synchronizer/index.js';
                 };
                 this.log.info(`Simulating transaction execution request to ${txRequest.functionSelector} at ${txRequest.origin}`, txInfo);
                 const syncTimer = new Timer();
-                await this.synchronizer.sync();
+                await this.blockStateSynchronizer.sync();
                 const syncTime = syncTimer.ms();
                 const contractFunctionSimulator = this.#getSimulatorForTx(overrides);
                 // Temporary: in case there are overrides, we have to skip the kernels or validations
@@ -640,10 +616,9 @@ import { Synchronizer } from './synchronizer/index.js';
                 let executionSteps = [];
                 if (skipKernels) {
                     // According to the protocol rules, the nonce generator for the note hashes
-                    // can either be the first nullifier in the tx or the hash of the initial tx request
-                    // if there are none.
-                    const nonceGenerator = privateExecutionResult.firstNullifier.equals(Fr.ZERO) ? await txRequest.toTxRequest().hash() : privateExecutionResult.firstNullifier;
-                    ({ publicInputs, executionSteps } = await generateSimulatedProvingResult(privateExecutionResult, nonceGenerator, this.contractDataProvider));
+                    // can either be the first nullifier in the tx or the protocol nullifier if there are none.
+                    const nonceGenerator = privateExecutionResult.firstNullifier.equals(Fr.ZERO) ? await computeProtocolNullifier(await txRequest.toTxRequest().hash()) : privateExecutionResult.firstNullifier;
+                    ({ publicInputs, executionSteps } = await generateSimulatedProvingResult(privateExecutionResult, nonceGenerator, this.contractStore));
                 } else {
                     // Kernel logic, plus proving of all private functions and kernels.
                     ({ publicInputs, executionSteps } = await this.#prove(txRequest, this.proofCreator, privateExecutionResult, {
@@ -710,15 +685,12 @@ import { Synchronizer } from './synchronizer/index.js';
     /**
    * Simulate the execution of a contract utility function.
    *
-   * @param functionName - The name of the utility contract function to be called.
-   * @param args - The arguments to be provided to the function.
-   * @param to - The address of the contract to be called.
+   * @param call - The function call containing the function details, arguments, and target contract address.
    * @param authwits - (Optional) The authentication witnesses required for the function call.
-   * @param from - (Optional) The msg sender to set for the call.
    * @param scopes - (Optional) The accounts whose notes we can access in this call. Currently optional and will
    * default to all.
    * @returns The result of the utility function call, structured based on the function ABI.
-   */ simulateUtility(functionName, args, to, authwits, _from, scopes) {
+   */ simulateUtility(call, authwits, scopes) {
         // We disable concurrent simulations since those might execute oracles which read and write to the PXE stores (e.g.
         // to the capsules), and we need to prevent concurrent runs from interfering with one another (e.g. attempting to
         // delete the same read value, or reading values that another simulation is currently modifying).
@@ -726,17 +698,16 @@ import { Synchronizer } from './synchronizer/index.js';
             try {
                 const totalTimer = new Timer();
                 const syncTimer = new Timer();
-                await this.synchronizer.sync();
+                await this.blockStateSynchronizer.sync();
                 const syncTime = syncTimer.ms();
-                const functionCall = await this.#getFunctionCall(functionName, args, to);
                 const functionTimer = new Timer();
                 const contractFunctionSimulator = this.#getSimulatorForTx();
-                const executionResult = await this.#simulateUtility(contractFunctionSimulator, functionCall, authwits ?? [], scopes);
+                const executionResult = await this.#simulateUtility(contractFunctionSimulator, call, authwits ?? [], scopes);
                 const functionTime = functionTimer.ms();
                 const totalTime = totalTimer.ms();
                 const perFunction = [
                     {
-                        functionName,
+                        functionName: call.name,
                         time: functionTime
                     }
                 ];
@@ -755,31 +726,31 @@ import { Synchronizer } from './synchronizer/index.js';
                     }
                 };
             } catch (err) {
+                const { to, name, args } = call;
                 const stringifiedArgs = args.map((arg)=>arg.toString()).join(', ');
-                throw this.#contextualizeError(err, `simulateUtility ${to}:${functionName}(${stringifiedArgs})`, `scopes=${scopes?.map((s)=>s.toString()).join(', ') ?? 'undefined'}`);
+                throw this.#contextualizeError(err, `simulateUtility ${to}:${name}(${stringifiedArgs})`, `scopes=${scopes?.map((s)=>s.toString()).join(', ') ?? 'undefined'}`);
             }
         });
     }
     /**
    * Returns the private events given search parameters.
-   * @param contractAddress - The address of the contract to get events from.
-   * @param eventMetadata - Metadata of the event. This should be the class generated from the contract. e.g. Contract.events.Event
-   * @param from - The block number to search from.
-   * @param numBlocks - The amount of blocks to search.
-   * @param recipients - The addresses that decrypted the logs.
-   * @returns - The deserialized events.
-   */ async getPrivateEvents(contractAddress, eventMetadataDef, from, numBlocks, recipients) {
-        if (recipients.length === 0) {
-            throw new Error('Recipients are required to get private events');
-        }
-        this.log.verbose(`Getting private events for ${contractAddress.toString()} from ${from} to ${from + numBlocks}`);
+   * @param eventSelector - Event selector to search for.
+   * @param filter
+   *  contractAddress - The address of the contract to get events from. Required.
+   *  scopes - One or more event scope addresses to filter by. Required.
+   *  fromBlock - The block number to search from (inclusive). Optional. If provided, it must be >= 0.
+   *    Defaults to 0.
+   *    If toBlock is defined but fromBlock is not, fromBlock defaults to toBlock - 1.
+   *  toBlock - The block number to search up to (exclusive). Optional. If provided, it must be > 0.
+   *    Defaults to the latest known block to PXE + 1.
+   * @returns - The packed events with block and tx metadata.
+   */ async getPrivateEvents(eventSelector, filter) {
         // We need to manually trigger private state sync to have a guarantee that all the events are available.
-        await this.simulateUtility('sync_private_state', [], contractAddress);
-        const events = await this.privateEventDataProvider.getPrivateEvents(contractAddress, from, numBlocks, recipients, eventMetadataDef.eventSelector);
-        const decodedEvents = events.map((event)=>decodeFromAbi([
-                eventMetadataDef.abiType
-            ], event.msgContent));
-        return decodedEvents;
+        const call = await this.contractStore.getFunctionCall('sync_private_state', [], filter.contractAddress);
+        await this.simulateUtility(call);
+        const sanitizedFilter = await new PrivateEventFilterValidator(this.anchorBlockStore).validate(filter);
+        this.log.debug(`Getting private events for ${sanitizedFilter.contractAddress.toString()} from ${sanitizedFilter.fromBlock} to ${sanitizedFilter.toBlock}`);
+        return this.privateEventStore.getPrivateEvents(eventSelector, sanitizedFilter);
     }
     /**
    * Stops the PXE's job queue.

package/dest/storage/address_store/address_store.d.ts

@@ -0,0 +1,11 @@
+import type { AztecAsyncKVStore } from '@aztec/kv-store';
+import type { AztecAddress } from '@aztec/stdlib/aztec-address';
+import { CompleteAddress } from '@aztec/stdlib/contract';
+export declare class AddressStore {
+    #private;
+    constructor(store: AztecAsyncKVStore);
+    addCompleteAddress(completeAddress: CompleteAddress): Promise<boolean>;
+    getCompleteAddress(account: AztecAddress): Promise<CompleteAddress | undefined>;
+    getCompleteAddresses(): Promise<CompleteAddress[]>;
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiYWRkcmVzc19zdG9yZS5kLnRzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vLi4vc3JjL3N0b3JhZ2UvYWRkcmVzc19zdG9yZS9hZGRyZXNzX3N0b3JlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUNBLE9BQU8sS0FBSyxFQUFtQixpQkFBaUIsRUFBaUIsTUFBTSxpQkFBaUIsQ0FBQztBQUN6RixPQUFPLEtBQUssRUFBRSxZQUFZLEVBQUUsTUFBTSw2QkFBNkIsQ0FBQztBQUNoRSxPQUFPLEVBQUUsZUFBZSxFQUFFLE1BQU0sd0JBQXdCLENBQUM7QUFFekQscUJBQWEsWUFBWTs7SUFLdkIsWUFBWSxLQUFLLEVBQUUsaUJBQWlCLEVBS25DO0lBRUQsa0JBQWtCLENBQUMsZUFBZSxFQUFFLGVBQWUsR0FBRyxPQUFPLENBQUMsT0FBTyxDQUFDLENBMEJyRTtJQVlELGtCQUFrQixDQUFDLE9BQU8sRUFBRSxZQUFZLEdBQUcsT0FBTyxDQUFDLGVBQWUsR0FBRyxTQUFTLENBQUMsQ0FFOUU7SUFFSyxvQkFBb0IsSUFBSSxPQUFPLENBQUMsZUFBZSxFQUFFLENBQUMsQ0FJdkQ7Q0FDRiJ9

package/dest/storage/address_store/address_store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"address_store.d.ts","sourceRoot":"","sources":["../../../src/storage/address_store/address_store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAmB,iBAAiB,EAAiB,MAAM,iBAAiB,CAAC;AACzF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAEzD,qBAAa,YAAY;;IAKvB,YAAY,KAAK,EAAE,iBAAiB,EAKnC;IAED,kBAAkB,CAAC,eAAe,EAAE,eAAe,GAAG,OAAO,CAAC,OAAO,CAAC,CA0BrE;IAYD,kBAAkB,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC,CAE9E;IAEK,oBAAoB,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC,CAIvD;CACF"}

package/dest/storage/{address_data_provider/address_data_provider.js → address_store/address_store.js}

@@ -1,6 +1,6 @@
 import { toArray } from '@aztec/foundation/iterable';
 import { CompleteAddress } from '@aztec/stdlib/contract';
-export class AddressDataProvider {
+export class AddressStore {
     #store;
     #completeAddresses;
     #completeAddressIndex;

package/dest/storage/address_store/index.d.ts

@@ -0,0 +1,2 @@
+export { AddressStore } from './address_store.js';
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uLy4uL3NyYy9zdG9yYWdlL2FkZHJlc3Nfc3RvcmUvaW5kZXgudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUEsT0FBTyxFQUFFLFlBQVksRUFBRSxNQUFNLG9CQUFvQixDQUFDIn0=

package/dest/storage/address_store/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/storage/address_store/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC"}