import { ProofId } from '../client_proofs/index.js';
import { RollupAccountProofData } from './rollup_account_proof_data.js';
import { RollupDefiClaimProofData } from './rollup_defi_claim_proof_data.js';
import { RollupDefiDepositProofData } from './rollup_defi_deposit_proof_data.js';
import { RollupDepositProofData } from './rollup_deposit_proof_data.js';
import { RollupPaddingProofData } from './rollup_padding_proof_data.js';
import { RollupSendProofData } from './rollup_send_proof_data.js';
import { RollupWithdrawProofData } from './rollup_withdraw_proof_data.js';

export const decodeProofId = (encoded: Buffer) => encoded.readUInt8(0);

const recoverProof = (encoded: Buffer) => {
  const proofId = decodeProofId(encoded);
  switch (proofId) {
    case ProofId.DEPOSIT:
      return RollupDepositProofData.decode(encoded);
    case ProofId.WITHDRAW:
      return RollupWithdrawProofData.decode(encoded);
    case ProofId.SEND:
      return RollupSendProofData.decode(encoded);
    case ProofId.ACCOUNT:
      return RollupAccountProofData.decode(encoded);
    case ProofId.DEFI_DEPOSIT:
      return RollupDefiDepositProofData.decode(encoded);
    case ProofId.DEFI_CLAIM:
      return RollupDefiClaimProofData.decode(encoded);
    case ProofId.PADDING:
      return RollupPaddingProofData.decode(encoded);
  }
};

export const getEncodedProofSizeForId = (proofId: ProofId) => {
  switch (proofId) {
    case ProofId.DEPOSIT:
      return RollupDepositProofData.ENCODED_LENGTH;
    case ProofId.WITHDRAW:
      return RollupWithdrawProofData.ENCODED_LENGTH;
    case ProofId.SEND:
      return RollupSendProofData.ENCODED_LENGTH;
    case ProofId.ACCOUNT:
      return RollupAccountProofData.ENCODED_LENGTH;
    case ProofId.DEFI_DEPOSIT:
      return RollupDefiDepositProofData.ENCODED_LENGTH;
    case ProofId.DEFI_CLAIM:
      return RollupDefiClaimProofData.ENCODED_LENGTH;
    case ProofId.PADDING:
      return RollupPaddingProofData.ENCODED_LENGTH;
  }
};

export const decodeInnerProof = (encoded: Buffer) => recoverProof(encoded)!;