import { Buffer } from 'buffer';
import { PublicKey } from '@solana/web3.js';
import {
  PROGRAM_ID,
  baseAssetInfoDiscriminator,
  mintInfoDiscriminator,
} from '@convergence-rfq/rfq';

import { Convergence } from '../../Convergence';
import { GpaBuilder } from '../../utils/GpaBuilder';

const BUMP = 8;
const INDEX = BUMP + 8;
const RISK_CATEGORY = INDEX + 8;
const PRICE_ORACLE = RISK_CATEGORY + 32;
const TICKER = PRICE_ORACLE + 32;

export class ProtocolGpaBuilder extends GpaBuilder {
  constructor(convergence: Convergence, programId?: PublicKey) {
    super(convergence, programId ?? PROGRAM_ID);
  }

  whereBaseAssets() {
    return this.where(0, Buffer.from(baseAssetInfoDiscriminator));
  }

  whereTicker(ticker: string) {
    return this.where(TICKER, Buffer.from(ticker));
  }

  whereRegisteredMints() {
    return this.where(0, Buffer.from(mintInfoDiscriminator));
  }
}