import { publicKey, seq, struct, u128, u16, u64, u8, vec } from "../../marshmallow";

export const LaunchpadConfig = struct([
  u64(),
  u64("epoch"),
  u8("curveType"),
  u16("index"),
  u64("migrateFee"),
  u64("tradeFeeRate"),

  u64("maxShareFeeRate"),
  u64("minSupplyA"),
  u64("maxLockRate"),
  u64("minSellRateA"),
  u64("minMigrateRateA"),
  u64("minFundRaisingB"),

  publicKey("mintB"),
  publicKey("protocolFeeOwner"),
  publicKey("migrateFeeOwner"),
  publicKey("migrateToAmmWallet"),
  publicKey("migrateToCpmmWallet"),
  seq(u64(), 16),
]);

export const VestingSchedule = struct([
  u64("totalLockedAmount"),
  u64("cliffPeriod"),
  u64("unlockPeriod"),
  u64("startTime"),
  u64("totalAllocatedShare"),
]);

export const LaunchpadPool = struct([
  u64(),
  u64("epoch"),
  u8("bump"),
  u8("status"),
  u8("mintDecimalsA"),
  u8("mintDecimalsB"),
  u8("migrateType"),

  u64("supply"),
  u64("totalSellA"),
  u64("virtualA"),
  u64("virtualB"),
  u64("realA"),
  u64("realB"),
  u64("totalFundRaisingB"),
  u64("protocolFee"),
  u64("platformFee"),
  u64("migrateFee"),

  VestingSchedule.replicate("vestingSchedule"),

  publicKey("configId"),
  publicKey("platformId"),
  publicKey("mintA"),
  publicKey("mintB"),
  publicKey("vaultA"),
  publicKey("vaultB"),

  publicKey("creator"),

  u8("mintProgramFlag"),
  u8("cpmmCreatorFeeOn"),

  u64('platformVestingShare'),

  seq(u8(), 54),
]);

export const LaunchpadVesting = struct([
  u64(),
  u64("epoch"),
  publicKey("poolId"),
  publicKey("beneficiary"),
  u64("claimedAmount"),
  u64("tokenShareAmount"),
  seq(u64(), 8),
]);

export const BondingCurveParam = struct([
  u8('migrateType'),
  u8('migrateCpmmFeeOn'),
  u64('supply'),
  u64('totalSellA'),
  u64('totalFundRaisingB'),

  u64('totalLockedAmount'),
  u64('cliffPeriod'),
  u64('unlockPeriod'),
]);

export const PlatformCurveParam = struct([
  u64("epoch"),
  u8("index"),
  publicKey("configId"),

  BondingCurveParam.replicate("bondingCurveParam"),
  seq(u64(), 50),
]);

export const PlatformConfig = struct([
  u64(),
  u64("epoch"),
  publicKey("platformClaimFeeWallet"),
  publicKey("platformLockNftWallet"),
  u64("platformScale"),
  u64("creatorScale"),
  u64("burnScale"),
  u64("feeRate"),
  seq(u8(), 64, "name"),
  seq(u8(), 256, "web"),
  seq(u8(), 256, "img"),
  publicKey("cpConfigId"),
  u64("creatorFeeRate"),
  publicKey("transferFeeExtensionAuth"),

  publicKey("platformVestingWallet"),
  u64("platformVestingScale"),

  publicKey("platformCpCreator"),

  u8("restrictGlobalConfig"),
  // 0: the platform does not restrict the launch params, 1: they must satisfy the
  // PlatformCurveRule account of the used config
  u8("restrictCurveParam"),
  // the wallet allowed to manage the PlatformCurveRule accounts of this platform, the
  // platform admin can always manage them too
  publicKey("curveRuleManager"),

  seq(u8(), 78),
]);

export const ParamConstraint = struct([u8("field"), u8("op"), u128("value")]);
export const CurveRuleGroup = struct([u16("groupId"), u64("epoch"), vec(ParamConstraint, "constraints")]);
export const PlatformCurveRule = struct([
  u64(),
  u8("bump"),
  u8("version"),
  publicKey("platformId"),
  publicKey("configId"),
  u64("epoch"),
  seq(u64(), 8),

  vec(CurveRuleGroup, "groups"),
]);