import type { Address } from "@solana/kit";

import type { Vault } from "../generated";

export const JUPITER_ADAPTOR_PROGRAM_ID =
  "EW35URAx3LiM13fFK3QxAXfGemHso9HWPixrv7YDY4AM" as Address;

export const DRIFT_ADAPTOR_PROGRAM_ID =
  "EBN93eXs5fHGBABuajQqdsKRkCgaqtJa8vEFD6vKXiP" as Address;

export const KAMINO_ADAPTOR_PROGRAM_ID =
  "to6Eti9CsC5FGkAtqiPphvKD2hiQiLsS8zWiDBqBPKR" as Address;

export const WHITELISTED_ADAPTOR_PROGRAM_IDS = [
  JUPITER_ADAPTOR_PROGRAM_ID,
  DRIFT_ADAPTOR_PROGRAM_ID,
  KAMINO_ADAPTOR_PROGRAM_ID,
] as const;

export function isWhitelistedAdaptorProgramId(
  adaptorProgram: Address
): boolean {
  return WHITELISTED_ADAPTOR_PROGRAM_IDS.some(
    (programId) => programId === adaptorProgram
  );
}

export function vaultAllowsAdaptorProgram(
  vault: Pick<Vault, "allowAnyAdaptor">,
  adaptorProgram: Address
): boolean {
  return (
    vault.allowAnyAdaptor === 1 || isWhitelistedAdaptorProgramId(adaptorProgram)
  );
}