import { useState } from 'react'
import { Card } from '../surface'
import { Stack, Row, Grid } from '../layout'
import { Heading, Text, Eyebrow, Highlight } from '../text'
import { Badge, Dot } from '../media'
import { Button } from '../Button'
import { Segmented } from '../Segmented'
import { Icon } from '../Icon'

interface Plan {
  name: string
  monthly: number
  blurb: string
  features: string[]
  cta: string
  tone: 'purple' | 'mint' | 'blue'
  featured?: boolean
}

const PLANS: Plan[] = [
  { name: 'Free', monthly: 0, blurb: 'For weekend projects.', tone: 'blue', cta: 'Start free', features: ['1 project', 'Community support', 'Core components'] },
  { name: 'Pro', monthly: 12, blurb: 'For serious builders.', tone: 'purple', featured: true, cta: 'Go Pro', features: ['Unlimited projects', 'Priority support', 'Every component', 'Example templates', 'Early features'] },
  { name: 'Team', monthly: 40, blurb: 'For the whole studio.', tone: 'mint', cta: 'Start a team', features: ['Everything in Pro', 'Shared workspace', 'SSO & roles', 'Audit log'] },
]

function Feature({ text }: { text: string }) {
  return (
    <Row gap={2} wrap={false}>
      <span style={{ display: 'inline-flex', flex: 'none', width: 22, height: 22, borderRadius: 999, alignItems: 'center', justifyContent: 'center', background: 'var(--mint)', color: 'var(--ink)' }}>
        <Icon name="ok" size="sm" />
      </span>
      <Text size="sm">{text}</Text>
    </Row>
  )
}

/** An example pricing page: three plans, one featured, each with its own
 * feature checklist. The monthly/annual Segmented toggle recomputes every
 * displayed price (and the per-plan savings line) rather than just relabeling
 * them, and picking a plan's CTA marks it as the current one. */
export function PricingBlock() {
  const [cycle, setCycle] = useState('monthly')
  const [selected, setSelected] = useState<string | null>(null)
  const annual = cycle === 'annual'
  return (
    <div style={{ maxWidth: 1000, margin: '0 auto', padding: 24 }}>
      <Stack gap={6}>
        <Stack gap={3}>
          <Eyebrow>Pricing</Eyebrow>
          <Heading level={1}>Pick a <Highlight>cushion</Highlight></Heading>
          <Text muted>Simple plans. Cancel anytime.</Text>
          <Row gap={3}>
            <Segmented
              label="Billing cycle"
              value={cycle}
              onChange={setCycle}
              options={[{ value: 'monthly', label: 'Monthly' }, { value: 'annual', label: 'Annual' }]}
            />
            {annual && <Badge tone="mint">Save 20%</Badge>}
          </Row>
        </Stack>

        <Grid cols={3}>
          {PLANS.map((p) => {
            const price = annual ? Math.round(p.monthly * 12 * 0.8) : p.monthly
            const savings = p.monthly * 12 - price
            const isCurrent = selected === p.name
            return (
            <Card key={p.name} variant={p.featured ? 'default' : 'tight'}>
              <Stack gap={4}>
                <Row justify="between">
                  <Row gap={2} wrap={false}>
                    <Dot tone={p.tone} />
                    <Heading level={3}>{p.name}</Heading>
                  </Row>
                  {p.featured && <Badge tone="yellow">Popular</Badge>}
                </Row>
                <Row gap={1} wrap={false}>
                  <span style={{ fontSize: 40, fontWeight: 900, letterSpacing: -1, color: 'var(--ink)' }}>${price}</span>
                  <Text muted>{annual ? '/yr' : '/mo'}</Text>
                </Row>
                {annual && savings > 0 && (
                  <Text size="sm" muted>Save ${savings}/yr versus paying monthly</Text>
                )}
                <Text size="sm" muted>{p.blurb}</Text>
                <Stack gap={2}>
                  {p.features.map((f) => <Feature key={f} text={f} />)}
                </Stack>
                <Button
                  block
                  tone={p.tone}
                  variant={isCurrent ? 'quiet' : p.featured ? 'solid' : 'quiet'}
                  disabled={isCurrent}
                  onClick={() => setSelected(p.name)}
                >
                  {isCurrent ? 'Current plan' : p.cta}
                </Button>
              </Stack>
            </Card>
          )})}
        </Grid>
      </Stack>
    </div>
  )
}