"use client"

import { useMemo, useState } from "react"
import toast from "react-hot-toast"
import { Button } from "../../ui/button"
import { Input } from "../../ui/input"
import { Textarea } from "../../ui/textarea"

type SubjectOption = 'Question - General' | 'Question - Billing' | 'Feature Recommendation' | 'Other'

export function SupportForm({
  session,
  productLabel = 'KYD Self-check',
}: {
  session?: { user?: { fullName?: string | null; name?: string | null; email?: string | null } | null } | null
  productLabel?: string
}) {
  const [subject, setSubject] = useState<SubjectOption>('Question - General')
  const [description, setDescription] = useState('')
  const [file, setFile] = useState<File | null>(null)
  const [isSubmitting, setIsSubmitting] = useState(false)
  const [isSent, setIsSent] = useState(false)

  const nameVal = useMemo(() => (session?.user?.fullName || session?.user?.name || ''), [session])
  const emailVal = useMemo(() => (session?.user?.email || ''), [session])
  const productVal = productLabel

  const onFileChange: React.ChangeEventHandler<HTMLInputElement> = (e) => {
    const f = e.target.files?.[0] || null
    if (!f) { setFile(null); return }
    const allowed = new Set(['image/png', 'image/jpeg', 'image/webp'])
    if (!allowed.has(f.type)) { toast.error('Only PNG, JPG, or WEBP allowed'); return }
    if (f.size > 5 * 1024 * 1024) { toast.error('Max file size is 5MB'); return }
    setFile(f)
  }

  async function handleSubmit(e: React.FormEvent) {
    e.preventDefault()
    if (!session) { toast.error('Please sign in to submit.'); return }
    if (!description.trim()) { toast.error('Please add a description.'); return }

    setIsSubmitting(true)
    try {
      let screenshotKey: string | undefined
      if (file) {
        const mime = file.type
        const body = JSON.stringify({ filename: file.name, mime })
        const res = await fetch('/api/support/presign', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body })
        if (!res.ok) throw new Error('Failed to prepare upload')
        const { uploadUrl, key } = await res.json()
        const up = await fetch(uploadUrl, { method: 'PUT', headers: { 'Content-Type': mime }, body: file })
        if (!up.ok) throw new Error('Upload failed')
        screenshotKey = key
      }

      const resp = await fetch('/api/support', {
        method: 'POST',
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ subject, description, screenshotKey })
      })
      if (!resp.ok) {
        const t = await resp.json().catch(() => ({}))
        throw new Error(t?.error || 'Failed to submit support request')
      }
      toast.success('Sent to support. Thank you!')
      setDescription('')
      setFile(null)
      setSubject('Question - General')
      setIsSent(true)
    } catch (err) {
      const e2 = err as Error
      toast.error(e2.message || 'Something went wrong')
    } finally {
      setIsSubmitting(false)
    }
  }

  if (isSent) {
    return (
      <div className="w-full">
        <div className="rounded-xl border p-6 md:p-10 text-center max-w-2xl mx-auto">
          <h2 className="text-xl md:text-2xl font-semibold mb-2">Thanks! Your message was sent.</h2>
          <p className="text-sm md:text-base mb-6">Our team will review and get back to you at your account email.</p>
          <div className="flex items-center justify-center gap-3">
            <Button onClick={() => setIsSent(false)} className="px-4">Send another</Button>
          </div>
        </div>
      </div>
    )
  }

  return (
    <form onSubmit={handleSubmit} className="w-full">
      <div className="grid md:grid-cols-2 gap-6">
        <div className="rounded-xl border p-4 sm:p-6">
          <h2 className="text-lg font-semibold mb-4">Contact Support</h2>

          <div className="space-y-4">
            <div>
              <label className="block text-sm mb-1">Name</label>
              <Input value={nameVal} readOnly />
            </div>
            <div>
              <label className="block text-sm mb-1">Email</label>
              <Input value={emailVal} readOnly />
            </div>
            <div>
              <label className="block text-sm mb-1">Product</label>
              <Input value={productVal} readOnly />
            </div>
            <div>
              <label className="block text-sm mb-1">Subject</label>
              <select
                value={subject}
                onChange={(e) => setSubject(e.target.value as SubjectOption)}
                className="w-full rounded-md border bg-transparent p-2"
              >
                <option>Question - General</option>
                <option>Question - Billing</option>
                <option>Feature Recommendation</option>
                <option>Other</option>
              </select>
            </div>
          </div>
        </div>

        <div className="rounded-xl border p-4 sm:p-6">
          <h2 className="text-lg font-semibold mb-4">Description</h2>
          <div className="space-y-4">
            <div>
              <Textarea
                value={description}
                onChange={(e) => setDescription(e.target.value)}
                placeholder="Tell us what's going on..."
                className="min-h-[160px]"
              />
            </div>
            <div>
              <label className="block text-sm mb-1">Optional screenshot (PNG/JPG/WEBP, ≤5MB)</label>
              <Input type="file" accept="image/png,image/jpeg,image/webp" onChange={onFileChange} />
              {file ? (
                <p className="text-xs mt-2">{file.name} ({Math.round(file.size/1024)} KB)</p>
              ) : null}
            </div>
          </div>

          <div className="mt-6">
            <Button type="submit" disabled={isSubmitting} className="w-full justify-center">
              {isSubmitting ? 'Sending…' : 'Send to Support'}
            </Button>
          </div>
        </div>
      </div>
    </form>
  )
}

export default SupportForm