import type { ParsedIdlAccountData } from "@cardinal/common";
import { emptyWallet } from "@cardinal/common";
import { AnchorProvider, Program } from "@coral-xyz/anchor";
import type { Wallet } from "@coral-xyz/anchor/dist/cjs/provider";
import type { ConfirmOptions, Connection } from "@solana/web3.js";
import { Keypair, PublicKey } from "@solana/web3.js";

import * as GROUP_REWARD_DISTRIBUTOR_TYPES from "../../idl/cardinal_group_reward_distributor";

export const GROUP_REWARD_DISTRIBUTOR_ADDRESS = new PublicKey(
  "grwDL1AZiCaBmTQHTQVhX6wxXKowAnisDZxH7866LUL"
);
export const GROUP_REWARD_MANAGER = new PublicKey(
  "crkdpVWjHWdggGgBuSyAqSmZUmAjYLzD435tcLDRLXr"
);

export const GROUP_REWARD_ENTRY_SEED = "group-reward-entry";

export const GROUP_REWARD_COUNTER_SEED = "group-reward-counter";

export const GROUP_REWARD_DISTRIBUTOR_SEED = "group-reward-distributor";

export type GROUP_REWARD_DISTRIBUTOR_PROGRAM =
  GROUP_REWARD_DISTRIBUTOR_TYPES.CardinalGroupRewardDistributor;

export const GROUP_REWARD_DISTRIBUTOR_IDL = GROUP_REWARD_DISTRIBUTOR_TYPES.IDL;

export type GroupRewardEntryData = ParsedIdlAccountData<
  "groupRewardEntry",
  GROUP_REWARD_DISTRIBUTOR_PROGRAM
>;
export type GroupRewardCounterData = ParsedIdlAccountData<
  "groupRewardCounter",
  GROUP_REWARD_DISTRIBUTOR_PROGRAM
>;
export type GroupRewardDistributorData = ParsedIdlAccountData<
  "groupRewardDistributor",
  GROUP_REWARD_DISTRIBUTOR_PROGRAM
>;

export enum GroupRewardDistributorKind {
  Mint = 1,
  Treasury = 2,
}
export const toGroupRewardDistributorKind = (value: { [key: string]: any }) =>
  Object.values(GroupRewardDistributorKind).findIndex(
    (x) => (x as string).toLowerCase() === Object.keys(value)[0]?.toLowerCase()
  ) + 1;

export enum GroupRewardDistributorMetadataKind {
  NoRestriction = 1,
  UniqueNames = 2,
  UniqueSymbols = 3,
}
export const toGroupRewardDistributorMetadataKind = (value: {
  [key: string]: any;
}) =>
  Object.values(GroupRewardDistributorMetadataKind).findIndex(
    (x) => (x as string).toLowerCase() === Object.keys(value)[0]?.toLowerCase()
  ) + 1;

export enum GroupRewardDistributorPoolKind {
  NoRestriction = 1,
  AllFromSinglePool = 2,
  EachFromSeparatePool = 3,
}
export const toGroupRewardDistributorPoolKind = (value: {
  [key: string]: any;
}) =>
  Object.values(GroupRewardDistributorPoolKind).findIndex(
    (x) => (x as string).toLowerCase() === Object.keys(value)[0]?.toLowerCase()
  ) + 1;

export const groupRewardDistributorProgram = (
  connection: Connection,
  wallet?: Wallet,
  confirmOptions?: ConfirmOptions
) => {
  return new Program<GROUP_REWARD_DISTRIBUTOR_PROGRAM>(
    GROUP_REWARD_DISTRIBUTOR_IDL,
    GROUP_REWARD_DISTRIBUTOR_ADDRESS,
    new AnchorProvider(
      connection,
      wallet ?? emptyWallet(Keypair.generate().publicKey),
      confirmOptions ?? {}
    )
  );
};