import { ProtocolVersion } from '@uniswap/client-pools/dist/pools/v1/types_pb'
import { CurrencyAmount, type Currency } from '@uniswap/sdk-core'
import { PositionInfo } from 'components/Liquidity/types'
import { LP_INCENTIVES_REWARD_TOKEN } from 'components/LpIncentives/constants'
import JSBI from 'jsbi'
import { useMemo } from 'react'
import { FeatureFlags } from 'uniswap/src/features/gating/flags'
import { useFeatureFlag } from 'uniswap/src/features/gating/hooks'
import { useLocalizationContext } from 'uniswap/src/features/language/LocalizationContext'
import { useUSDCPrice } from 'uniswap/src/features/transactions/hooks/useUSDCPrice'
import { NumberType } from 'utilities/src/format/types'

interface UseLpIncentivesFormattedEarningsProps {
  liquidityPosition: PositionInfo
  fiatFeeValue0?: CurrencyAmount<Currency>
  fiatFeeValue1?: CurrencyAmount<Currency>
}

interface LpIncentivesEarningsResult {
  uniLpRewardsCurrencyAmount?: CurrencyAmount<Currency>
  uniLpRewardsFiatValue?: CurrencyAmount<Currency>
  totalEarningsFiatValue?: CurrencyAmount<Currency>
  totalFormattedEarnings?: string
  totalFeesFiatValue?: CurrencyAmount<Currency>
  formattedFeesValue?: string
  formattedRewardsValue?: string
  hasRewards: boolean
  hasFees: boolean
}

export function useLpIncentivesFormattedEarnings({
  liquidityPosition,
  fiatFeeValue0,
  fiatFeeValue1,
}: UseLpIncentivesFormattedEarningsProps): LpIncentivesEarningsResult {
  const { formatCurrencyAmount } = useLocalizationContext()
  const isLPIncentivesEnabled = useFeatureFlag(FeatureFlags.LpIncentives)
  const { price: uniPrice } = useUSDCPrice(LP_INCENTIVES_REWARD_TOKEN)

  return useMemo(() => {
    const formatCurrency = (value: CurrencyAmount<Currency>) => {
      return formatCurrencyAmount({
        value,
        type: NumberType.FiatStandard,
      })
    }

    const result: LpIncentivesEarningsResult = {
      uniLpRewardsCurrencyAmount: undefined,
      uniLpRewardsFiatValue: undefined,
      totalFeesFiatValue: undefined,
      formattedFeesValue: undefined,
      formattedRewardsValue: undefined,
      hasRewards: false,
      hasFees: false,
      totalEarningsFiatValue: undefined,
      totalFormattedEarnings: undefined,
    }

    if (fiatFeeValue0 && fiatFeeValue1) {
      result.totalFeesFiatValue = fiatFeeValue0.add(fiatFeeValue1)
      result.formattedFeesValue = formatCurrency(result.totalFeesFiatValue)
      result.hasFees = result.totalFeesFiatValue.greaterThan(0)
    } else if (fiatFeeValue0) {
      result.totalFeesFiatValue = fiatFeeValue0
      result.formattedFeesValue = formatCurrency(result.totalFeesFiatValue)
      result.hasFees = result.totalFeesFiatValue.greaterThan(0)
    } else if (fiatFeeValue1) {
      result.totalFeesFiatValue = fiatFeeValue1
      result.formattedFeesValue = formatCurrency(result.totalFeesFiatValue)
      result.hasFees = result.totalFeesFiatValue.greaterThan(0)
    }

    const shouldIncludeRewards =
      liquidityPosition.version === ProtocolVersion.V4 &&
      liquidityPosition.unclaimedRewardsAmountUni &&
      liquidityPosition.unclaimedRewardsAmountUni !== '0' &&
      isLPIncentivesEnabled

    result.totalEarningsFiatValue = result.totalFeesFiatValue
    result.totalFormattedEarnings = result.formattedFeesValue

    // If no rewards and no LP incentives enabled, return early with just fees data
    if (!shouldIncludeRewards) {
      return result
    }

    // Create a UNI token amount from the unclaimed rewards (using Mainnet UNI)
    const uniLpRewardsCurrencyAmount = CurrencyAmount.fromRawAmount(
      LP_INCENTIVES_REWARD_TOKEN,
      liquidityPosition.unclaimedRewardsAmountUni as string,
    )

    // Set the UNI rewards currency amount in the result
    result.uniLpRewardsCurrencyAmount = uniLpRewardsCurrencyAmount
    result.hasRewards = uniLpRewardsCurrencyAmount.greaterThan(0)

    const uniLpRewardsFiatValue = uniPrice ? uniPrice.quote(uniLpRewardsCurrencyAmount) : undefined
    result.uniLpRewardsFiatValue = uniLpRewardsFiatValue

    if (uniLpRewardsFiatValue) {
      result.formattedRewardsValue = formatCurrency(uniLpRewardsFiatValue)
      // Note: Both rewards are the same currency, but can be across different chains
      result.totalEarningsFiatValue = CurrencyAmount.fromRawAmount(
        uniLpRewardsFiatValue.currency,
        result.totalFeesFiatValue
          ? JSBI.add(result.totalFeesFiatValue.quotient, uniLpRewardsFiatValue.quotient)
          : uniLpRewardsFiatValue.quotient,
      )
      result.totalFormattedEarnings = formatCurrency(result.totalEarningsFiatValue)
    }

    return result
  }, [fiatFeeValue0, fiatFeeValue1, liquidityPosition, isLPIncentivesEnabled, uniPrice, formatCurrencyAmount])
}
