import React, { useEffect, useMemo, useState } from 'react'
import { extraUsage } from 'src/commands/extra-usage/index.js'
import { Box, Text } from '@anthropic/ink'
import { useClaudeAiLimits } from 'src/services/claudeAiLimitsHook.js'
import { shouldProcessMockLimits } from 'src/services/rateLimitMocking.js' // Used for /mock-limits command
import {
  getRateLimitTier,
  getSubscriptionType,
  isClaudeAISubscriber,
} from 'src/utils/auth.js'
import { hasClaudeAiBillingAccess } from 'src/utils/billing.js'
import { MessageResponse } from '../MessageResponse.js'

type UpsellParams = {
  shouldShowUpsell: boolean
  isMax20x: boolean
  isExtraUsageCommandEnabled: boolean
  shouldAutoOpenRateLimitOptionsMenu: boolean
  isTeamOrEnterprise: boolean
  hasBillingAccess: boolean
}

export function getUpsellMessage({
  shouldShowUpsell,
  isMax20x,
  isExtraUsageCommandEnabled,
  shouldAutoOpenRateLimitOptionsMenu,
  isTeamOrEnterprise,
  hasBillingAccess,
}: UpsellParams): string | null {
  if (!shouldShowUpsell) return null

  if (isMax20x) {
    if (isExtraUsageCommandEnabled) {
      return '/extra-usage to finish what you\u2019re working on.'
    }
    return '/login to switch to an API usage-billed account.'
  }

  if (shouldAutoOpenRateLimitOptionsMenu) {
    return 'Opening your options\u2026'
  }

  if (!isTeamOrEnterprise && !isExtraUsageCommandEnabled) {
    return '/upgrade to increase your usage limit.'
  }

  if (isTeamOrEnterprise) {
    if (!isExtraUsageCommandEnabled) return null

    if (hasBillingAccess) {
      return '/extra-usage to finish what you\u2019re working on.'
    }

    return '/extra-usage to request more usage from your admin.'
  }

  return '/upgrade or /extra-usage to finish what you\u2019re working on.'
}

type RateLimitMessageProps = {
  text: string
  onOpenRateLimitOptions?: () => void
}

export function RateLimitMessage({
  text,
  onOpenRateLimitOptions,
}: RateLimitMessageProps): React.ReactNode {
  const subscriptionType = getSubscriptionType()
  const rateLimitTier = getRateLimitTier()
  const isTeamOrEnterprise =
    subscriptionType === 'team' || subscriptionType === 'enterprise'
  const isMax20x = rateLimitTier === 'default_claude_max_20x'
  // Always show upsell when using /mock-limits command, otherwise show for subscribers
  const shouldShowUpsell = shouldProcessMockLimits() || isClaudeAISubscriber()

  const canSeeRateLimitOptionsUpsell = shouldShowUpsell && !isMax20x

  const [hasOpenedInteractiveMenu, setHasOpenedInteractiveMenu] =
    useState(false)

  // Check actual rate limit status - only auto-open if user is currently rate limited
  // AND we've verified this with the API (resetsAt is only set after API response).
  // This prevents false alerts when resuming sessions with old rate limit messages.
  const claudeAiLimits = useClaudeAiLimits()
  const isCurrentlyRateLimited =
    claudeAiLimits.status === 'rejected' &&
    claudeAiLimits.resetsAt !== undefined &&
    !claudeAiLimits.isUsingOverage

  const shouldAutoOpenRateLimitOptionsMenu =
    canSeeRateLimitOptionsUpsell &&
    !hasOpenedInteractiveMenu &&
    isCurrentlyRateLimited &&
    onOpenRateLimitOptions

  useEffect(() => {
    if (shouldAutoOpenRateLimitOptionsMenu) {
      setHasOpenedInteractiveMenu(true)
      onOpenRateLimitOptions()
    }
  }, [shouldAutoOpenRateLimitOptionsMenu, onOpenRateLimitOptions])

  const upsell = useMemo(() => {
    const message = getUpsellMessage({
      shouldShowUpsell,
      isMax20x,
      isExtraUsageCommandEnabled: extraUsage.isEnabled(),
      shouldAutoOpenRateLimitOptionsMenu: !!shouldAutoOpenRateLimitOptionsMenu,
      isTeamOrEnterprise,
      hasBillingAccess: hasClaudeAiBillingAccess(),
    })
    if (!message) return null
    return <Text dimColor>{message}</Text>
  }, [
    shouldShowUpsell,
    isMax20x,
    isTeamOrEnterprise,
    shouldAutoOpenRateLimitOptionsMenu,
  ])

  return (
    <MessageResponse>
      <Box flexDirection="column">
        <Text color="error">{text}</Text>
        {hasOpenedInteractiveMenu ? null : upsell}
      </Box>
    </MessageResponse>
  )
}