import { createTxId, ProofId } from '../client_proofs/index.js';
import { numToUInt32BE } from '../serialize/index.js';

export class InnerProofData {
  static NUM_PUBLIC_INPUTS = 8;
  static LENGTH = InnerProofData.NUM_PUBLIC_INPUTS * 32;
  static PADDING = InnerProofData.fromBuffer(Buffer.alloc(InnerProofData.LENGTH));

  public txId_?: Buffer;

  constructor(
    public proofId: ProofId,
    public noteCommitment1: Buffer,
    public noteCommitment2: Buffer,
    public nullifier1: Buffer,
    public nullifier2: Buffer,
    public publicValue: Buffer,
    public publicOwner: Buffer,
    public publicAssetId: Buffer,
  ) {}

  public get txId(): Buffer {
    if (!this.txId_) {
      this.txId_ = createTxId(this.toBuffer());
    }
    return this.txId_;
  }

  getDepositSigningData() {
    return this.toBuffer();
  }

  toBuffer() {
    return Buffer.concat([
      numToUInt32BE(this.proofId, 32),
      this.noteCommitment1,
      this.noteCommitment2,
      this.nullifier1,
      this.nullifier2,
      this.publicValue,
      this.publicOwner,
      this.publicAssetId,
    ]);
  }

  isPadding() {
    return this.proofId === ProofId.PADDING;
  }

  static fromBuffer(innerPublicInputs: Buffer) {
    let dataStart = 0;
    const proofId = innerPublicInputs.readUInt32BE(dataStart + 28);
    dataStart += 32;
    const noteCommitment1 = innerPublicInputs.slice(dataStart, dataStart + 32);
    dataStart += 32;
    const noteCommitment2 = innerPublicInputs.slice(dataStart, dataStart + 32);
    dataStart += 32;
    const nullifier1 = innerPublicInputs.slice(dataStart, dataStart + 32);
    dataStart += 32;
    const nullifier2 = innerPublicInputs.slice(dataStart, dataStart + 32);
    dataStart += 32;
    const publicValue = innerPublicInputs.slice(dataStart, dataStart + 32);
    dataStart += 32;
    const publicOwner = innerPublicInputs.slice(dataStart, dataStart + 32);
    dataStart += 32;
    const publicAssetId = innerPublicInputs.slice(dataStart, dataStart + 32);
    dataStart += 32;

    return new InnerProofData(
      proofId,
      noteCommitment1,
      noteCommitment2,
      nullifier1,
      nullifier2,
      publicValue,
      publicOwner,
      publicAssetId,
    );
  }
}