index.ts

import {
  initVerkleWasm,
  getTreeKey as getTreeKeyBase,
  getTreeKeyHash as getTreeKeyHashBase,
  updateCommitment as updateCommitmentBase,
  zeroCommitment as zeroCommitmentBase,
  verifyExecutionWitnessPreState as verifyExecutionWitnessPreStateBase,
  createProof as createProofBase,
  verifyProof as verifyProofBase,
  commitToScalars as commitToScalarsBase,
  type ProverInput as ProverInputBase,
  type VerifierInput as VerifierInputBase,
} from './verkleFFIBindings/index.js'
import { Context as VerkleFFI } from './wasm/rust_verkle_wasm.js'

export const loadVerkleCrypto = async () => {
  await initVerkleWasm()

  const verkleFFI = new VerkleFFI()

  const getTreeKey = (address: Uint8Array, treeIndex: Uint8Array, subIndex: number): Uint8Array =>
    getTreeKeyBase(verkleFFI, address, treeIndex, subIndex)

  const getTreeKeyHash = (address: Uint8Array, treeIndexLE: Uint8Array): Uint8Array =>
    getTreeKeyHashBase(verkleFFI, address, treeIndexLE)

  const updateCommitment = (
    commitment: Uint8Array,
    commitmentIndex: number,
    oldScalarValue: Uint8Array,
    newScalarValue: Uint8Array,
  ): Commitment =>
    updateCommitmentBase(verkleFFI, commitment, commitmentIndex, oldScalarValue, newScalarValue)

  const verifyExecutionWitnessPreState = (
    prestateRoot: string,
    execution_witness_json: string,
  ): boolean => verifyExecutionWitnessPreStateBase(prestateRoot, execution_witness_json)

  const zeroCommitment = zeroCommitmentBase()

  const hashCommitment = (commitment: Uint8Array) => verkleFFI.hashCommitment(commitment)
  const serializeCommitment = (commitment: Uint8Array) => verkleFFI.serializeCommitment(commitment)
  const createProof = (proverInputs: ProverInput[]) => createProofBase(verkleFFI, proverInputs)
  const verifyProof = (proof: Uint8Array, verifierInputs: VerifierInput[]) =>
    verifyProofBase(verkleFFI, proof, verifierInputs)
  const commitToScalars = (vector: Uint8Array[]) => commitToScalarsBase(verkleFFI, vector)
  return {
    commitToScalars, 
    getTreeKey,
    getTreeKeyHash,
    updateCommitment,
    zeroCommitment,
    verifyExecutionWitnessPreState,
    hashCommitment,
    serializeCommitment,
    createProof,
    verifyProof,
  }
}

// Input used to create proofs over vectors
export type ProverInput = ProverInputBase
// Input needed to verify proofs over vectors
// alongside the proof.
export type VerifierInput = VerifierInputBase

// This is a 32 byte serialized field element
export type Scalar = Uint8Array

// This is a 64 byte serialized point.
// It is 64 bytes because the point is serialized in uncompressed format.
export type Commitment = Uint8Array