import { BridgeCallData } from '../bridge_call_data/index.js';
import { ProofId } from '../client_proofs/index.js';
import { InnerProofData } from './inner_proof.js';

export class RollupDefiClaimProofData {
  static ENCODED_LENGTH = 1 + 4 * 32;

  constructor(public readonly proofData: InnerProofData) {
    if (proofData.proofId !== ProofId.DEFI_CLAIM) {
      throw new Error('Not a defi claim proof.');
    }
  }

  get ENCODED_LENGTH() {
    return RollupDefiClaimProofData.ENCODED_LENGTH;
  }

  get bridgeCallData() {
    return BridgeCallData.fromBuffer(this.proofData.publicAssetId);
  }

  static decode(encoded: Buffer) {
    const proofId = encoded.readUInt8(0);
    if (proofId !== ProofId.DEFI_CLAIM) {
      throw new Error('Not a defi claim proof.');
    }

    let offset = 1;
    const noteCommitment1 = encoded.slice(offset, offset + 32);
    offset += 32;
    const noteCommitment2 = encoded.slice(offset, offset + 32);
    offset += 32;
    const nullifier1 = encoded.slice(offset, offset + 32);
    offset += 32;
    const nullifier2 = encoded.slice(offset, offset + 32);
    return new RollupDefiClaimProofData(
      new InnerProofData(
        ProofId.DEFI_CLAIM,
        noteCommitment1,
        noteCommitment2,
        nullifier1,
        nullifier2,
        Buffer.alloc(32),
        Buffer.alloc(32),
        Buffer.alloc(32),
      ),
    );
  }

  encode() {
    const { noteCommitment1, noteCommitment2, nullifier1, nullifier2 } = this.proofData;
    return Buffer.concat([Buffer.from([ProofId.DEFI_CLAIM]), noteCommitment1, noteCommitment2, nullifier1, nullifier2]);
  }
}