@unloc-xyz/unloc-sdk/src/loan/index.ts

Unloc SDK

import * as anchor from '@project-serum/anchor'
import { bool, publicKey, struct, u32, u64, u8 } from '@project-serum/borsh'
import { NATIVE_MINT, Token, TOKEN_PROGRAM_ID } from '@solana/spl-token'
import { IDL as idl, UnlocLoan } from '../types/unloc_loan'
import { NFT_LOAN_PID, TOKEN_META_PID, UNLOC_MINT } from '../global-config'
import {
  Connection,
  Keypair,
  MemcmpFilter,
  PublicKey,
  SystemInstruction,
  SystemProgram,
  TransactionInstruction
} from '@solana/web3.js'
import { bs58 } from '@project-serum/anchor/dist/cjs/utils/bytes'
import { Edition, MasterEdition, Metadata, MetadataData, MetadataKey } from '@metaplex-foundation/mpl-token-metadata'
import axios from 'axios'
import { ArweaveMetadata, IMasterEdition, IMetadata, OnChainMetadata } from './IOfferData'
import { VOTING_ITEM_TAG, VOTING_TAG } from './../voting'
import { clock, defaults, discriminatorLen, pda, SOLANA_CONNECTION, systemProgram, tokenProgram } from './../utils'
import { getTokenAccount, parseTokenAccount } from '@project-serum/common'


export let program: anchor.Program<UnlocLoan> = null as unknown as anchor.Program<UnlocLoan>
export let programProvider: anchor.AnchorProvider = null as unknown as anchor.AnchorProvider
export let programId: anchor.web3.PublicKey = null as unknown as anchor.web3.PublicKey

const GLOBAL_STATE_TAG = Buffer.from('GLOBAL_STATE_SEED')
const OFFER_TAG = Buffer.from('OFFER_SEED')
const SUB_OFFER_TAG = Buffer.from('SUB_OFFER_SEED')
const NFT_VAULT_TAG = Buffer.from('NFT_VAULT_SEED')
const OFFER_VAULT_TAG = Buffer.from('OFFER_VAULT_SEED')
const TREASURY_VAULT_TAG = Buffer.from('TREASURY_VAULT_SEED')
const REWARD_VAULT_TAG = Buffer.from('REWARD_VAULT_SEED')
const USER_REWARD_TAG = Buffer.from('LENDER_REWARD_SEED')

const WSOL_MINT = new anchor.web3.PublicKey('So11111111111111111111111111111111111111112')
// This command makes an Lottery
export const initLoanProgram = (
  wallet: any,
  connection: anchor.web3.Connection = SOLANA_CONNECTION,
  pid: anchor.web3.PublicKey = NFT_LOAN_PID
) => {
  if (program != null) {
    return
  }
  programId = pid
  const provider = new anchor.AnchorProvider(connection, wallet, { skipPreflight: true })
  programProvider = provider;

  // Generate the program client from IDL.
  program = new (anchor as any).Program(idl, programId, provider) as anchor.Program<UnlocLoan>
}
// initLoanProgram(SOLANA_CONNECTION, Keypair.generate())

export const getLoanGlobalState = async () => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  return await program.account.globalState.fetchNullable(globalState)
}

export const getCurrentVotingKey = async () => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const globalStateData = await program.account.globalState.fetch(globalState)
  const currentVotingNum = globalStateData?.currentVotingNum;
  const votingPid = globalStateData?.votingPid;
  const voting = await pda(
    [VOTING_TAG, currentVotingNum.toArrayLike(Buffer, "be", 8)],
    votingPid
  );
  return voting;
};


export const getLoanOfferList = async () => {
  return await program.account.offer.all()
}

export const getLoanOfferMultiple = async (keys: anchor.web3.PublicKey[]) => {
  return await program.account.offer.fetchMultiple(keys)
}

export const getLoanOffer = async (key: anchor.web3.PublicKey) => {
  return await program.account.offer.fetchNullable(key)
}

export const getLoanOffersBy = async (
  owner?: anchor.web3.PublicKey,
  nftMint?: anchor.web3.PublicKey,
  state?: OfferState
) => {
  const filters: any[] = []
  if (owner) {
    const filter: MemcmpFilter = {
      memcmp: {
        offset: discriminatorLen,
        bytes: owner.toBase58()
      }
    }
    filters.push(filter)
  }
  if (nftMint) {
    const filter: MemcmpFilter = {
      memcmp: {
        offset: discriminatorLen + 32,
        bytes: nftMint.toBase58()
      }
    }
    filters.push(filter)
  }
  if (state) {
    const filter: MemcmpFilter = {
      memcmp: {
        offset: discriminatorLen + 96,
        bytes: bs58.encode([state])
      }
    }
    filters.push(filter)
  }

  return await program.account.offer.all(filters)
}

export const getLoanSubOffer = async (key: anchor.web3.PublicKey) => {
  const subOffer = await program.account.subOffer.fetchNullable(key)
  if (subOffer) {
    const offerKey = subOffer.offer
    const offerData = await program.account.offer.fetch(offerKey)
    if (offerData.startSubOfferNum.toNumber() > subOffer.subOfferNumber.toNumber()) {
      return null
    }
  }
  return subOffer
}

export const getLoanSubOfferList = async (
  offer?: anchor.web3.PublicKey,
  nftMint?: anchor.web3.PublicKey,
  state?: SubOfferState
) => {
  const accountName = 'subOffer'

  const filters: any[] = []
  if (offer) {
    const filter: MemcmpFilter = {
      memcmp: {
        offset: discriminatorLen + 97,
        bytes: offer.toBase58()
      }
    }
    filters.push(filter)
  }
  if (nftMint) {
    const filter: MemcmpFilter = {
      memcmp: {
        offset: discriminatorLen + 32,
        bytes: nftMint.toBase58()
      }
    }
    filters.push(filter)
  }
  if (state) {
    const filter: MemcmpFilter = {
      memcmp: {
        offset: discriminatorLen + 96,
        bytes: bs58.encode([state])
      }
    }
    filters.push(filter)
  }

  const subOffers = await program.account.subOffer.all(filters)
  const result: any[] = []
  for (let i = 0; i < subOffers.length; i++) {
    const offerKey = subOffers[i].account.offer
    const offerData = await program.account.offer.fetch(offerKey)
    if (offerData.startSubOfferNum.toNumber() <= subOffers[i].account.subOfferNumber.toNumber()) {
      result.push(subOffers[i])
    }
  }
  return result
}

export const getAllLoanSubOffers = async () => {
  const subOffers = await program.account.subOffer.all()
  const result: any[] = []
  for (let i = 0; i < subOffers.length; i++) {
    const offerKey = subOffers[i].account.offer
    const offerData = await program.account.offer.fetch(offerKey)
    if (offerData.startSubOfferNum.toNumber() <= subOffers[i].account.subOfferNumber.toNumber()) {
      result.push(subOffers[i])
    }
  }
  return result
}

export const getLoanSubOfferMultiple = async (keys: anchor.web3.PublicKey[], offerState?: number) => {
  const subOffers: any[] = await program.account.subOffer.fetchMultiple(keys)

  // Filter for nulls and get the offer address
  const offerAddresses = subOffers.filter((subOffer) => subOffer).map((subOffer) => subOffer.offer)

  // Get all offer data in a single call instead of a loop, filter for nulls
  const offers: any[] = (await program.account.offer.fetchMultiple(offerAddresses)).filter((data) => data)

  const result: any[] = []
  for (let i = 0; i < subOffers.length; i++) {
    if (subOffers[i]) {
      const offerKey = subOffers[i].offer
      const offerKeyIndex = offerAddresses.findIndex((address) => address.equals(offerKey))
      const offerData = offers[offerKeyIndex]

      // If we want to, check the offer state too.
      // Suboffer can be proposed when the offer isn't, so that's why we check the state on the offer, not just the suboffer.
      // Be careful, offerState = 0 can coerce to false.
      if (typeof offerState === 'number') {
        if (offerData.startSubOfferNum.toNumber() <= subOffers[i].subOfferNumber.toNumber() && offerData.state === offerState) {
          result.push({ ...subOffers[i], subOfferKey: keys[i] })
        }
      } else {
        if (offerData.startSubOfferNum.toNumber() <= subOffers[i].subOfferNumber.toNumber()) {
          result.push({ ...subOffers[i], subOfferKey: keys[i] })
        }
      }
    }
  }
  return result
}

export const setLoanGlobalState = async (
  accruedInterestNumerator: anchor.BN,
  denominator: anchor.BN,
  aprNumerator: anchor.BN,
  expireLoanDuration: anchor.BN,
  rewardPerSol: anchor.BN,
  rewardPerUsdc: anchor.BN,
  lenderRewardsPercentage: anchor.BN,
  rewardMint: anchor.web3.PublicKey,
  treasury: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const rewardVault = await pda([REWARD_VAULT_TAG], programId)
  const superOwner = signer

  try {
    const tx = await program.rpc.setGlobalState(
      accruedInterestNumerator,
      denominator,
      aprNumerator,
      expireLoanDuration,
      rewardPerSol,
      rewardPerUsdc,
      lenderRewardsPercentage,
      {
        accounts: {
          superOwner,
          globalState,
          rewardMint,
          rewardVault,
          newSuperOwner: superOwner,
          treasuryWallet: treasury,
          ...defaults
        },
        signers
      }
    )

    // eslint-disable-next-line no-console
    console.log('setGlobalState tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}

export const setLoanInternalInfo = async (
  unlocStakingPid: anchor.web3.PublicKey,
  unlocStakingPoolId: anchor.web3.PublicKey,
  votingPid: anchor.web3.PublicKey,
  tokenMetadataPid: anchor.web3.PublicKey,
  currentVotingNum: anchor.BN,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const superOwner = signer

  try {
    const tx = await program.rpc.setInternalInfo(
      unlocStakingPid,
      unlocStakingPoolId,
      votingPid,
      tokenMetadataPid,
      currentVotingNum,
      {
        accounts: {
          superOwner,
          globalState,
        },
        signers
      }
    )

    // eslint-disable-next-line no-console
    console.log('setInternalInfo tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}

export const depositRewards = async (
  amount: anchor.BN,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const rewardMint = UNLOC_MINT
  const rewardVault = await pda([REWARD_VAULT_TAG], programId)
  const authority = signer
  const userRewardVault = await checkWalletATA(rewardMint.toBase58(), programProvider.connection, authority)

  const tx = await program.rpc.depositRewards(
    amount,
    {
      accounts: {
        authority,
        rewardVault,
        userRewardVault,
        ...defaults
      },
      signers
    }
  )

  // eslint-disable-next-line no-console
  console.log('deposit rewards tx = ', tx)
}

export const claimRewards = async (
  userReward: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const rewardMint = UNLOC_MINT
  const rewardVault = await pda([REWARD_VAULT_TAG], programId)
  const authority = signer
  let userRewardVault = await checkWalletATA(rewardMint.toBase58(), programProvider.connection, authority)
  const preInstructions: TransactionInstruction[] = []
  if (!userRewardVault) {
    userRewardVault = await addTokenAccountInstruction(rewardMint, authority, preInstructions, signer, signers)
  }
  const tx = await program.rpc.claimRewards(
    {
      accounts: {
        authority,
        globalState,
        rewardVault,
        userReward,
        userRewardVault,
        ...defaults
      },
      preInstructions,
      signers
    }
  )

  // eslint-disable-next-line no-console
  console.log('claim rewards tx = ', tx)
}

export const setLoanOffer = async (
  nftMint: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const borrowerNftVault: anchor.web3.PublicKey = await checkWalletATA(
    nftMint.toBase58(),
    programProvider.connection,
    signer
  )

  const borrower = signer
  const offer = await pda([OFFER_TAG, borrower.toBuffer(), nftMint.toBuffer()], programId)
  const nftMetadata = await Metadata.getPDA(nftMint)
  const edition = await Edition.getPDA(nftMint);

  try {
    const tx = await program.rpc.setOffer({
      accounts: {
        borrower,
        globalState,
        offer,
        nftMint,
        nftMetadata,
        edition,
        userVault: borrowerNftVault,
        metadataProgram: TOKEN_META_PID,
        ...defaults
      },
      signers
    })

    // eslint-disable-next-line no-console
    console.log('setOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const cancelLoanOffer = async (
  nftMint: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const borrower = signer
  const offer = await pda([OFFER_TAG, borrower.toBuffer(), nftMint.toBuffer()], programId)

  let borrowerNftVault = await checkWalletATA(nftMint.toBase58(), programProvider.connection, borrower)
  const preInstructions: TransactionInstruction[] = []
  if (!borrowerNftVault) {
    borrowerNftVault = await addTokenAccountInstruction(nftMint, borrower, preInstructions, signer, signers)
  }
  const edition = await Edition.getPDA(nftMint);
  try {
    const tx = await program.rpc.cancelOffer({
      accounts: {
        borrower,
        offer,
        nftMint,
        edition,
        userVault: borrowerNftVault,
        metadataProgram: TOKEN_META_PID,
        ...defaults
      },
      preInstructions,
      signers
    })

    // eslint-disable-next-line no-console
    console.log('cancelOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const createLoanSubOffer = async (
  offerAmount: anchor.BN,
  loanDuration: anchor.BN,
  minRepaidNumerator: anchor.BN,
  aprNumerator: anchor.BN,
  nftMint: anchor.web3.PublicKey,
  offerMint: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  if (offerMint.equals(NATIVE_MINT)) {
    offerMint = WSOL_MINT
  }
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const borrower = signer
  const offer = await pda([OFFER_TAG, borrower.toBuffer(), nftMint.toBuffer()], programId)
  const offerData = await program.account.offer.fetch(offer)
  const subOfferNumer = offerData.subOfferCount
  const subOffer = await pda([SUB_OFFER_TAG, offer.toBuffer(), subOfferNumer.toArrayLike(Buffer, 'be', 8)], programId)
  const treasuryVault = await pda([TREASURY_VAULT_TAG, offerMint.toBuffer()], programId)
  const globalStateData = await program.account.globalState.fetch(globalState)
  const treasuryWallet = globalStateData.treasuryWallet
  try {
    const tx = await program.rpc.setSubOffer(offerAmount, subOfferNumer, loanDuration, minRepaidNumerator, aprNumerator, {
      accounts: {
        borrower,
        globalState,
        offer,
        subOffer,
        offerMint,
        treasuryWallet,
        treasuryVault,
        ...defaults
      },
      signers
    })

    // eslint-disable-next-line no-console
    console.log('createSubOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const createLoanSubOfferByStaking = async (
  offerAmount: anchor.BN,
  loanDuration: anchor.BN,
  minRepaidNumerator: anchor.BN,
  aprNumerator: anchor.BN,
  nftMint: anchor.web3.PublicKey,
  offerMint: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  if (offerMint.equals(NATIVE_MINT)) {
    offerMint = WSOL_MINT
  }
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const borrower = signer
  const offer = await pda([OFFER_TAG, borrower.toBuffer(), nftMint.toBuffer()], programId)
  const offerData = await program.account.offer.fetch(offer)
  const subOfferNumer = offerData.subOfferCount
  const subOffer = await pda([SUB_OFFER_TAG, offer.toBuffer(), subOfferNumer.toArrayLike(Buffer, 'be', 8)], programId)
  const treasuryVault = await pda([TREASURY_VAULT_TAG, offerMint.toBuffer()], programId)
  const globalStateData = await program.account.globalState.fetch(globalState)
  const stakingUser = await pda([globalStateData.unlocStakingPoolId.toBuffer(), borrower.toBuffer()], globalStateData.unlocStakingPid)
  const treasuryWallet = globalStateData.treasuryWallet
  try {
    const tx = await program.rpc.setSubOfferByStaking(offerAmount, subOfferNumer, loanDuration, minRepaidNumerator, aprNumerator, {
      accounts: {
        borrower,
        globalState,
        offer,
        subOffer,
        stakingUser,
        offerMint,
        treasuryWallet,
        treasuryVault,
        ...defaults
      },
      signers
    })

    // eslint-disable-next-line no-console
    console.log('createSubOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const updateLoanSubOffer = async (
  offerAmount: anchor.BN,
  loanDuration: anchor.BN,
  minRepaidNumerator: anchor.BN,
  aprNumerator: anchor.BN,

  subOffer: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const borrower = signer
  const subOfferData = await program.account.subOffer.fetch(subOffer)
  const offer = await pda([OFFER_TAG, borrower.toBuffer(), subOfferData.nftMint.toBuffer()], programId)
  const subOfferNumer = subOfferData.subOfferNumber
  const offerMint = subOfferData.offerMint
  const treasuryVault = await pda([TREASURY_VAULT_TAG, subOfferData.offerMint.toBuffer()], programId)
  const globalStateData = await program.account.globalState.fetch(globalState)

  const treasuryWallet = globalStateData.treasuryWallet
  try {
    const tx = await program.rpc.setSubOffer(offerAmount, subOfferNumer, loanDuration, minRepaidNumerator, aprNumerator, {
      accounts: {
        borrower,
        globalState,
        offer,
        subOffer,
        offerMint,
        treasuryWallet,
        treasuryVault,
        ...defaults
      },
      signers
    })

    // eslint-disable-next-line no-console
    console.log('updateSubOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const updateLoanSubOfferByStaking = async (
  offerAmount: anchor.BN,
  loanDuration: anchor.BN,
  minRepaidNumerator: anchor.BN,
  aprNumerator: anchor.BN,

  subOffer: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const borrower = signer
  const subOfferData = await program.account.subOffer.fetch(subOffer)
  const offer = await pda([OFFER_TAG, borrower.toBuffer(), subOfferData.nftMint.toBuffer()], programId)
  const subOfferNumer = subOfferData.subOfferNumber
  const offerMint = subOfferData.offerMint
  const treasuryVault = await pda([TREASURY_VAULT_TAG, subOfferData.offerMint.toBuffer()], programId)
  const globalStateData = await program.account.globalState.fetch(globalState)
  const stakingUser = await pda([globalStateData.unlocStakingPoolId.toBuffer(), borrower.toBuffer()], globalStateData.unlocStakingPid)

  const treasuryWallet = globalStateData.treasuryWallet
  try {
    const tx = await program.rpc.setSubOfferByStaking(offerAmount, subOfferNumer, loanDuration, minRepaidNumerator, aprNumerator, {
      accounts: {
        borrower,
        globalState,
        offer,
        subOffer,
        stakingUser,
        offerMint,
        treasuryWallet,
        treasuryVault,
        ...defaults
      },
      signers
    })

    // eslint-disable-next-line no-console
    console.log('updateSubOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const cancelLoanSubOffer = async (
  subOffer: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const borrower = signer
  const subOfferData = await program.account.subOffer.fetch(subOffer)
  const offer = await pda([OFFER_TAG, borrower.toBuffer(), subOfferData.nftMint.toBuffer()], programId)
  try {
    const tx = await program.rpc.cancelSubOffer({
      accounts: {
        borrower,
        offer,
        subOffer
      },
      signers
    })

    // eslint-disable-next-line no-console
    console.log('cancelSubOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const acceptLoanOfferByVoting = async (
  subOffer: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const lender = signer
  const subOfferData = await program.account.subOffer.fetch(subOffer)
  const offer = subOfferData.offer
  const offerMint = subOfferData.offerMint
  const offerData = await program.account.offer.fetch(subOfferData.offer)
  const borrower = offerData.borrower
  let borrowerOfferVault = await checkWalletATA(offerMint.toBase58(), programProvider.connection, borrower)
  let lenderOfferVault = await checkWalletATA(offerMint.toBase58(), programProvider.connection, lender)
  const preInstructions: TransactionInstruction[] = []
  const postInstructions: TransactionInstruction[] = []
  if (offerMint.equals(WSOL_MINT)) {
    const treasuryVault = await pda([TREASURY_VAULT_TAG, offerMint.toBuffer()], programId)
    lenderOfferVault = treasuryVault
    borrowerOfferVault = treasuryVault
  } else {
    if (!borrowerOfferVault) {
      console.log("borrower doesn't have offer token account!")
      borrowerOfferVault = await addTokenAccountInstruction(offerMint, borrower, preInstructions, signer, signers)
    }
    if (!lenderOfferVault) {
      console.log("lender doesn't have offer token!")
      return
    }
  }
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const globalStateData = await program.account.globalState.fetch(globalState)
  const userReward = await pda([USER_REWARD_TAG, lender.toBuffer(), borrower.toBuffer(), subOffer.toBuffer()], programId)
  const votingPid = globalStateData.votingPid;
  const currentVotingKey = await getCurrentVotingKey();
  const votingItemKey = await pda([VOTING_ITEM_TAG, currentVotingKey.toBuffer(), offerData.collection.toBuffer()], votingPid)
  try {
    const tx = await program.rpc.acceptOfferByVoting({
      accounts: {
        lender,
        borrower,
        globalState,
        offer,
        subOffer,
        offerMint,
        userReward,
        voting: currentVotingKey,
        votingItem: votingItemKey,
        borrowerOfferVault,
        lenderOfferVault,
        ...defaults
      },
      preInstructions,
      postInstructions,
      signers
    })

    // eslint-disable-next-line no-console
    console.log('acceptOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }
}
export const acceptLoanOffer = async (
  subOffer: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const lender = signer
  const subOfferData = await program.account.subOffer.fetch(subOffer)
  const offer = subOfferData.offer
  const offerMint = subOfferData.offerMint
  const offerData = await program.account.offer.fetch(subOfferData.offer)
  const borrower = offerData.borrower
  let borrowerOfferVault = await checkWalletATA(offerMint.toBase58(), programProvider.connection, borrower)
  let lenderOfferVault = await checkWalletATA(offerMint.toBase58(), programProvider.connection, lender)
  const preInstructions: TransactionInstruction[] = []
  const postInstructions: TransactionInstruction[] = []
  if (offerMint.equals(WSOL_MINT)) {
    const treasuryVault = await pda([TREASURY_VAULT_TAG, offerMint.toBuffer()], programId)
    lenderOfferVault = treasuryVault
    borrowerOfferVault = treasuryVault
  } else {
    if (!borrowerOfferVault) {
      console.log("borrower doesn't have offer token account!")
      borrowerOfferVault = await addTokenAccountInstruction(offerMint, borrower, preInstructions, signer, signers)
    }
    if (!lenderOfferVault) {
      console.log("lender doesn't have offer token!")
      return
    }
  }
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const userReward = await pda([USER_REWARD_TAG, lender.toBuffer(), borrower.toBuffer(), subOffer.toBuffer()], programId)
  try {
    const tx = await program.rpc.acceptOffer({
      accounts: {
        lender,
        borrower,
        globalState,
        offer,
        subOffer,
        offerMint,
        userReward,
        borrowerOfferVault,
        lenderOfferVault,
        ...defaults
      },
      preInstructions,
      postInstructions,
      signers
    })

    // eslint-disable-next-line no-console
    console.log('acceptOffer tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const repayLoan = async (
  subOffer: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const borrower = signer
  const globalStateData = await program.account.globalState.fetch(globalState)
  const treasuryWallet = globalStateData.treasuryWallet
  const subOfferData = await program.account.subOffer.fetch(subOffer)
  const offer = subOfferData.offer
  const lender = subOfferData.lender
  const offerData = await program.account.offer.fetch(offer)
  const offerMint = subOfferData.offerMint
  const nftMint = offerData.nftMint
  const treasuryVault = await pda([TREASURY_VAULT_TAG, offerMint.toBuffer()], programId)

  let borrowerOfferVault = await checkWalletATA(offerMint.toBase58(), programProvider.connection, borrower)
  let borrowerNftVault = await checkWalletATA(nftMint.toBase58(), programProvider.connection, borrower)
  let lenderOfferVault = await checkWalletATA(offerMint.toBase58(), programProvider.connection, lender)
  const preInstructions: TransactionInstruction[] = []
  const postInstructions: TransactionInstruction[] = []
  if (offerMint.equals(WSOL_MINT)) {
    const treasuryVault = await pda([TREASURY_VAULT_TAG, offerMint.toBuffer()], programId)
    lenderOfferVault = treasuryVault
    borrowerOfferVault = treasuryVault
  }

  if (!borrowerOfferVault) {
    console.log("borrower doesn't have offer token!")
    return
  }
  if (!borrowerNftVault) {
    console.log("borrower doesn't have nft token token account!")
    borrowerNftVault = await addTokenAccountInstruction(nftMint, borrower, preInstructions, signer, signers)
  }
  if (!lenderOfferVault) {
    console.log("lender doesn't have offer token account!", lender.toBase58())
    lenderOfferVault = await addTokenAccountInstruction(offerMint, lender, preInstructions, signer, signers)
  }

  const userReward = await pda([USER_REWARD_TAG, lender.toBuffer(), borrower.toBuffer(), subOffer.toBuffer()], programId)
  const edition = await Edition.getPDA(nftMint);

  try {
    const tx = await program.rpc.repayLoan({
      accounts: {
        borrower,
        lender,
        globalState,
        treasuryWallet,
        offer,
        subOffer,
        nftMint,
        edition,
        userReward,
        borrowerNftVault,
        lenderOfferVault,
        borrowerOfferVault,
        treasuryVault,
        metadataProgram: TOKEN_META_PID,
        ...defaults
      },
      preInstructions,
      postInstructions,
      signers
    })

    // eslint-disable-next-line no-console
    console.log('repayLoan tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
export const claimLoanCollateral = async (
  subOffer: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const lender = signer
  const globalStateData = await program.account.globalState.fetch(globalState)
  const treasuryWallet = globalStateData.treasuryWallet
  const subOfferData = await program.account.subOffer.fetch(subOffer)
  const offer = subOfferData.offer
  const offerData = await program.account.offer.fetch(offer)
  const offerMint = subOfferData.offerMint
  const nftMint = offerData.nftMint
  const treasuryVault = await pda([TREASURY_VAULT_TAG, offerMint.toBuffer()], programId)

  let borrowerNftVault = await checkWalletATA(nftMint.toBase58(), programProvider.connection, offerData.borrower)
  let lenderNftVault = await checkWalletATA(nftMint.toBase58(), programProvider.connection, lender)
  let lenderOfferVault = await checkWalletATA(offerMint.toBase58(), programProvider.connection, lender)
  const preInstructions: TransactionInstruction[] = []
  const postInstructions: TransactionInstruction[] = []
  if (offerMint.equals(WSOL_MINT)) {
    const treasuryVault = await pda([TREASURY_VAULT_TAG, offerMint.toBuffer()], programId)
    lenderOfferVault = treasuryVault
  }

  if (!lenderNftVault) {
    console.log("lender doesn't have nft token token account!")
    lenderNftVault = await addTokenAccountInstruction(nftMint, lender, preInstructions, signer, signers)
  }
  if (!lenderOfferVault) {
    console.log("lender doesn't have offer token account!", lender.toBase58())
    lenderOfferVault = await addTokenAccountInstruction(offerMint, lender, preInstructions, signer, signers)
  }

  const edition = await Edition.getPDA(nftMint);
  try {
    const tx = await program.rpc.claimCollateral({
      accounts: {
        lender,
        globalState,
        treasuryWallet,
        offer,
        subOffer,
        nftMint,
        edition,
        lenderNftVault,
        borrowerNftVault,
        lenderOfferVault,
        treasuryVault,
        metadataProgram: TOKEN_META_PID,
        ...defaults
      },
      preInstructions,
      postInstructions,
      signers
    })

    // eslint-disable-next-line no-console
    console.log('claimCollateral tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}
// super owner can call this function
export const claimExpiredCollateral = async (
  subOffer: anchor.web3.PublicKey,
  signer: anchor.web3.PublicKey = programProvider.wallet.publicKey,
  signers: anchor.web3.Keypair[] = []
) => {
  const globalState = await pda([GLOBAL_STATE_TAG], programId)
  const globalStateData = await program.account.globalState.fetch(globalState)
  const treasuryWallet = globalStateData.treasuryWallet
  const subOfferData = await program.account.subOffer.fetch(subOffer)
  const offer = subOfferData.offer
  const offerData = await program.account.offer.fetch(offer)
  const nftMint = offerData.nftMint

  let userNftVault = await checkWalletATA(nftMint.toBase58(), programProvider.connection, signer)
  const preInstructions: TransactionInstruction[] = []

  if (!userNftVault) {
    userNftVault = await addTokenAccountInstruction(nftMint, signer, preInstructions, signer, signers)
  }

  let borrowerNftVault = await checkWalletATA(nftMint.toBase58(), programProvider.connection, offerData.borrower)
  const edition = await Edition.getPDA(nftMint);
  try {
    const tx = await program.rpc.claimExpiredCollateral({
      accounts: {
        superOwner: signer,
        globalState,
        treasuryWallet,
        offer,
        subOffer,
        nftMint,
        edition,
        userNftVault,
        borrowerNftVault,
        metadataProgram: TOKEN_META_PID,
        ...defaults
      },
      preInstructions,
      signers
    })

    // eslint-disable-next-line no-console
    console.log('claimExpiredCollateral tx = ', tx)
  } catch (e) {
    console.log(e);
  }

}

export enum OfferState {
  Proposed,
  Accepted,
  Expired,
  Fulfilled,
  NFTClaimed,
  Canceled
}

export enum SubOfferState {
  Proposed,
  Accepted,
  Expired,
  Fulfilled,
  LoanPaymentClaimed,
  Canceled
}

export const getLoanSubOffersKeysByState = async (state: SubOfferState[]) => {
  try {
    const accountName = 'subOffer'
    const data = await programProvider.connection.getProgramAccounts(programId, {
      dataSlice: { offset: 0, length: 0 }, // Fetch without any data.
      filters: [
        { memcmp: program.account.subOffer.coder.accounts.memcmp(accountName) },
        { memcmp: { offset: discriminatorLen + 96, bytes: bs58.encode(state) } } // add filters for nftMint, APR, Duration, Amount and this function will be glorious
      ]
    })
    return data.map((offer) => offer.pubkey)
  } catch (e) {
    console.log(e)
  }
}

export const addTokenAccountInstruction = async (
  mint: anchor.web3.PublicKey,
  owner: anchor.web3.PublicKey,
  instructions: TransactionInstruction[],
  signer: anchor.web3.PublicKey,
  signers: anchor.web3.Keypair[],
  rent: number = 0
) => {
  const newKeypair = Keypair.generate()
  const rentForTokenAccount = await Token.getMinBalanceRentForExemptAccount(programProvider.connection)
  instructions.push(
    SystemProgram.createAccount({
      fromPubkey: signer,
      newAccountPubkey: newKeypair.publicKey,
      lamports: rent > 0 ? rent : rentForTokenAccount,
      space: ACCOUNT_LAYOUT.span,
      programId: TOKEN_PROGRAM_ID
    })
  )
  const instruction = Token.createInitAccountInstruction(TOKEN_PROGRAM_ID, mint, newKeypair.publicKey, owner)
  instructions.push(instruction)
  signers.push(newKeypair)
  return newKeypair.publicKey
}

export const checkWalletATA = async (
  mint: string,
  connection: anchor.web3.Connection = programProvider.connection,
  walletPubkey: anchor.web3.PublicKey = programProvider.wallet.publicKey
) => {
  const parsedTokenAccounts = await connection.getParsedTokenAccountsByOwner(
    walletPubkey,
    {
      programId: TOKEN_PROGRAM_ID
    },
    'confirmed'
  )
  let result: any = null
  let maxAmount = 0
  parsedTokenAccounts.value.forEach(async (tokenAccountInfo) => {
    const tokenAccountPubkey = tokenAccountInfo.pubkey
    const parsedInfo = tokenAccountInfo.account.data.parsed.info
    const mintAddress = parsedInfo.mint
    const amount = parsedInfo.tokenAmount.uiAmount
    if (mintAddress === mint && amount >= maxAmount) {
      result = tokenAccountPubkey
      maxAmount = amount
    }
  })

  return result
}

export const logObject = (title: string, obj: any) => {
  console.log(title, obj)
}
export const getOfferBalance = async (
  subOffer: anchor.web3.PublicKey,
  connection: anchor.web3.Connection = programProvider.connection,
  walletPubkey: anchor.web3.PublicKey = programProvider.wallet.publicKey
) => {
  try {
    const tokenAccount = await pda([OFFER_VAULT_TAG, subOffer.toBuffer()], programId)
    const balance = (await connection.getTokenAccountBalance(tokenAccount)).value.uiAmount as number
    return balance
  } catch (e) {
    console.log(e)
  }
  return 0
}
// --------------------utilities---------------------

export class MultipleNFT {
  public mints: anchor.web3.PublicKey[]
  public metadatas: NFTMetadata[] = []
  constructor(keys: anchor.web3.PublicKey[]) {
    this.mints = keys
  }
  public async initialize() {
    if (this.metadatas.length == 0) {
      const metadataPDAs: PublicKey[] = []
      for (let i = 0; i < this.mints.length; i++) {
        const metadataPDA = await Metadata.getPDA(this.mints[i])
        metadataPDAs.push(metadataPDA)
      }
      let i = 0
      const metaInfos = await Metadata.getInfos(programProvider.connection, metadataPDAs)
      metaInfos.forEach((value: anchor.web3.AccountInfo<Buffer>, key: any, map: any) => {
        const metadata = MetadataData.deserialize(value.data)
        const nftMetadata = new NFTMetadata(this.mints[i], metadataPDAs[i], metadata)
        this.metadatas.push(nftMetadata)
        i++
      })
    }
  }
  public getNftMeta(mint: PublicKey) {
    for (let i = 0; i < this.metadatas.length; i++) {
      if (this.metadatas[i].mint === mint.toBase58()) {
        return this.metadatas[i]
      }
    }
    return null
  }
  public async initArweavedata() {
    for (let i = 0; i < this.metadatas.length; i++) {
      await this.metadatas[i].getAreaveMetadata()
    }
  }
}

export class NFTMetadata implements IMetadata {
  mint = ''
  metadataPDA = ''
  onChainMetadata: OnChainMetadata = null as any
  arweaveMetadata: ArweaveMetadata = null as any
  masterEdition: IMasterEdition = null as any
  constructor(mint: anchor.web3.PublicKey, metadataPDA: anchor.web3.PublicKey, onChainMetadata: OnChainMetadata) {
    this.mint = mint.toBase58()
    this.metadataPDA = metadataPDA.toBase58()
    this.onChainMetadata = onChainMetadata
  }

  public async getAreaveMetadata() {
    if (this.arweaveMetadata === null) {
      try {
        const arweaveData = await axios.get(this.onChainMetadata.data.uri)
        if (arweaveData) {
          this.arweaveMetadata = arweaveData.data
        }
      } catch (e) {
        // eslint-disable-next-line no-console
        console.log(e)
      }
    }
    return this.arweaveMetadata
  }
  public currentAreaveMetadata() {
    return this.arweaveMetadata
  }
  public async getMasterEdition() {
    if (this.arweaveMetadata === null) {
      const masterEditionPDA = await Edition.getPDA(new PublicKey(this.mint))
      const editionAccountInfo = await programProvider.connection.getAccountInfo(masterEditionPDA)

      if (editionAccountInfo) {
        const key = editionAccountInfo?.data[0]
        let masterEditionData
        let data

        switch (key) {
          case MetadataKey.MasterEditionV1:
          case MetadataKey.MasterEditionV2:
            ; ({ data } = new MasterEdition(masterEditionPDA, editionAccountInfo))
            masterEditionData = data
            break
          default:
            masterEditionData = undefined
            break
        }

        this.masterEdition = {
          masterEditionPDA: masterEditionPDA.toString(),
          masterEditionData
        }
      }
    }
    return this.masterEdition
  }
}
export const ACCOUNT_LAYOUT = struct([
  publicKey('mint'),
  publicKey('owner'),
  u64('amount'),
  u32('delegateOption'),
  publicKey('delegate'),
  u8('state'),
  u32('isNativeOption'),
  u64('isNative'),
  u64('delegatedAmount'),
  u32('closeAuthorityOption'),
  publicKey('closeAuthority')
])

export const MINT_LAYOUT = struct([
  u32('mintAuthorityOption'),
  publicKey('mintAuthority'),
  u64('supply'),
  u8('decimals'),
  bool('initialized'),
  u32('freezeAuthorityOption'),
  publicKey('freezeAuthority')
])