import { useMemo } from 'react'
import useStakeRates from './useStakeRates'
import useMangoGroup from './useMangoGroup'
// import mangoStore from '@store/mangoStore'
import useLeverageMax from './useLeverageMax'
import { JLP_BORROW_TOKEN, LST_BORROW_TOKEN } from 'utils/constants'

// const set = mangoStore.getState().set

export default function useBankRates(selectedToken: string, leverage: number) {
  const { data: stakeRates } = useStakeRates()
  const { jlpGroup, lstGroup } = useMangoGroup()

  // const estimatedMaxAPY = mangoStore((s) => s.estimatedMaxAPY.current)
  const leverageMax = useLeverageMax(selectedToken)

  const [stakeBank, borrowBank] = useMemo(() => {
    const isJlpGroup = selectedToken === 'JLP' || selectedToken === 'USDC'
    const stakeBank = isJlpGroup
      ? jlpGroup?.banksMapByName.get(selectedToken)?.[0]
      : lstGroup?.banksMapByName.get(selectedToken)?.[0]
    const borrowBank = isJlpGroup
      ? jlpGroup?.banksMapByName.get(JLP_BORROW_TOKEN)?.[0]
      : lstGroup?.banksMapByName.get(LST_BORROW_TOKEN)?.[0]
    return [stakeBank, borrowBank]
  }, [selectedToken, jlpGroup, lstGroup])

  const stakeBankDepositRate = useMemo(() => {
    return stakeBank ? stakeBank.getDepositRate() : 0
  }, [stakeBank])

  const borrowBankBorrowRate = useMemo(() => {
    return borrowBank ? Number(borrowBank.getBorrowRate()) : 0
  }, [borrowBank])

  const tokenStakeRateAPY = useMemo(() => {
    return stakeRates ? stakeRates[selectedToken.toLowerCase()] : 0
  }, [stakeRates, selectedToken])

  const financialMetrics = useMemo(() => {
    // Collateral fee is charged on the assets needed to back borrows and
    // 1 deposited JLP can back maintAssetWeight * (1 JLP-value) USDC borrows.
    const collateralFeePerBorrowPerDay =
      Number(stakeBank?.collateralFeePerDay) /
      Number(stakeBank?.maintAssetWeight)

    // Convert the borrow APR to a daily rate
    const borrowRatePerDay = Number(borrowBankBorrowRate) / 365

    // Convert the JLP APY to a daily rate
    const tokenRatePerDay = (1 + tokenStakeRateAPY) ** (1 / 365) - 1

    // Assume the user deposits 1 JLP, then these are the starting deposits and
    // borrows for the desired leverage (in terms of starting-value JLP)
    const initialBorrows = leverage - 1
    const initialDeposits = leverage

    // In the following, we'll simulate time passing and how the deposits and
    // borrows evolve.
    // Note that these will be in terms of starting-value JLP, meaning that JLP
    // price increases will be modelled as deposits increasing in amount.
    let borrows = initialBorrows
    let deposits = initialDeposits

    let collectedCollateralFees = 0
    let collectedReturns = 0

    for (let day = 1; day <= 365; day++) {
      borrows *= 1 + borrowRatePerDay

      const collateralFees = collateralFeePerBorrowPerDay * borrows
      deposits -= collateralFees
      collectedCollateralFees += collateralFees

      const tokenReturns = tokenRatePerDay * deposits
      deposits += tokenReturns
      collectedReturns += tokenReturns
    }

    // APY's for the calculation
    const depositAPY = (deposits - initialDeposits) * 100
    const collateralFeeAPY = collectedCollateralFees * 100
    const collectedReturnsAPY = collectedReturns * 100

    // Interest Fee APY: Reflecting borrowing cost as an annual percentage yield
    const borrowsAPY = (borrows - initialBorrows) * 100

    // Total APY, comparing the end value (deposits - borrows) to the starting value (1)
    const APY = (deposits - borrows - 1) * 100

    // Comparisons to outside
    const nonMangoAPY = tokenStakeRateAPY * leverage * 100
    const diffToNonMango = APY - nonMangoAPY
    const diffToNonLeveraged = APY - tokenStakeRateAPY * 100

    return {
      APY,
      depositAPY,
      collectedReturnsAPY,
      collateralFeeAPY,
      borrowsAPY,
      nonMangoAPY,
      diffToNonMango,
      diffToNonLeveraged,
    }
  }, [
    stakeBank?.collateralFeePerDay,
    stakeBank?.maintAssetWeight,
    borrowBankBorrowRate,
    tokenStakeRateAPY,
    leverage,
  ])

  const estimatedMaxAPY = useMemo(() => {
    return (
      tokenStakeRateAPY * leverageMax -
      Number(borrowBankBorrowRate) * (leverageMax - 1)
    )
  }, [tokenStakeRateAPY, borrowBankBorrowRate, leverageMax])

  return {
    financialMetrics,
    stakeBankDepositRate,
    borrowBankBorrowRate,
    jlpStakeRateAPY: tokenStakeRateAPY,
    estimatedMaxAPY,
  }
}