import { BarretenbergWorker } from '../../wasm/index.js';
import { SinglePippenger } from '../../pippenger/index.js';

export class JoinSplitVerifier {
  private worker!: BarretenbergWorker;

  public async computeKey(pippenger: SinglePippenger, g2Data: Uint8Array) {
    this.worker = pippenger.getWorker();
    await this.worker.transferToHeap(g2Data, 0);
    await this.worker.asyncCall('join_split__init_verification_key', pippenger.getPointer(), 0);
  }

  public async getKey() {
    const keySize = await this.worker.call('join_split__get_new_verification_key_data', 0);
    const keyPtr = Buffer.from(await this.worker.sliceMemory(0, 4)).readUInt32LE(0);
    const buf = Buffer.from(await this.worker.sliceMemory(keyPtr, keyPtr + keySize));
    await this.worker.call('bbfree', keyPtr);
    return buf;
  }

  public async loadKey(worker: BarretenbergWorker, keyBuf: Buffer, g2Data: Uint8Array) {
    this.worker = worker;
    const keyPtr = await this.worker.call('bbmalloc', keyBuf.length);
    await this.worker.transferToHeap(g2Data, 0);
    await this.worker.transferToHeap(keyBuf, keyPtr);
    await this.worker.call('join_split__init_verification_key_from_buffer', keyPtr, 0);
    await this.worker.call('bbfree', keyPtr);
  }

  public async verifyProof(proof: Buffer) {
    const proofPtr = await this.worker.call('bbmalloc', proof.length);
    await this.worker.transferToHeap(proof, proofPtr);
    const verified = (await this.worker.call('join_split__verify_proof', proofPtr, proof.length)) ? true : false;
    await this.worker.call('bbfree', proofPtr);
    return verified;
  }
}