ts_callback_contract_db.hpp

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <napi.h>
#include <optional>
 
#include "barretenberg/vm2/common/aztec_types.hpp"
#include "barretenberg/vm2/simulation/interfaces/db.hpp"
 
namespace bb::nodejs {
 
class TsCallbackContractDB : public avm2::simulation::ContractDBInterface {
  public:
    TsCallbackContractDB(Napi::ThreadSafeFunction instanceCallback,
                         Napi::ThreadSafeFunction classCallback,
                         Napi::ThreadSafeFunction addContractsCallback,
                         Napi::ThreadSafeFunction bytecodeCommitmentCallback,
                         Napi::ThreadSafeFunction debugNameCallback,
                         Napi::ThreadSafeFunction createCheckpointCallback,
                         Napi::ThreadSafeFunction commitCheckpointCallback,
                         Napi::ThreadSafeFunction revertCheckpointCallback);
 
    std::optional<bb::avm2::ContractInstance> get_contract_instance(
        const bb::avm2::AztecAddress& address) const override;
 
    std::optional<bb::avm2::ContractClass> get_contract_class(const bb::avm2::ContractClassId& class_id) const override;
 
    void add_contracts(const bb::avm2::ContractDeploymentData& contract_deployment_data) override;
 
    std::optional<bb::avm2::FF> get_bytecode_commitment(const bb::avm2::ContractClassId& class_id) const override;
 
    std::optional<std::string> get_debug_function_name(const bb::avm2::AztecAddress& address,
                                                       const bb::avm2::FF& selector) const override;
 
    void create_checkpoint() override;
 
    void commit_checkpoint() override;
 
    void revert_checkpoint() override;
 
    void release();
 
  private:
    Napi::ThreadSafeFunction contract_instance_callback_;
    Napi::ThreadSafeFunction contract_class_callback_;
    Napi::ThreadSafeFunction add_contracts_callback_;
    Napi::ThreadSafeFunction bytecode_commitment_callback_;
    Napi::ThreadSafeFunction debug_name_callback_;
    Napi::ThreadSafeFunction create_checkpoint_callback_;
    Napi::ThreadSafeFunction commit_checkpoint_callback_;
    Napi::ThreadSafeFunction revert_checkpoint_callback_;
 
    // Track whether TSFNs have been released to avoid double-release
    mutable bool released_ = false;
};
 
} // namespace bb::nodejs