"use client"

import { useState, useCallback, useRef } from "react"
import { AIChat, type AIChatHandle } from "@/components/ai/ai-chat"
import { JobSelector } from "@/components/ai/job-selector"
import { Bot, ChevronRight } from "lucide-react"

const tips = [
  "Ask for cold email templates for a specific job",
  "Request objection handling tips",
  "Ask for outreach strategies per industry",
  "Generate a follow up sequence",
  "Get LinkedIn connection message templates",
]

export default function AIAssistantPage() {
  const [selectedJob, setSelectedJob] = useState<{ job_title: string; keywords: string[]; industry: string; description: string } | null>(null)
  const [recentPrompts, setRecentPrompts] = useState<string[]>([])
  const [searching, setSearching] = useState(false)
  const aiChatRef = useRef<AIChatHandle | null>(null)

  const handleJobSelect = useCallback((job: typeof selectedJob) => {
    setSelectedJob(job)
  }, [])

  const handleSearch = useCallback(async (job: NonNullable<typeof selectedJob>) => {
    setSearching(true)
    const keyword = job.keywords[0]
    aiChatRef.current?.addAssistantMessage(`🔍 Searching Facebook for **${job.job_title}** leads...`)

    const controller = new AbortController()
    const timeoutId = setTimeout(() => controller.abort(), 360000)
    const statusId = setTimeout(() => {
      aiChatRef.current?.addAssistantMessage("⏳ Still searching Facebook (this can take up to 5 minutes)...")
    }, 45000)

    try {
      const res = await fetch(`http://localhost:3008/scrape/facebook?force=true&query=${encodeURIComponent(keyword)}`, { method: "POST", signal: controller.signal })
      clearTimeout(timeoutId)
      clearTimeout(statusId)
      const data = await res.json()
      if (data.success && data.leads?.length > 0) {
        const leadsText = data.leads.map((lead: any, i: number) =>
          `**${i + 1}.** ${lead.author || "Unknown"}\n> ${(lead.content || "").slice(0, 300)}\n> 🔗 ${lead.url || "(no link available)"}`
        ).join("\n\n")
        aiChatRef.current?.addAssistantMessage(`✅ Found **${data.leads.length}** leads:\n\n${leadsText}`)
      } else {
        const reason = data.error || data.flag_reason || "No leads found this time"
        aiChatRef.current?.addAssistantMessage(`⚠️ ${reason}`)
      }
    } catch (err: any) {
      clearTimeout(timeoutId)
      clearTimeout(statusId)
      const msg = err.name === "AbortError"
        ? "❌ Scraper timed out after 5 minutes. Try again or check the scraper service."
        : `❌ ${err instanceof Error ? err.message : "Search failed"}`
      aiChatRef.current?.addAssistantMessage(msg)
    } finally {
      setSearching(false)
    }
  }, [])

  const handleTipClick = useCallback((tip: string) => {
    aiChatRef.current?.fillInput(tip)
  }, [])

  const handleRecentPromptClick = useCallback((prompt: string) => {
    aiChatRef.current?.fillInput(prompt)
  }, [])

  const handleMessageSent = useCallback((msg: string) => {
    setRecentPrompts((prev) => {
      const next = [msg, ...prev.filter((p) => p !== msg)].slice(0, 3)
      return next
    })
  }, [])

  return (
    <div className="flex h-[calc(100vh-3.5rem)] bg-background">
      <div className="flex-1 flex flex-col min-w-0 border border-foreground transition-colors overflow-hidden">
        <div className="bg-card/80 backdrop-blur-md border-b border-border px-6 py-4">
          <div className="flex items-center justify-between">
            <div className="flex items-center gap-4">
              <div className="w-10 h-10 rounded-xl bg-gradient-to-br from-primary to-primary flex items-center justify-center" style={{boxShadow: "0 0 40px hsl(var(--primary) / 0.3)"}}>
                <Bot className="h-5 w-5 text-white" />
              </div>
              <div>
                <h1 className="text-foreground font-bold text-lg">AI Sales Assistant</h1>
                <p className="text-muted-foreground text-xs mt-0.5">Powered by local AI</p>
              </div>
            </div>
            <div className="flex items-center gap-4">
              <div className="flex items-center gap-2">
                <span className="w-2 h-2 rounded-full bg-[#22c55e] animate-pulse" />
                <span className="text-[#22c55e] text-xs font-medium">Online</span>
              </div>
              <div className="w-px h-5 bg-border" />
              <div className="bg-muted/50 rounded-lg px-3 py-1.5">
                <span className="text-muted-foreground text-xs">Local AI Model</span>
              </div>
            </div>
          </div>
        </div>
        <AIChat ref={aiChatRef} onMessageSent={handleMessageSent} />
      </div>
      <div className="w-[300px] flex-none bg-sidebar border border-foreground transition-colors p-5 overflow-y-auto h-full flex flex-col">
        <div>
          <div className="flex items-center gap-2 mb-3">
            <span className="w-1.5 h-1.5 rounded-full bg-primary" />
            <span className="text-primary text-[10px] font-bold uppercase tracking-[0.15em]">Target Job</span>
          </div>
          <JobSelector onSelect={handleJobSelect} onSearch={handleSearch} searching={searching} />
          {selectedJob && (
            <div className="bg-card/50 border border-border rounded-xl p-3.5 mt-3 space-y-2">
              <h4 className="text-sm font-semibold text-foreground">{selectedJob.job_title}</h4>
              <span className="text-xs px-2 py-0.5 rounded-md bg-primary/10 text-primary font-medium inline-block">{selectedJob.industry}</span>
              <p className="text-xs text-muted-foreground leading-relaxed">{selectedJob.description}</p>
              <div className="flex flex-wrap gap-1.5">
                {selectedJob.keywords.map((kw, i) => (
                  <span key={i} className="text-xs px-2 py-0.5 rounded-md bg-muted/50 text-muted-foreground">{kw}</span>
                ))}
              </div>
            </div>
          )}
        </div>
        <div className="mt-6">
          <div className="flex items-center gap-2 mb-3">
            <span className="w-1.5 h-1.5 rounded-full bg-primary" />
            <span className="text-primary text-[10px] font-bold uppercase tracking-[0.15em]">Quick Tips</span>
          </div>
          {tips.map((tip, i) => (
            <div
              key={i}
              onClick={() => handleTipClick(tip)}
              className="flex items-start gap-2.5 bg-card/50 hover:bg-card border border-border hover:border-primary/20 rounded-xl p-3.5 mb-2 cursor-pointer transition-all duration-200 group"
            >
              <ChevronRight className="h-3.5 w-3.5 mt-0.5 text-muted-foreground group-hover:text-primary transition-colors duration-200 flex-shrink-0" />
              <span className="text-muted-foreground text-xs leading-5 group-hover:text-foreground transition-colors duration-200">{tip}</span>
            </div>
          ))}
        </div>
        <div className="mt-6">
          <div className="flex items-center gap-2 mb-3">
            <span className="w-1.5 h-1.5 rounded-full bg-primary" />
            <span className="text-primary text-[10px] font-bold uppercase tracking-[0.15em]">Recent Prompts</span>
          </div>
          {recentPrompts.length > 0 ? (
            recentPrompts.map((prompt, i) => (
              <div
                key={i}
                onClick={() => handleRecentPromptClick(prompt)}
                className="bg-card/50 rounded-xl p-3 mb-2 border border-border text-muted-foreground text-xs truncate hover:text-foreground cursor-pointer transition-colors duration-200"
              >
                {prompt}
              </div>
            ))
          ) : (
            <div className="text-muted-foreground text-xs text-center py-4">Your recent prompts will appear here</div>
          )}
        </div>
        <div className="mt-auto border-t border-border pt-4">
          <div className="flex gap-4">
            <div className="flex-1">
              <div className="text-muted-foreground text-[10px] uppercase tracking-wide">Messages today</div>
              <div className="text-foreground text-sm font-semibold mt-0.5">24</div>
            </div>
            <div className="flex-1">
              <div className="text-muted-foreground text-[10px] uppercase tracking-wide">Tokens used</div>
              <div className="text-foreground text-sm font-semibold mt-0.5">12.4k</div>
            </div>
          </div>
        </div>
      </div>
    </div>
  )
}