writer-work-flow/apps/web/components/generation/CharacterGenerator.tsx

"use client";

import { useCallback, useEffect, useMemo, useState } from "react";

import type { CharacterCardView } from "@/lib/api/types";
import {
  MAX_CHARACTER_COUNT,
  MIN_CHARACTER_COUNT,
  updateCard,
} from "@/lib/generation/cards";
import { useCharacterGen } from "@/lib/generation/useCharacterGen";
import { CharacterCardItem } from "./CharacterCardItem";
import { ConflictAdjudication } from "./ConflictAdjudication";

interface CharacterGeneratorProps {
  projectId: string;
  // 入库成功后回调（如刷新 Codex 本地列表）。
  onIngested?: (created: string[], cards: CharacterCardView[]) => void;
}

// 角色生成器（UX §6.6）：一句话需求 + 数量 + 定位 → 预览卡 → 选/改 → 入库（处理 409 裁决）。
export function CharacterGenerator({
  projectId,
  onIngested,
}: CharacterGeneratorProps) {
  const gen = useCharacterGen();
  const [brief, setBrief] = useState("");
  const [count, setCount] = useState(MIN_CHARACTER_COUNT);
  const [role, setRole] = useState("");
  // 编辑态卡片（预览返回后拷贝进本地，允许就地改）。
  const [drafts, setDrafts] = useState<CharacterCardView[]>([]);
  const [selected, setSelected] = useState<Set<number>>(new Set());

  // 预览返回 → 同步进本地编辑态 + 默认全选。
  const syncDrafts = useCallback((cards: CharacterCardView[]) => {
    setDrafts(cards);
    setSelected(new Set(cards.map((_, i) => i)));
  }, []);

  const onGenerate = useCallback(async () => {
    await gen.generate({ projectId, brief, count, role });
  }, [gen, projectId, brief, count, role]);

  // gen.cards 变化（新预览）→ 重新同步本地草稿。
  const previewKey = gen.cards.map((c) => c.name).join("|");
  useEffect(() => {
    if (gen.genStatus === "preview") syncDrafts(gen.cards);
    // eslint-disable-next-line react-hooks/exhaustive-deps
  }, [previewKey, gen.genStatus]);

  const toggle = (i: number): void => {
    setSelected((prev) => {
      const next = new Set(prev);
      if (next.has(i)) next.delete(i);
      else next.add(i);
      return next;
    });
  };

  const selectedCards = useMemo(
    () => drafts.filter((_, i) => selected.has(i)),
    [drafts, selected],
  );

  const doIngest = useCallback(
    async (acknowledge: boolean) => {
      const ok = await gen.ingest(projectId, selectedCards, acknowledge);
      if (ok) {
        onIngested?.(gen.created, selectedCards);
        setDrafts([]);
        setSelected(new Set());
      }
    },
    [gen, projectId, selectedCards, onIngested],
  );

  const generating = gen.genStatus === "generating";
  const ingesting = gen.ingestStatus === "ingesting";

  return (
    <section className="flex flex-col gap-4" aria-label="角色生成器">
      <div className="rounded border border-line bg-panel p-4">
        <h2 className="mb-3 font-serif text-base text-ink">角色生成器</h2>
        <label className="mb-2 block text-sm text-ink-soft">
          一句话需求
          <input
            value={brief}
            onChange={(e) => setBrief(e.target.value)}
            placeholder="如：一个亦正亦邪的剑修，背负灭门血仇"
            className="mt-1 w-full rounded border border-line bg-bg px-3 py-2 text-sm text-ink"
          />
        </label>
        <div className="flex flex-wrap items-end gap-3">
          <label className="text-sm text-ink-soft">
            数量
            <input
              type="number"
              min={MIN_CHARACTER_COUNT}
              max={MAX_CHARACTER_COUNT}
              value={count}
              onChange={(e) => setCount(Number(e.target.value))}
              className="mt-1 block w-20 rounded border border-line bg-bg px-2 py-1.5 text-sm text-ink"
            />
          </label>
          <label className="flex-1 text-sm text-ink-soft">
            定位（可选）
            <input
              value={role}
              onChange={(e) => setRole(e.target.value)}
              placeholder="主角 / CP / 对手 / 导师 / 工具人"
              className="mt-1 block w-full rounded border border-line bg-bg px-2 py-1.5 text-sm text-ink"
            />
          </label>
          <button
            type="button"
            onClick={() => void onGenerate()}
            disabled={generating || brief.trim().length === 0}
            className="rounded bg-cinnabar px-4 py-2 text-sm text-white disabled:opacity-50"
          >
            {generating ? "生成中…" : "生成"}
          </button>
        </div>
      </div>

      {gen.genStatus === "preview" && drafts.length > 0 ? (
        <div className="flex flex-col gap-3">
          <p className="text-xs text-ink-soft">
            已生成 {drafts.length} 张，选 {selectedCards.length} 张入库（可就地编辑名/弧光）。
          </p>
          <div className="grid grid-cols-1 gap-3 lg:grid-cols-2">
            {drafts.map((card, i) => (
              <CharacterCardItem
                key={i}
                card={card}
                selected={selected.has(i)}
                onToggle={() => toggle(i)}
                onEdit={(patch) =>
                  setDrafts((prev) =>
                    prev.map((c, j) => (j === i ? updateCard(c, patch) : c)),
                  )
                }
              />
            ))}
          </div>

          {gen.ingestStatus === "conflict" && gen.conflicts ? (
            <ConflictAdjudication
              conflicts={gen.conflicts}
              busy={ingesting}
              onAcknowledge={() => void doIngest(true)}
              onCancel={() => gen.reset()}
            />
          ) : (
            <button
              type="button"
              onClick={() => void doIngest(false)}
              disabled={ingesting || selectedCards.length === 0}
              className="self-start rounded bg-cinnabar px-4 py-2 text-sm text-white disabled:opacity-50"
            >
              {ingesting ? "入库中…" : `入库选中 ${selectedCards.length} 张`}
            </button>
          )}
        </div>
      ) : null}

      {gen.ingestStatus === "done" && gen.created.length > 0 ? (
        <p className="text-sm text-pass">已入库：{gen.created.join("、")}</p>
      ) : null}
    </section>
  );
}