"use client";

import { useMemo } from "react";
import { motion } from "framer-motion";
import { ArrowRight, Check, Copy, Sparkles, ChevronRight } from "lucide-react";
import { toast } from "sonner";
import { Button } from "@/components/ui/button";
import { SearchableSelect } from "@/components/ui/searchable-select";
import { Textarea } from "@/components/ui/textarea";
import { cn } from "@/lib/utils";
import {
  defaultAgentWorkflowUiConfig,
  flattenAgentWorkflowSteps,
  getEnabledAgentWorkflowStepTree,
} from "@/lib/agents/workflow-ui-config";
import { AgentCanvasTopNavDark } from "@/components/agents/AgentCanvasTopNavDark";
import { AgentWorkflowShowcaseBanner } from "@/components/agents/AgentWorkflowShowcaseBanner";
import {
  buildAgentWorkflowComposerContext,
  getAgentWorkflowProgress,
} from "@/lib/agents/workflow-composer";
import type {
  AgentWorkflowComposerSelection,
  AgentWorkflowMaterialSelections,
  AgentWorkflowShowcaseConfig,
  AgentWorkflowStepKey,
  AgentWorkflowUiConfig,
} from "@/types/agent";

export interface AgentWorkflowInputs {
  primarySku: string;
  secondarySku: string;
  brand: string;
  store: string;
}

interface AgentWorkflowPanelDarkV2Props {
  config?: AgentWorkflowUiConfig | null;
  inputs: AgentWorkflowInputs;
  onInputsChange: (next: AgentWorkflowInputs) => void;
  materials: AgentWorkflowMaterialSelections;
  onMaterialsChange: (next: AgentWorkflowMaterialSelections) => void;
  selections: Partial<Record<AgentWorkflowStepKey, string>>;
  onSelectionsChange: (next: Partial<Record<AgentWorkflowStepKey, string>>) => void;
  onReset?: () => void;
  onNextStep?: () => void;
  onSkipStep?: () => void;
  siteName?: string | null;
  siteLogo?: string | null;
  siteLogoDark?: string | null;
  showcaseConfig?: AgentWorkflowShowcaseConfig | null;
}

function buildSelectionSummary(params: {
  config: AgentWorkflowUiConfig;
  selections: Partial<Record<AgentWorkflowStepKey, string>>;
}) {
  const result: AgentWorkflowComposerSelection[] = [];
  for (const { step } of flattenAgentWorkflowSteps(params.config.steps)) {
    const selectedOptionKey = params.selections[step.key];
    if (!selectedOptionKey) continue;
    const option = step.options.find((item) => item.key === selectedOptionKey && item.enabled);
    if (!option) continue;
    result.push({
      stepKey: step.key,
      stepTitle: step.title,
      optionKey: option.key,
      optionName: option.buttonName,
      promptFragment: option.promptFragment || "",
    });
  }
  return result;
}

function StepOptionCard({
  option,
  step,
  selected,
  onClick,
  index,
}: {
  option: { key: string; buttonName: string; description?: string | null; previewImageUrl?: string | null };
  step: { title: string };
  selected: boolean;
  onClick: () => void;
  index: number;
}) {
  return (
    <motion.button
      type="button"
      onClick={onClick}
      initial={{ opacity: 0, y: 20 }}
      animate={{ opacity: 1, y: 0 }}
      transition={{ duration: 0.4, delay: index * 0.05 }}
      whileHover={{ y: -4, scale: 1.02 }}
      whileTap={{ scale: 0.98 }}
      className={cn(
        "group relative overflow-hidden rounded-2xl text-left",
        "aspect-[16/10] w-full",
        "transition-all duration-500",
        selected
          ? "ring-2 ring-white/30 shadow-[0_0_30px_rgba(255,255,255,0.1)]"
          : "ring-1 ring-white/12 hover:ring-white/24"
      )}
    >
      {/* 背景 */}
      <div className="absolute inset-0 bg-zinc-900">
        {option.previewImageUrl ? (
          <>
            <img
              src={option.previewImageUrl}
              alt={option.buttonName}
              className="h-full w-full object-cover transition-transform duration-700 group-hover:scale-110"
            />
            <div className="absolute inset-0 bg-gradient-to-t from-black/80 via-black/40 to-transparent" />
          </>
        ) : (
          <div className="flex h-full w-full items-center justify-center bg-gradient-to-br from-zinc-800 to-zinc-900">
            <Sparkles className="h-10 w-10 text-white/20" />
          </div>
        )}
      </div>

      {/* 选中标记 */}
      {selected && (
        <motion.div
          initial={{ scale: 0 }}
          animate={{ scale: 1 }}
          className="absolute right-3 top-3 flex h-6 w-6 items-center justify-center rounded-full bg-white text-black"
        >
          <Check className="h-4 w-4" />
        </motion.div>
      )}

      {/* 步骤标签 */}
      <div className="absolute left-3 top-3">
        <span className="rounded-full bg-black/60 px-2.5 py-1 text-[10px] font-medium text-white/80 backdrop-blur-md">
          {step.title}
        </span>
      </div>

      {/* 底部信息 */}
      <div className="absolute bottom-0 left-0 right-0 p-4">
        <h4 className="text-sm font-semibold text-white drop-shadow-lg">
          {option.buttonName}
        </h4>
        {option.description && (
          <p className="mt-1 line-clamp-1 text-xs text-white/60">
            {option.description}
          </p>
        )}
      </div>

      {/* 选中边框 */}
      {selected && (
        <div className="absolute inset-0 rounded-2xl ring-2 ring-inset ring-white/30" />
      )}
    </motion.button>
  );
}

export function AgentWorkflowPanelDarkV2({
  config,
  inputs,
  onInputsChange,
  materials,
  onMaterialsChange,
  selections,
  onSelectionsChange,
  onNextStep,
  onSkipStep,
  siteName,
  siteLogo,
  siteLogoDark,
  showcaseConfig,
}: AgentWorkflowPanelDarkV2Props) {
  const brandOptions = useMemo(
    () => config?.inputOptions?.brands || defaultAgentWorkflowUiConfig.inputOptions?.brands || [],
    [config?.inputOptions?.brands]
  );
  
  const storeOptions = useMemo(
    () => config?.inputOptions?.stores || defaultAgentWorkflowUiConfig.inputOptions?.stores || [],
    [config?.inputOptions?.stores]
  );

  const resolvedConfig = useMemo(() => {
    const source = config || defaultAgentWorkflowUiConfig;
    return {
      steps: getEnabledAgentWorkflowStepTree(source.steps || []),
    };
  }, [config]);
  const flattenedResolvedSteps = useMemo(
    () =>
      flattenAgentWorkflowSteps(resolvedConfig.steps).filter(
        ({ step }) => Array.isArray(step.options) && step.options.length > 0
      ),
    [resolvedConfig.steps]
  );

  const summary = useMemo(
    () => buildSelectionSummary({ config: resolvedConfig, selections }),
    [resolvedConfig, selections]
  );

  const composerContext = useMemo(
    () =>
      buildAgentWorkflowComposerContext({
        config: resolvedConfig,
        inputs,
        materials,
        selections,
      }),
    [inputs, materials, resolvedConfig, selections]
  );

  const progress = useMemo(
    () => getAgentWorkflowProgress({ config: resolvedConfig, selections }),
    [resolvedConfig, selections]
  );

  const handleCopyPrompt = async () => {
    if (!composerContext?.promptText) {
      toast.error("当前还没有可复制的 Prompt");
      return;
    }
    try {
      await navigator.clipboard.writeText(composerContext.promptText);
      toast.success("Prompt 已复制");
    } catch {
      toast.error("复制失败");
    }
  };

  return (
    <div className="flex h-dvh flex-col overflow-hidden bg-black">
      <AgentCanvasTopNavDark
        siteName={siteName}
        siteLogo={siteLogo}
        siteLogoDark={siteLogoDark}
        activeKey="agent"
      />

      <div className="flex-1 overflow-y-auto overscroll-contain pt-16">
        {/* Hero Section */}
        <section className="relative">
          <div className="mx-auto max-w-[1480px] px-6 py-8 lg:px-8 xl:px-10">
            <AgentWorkflowShowcaseBanner
              config={showcaseConfig}
              actionLabel="开始选择风格"
              secondaryActionLabel={onSkipStep ? "进入风格生图" : undefined}
              onActionClick={() => {
                const target = document.getElementById("agent-workflow-content");
                if (!target) return;
                target.scrollIntoView({ behavior: "smooth", block: "start" });
              }}
              onSecondaryActionClick={onSkipStep}
            />
          </div>
        </section>

        {/* Main Content */}
        <main className="pb-40">
          <div
            id="agent-workflow-content"
            className="mx-auto max-w-[1480px] px-6 lg:px-8 xl:px-10"
          >
            <div className="space-y-8">
              {/* 基础配置 */}
              <motion.section
                className="rounded-2xl border border-white/12 bg-white/[0.04] p-5 shadow-[inset_0_1px_0_rgba(255,255,255,0.03)]"
                initial={{ opacity: 0, y: 20 }}
                animate={{ opacity: 1, y: 0 }}
                transition={{ delay: 0.1 }}
              >
                <h2 className="mb-4 flex items-center gap-2 text-lg font-semibold text-white">
                  <span className="flex h-6 w-6 items-center justify-center rounded-md bg-emerald-500/20 text-xs font-bold text-emerald-400">
                    1
                  </span>
                  基础配置
                </h2>
                <div className="grid grid-cols-1 gap-4 md:grid-cols-2 max-w-2xl">
                  <SearchableSelect
                    value={inputs.brand}
                    options={brandOptions}
                    placeholder="选择品牌"
                    searchPlaceholder="输入品牌名筛选"
                    emptyText="暂无品牌可选"
                    theme="dark"
                    onValueChange={(value) => onInputsChange({ ...inputs, brand: value })}
                  />
                  <SearchableSelect
                    value={inputs.store}
                    options={storeOptions}
                    placeholder="选择店铺"
                    searchPlaceholder="输入店铺名筛选"
                    emptyText="暂无店铺可选"
                    theme="dark"
                    onValueChange={(value) => onInputsChange({ ...inputs, store: value })}
                  />
                </div>
              </motion.section>

              {/* 工作流步骤 */}
              {flattenedResolvedSteps.map(({ step, depth }, stepIndex) => (
                <motion.section
                  key={step.key}
                  className={cn(
                    "rounded-2xl border border-white/12 bg-white/[0.04] p-5 shadow-[inset_0_1px_0_rgba(255,255,255,0.03)]",
                    depth > 1 && "ml-3 border-l-4 border-l-emerald-500/30 sm:ml-6"
                  )}
                  initial={{ opacity: 0, y: 20 }}
                  animate={{ opacity: 1, y: 0 }}
                  transition={{ delay: 0.15 + stepIndex * 0.05 }}
                >
                  <h2 className="mb-4 flex items-center gap-2 text-lg font-semibold text-white">
                    <span className="flex h-6 w-6 items-center justify-center rounded-md bg-emerald-500/20 text-xs font-bold text-emerald-400">
                      {stepIndex + 2}
                    </span>
                    {step.title}
                    {depth > 1 ? (
                      <span className="rounded-full border border-white/12 px-2 py-0.5 text-[11px] font-medium text-white/45">
                        {depth === 2 ? "二级模块" : "三级模块"}
                      </span>
                    ) : null}
                  </h2>
                  {step.description && (
                    <p className="mb-4 text-sm text-white/40">{step.description}</p>
                  )}
                  <div className="grid grid-cols-2 gap-4 sm:grid-cols-3 lg:grid-cols-4 xl:grid-cols-5">
                    {step.options.map((option, optionIndex) => (
                      <StepOptionCard
                        key={option.key}
                        option={option}
                        step={step}
                        selected={selections[step.key] === option.key}
                        onClick={() =>
                          onSelectionsChange({
                            ...selections,
                            [step.key]: selections[step.key] === option.key ? undefined : option.key,
                          })
                        }
                        index={optionIndex}
                      />
                    ))}
                  </div>
                </motion.section>
              ))}

              {/* 当前选择摘要 */}
              {summary.length > 0 && (
                <motion.section
                  className="rounded-2xl border border-white/12 bg-white/[0.04] p-5 shadow-[inset_0_1px_0_rgba(255,255,255,0.03)]"
                  initial={{ opacity: 0, y: 20 }}
                  animate={{ opacity: 1, y: 0 }}
                >
                  <h3 className="mb-4 text-sm font-semibold text-white/80">当前选择</h3>
                  <div className="grid grid-cols-1 gap-3 md:grid-cols-2 lg:grid-cols-3">
                    {summary.map((item) => (
                      <div
                        key={`${item.stepKey}-${item.optionKey}`}
                        className="flex items-center justify-between rounded-xl border border-white/10 bg-black/20 px-4 py-3"
                      >
                        <span className="text-sm text-white/50">{item.stepTitle}</span>
                        <span className="text-sm font-medium text-white">{item.optionName}</span>
                      </div>
                    ))}
                  </div>
                </motion.section>
              )}

              {/* Prompt 预览 */}
              <motion.section
                className="rounded-2xl border border-white/10 bg-white/[0.035] p-5 shadow-[inset_0_1px_0_rgba(255,255,255,0.03)]"
                initial={{ opacity: 0, y: 20 }}
                animate={{ opacity: 1, y: 0 }}
              >
                <div className="mb-4 flex items-center justify-between">
                  <h3 className="text-sm font-semibold text-white/80">Prompt 预览</h3>
                  <Button
                    variant="ghost"
                    size="sm"
                    onClick={handleCopyPrompt}
                    className="h-8 gap-1.5 text-xs text-white/60 hover:bg-white/10 hover:text-white"
                  >
                    <Copy className="h-3.5 w-3.5" />
                    复制
                  </Button>
                </div>
                <Textarea
                  value={composerContext?.promptText || ""}
                  readOnly
                  rows={6}
                  placeholder="选择工作流后，这里会自动生成结构化 Prompt"
                  className="resize-none rounded-xl border-white/12 bg-black/20 text-sm text-white/80 placeholder:text-white/20 focus:border-white/15"
                />
              </motion.section>
            </div>
          </div>
        </main>

      {onNextStep ? (
        <motion.div
          initial={{ opacity: 0, y: 18 }}
          animate={{ opacity: 1, y: 0 }}
          transition={{ delay: 0.2 }}
          className="mx-auto mt-8 w-full max-w-[1480px] px-6 lg:px-8 xl:px-10"
        >
          <div className="flex w-full flex-col gap-3 rounded-[24px] border border-white/12 bg-black/88 px-3 py-3 shadow-[0_18px_54px_rgba(0,0,0,0.42)] backdrop-blur-2xl sm:px-4">
            <div className="flex flex-col gap-1.5 sm:flex-row sm:items-center sm:justify-between">
              <div className="min-w-0">
                <div className="text-[13px] font-medium text-white">工作流进度</div>
                <div className="text-[11px] text-white/45">
                  已完成 {progress.completed} / {progress.total} 个步骤
                  {progress.isComplete ? "，可以直接进入下一步" : "，也可以选择跳过直接进入下一步"}
                </div>
              </div>
              <div className="text-[13px] font-semibold text-white">
                {progress.total > 0 ? Math.round((progress.completed / progress.total) * 100) : 0}%
              </div>
            </div>

            <div className="h-1.5 overflow-hidden rounded-full bg-white/10">
              <motion.div
                className="h-full rounded-full bg-gradient-to-r from-emerald-500 to-teal-400"
                initial={{ width: 0 }}
                animate={{ width: `${progress.total > 0 ? (progress.completed / progress.total) * 100 : 0}%` }}
                transition={{ duration: 0.45 }}
              />
            </div>

            <div className="flex flex-col gap-3 sm:flex-row sm:justify-end">
              {onSkipStep ? (
                <Button
                  variant="ghost"
                  onClick={onSkipStep}
                  className="h-10 gap-2 rounded-full border border-white/10 bg-white/[0.03] px-5 text-white/75 hover:bg-white/10 hover:text-white"
                >
                  跳过并进入下一步
                  <ChevronRight className="h-4 w-4" />
                </Button>
              ) : null}
              <Button
                onClick={onNextStep}
                disabled={progress.total > 0 && !progress.isComplete}
                className={cn(
                  "h-10 gap-2 rounded-full px-6",
                  "bg-gradient-to-r from-emerald-500 to-teal-500",
                  "text-sm font-medium text-white",
                  "hover:from-emerald-400 hover:to-teal-400",
                  "shadow-[0_0_30px_rgba(16,185,129,0.3)]",
                  "disabled:opacity-30"
                )}
              >
                下一步：选择生图模式
                <ArrowRight className="h-4 w-4" />
              </Button>
            </div>
          </div>
        </motion.div>
      ) : null}

        {/* 页脚 */}
        <footer className="mt-8 border-t border-white/5 pb-8 pt-6">
          <div className="mx-auto max-w-[1480px] px-6 lg:px-8 xl:px-10">
            <div className="flex items-center justify-center gap-6 text-sm text-white/30">
              <a href="#" className="hover:text-white/60 transition-colors">使用协议</a>
              <a href="#" className="hover:text-white/60 transition-colors">隐私条款</a>
              <span>© 2024 AI创作平台</span>
            </div>
          </div>
        </footer>
      </div>
    </div>
  );
}