import { useMemo } from 'react'
import useJupiterMints from 'hooks/useJupiterMints'
import mangoStore from '@store/mangoStore'

const useJupiterSwapData = () => {
  const inputBank = mangoStore((s) => s.swap.inputBank)
  const outputBank = mangoStore((s) => s.swap.outputBank)
  const { mangoTokens } = useJupiterMints()

  const [inputTokenInfo, outputTokenInfo] = useMemo(() => {
    if (inputBank && outputBank) {
      return [
        mangoTokens?.find(
          (item) => item?.address === inputBank.mint.toString() || '',
        ),
        mangoTokens?.find(
          (item) => item?.address === outputBank.mint.toString() || '',
        ),
      ]
    } else {
      return []
    }
  }, [inputBank, outputBank, mangoTokens])

  let inputCoingeckoId = inputTokenInfo?.extensions?.coingeckoId
  let outputCoingeckoId = outputTokenInfo?.extensions?.coingeckoId

  if (inputBank?.name.toLocaleLowerCase() === 'dai') {
    inputCoingeckoId = 'dai'
  }
  if (outputBank?.name.toLocaleLowerCase() === 'dai') {
    outputCoingeckoId = 'dai'
  }

  return {
    inputTokenInfo,
    inputCoingeckoId,
    outputTokenInfo,
    outputCoingeckoId,
  }
}

export default useJupiterSwapData