import { useMemo } from 'react'
import useMangoGroup from './useMangoGroup'
import { floorToDecimal } from 'utils/numbers'

export default function useLeverageMax(selectedToken: string) {
  const { group } = useMangoGroup()

  const stakeBank = useMemo(() => {
    return group?.banksMapByName.get(selectedToken)?.[0]
  }, [selectedToken, group])

  const borrowBank = useMemo(() => {
    return group?.banksMapByName.get('SOL')?.[0]
  }, [group])

  const leverageMax = useMemo(() => {
    if (!stakeBank || !borrowBank) return 0
    const borrowInitLiabWeight = borrowBank.scaledInitLiabWeight(
      borrowBank.price,
    )
    const stakeInitAssetWeight = stakeBank.scaledInitAssetWeight(
      stakeBank.price,
    )
    if (!borrowInitLiabWeight || !stakeInitAssetWeight) return 1

    const x = floorToDecimal(stakeInitAssetWeight.toNumber(), 2).div(
      floorToDecimal(borrowInitLiabWeight.toNumber(), 2),
    )
    const conversionRate = borrowBank.uiPrice / stakeBank.uiPrice
    const y =
      1 -
      conversionRate *
        floorToDecimal(stakeInitAssetWeight.toNumber(), 2).toNumber()

    return 1 + x.toNumber() / y
  }, [stakeBank, borrowBank])

  return leverageMax
}