import { ChainId } from '@uniswap/sdk-core'
import { ProtocolVersion, Token, useV2PairQuery, useV3PoolQuery } from 'graphql/data/__generated__/types-and-hooks'
import { V2_BIPS } from 'graphql/data/pools/useTopPools'
import { chainIdToBackendName } from 'graphql/data/util'
import ms from 'ms'
import { useMemo } from 'react'

export interface PoolData {
  // basic pool info
  address: string
  feeTier?: number
  txCount?: number
  protocolVersion?: ProtocolVersion

  // token info
  token0: Token
  tvlToken0?: number
  token0Price?: number

  token1: Token
  tvlToken1?: number
  token1Price?: number

  // volume
  volumeUSD24H?: number
  volumeUSD24HChange?: number

  // liquidity
  tvlUSD?: number
  tvlUSDChange?: number
}

/**
 * Given an array of historical volume, calculate the 24h % change in volume
 */
function calc24HVolChange(historicalVolume?: { value: number; timestamp: number }[]) {
  if (!historicalVolume) {
    return undefined
  }
  const currentTime = new Date().getTime()
  const dayAgo = (currentTime - ms('1d')) / 1000
  const twoDaysAgo = (currentTime - ms('2d')) / 1000

  const volume24h = historicalVolume
    .filter((entry) => entry.timestamp >= dayAgo)
    .reduce((acc, cur) => acc + cur.value, 0)
  const volume48h = historicalVolume
    .filter((entry) => entry.timestamp >= twoDaysAgo && entry.timestamp < dayAgo)
    .reduce((acc, cur) => acc + cur.value, 0)
  return ((volume24h - volume48h) / volume48h) * 100
}

/**
 * Queries both v3 and v2 for pool data
 * @param poolAddress
 * @param chainId
 * @returns
 */
export function usePoolData(
  poolAddress: string,
  chainId?: ChainId
): {
  loading: boolean
  error: boolean
  data?: PoolData
} {
  const {
    loading: loadingV3,
    error: errorV3,
    data: dataV3,
  } = useV3PoolQuery({
    variables: { chain: chainIdToBackendName(chainId), address: poolAddress },
  })
  const {
    loading: loadingV2,
    error: errorV2,
    data: dataV2,
  } = useV2PairQuery({
    variables: { address: poolAddress },
    skip: chainId !== ChainId.MAINNET,
  })

  return useMemo(() => {
    const anyError = Boolean(errorV3 || (errorV2 && chainId === ChainId.MAINNET))
    const anyLoading = Boolean(loadingV3 || (loadingV2 && chainId === ChainId.MAINNET))

    // return early if not all data yet
    if (anyError || anyLoading) {
      return {
        loading: anyLoading,
        error: anyError,
        data: undefined,
      }
    }

    const pool = dataV3?.v3Pool ?? dataV2?.v2Pair ?? undefined
    const feeTier = dataV3?.v3Pool?.feeTier ?? V2_BIPS

    return {
      data: pool
        ? {
            address: pool.address,
            txCount: pool.txCount,
            protocolVersion: pool.protocolVersion,
            token0: pool.token0 as Token,
            tvlToken0: pool.token0Supply,
            token0Price: pool.token0?.project?.markets?.[0]?.price?.value,
            token1: pool.token1 as Token,
            tvlToken1: pool.token1Supply,
            token1Price: pool.token1?.project?.markets?.[0]?.price?.value,
            feeTier,
            volumeUSD24H: pool.volume24h?.value,
            volumeUSD24HChange: calc24HVolChange(pool.historicalVolume?.concat()),
            tvlUSD: pool.totalLiquidity?.value,
            tvlUSDChange: pool.totalLiquidityPercentChange24h?.value,
          }
        : undefined,
      error: anyError,
      loading: anyLoading,
    }
  }, [chainId, dataV2?.v2Pair, dataV3?.v3Pool, errorV2, errorV3, loadingV2, loadingV3])
}
