'use client'

import { useState } from 'react'
import useSWR from 'swr'
import { api } from '@/lib/api-client'
import type { PromptTemplate, PromptVersion } from '@/lib/types'
import { EmptyState } from '@/components/ui/state'
import { TableSkeleton } from '@/components/ui/skeleton'

export default function PromptsPage() {
  const [page, setPage] = useState(1)
  const [selectedName, setSelectedName] = useState<string | null>(null)
  const [versions, setVersions] = useState<PromptVersion[]>([])
  const [showCreate, setShowCreate] = useState(false)
  const [showNewVersion, setShowNewVersion] = useState(false)
  const [filter, setFilter] = useState<{ source?: string; status?: string }>({})

  const { data, error, isLoading, mutate } = useSWR(
    ['prompts.list', page, filter.source, filter.status],
    () => api.prompts.list({ page, page_size: 50, ...filter }),
  )

  const templates = data?.items ?? []
  const total = data?.total ?? 0

  const fetchVersions = async (name: string) => {
    try {
      const res = await api.prompts.listVersions(name)
      setVersions(res)
      setSelectedName(name)
    } catch (err) {
      console.error('Failed to fetch versions:', err)
    }
  }

  const handleCreate = async (e: React.FormEvent<HTMLFormElement>) => {
    e.preventDefault()
    const fd = new FormData(e.currentTarget)
    try {
      await api.prompts.create({
        name: fd.get('name') as string,
        category: fd.get('category') as string,
        description: (fd.get('description') as string) || undefined,
        source: 'custom',
        system_prompt: fd.get('system_prompt') as string,
      })
      setShowCreate(false)
      mutate()
    } catch (err) {
      console.error('Failed to create prompt:', err)
    }
  }

  const handleNewVersion = async (e: React.FormEvent<HTMLFormElement>) => {
    e.preventDefault()
    if (!selectedName) return
    const fd = new FormData(e.currentTarget)
    try {
      await api.prompts.createVersion(selectedName, {
        system_prompt: fd.get('system_prompt') as string,
        changelog: (fd.get('changelog') as string) || undefined,
      })
      setShowNewVersion(false)
      fetchVersions(selectedName)
    } catch (err) {
      console.error('Failed to create version:', err)
    }
  }

  const handleRollback = async (name: string, version: number) => {
    if (!confirm(`确认回退到版本 ${version}？`)) return
    try {
      await api.prompts.rollback(name, version)
      fetchVersions(name)
      mutate()
    } catch (err) {
      console.error('Failed to rollback:', err)
    }
  }

  const handleArchive = async (name: string) => {
    if (!confirm(`确认归档 ${name}？`)) return
    try {
      await api.prompts.archive(name)
      mutate()
    } catch (err) {
      console.error('Failed to archive:', err)
    }
  }

  const statusBadge = (status: string) => {
    const colors: Record<string, string> = {
      active: 'bg-emerald-500/20 text-emerald-400',
      deprecated: 'bg-amber-500/20 text-amber-400',
      archived: 'bg-zinc-500/20 text-zinc-400',
    }
    return (
      <span className={`px-2 py-0.5 text-xs rounded-full ${colors[status] || colors.archived}`}>
        {status}
      </span>
    )
  }

  const sourceBadge = (source: string) => {
    const colors: Record<string, string> = {
      builtin: 'bg-blue-500/20 text-blue-400',
      custom: 'bg-purple-500/20 text-purple-400',
    }
    return (
      <span className={`px-2 py-0.5 text-xs rounded-full ${colors[source] || ''}`}>
        {source === 'builtin' ? '内置' : '自定义'}
      </span>
    )
  }

  return (
    <div className="space-y-6">
      {/* Header */}
      <div className="flex items-center justify-between">
        <div>
          <h1 className="text-2xl font-bold text-white">提示词管理</h1>
          <p className="text-sm text-zinc-400 mt-1">管理内置和自定义提示词模板，支持版本控制和 OTA 分发</p>
        </div>
        <button
          onClick={() => setShowCreate(true)}
          className="px-4 py-2 bg-blue-600 text-white rounded-lg hover:bg-blue-700 transition-colors text-sm"
        >
          + 新建模板
        </button>
      </div>

      {/* Filters */}
      <div className="flex gap-2">
        {(['all', 'builtin', 'custom'] as const).map(s => (
          <button
            key={s}
            onClick={() => setFilter(s === 'all' ? {} : { source: s })}
            className={`px-3 py-1 text-sm rounded-lg transition-colors ${
              (filter.source || 'all') === s
                ? 'bg-zinc-700 text-white'
                : 'bg-zinc-800 text-zinc-400 hover:text-white'
            }`}
          >
            {s === 'all' ? '全部' : s === 'builtin' ? '内置' : '自定义'}
          </button>
        ))}
      </div>

      {/* Template List */}
      <div className="bg-zinc-900 rounded-xl border border-zinc-800 overflow-hidden">
        <table className="w-full text-sm">
          <thead>
            <tr className="border-b border-zinc-800">
              <th className="text-left px-4 py-3 text-zinc-400 font-medium">名称</th>
              <th className="text-left px-4 py-3 text-zinc-400 font-medium">分类</th>
              <th className="text-left px-4 py-3 text-zinc-400 font-medium">来源</th>
              <th className="text-left px-4 py-3 text-zinc-400 font-medium">版本</th>
              <th className="text-left px-4 py-3 text-zinc-400 font-medium">状态</th>
              <th className="text-left px-4 py-3 text-zinc-400 font-medium">更新时间</th>
              <th className="text-right px-4 py-3 text-zinc-400 font-medium">操作</th>
            </tr>
          </thead>
          <tbody>
            {isLoading ? (
              <tr>
                <td colSpan={7}>
                  <TableSkeleton rows={5} cols={7} hasToolbar={false} />
                </td>
              </tr>
            ) : error ? (
              <tr><td colSpan={7} className="px-4 py-8 text-center text-red-400">加载失败</td></tr>
            ) : templates.length === 0 ? (
              <tr><td colSpan={7}><EmptyState message="暂无提示词模板" /></td></tr>
            ) : (
              templates.map(t => (
                <tr key={t.id} className="border-b border-zinc-800/50 hover:bg-zinc-800/30">
                  <td className="px-4 py-3">
                    <button
                      onClick={() => fetchVersions(t.name)}
                      className="text-blue-400 hover:text-blue-300 font-mono"
                    >
                      {t.name}
                    </button>
                  </td>
                  <td className="px-4 py-3 text-zinc-400">{t.category}</td>
                  <td className="px-4 py-3">{sourceBadge(t.source)}</td>
                  <td className="px-4 py-3 text-zinc-300">v{t.current_version}</td>
                  <td className="px-4 py-3">{statusBadge(t.status)}</td>
                  <td className="px-4 py-3 text-zinc-500 text-xs">
                    {new Date(t.updated_at).toLocaleString('zh-CN')}
                  </td>
                  <td className="px-4 py-3 text-right">
                    <button
                      onClick={() => fetchVersions(t.name)}
                      className="text-zinc-400 hover:text-white mr-2"
                    >
                      历史
                    </button>
                    {t.source === 'custom' && (
                      <button
                        onClick={() => handleArchive(t.name)}
                        className="text-red-400 hover:text-red-300"
                      >
                        归档
                      </button>
                    )}
                  </td>
                </tr>
              ))
            )}
          </tbody>
        </table>
        <div className="px-4 py-2 text-xs text-zinc-500 border-t border-zinc-800">
          共 {total} 个模板
        </div>
      </div>

      {/* Version History Panel */}
      {selectedName && (
        <div className="bg-zinc-900 rounded-xl border border-zinc-800 p-4">
          <div className="flex items-center justify-between mb-4">
            <h2 className="text-lg font-semibold text-white">
              {selectedName} — 版本历史
            </h2>
            <div className="flex gap-2">
              <button
                onClick={() => setShowNewVersion(true)}
                className="px-3 py-1.5 bg-blue-600 text-white rounded-lg hover:bg-blue-700 text-xs"
              >
                发布新版本
              </button>
              <button
                onClick={() => { setSelectedName(null); setVersions([]) }}
                className="px-3 py-1.5 bg-zinc-700 text-white rounded-lg hover:bg-zinc-600 text-xs"
              >
                关闭
              </button>
            </div>
          </div>
          <div className="space-y-3">
            {versions.map(v => (
              <div key={v.id} className="bg-zinc-800/50 rounded-lg p-3">
                <div className="flex items-center justify-between mb-2">
                  <span className="text-sm font-mono text-zinc-300">v{v.version}</span>
                  <div className="flex items-center gap-2">
                    <span className="text-xs text-zinc-500">
                      {new Date(v.created_at).toLocaleString('zh-CN')}
                    </span>
                    {v.changelog && (
                      <span className="text-xs text-zinc-400">— {v.changelog}</span>
                    )}
                    {v.min_app_version && (
                      <span className="text-xs text-amber-400">最低版本: {v.min_app_version}</span>
                    )}
                  </div>
                </div>
                <pre className="text-xs text-zinc-400 bg-zinc-900 rounded p-2 overflow-x-auto max-h-32">
                  {v.system_prompt.substring(0, 300)}{v.system_prompt.length > 300 ? '...' : ''}
                </pre>
                <div className="mt-2 flex gap-2">
                  <button
                    onClick={() => {
                      navigator.clipboard.writeText(v.system_prompt)
                    }}
                    className="text-xs text-zinc-500 hover:text-white"
                  >
                    复制
                  </button>
                  <button
                    onClick={() => handleRollback(selectedName, v.version)}
                    className="text-xs text-amber-500 hover:text-amber-400"
                  >
                    回退到此版本
                  </button>
                </div>
              </div>
            ))}
            {versions.length === 0 && (
              <EmptyState message="暂无版本历史" />
            )}
          </div>
        </div>
      )}

      {/* Create Modal */}
      {showCreate && (
        <div className="fixed inset-0 bg-black/50 flex items-center justify-center z-50">
          <form onSubmit={handleCreate} className="bg-zinc-900 rounded-xl border border-zinc-700 p-6 w-full max-w-lg space-y-4">
            <h2 className="text-lg font-semibold text-white">新建提示词模板</h2>
            <div>
              <label className="block text-sm text-zinc-400 mb-1">名称</label>
              <input name="name" required className="w-full px-3 py-2 bg-zinc-800 border border-zinc-700 rounded-lg text-white text-sm" placeholder="my_prompt" />
            </div>
            <div>
              <label className="block text-sm text-zinc-400 mb-1">分类</label>
              <select name="category" className="w-full px-3 py-2 bg-zinc-800 border border-zinc-700 rounded-lg text-white text-sm">
                <option value="custom_system">系统提示词</option>
                <option value="custom_extraction">提取提示词</option>
                <option value="custom_compaction">压缩提示词</option>
                <option value="custom_other">其他</option>
              </select>
            </div>
            <div>
              <label className="block text-sm text-zinc-400 mb-1">描述</label>
              <input name="description" className="w-full px-3 py-2 bg-zinc-800 border border-zinc-700 rounded-lg text-white text-sm" placeholder="可选" />
            </div>
            <div>
              <label className="block text-sm text-zinc-400 mb-1">系统提示词</label>
              <textarea name="system_prompt" required rows={6} className="w-full px-3 py-2 bg-zinc-800 border border-zinc-700 rounded-lg text-white text-sm font-mono" />
            </div>
            <div className="flex gap-2 justify-end">
              <button type="button" onClick={() => setShowCreate(false)} className="px-4 py-2 bg-zinc-700 text-white rounded-lg hover:bg-zinc-600 text-sm">取消</button>
              <button type="submit" className="px-4 py-2 bg-blue-600 text-white rounded-lg hover:bg-blue-700 text-sm">创建</button>
            </div>
          </form>
        </div>
      )}

      {/* New Version Modal */}
      {showNewVersion && selectedName && (
        <div className="fixed inset-0 bg-black/50 flex items-center justify-center z-50">
          <form onSubmit={handleNewVersion} className="bg-zinc-900 rounded-xl border border-zinc-700 p-6 w-full max-w-lg space-y-4">
            <h2 className="text-lg font-semibold text-white">发布 {selectedName} 新版本</h2>
            <div>
              <label className="block text-sm text-zinc-400 mb-1">系统提示词</label>
              <textarea name="system_prompt" required rows={6} className="w-full px-3 py-2 bg-zinc-800 border border-zinc-700 rounded-lg text-white text-sm font-mono" />
            </div>
            <div>
              <label className="block text-sm text-zinc-400 mb-1">变更说明</label>
              <input name="changelog" className="w-full px-3 py-2 bg-zinc-800 border border-zinc-700 rounded-lg text-white text-sm" placeholder="描述本次变更" />
            </div>
            <div className="flex gap-2 justify-end">
              <button type="button" onClick={() => setShowNewVersion(false)} className="px-4 py-2 bg-zinc-700 text-white rounded-lg hover:bg-zinc-600 text-sm">取消</button>
              <button type="submit" className="px-4 py-2 bg-blue-600 text-white rounded-lg hover:bg-blue-700 text-sm">发布</button>
            </div>
          </form>
        </div>
      )}
    </div>
  )
}