import { PublicKey, Connection } from "@solana/web3.js"
import BN from "bn.js" // eslint-disable-line @typescript-eslint/no-unused-vars
import * as borsh from "@coral-xyz/borsh" // eslint-disable-line @typescript-eslint/no-unused-vars
import * as types from "../types" // eslint-disable-line @typescript-eslint/no-unused-vars
import { PROGRAM_ID } from "../programId"

export interface ReferrerTokenStateFields {
  /** Pubkey of the referrer/owner */
  referrer: PublicKey
  /** Token mint for the account */
  mint: PublicKey
  /** Amount that has been accumulated and not claimed yet -> available to claim (scaled fraction) */
  amountUnclaimedSf: BN
  /** Amount that has been accumulated in total -> both already claimed and unclaimed (scaled fraction) */
  amountCumulativeSf: BN
  /** Referrer token state bump, used for address validation */
  bump: BN
  padding: Array<BN>
}

export interface ReferrerTokenStateJSON {
  /** Pubkey of the referrer/owner */
  referrer: string
  /** Token mint for the account */
  mint: string
  /** Amount that has been accumulated and not claimed yet -> available to claim (scaled fraction) */
  amountUnclaimedSf: string
  /** Amount that has been accumulated in total -> both already claimed and unclaimed (scaled fraction) */
  amountCumulativeSf: string
  /** Referrer token state bump, used for address validation */
  bump: string
  padding: Array<string>
}

/** Referrer account -> each owner can have multiple accounts for specific reserves */
export class ReferrerTokenState {
  /** Pubkey of the referrer/owner */
  readonly referrer: PublicKey
  /** Token mint for the account */
  readonly mint: PublicKey
  /** Amount that has been accumulated and not claimed yet -> available to claim (scaled fraction) */
  readonly amountUnclaimedSf: BN
  /** Amount that has been accumulated in total -> both already claimed and unclaimed (scaled fraction) */
  readonly amountCumulativeSf: BN
  /** Referrer token state bump, used for address validation */
  readonly bump: BN
  readonly padding: Array<BN>

  static readonly discriminator = Buffer.from([
    39, 15, 208, 77, 32, 195, 105, 56,
  ])

  static readonly layout = borsh.struct([
    borsh.publicKey("referrer"),
    borsh.publicKey("mint"),
    borsh.u128("amountUnclaimedSf"),
    borsh.u128("amountCumulativeSf"),
    borsh.u64("bump"),
    borsh.array(borsh.u64(), 31, "padding"),
  ])

  constructor(fields: ReferrerTokenStateFields) {
    this.referrer = fields.referrer
    this.mint = fields.mint
    this.amountUnclaimedSf = fields.amountUnclaimedSf
    this.amountCumulativeSf = fields.amountCumulativeSf
    this.bump = fields.bump
    this.padding = fields.padding
  }

  static async fetch(
    c: Connection,
    address: PublicKey,
    programId: PublicKey = PROGRAM_ID
  ): Promise<ReferrerTokenState | null> {
    const info = await c.getAccountInfo(address)

    if (info === null) {
      return null
    }
    if (!info.owner.equals(programId)) {
      throw new Error("account doesn't belong to this program")
    }

    return this.decode(info.data)
  }

  static async fetchMultiple(
    c: Connection,
    addresses: PublicKey[],
    programId: PublicKey = PROGRAM_ID
  ): Promise<Array<ReferrerTokenState | null>> {
    const infos = await c.getMultipleAccountsInfo(addresses)

    return infos.map((info) => {
      if (info === null) {
        return null
      }
      if (!info.owner.equals(programId)) {
        throw new Error("account doesn't belong to this program")
      }

      return this.decode(info.data)
    })
  }

  static decode(data: Buffer): ReferrerTokenState {
    if (!data.slice(0, 8).equals(ReferrerTokenState.discriminator)) {
      throw new Error("invalid account discriminator")
    }

    const dec = ReferrerTokenState.layout.decode(data.slice(8))

    return new ReferrerTokenState({
      referrer: dec.referrer,
      mint: dec.mint,
      amountUnclaimedSf: dec.amountUnclaimedSf,
      amountCumulativeSf: dec.amountCumulativeSf,
      bump: dec.bump,
      padding: dec.padding,
    })
  }

  toJSON(): ReferrerTokenStateJSON {
    return {
      referrer: this.referrer.toString(),
      mint: this.mint.toString(),
      amountUnclaimedSf: this.amountUnclaimedSf.toString(),
      amountCumulativeSf: this.amountCumulativeSf.toString(),
      bump: this.bump.toString(),
      padding: this.padding.map((item) => item.toString()),
    }
  }

  static fromJSON(obj: ReferrerTokenStateJSON): ReferrerTokenState {
    return new ReferrerTokenState({
      referrer: new PublicKey(obj.referrer),
      mint: new PublicKey(obj.mint),
      amountUnclaimedSf: new BN(obj.amountUnclaimedSf),
      amountCumulativeSf: new BN(obj.amountCumulativeSf),
      bump: new BN(obj.bump),
      padding: obj.padding.map((item) => new BN(item)),
    })
  }
}