@allbluecn/web-app 0.3.0

package/src/components/story/ai/split-dialog.tsx

@@ -0,0 +1,467 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { useEffect, useRef, useState } from "react"
+import { Loader2, WandSparkles, X, Check, SquareStop, CheckCheck, RotateCcw, Sparkles } from "lucide-react"
+import { useTranslation } from "react-i18next"
+import { Button } from "@/components/ui/button"
+import { Dialog, DialogContent, DialogHeader, DialogTitle, DialogFooter } from "@/components/ui/dialog"
+import { Badge } from "@/components/ui/badge"
+import { Textarea } from "@/components/ui/textarea"
+import { toast } from "sonner"
+import { Checkbox } from "@/components/ui/checkbox"
+import { createStoryFn, addChildFn } from "@/server/serverFns/story"
+import { useSession } from "@/lib/auth/auth-client"
+import { fetchStoryAiStream, type StoryAiMode, type StoryAiSplitMode } from "@/lib/ai/story-ai-stream"
+import { parseJsonlBuffer, parseFinalBuffer, type ParsedStory } from "@/lib/ai/story-jsonl"
+
+let localIdSeq = 0
+
+interface SplitItem {
+  localId: string
+  name: string
+  description?: string
+  priority?: string
+  checked: boolean
+}
+
+/**
+ * 单个需求的拆解快照——保存用户在弹窗里产生但尚未确认创建的拆解结果，
+ * 让用户关闭后重新打开（甚至切换到别的需求再切回来）仍能看到上次的拆解列表。
+ *
+ * 仅缓存 items / prompt / splitMode 三个用户可见且影响「确认创建」的字段；
+ * 流式中间状态（bufferRef / nameIdMapRef / streaming / creating）不缓存，
+ * 重新打开后用户需要主动点「重新拆解」才能继续生成。
+ */
+interface SplitSnapshot {
+  items: SplitItem[]
+  prompt: string
+  splitMode: StoryAiSplitMode
+}
+
+/** LRU 上限，避免查看过多需求后内存无限增长 */
+const SPLIT_SNAPSHOT_LIMIT = 20
+const splitSnapshots = new Map<string, SplitSnapshot>()
+
+/** 读取并刷新 LRU 顺序（最近访问的放到 Map 末尾） */
+function loadSplitSnapshot(storyId: string): SplitSnapshot | undefined {
+  const snap = splitSnapshots.get(storyId)
+  if (!snap) return undefined
+  splitSnapshots.delete(storyId)
+  splitSnapshots.set(storyId, snap)
+  return snap
+}
+
+/** 写入快照，超出上限时淘汰 Map 头部（最久未访问） */
+function saveSplitSnapshot(storyId: string, snap: SplitSnapshot) {
+  if (splitSnapshots.has(storyId)) splitSnapshots.delete(storyId)
+  splitSnapshots.set(storyId, snap)
+  while (splitSnapshots.size > SPLIT_SNAPSHOT_LIMIT) {
+    const oldest = splitSnapshots.keys().next().value
+    if (oldest === undefined) break
+    splitSnapshots.delete(oldest)
+  }
+}
+
+/** 清空指定需求的快照（确认创建成功后调用） */
+function clearSplitSnapshot(storyId: string) {
+  splitSnapshots.delete(storyId)
+}
+
+export interface StorySplitDialogProps {
+  storyId: string
+  storyName: string
+  storyDescription: string
+  open: boolean
+  onOpenChange: (open: boolean) => void
+  onComplete: () => void
+}
+
+export function StorySplitDialog({
+  storyId, storyName, storyDescription, open, onOpenChange, onComplete,
+}: StorySplitDialogProps) {
+  const { data: session } = useSession()
+  const { t } = useTranslation("story")
+  const hasAiKey = session?.user?.hasAiKey ?? false
+
+  const [prompt, setPrompt] = useState("")
+  const [splitMode, setSplitMode] = useState<StoryAiSplitMode>("children")
+  const [streaming, setStreaming] = useState(false)
+  const [creating, setCreating] = useState(false)
+  const [items, setItems] = useState<SplitItem[]>([])
+  const abortRef = useRef<AbortController | null>(null)
+  const bufferRef = useRef("")
+  // 按 name 稳定映射 localId，避免流式增量解析时 React key 抖动
+  const nameIdMapRef = useRef<Map<string, string>>(new Map())
+
+  const reset = () => {
+    abortRef.current?.abort()
+    setPrompt("")
+    setStreaming(false)
+    setCreating(false)
+    setItems([])
+    bufferRef.current = ""
+    nameIdMapRef.current.clear()
+  }
+
+  // 关闭弹窗时仅中断进行中的流，保留 items / prompt / splitMode 给下次打开，
+  // 让用户在未确认创建就关闭后，重新打开仍能看到上次的拆解结果。
+  const cancelStreamingOnClose = () => {
+    abortRef.current?.abort()
+    setStreaming(false)
+    setCreating(false)
+  }
+
+  // 流式增量解析：每次都重新解析完整 buffer（包含已积累的全部片段），
+  // 用 name 复用 localId 保持 React key 稳定。
+  const applyParsed = (parsed: ParsedStory[]) => {
+    const map = nameIdMapRef.current
+    setItems(
+      // 防御性过滤：zod 已校验 name 非空，此处再兜底过滤空格串，避免空标题项进入列表
+      parsed
+        .filter((p) => typeof p.name === "string" && p.name.trim().length > 0)
+        .map((p) => {
+          let id = map.get(p.name)
+          if (!id) {
+            id = `ai-${++localIdSeq}`
+            map.set(p.name, id)
+          }
+          return {
+            localId: id,
+            name: p.name,
+            description: p.description,
+            priority: p.priority ?? "none",
+            checked: true,
+          }
+        }),
+    )
+  }
+
+  const generate = () => {
+    const mode: StoryAiMode = "split"
+    abortRef.current?.abort()
+    setStreaming(true)
+    setItems([])
+    bufferRef.current = ""
+    nameIdMapRef.current.clear()
+
+    abortRef.current = fetchStoryAiStream(
+      {
+        mode,
+        prompt: prompt.trim()
+          ? `${storyDescription}\n\n${t("storyAi.split.extraPromptPrefix")}${prompt}`
+          : storyDescription,
+        context: {
+          story: { id: storyId, name: storyName, description: storyDescription },
+        },
+        options: {
+          splitMode,
+        },
+      },
+      {
+        onDelta: (delta) => {
+          bufferRef.current += delta
+          // 关键：不清空 buffer 到 rest，每次重新解析完整缓冲区，
+          // 避免 LLM 分片输出多行时 setItems 覆盖丢失历史行。
+          const { items: parsed } = parseJsonlBuffer(bufferRef.current)
+          applyParsed(parsed)
+        },
+        onDone: () => {
+          // 处理残尾（最后一行无换行的情况）
+          const { items: final } = parseFinalBuffer(bufferRef.current)
+          if (final.length > 0) {
+            // 合并去重（按 name）
+            setItems((prev) => {
+              const merged = [...prev]
+              const existing = new Set(merged.map((m) => m.name))
+              for (const item of final) {
+                // 防御性跳过：name 非字符串或 trim 后为空的项
+                if (typeof item.name !== "string" || item.name.trim().length === 0) continue
+                if (existing.has(item.name)) continue
+                let id = nameIdMapRef.current.get(item.name)
+                if (!id) {
+                  id = `ai-${++localIdSeq}`
+                  nameIdMapRef.current.set(item.name, id)
+                }
+                merged.push({
+                  localId: id,
+                  name: item.name,
+                  description: item.description,
+                  priority: item.priority ?? "none",
+                  checked: true,
+                })
+              }
+              return merged
+            })
+          }
+          setStreaming(false)
+          // 完全无产出
+          if (final.length === 0 && items.length === 0 && bufferRef.current.trim() === "") {
+            toast.error(t("storyAi.split.noValidResult"))
+          }
+        },
+        onError: (err) => {
+          setStreaming(false)
+          toast.error(t("storyAi.split.generationFailed"), { description: err.message })
+        },
+      },
+    )
+  }
+
+  const confirmCreate = async () => {
+    const checked = items.filter((i) => i.checked)
+    // 防御性过滤：跳过 name 非字符串或 trim 后为空的项（zod 已保证，此处兜底运行时异常）
+    const selected = checked.filter(
+      (i) => typeof i.name === "string" && i.name.trim().length > 0,
+    )
+    const invalid = checked.length - selected.length
+    if (invalid > 0) {
+      console.error(
+        "[SplitDialog] invalid checked items:",
+        items.filter((i) => i.checked && !(typeof i.name === "string" && i.name.trim().length > 0)),
+      )
+    }
+    if (selected.length === 0) {
+      toast.error(t("storyAi.split.selectAtLeastOne"))
+      return
+    }
+    setCreating(true)
+    try {
+      const results = await Promise.allSettled(
+        selected.map(async (item) => {
+           const created = await createStoryFn({
+             data: {
+               name: item.name.trim(),
+               description: item.description ?? undefined,
+               state: "backlog",
+               priority: (item.priority ?? "none") as "urgent" | "high" | "medium" | "low" | "none",
+               isAI: true,
+             },
+           })
+          if (splitMode === "children") {
+            await addChildFn({ data: { parentId: storyId, childId: created.id } })
+          }
+          return created
+        }),
+      )
+      const failed = results.filter((r) => r.status === "rejected").length
+      const done = selected.length - failed
+      toast.success(t("storyAi.split.createdSuccess", { done, total: selected.length }), {
+        description: failed > 0 ? t("storyAi.split.failuresDesc", { count: failed }) : undefined,
+      })
+      clearSplitSnapshot(storyId)
+      reset()
+      onComplete()
+    } catch {
+      toast.error(t("storyAi.split.createFailed"))
+    } finally {
+      setCreating(false)
+    }
+  }
+
+  const stop = () => {
+    abortRef.current?.abort()
+    setStreaming(false)
+  }
+
+  const isOpen = open && hasAiKey
+  const hasItems = items.length > 0
+  const allChecked = hasItems && items.every((i) => i.checked)
+
+  // 弹窗打开时（open 由 false → true）按当前 storyId 恢复上次的拆解快照。
+  // 不同 storyId 维护独立快照，切换需求互不干扰；同一需求关闭再开仍能看到上次结果。
+  // 仅在打开的瞬间恢复，避免每次 setState 都被恢复逻辑覆盖。
+  const prevOpenRef = useRef(false)
+  useEffect(() => {
+    const opened = isOpen && !prevOpenRef.current
+    prevOpenRef.current = isOpen
+    if (!opened) return
+    const snap = loadSplitSnapshot(storyId)
+    if (snap) {
+      setItems(snap.items)
+      setPrompt(snap.prompt)
+      setSplitMode(snap.splitMode)
+      bufferRef.current = ""
+      nameIdMapRef.current.clear()
+    }
+  }, [isOpen, storyId])
+
+  // 拆解结果或补充要求变化时实时写入快照（恢复时的 setItems 会触发本 effect，
+  // 写回的就是刚恢复的内容，等价覆盖，无副作用）。空状态不保存，避免把
+  // 「未拆解过」的默认状态误存为快照。
+  useEffect(() => {
+    if (!isOpen) return
+    if (items.length === 0 && !prompt.trim()) return
+    saveSplitSnapshot(storyId, { items, prompt, splitMode })
+  }, [items, prompt, splitMode, isOpen, storyId])
+
+  return (
+    <Dialog open={isOpen} onOpenChange={(o) => { if (!o) cancelStreamingOnClose(); onOpenChange(o) }}>
+      <DialogContent className="flex flex-col gap-3 p-4 h-160 max-h-[85vh] max-w-lg sm:max-w-260">
+        <DialogHeader className="shrink-0">
+          <DialogTitle>{t("storyAi.split.title")}</DialogTitle>
+        </DialogHeader>
+
+        <div className="flex-1 min-h-0 space-y-3 overflow-y-auto overscroll-contain pr-1">
+          <div className="grid grid-cols-2 gap-2">
+            <button
+              type="button"
+              className={`flex flex-col items-start gap-1.5 rounded-lg border-2 p-3 text-left transition-all border
+                ${splitMode === "children"
+                  ? "border-primary bg-primary/5"
+                  : "border-border bg-background hover:border-primary/50"
+                }`}
+              onClick={() => setSplitMode("children")}
+            >
+              <div className="flex items-center gap-2">
+                <span className={`inline-flex items-center justify-center rounded-full border-2 h-4 w-4 shrink-0 border
+                  ${splitMode === "children" ? "border-primary bg-primary" : "border-border"}`}>
+                  {splitMode === "children" && <Check className="size-2.5 text-primary-foreground" />}
+                </span>
+                <span className="text-sm font-medium">{t("storyAi.split.asChildTitle")}</span>
+                <Badge variant="outline" className="text-[10px]">{t("storyAi.split.asChildBadge")}</Badge>
+              </div>
+              <p className="text-xs text-muted-foreground ml-6 mt-1 border-l-2 border-border truncate">
+                {t("storyAi.split.parentLabel")}{storyName}
+              </p>
+            </button>
+
+            <button
+              type="button"
+              className={`flex flex-col items-start gap-1.5 rounded-lg border-2 p-3 text-left transition-all border
+                ${splitMode === "independent"
+                  ? "border-primary bg-primary/5"
+                  : "border-border bg-background hover:border-primary/50"
+                }`}
+              onClick={() => setSplitMode("independent")}
+            >
+              <div className="flex items-center gap-2">
+                <span className={`inline-flex items-center justify-center rounded-full border-2 h-4 w-4 shrink-0 border
+                  ${splitMode === "independent" ? "border-primary bg-primary" : "border-border"}`}>
+                  {splitMode === "independent" && <Check className="size-2.5 text-primary-foreground" />}
+                </span>
+                <span className="text-sm font-medium">{t("storyAi.split.asIndependentTitle")}</span>
+              </div>
+              <p className="text-xs text-muted-foreground ml-6">{t("storyAi.split.asIndependentDesc")}</p>
+            </button>
+          </div>
+
+          <Textarea
+            value={prompt}
+            onChange={(e) => setPrompt(e.target.value)}
+            placeholder={t("storyAi.split.extraPromptPlaceholder")}
+            rows={3}
+            className="h-35"
+          />
+
+          <div className="space-y-2">
+            <div className="flex items-center justify-between">
+              {hasItems && (
+                <span className="text-sm text-muted-foreground">
+                  {t("storyAi.split.totalPrefix")}{items.length}{t("storyAi.split.totalSuffix")}
+                </span>
+              )}
+              {hasItems && (
+                <div className="flex gap-1">
+                  <Button
+                    size="sm"
+                    variant="ghost"
+                    onClick={() => setItems((prev) => prev.map((i) => ({ ...i, checked: !allChecked })))}
+                  >
+                    <CheckCheck className="size-3.5" /> {allChecked ? t("storyAi.split.deselectAll") : t("storyAi.split.selectAll")}
+                  </Button>
+                </div>
+              )}
+            </div>
+            <div className="max-h-60 overflow-y-auto overscroll-contain space-y-1.5">
+              {hasItems ? (
+                items.map((item) => (
+                  <div key={item.localId} className="flex items-start gap-2 rounded border p-2">
+                    <Checkbox
+                      checked={item.checked}
+                      onCheckedChange={(v) =>
+                        setItems((prev) =>
+                          prev.map((i) => (i.localId === item.localId ? { ...i, checked: v as boolean } : i)),
+                        )
+                      }
+                      className="mt-1 data-checked:bg-black data-checked:text-white data-checked:border-black dark:data-checked:bg-white dark:data-checked:text-black dark:data-checked:border-white"
+                    />
+                    <div className="flex-1 min-w-0">
+                      <p className="text-sm font-medium truncate">{item.name}</p>
+                      {item.description && (
+                        <p className="text-xs text-muted-foreground line-clamp-2 mt-0.5">{item.description}</p>
+                      )}
+                    </div>
+                    <Button
+                      size="sm"
+                      variant="ghost"
+                      onClick={() => setItems((prev) => prev.filter((i) => i.localId !== item.localId))}
+                    >
+                      <X className="size-3.5" />
+                    </Button>
+                  </div>
+                ))
+              ) : (
+                <div className="flex flex-col items-center justify-center gap-1.5 py-8 text-center">
+                  {streaming ? (
+                    <>
+                      <Loader2 className="size-7 animate-spin text-muted-foreground" />
+                      <p className="text-xs text-muted-foreground">{t("storyAi.split.splittingMessage")}</p>
+                    </>
+                  ) : (
+                    <>
+                      <Sparkles className="size-7 text-muted-foreground" />
+                      <p className="text-xs text-muted-foreground">{t("storyAi.split.clickToSplit")}</p>
+                    </>
+                  )}
+                </div>
+              )}
+            </div>
+          </div>
+        </div>
+
+        <DialogFooter className="shrink-0">
+          {streaming ? (
+            <Button onClick={stop} variant="outline">
+              <SquareStop className="mr-1 size-3.5" /> {t("storyAi.split.stop")}
+            </Button>
+          ) : (
+            <Button
+              onClick={generate}
+              disabled={creating || !storyId}
+              size="lg"
+              variant={hasItems ? "outline" : "default"}
+            >
+              {hasItems ? (
+                <WandSparkles className="mr-1 size-3.5" />
+              ) : (
+                <WandSparkles className="mr-1 size-3.5" />
+              )}
+              {hasItems ? t("storyAi.split.redo") : t("storyAi.split.split")}
+            </Button>
+          )}
+          {hasItems && (
+            <Button onClick={confirmCreate} disabled={creating} size="lg">
+              {creating ? <Loader2 className="mr-1 size-3.5 animate-spin" /> : <Check className="mr-1 size-3.5" />}
+              {t("storyAi.split.confirmCreate")}
+            </Button>
+          )}
+        </DialogFooter>
+      </DialogContent>
+    </Dialog>
+  )
+}