mnote/wolai-frontend/src/components/editor/blocks/MindmapBlock.tsx

"use client";

import "simple-mind-map/dist/simpleMindMap.esm.css";
import React, {
  useCallback,
  useEffect,
  useLayoutEffect,
  useMemo,
  useRef,
  useState,
} from "react";
import { createPortal } from "react-dom";
import { createReactBlockSpec } from "@blocknote/react";
import type {
  BlockConfig,
  BlockNoteEditor,
  DefaultInlineContentSchema,
  DefaultStyleSchema,
  PropSchema,
  SpecificBlock,
} from "@blocknote/core";
import { useDebouncedCallback } from "@/hooks/use-debounced-callback";
import type { CustomBlockSchema } from "../schema";
import { MindmapToolbar } from "./MindmapToolbar";
import { MindmapSidebar } from "./MindmapSidebar";
import { MindmapSidebarTrigger } from "./MindmapSidebarTrigger";
import { MindmapNavigator } from "./MindmapNavigator";
import { MindmapMiniMap } from "./MindmapMiniMap";
import { MindmapCount } from "./MindmapCount";
import type { SidebarPanel } from "./mindmapSidebarConfig";
import type { MindMapNode } from "./mindmapTypes";
import { Label } from "@/components/ui/label";
import { Input } from "@/components/ui/input";
import { ArrowUp, ArrowDown, ArrowLeft, ArrowRight } from "lucide-react";       
import iconConfig from "./mindmapIconConfig";
import { emitAssetsChanged } from "@/lib/events";

// 避免 SSR/Node 环境提前评估 simple-mind-map 内部依赖 document
const loadIconModules = async () => {
  const { nodeIconList } = await import("simple-mind-map/src/svg/icons.js");    
  const { mergerIconList } = await import("simple-mind-map/src/utils/index.js");
  return { nodeIconList, mergerIconList };
};

// 安全获取图片尺寸
const getImageSizeSafe = (url: string): Promise<{ width: number; height: number } | null> =>
  new Promise((resolve) => {
    if (!url) return resolve(null);
    const img = new Image();
    img.onload = () => resolve({ width: img.naturalWidth, height: img.naturalHeight });
    img.onerror = () => resolve(null);
    img.src = url;
  });

type MindMapInstance = {
  execCommand: (command: string, ...args: unknown[]) => void;
  destroy: () => void;
  setData: (data: unknown) => void;
  getData: (withConfig?: boolean) => unknown;
  on?: <TArgs extends unknown[]>(event: string, handler: (...args: TArgs) => void) => void;
  emit?: (event: string, ...args: unknown[]) => void;
  off?: <TArgs extends unknown[]>(event: string, handler: (...args: TArgs) => void) => void;
  setMode?: (mode: string) => void;
  command: { clearHistory: () => void };
  editNodeClassList?: string[];
  view: {
    fit: () => void;
    scale: number;
    enlarge: () => void;
    narrow: () => void;
    setScale: (scale: number, cx: number, cy: number) => void;
  };
  renderer: {
    activeNodeList: unknown[];
    lastActiveNodeList?: unknown[];
    root?: unknown;
    renderTree: { _node: unknown };
    setRootNodeCenter?: () => void;
    clearActiveNodeList?: () => void;
    addNodeToActiveList?: (node: unknown, isActive?: boolean) => void;
    emitNodeActiveEvent?: (node: unknown) => void;
    findNodeByUid?: (uid: string) => unknown;
    copy?: () => void;
    paste?: () => void;
    beingCopyData?: unknown;
  };
  painter?: { startPainter: () => void };
  doExport?: { export: (type: string, isDownload?: boolean, name?: string) => Promise<unknown> };
  width: number;
  height: number;
  miniMap?: unknown;
  getThemeConfig: (key: string) => unknown;
  setThemeConfig: (config: unknown) => void;
  setTheme: (theme: string) => void;
  setLayout: (layout: string) => void;
};

type MindMapData = {
  data: Record<string, unknown>;
  children?: unknown[];
};

export const defaultMindmapData = {
  data: { text: "中心主题" },
  children: [],
};

const STORAGE_PREFIX = "wolai-mindmap-autosave-";

function downloadJson(data: unknown, name: string) {
  const blob = new Blob([JSON.stringify(data, null, 2)], { type: "application/json" });
  const url = URL.createObjectURL(blob);
  const anchor = document.createElement("a");
  anchor.href = url;
  anchor.download = `${name}.json`;
  anchor.click();
  URL.revokeObjectURL(url);
}

const createSimplePrompt = (title: string, placeholder = "") => {
  const value = window.prompt(title, placeholder);
  if (!value || !value.trim()) return null;
  return value.trim();
};

// 修补 svg.js rbox 在节点未挂载时抛出的异常
const patchSvgRbox = async () => {
  // 仅在浏览器环境生效
  if (typeof window === "undefined") return;
  const svgModule = await import("@svgdotjs/svg.js");
  type SvgCtorPrototype = {
    rbox?: (ref?: unknown) => unknown;
    __wolaiRboxPatched?: boolean;
  };
  type SvgCtor = { prototype?: SvgCtorPrototype } | undefined;
  const svgModuleTyped = svgModule as unknown as { Element?: SvgCtor; G?: SvgCtor };
  const candidates = [svgModuleTyped.Element, svgModuleTyped.G];
  let warned = false;
  candidates.forEach((Ctor) => {
    if (!Ctor?.prototype) return;
    if (Ctor.prototype.__wolaiRboxPatched) return;
    const original = Ctor.prototype.rbox;
    if (typeof original !== "function") return;
    Ctor.prototype.rbox = function patchedRbox(ref?: unknown) {
      try {
        return original.call(this, ref);
      } catch (error) {
        // 退化到 DOM 的 getBoundingClientRect 计算，避免初次渲染时出现 (0,0)
        type SvgElementLike = { node?: Element | null; el?: Element | null };
        const maybeThis = this as unknown as SvgElementLike;
        const el = maybeThis?.node ?? maybeThis?.el ?? null;
        const rect =
          el && typeof el.getBoundingClientRect === "function"
            ? el.getBoundingClientRect()
            : null;
        const viewportWidth = typeof window !== "undefined" ? window.innerWidth : 0;
        const viewportHeight =
          typeof window !== "undefined" ? window.innerHeight : 0;
        const defaultWidth = Math.max(120, Math.floor(viewportWidth * 0.2));
        const defaultHeight = Math.max(60, Math.floor(viewportHeight * 0.1));
        const fallback = rect && rect.width && rect.height
          ? {
              x: rect.x,
              y: rect.y,
              width: rect.width,
              height: rect.height,
              x2: rect.x + rect.width,
              y2: rect.y + rect.height,
              cx: rect.x + rect.width / 2,
              cy: rect.y + rect.height / 2,
            }
          : {
              x: Math.max(0, (viewportWidth - defaultWidth) / 2),
              y: Math.max(0, (viewportHeight - defaultHeight) / 2),
              width: defaultWidth,
              height: defaultHeight,
              x2: Math.max(0, (viewportWidth + defaultWidth) / 2),
              y2: Math.max(0, (viewportHeight + defaultHeight) / 2),
              cx: Math.max(0, viewportWidth / 2),
              cy: Math.max(0, viewportHeight / 2),
            };
        if (!warned) {
          console.warn("rbox 失败，使用 DOM 边界框降级避免崩溃", error, fallback);
          warned = true;
        }
        return fallback;
      }
    };
    Ctor.prototype.__wolaiRboxPatched = true;
  });
};

const applyToActiveNodes = (
  mindmap: MindMapInstance | null,
  handler: (node: unknown) => void,
) => {
  if (!mindmap) {
    window.alert("思维导图尚未初始化");
    return;
  }
  const renderer = mindmap.renderer;
  const list = renderer?.activeNodeList;
  if (!list || list.length === 0) {
    // 优先尝试 lastActiveNodeList（部分操作会短暂清空 activeNodeList）
    const lastActive = renderer?.lastActiveNodeList?.[0];
    const root = renderer?.root ?? renderer?.renderTree?._node;
    const fallback = lastActive ?? root;
    if (fallback) {
      renderer?.addNodeToActiveList?.(fallback);
      mindmap.execCommand?.("SET_NODE_ACTIVE", fallback, true);
    } else {
      window.alert("请选择至少一个节点再执行该操作");
      return;
    }
  }
  (renderer?.activeNodeList ?? []).forEach(handler);
};

// 确保存在激活节点后再执行命令，若无则自动选中根节点
const ensureActiveBefore = (mindmap?: MindMapInstance | null) => {
  const mm = mindmap ?? null;
  if (!mm) {
    window.alert("思维导图尚未初始化");
    return null;
  }
  const renderer = mm.renderer;
  const list = renderer?.activeNodeList ?? [];
  if (!list || list.length === 0) {
    const lastActive = renderer?.lastActiveNodeList?.[0];
    const root = renderer?.root ?? renderer?.renderTree?._node;
    const fallback = lastActive ?? root;
    if (fallback) {
      // 主动维护激活列表，避免首次插入子节点时因无激活节点导致插入位置错误
      // 先确保根节点居中，避免后续插入节点跑到 (0,0)
      if (renderer?.setRootNodeCenter) renderer.setRootNodeCenter();
      mm.view?.fit?.();
      if (renderer?.clearActiveNodeList) renderer.clearActiveNodeList();
      if (renderer?.addNodeToActiveList) renderer.addNodeToActiveList(fallback, true);
      if (renderer?.emitNodeActiveEvent) renderer.emitNodeActiveEvent(fallback);
      mm.execCommand?.("SET_NODE_ACTIVE", fallback, true);
    } else {
      window.alert("请先选中一个节点");
      return null;
    }
  }
  return mm;
};

const MindmapBlockView = ({
  block,
  editor,
  fullscreen = false,
}: {
  block: SpecificBlock<
    CustomBlockSchema,
    "mindmap",
    DefaultInlineContentSchema,
    DefaultStyleSchema
  >;
  editor: BlockNoteEditor<CustomBlockSchema>;
  fullscreen?: boolean;
}) => {
  const containerRef = useRef<HTMLDivElement | null>(null);
  const fileInputRef = useRef<HTMLInputElement | null>(null);
  const [mindmap, setMindmap] = useState<MindMapInstance | null>(null);   
  const mindmapReadyRef = useRef(false);
  const mindmapRef = useRef<MindMapInstance | null>(null);
  const hasLocalEditsRef = useRef(false);
  const applyingRemoteRef = useRef(false);
  const [canBack, setCanBack] = useState(false);
  const [canForward, setCanForward] = useState(false);
  const [activeNodes, setActiveNodes] = useState<MindMapNode[]>([]);
  const [painterMode, setPainterMode] = useState(false);
  const [showMiniMap, setShowMiniMap] = useState(false);
  const [activeSidebar, setActiveSidebar] = useState<SidebarPanel | null>(null);
  const [showNoteModal, setShowNoteModal] = useState(false);
  const [noteContent, setNoteContent] = useState("");
  const [localFullscreen, setLocalFullscreen] = useState(false);
  const effectiveFullscreen = fullscreen || localFullscreen;
  const wrapperRef = useRef<HTMLDivElement | null>(null);
  const hotkeyScopeRef = useRef(false);
  const recentNodeDblclickRef = useRef(false);
  const [fullscreenApiActive, setFullscreenApiActive] = useState(false);

  const getRootNode = useCallback((mm?: MindMapInstance | null) => {
    const instance = mm ?? mindmap;
    return instance?.renderer?.root ?? instance?.renderer?.renderTree?._node ?? null;
  }, [mindmap]);

  const docId = useMemo(
    () =>
      block.props.docId ||
      (typeof window !== "undefined"
        ? window.location.pathname.split("/").pop() ?? ""
        : ""),
    [block.props.docId],
  );

  useEffect(() => {
    hasLocalEditsRef.current = false;
    applyingRemoteRef.current = false;
  }, [docId]);

  const autosaveKey = useMemo(
    () => `${STORAGE_PREFIX}${docId || block.id}`,
    [block.id, docId],
  );
  const initialDataRef = useRef<unknown>(null);
  if (initialDataRef.current === null) {
    const cached = typeof window !== "undefined" ? window.localStorage.getItem(autosaveKey) : null;
    if (cached) {
      try {
        initialDataRef.current = JSON.parse(cached);
      } catch {
        initialDataRef.current = block.props.data ?? defaultMindmapData;        
      }
    } else {
      initialDataRef.current = block.props.data ?? defaultMindmapData;
    }
  }

  // 优先加载本地文件，其次 Supabase（通过后端 API）
  useEffect(() => {
    let cancelled = false;
    if (!docId) return;
    (async () => {
      try {
        const resp = await fetch(`/api/mindmap/${docId}`);
        if (!resp.ok) return;
        const payload = await resp.json().catch(() => null);
        const data = payload?.data;
        if (!data || cancelled) return;
        // 若用户已在本地做过编辑，则不再用远端数据覆盖，避免“插入节点后又被重置”
        if (hasLocalEditsRef.current) return;
        initialDataRef.current = data;
        if (mindmap) {
          applyingRemoteRef.current = true;
          try {
            mindmap.setData(data);
            mindmap.command.clearHistory();
          } finally {
            window.setTimeout(() => {
              applyingRemoteRef.current = false;
            }, 0);
          }
        }
      } catch (error) {
        console.warn("加载本地/远端思维导图失败", error);
      }
    })();
    return () => {
      cancelled = true;
    };
  }, [docId, mindmap]);

  // 记录“最近一次指针交互是否发生在思维导图块内”，用于键盘快捷键作用域
  useEffect(() => {
    const onPointerDownCapture = (e: Event) => {
      const wrapper = wrapperRef.current;
      if (!wrapper) return;
      const target = e.target as Node | null;
      hotkeyScopeRef.current = !!(target && wrapper.contains(target));
    };
    document.addEventListener("pointerdown", onPointerDownCapture, true);
    document.addEventListener("mousedown", onPointerDownCapture, true);
    return () => {
      document.removeEventListener("pointerdown", onPointerDownCapture, true);
      document.removeEventListener("mousedown", onPointerDownCapture, true);
    };
  }, []);

  // 标记“节点双击”事件，避免和“画布双击进入全屏”产生冲突
  useEffect(() => {
    if (!mindmap) return;
    const mark = () => {
      recentNodeDblclickRef.current = true;
      window.setTimeout(() => {
        recentNodeDblclickRef.current = false;
      }, 0);
    };
    mindmap.on?.("node_dblclick", mark);
    return () => {
      mindmap.off?.("node_dblclick", mark);
    };
  }, [mindmap]);

  const exitLocalFullscreen = useCallback(() => {
    setActiveSidebar(null);
    if (fullscreen) return;
    if (typeof document === "undefined") {
      setLocalFullscreen(false);
      return;
    }
    if (document.fullscreenElement) {
      document
        .exitFullscreen()
        .catch(() => {
          // ignore
        })
        .finally(() => setLocalFullscreen(false));
      return;
    }
    setLocalFullscreen(false);
  }, [fullscreen]);

  const enterLocalFullscreen = useCallback(() => {
    if (fullscreen) return;
    setLocalFullscreen(true);
    setActiveSidebar(null);
    if (typeof document === "undefined") return;
    if (!document.fullscreenEnabled) return;
    if (document.fullscreenElement) return;
    // 必须在“用户手势回调”中同步调用，避免 Fullscreen API 被浏览器拒绝
    try {
      const target = document.documentElement as unknown as {
        requestFullscreen?: () => Promise<void>;
      };
      const p = target.requestFullscreen?.();
      if (p && typeof p.catch === "function") {
        p.catch(() => {
          // ignore：失败则保持 Portal 伪全屏
        });
      }
    } catch {
      // ignore：失败则保持 Portal 伪全屏
    }
  }, [fullscreen]);

  // 沉浸式全屏：锁定页面滚动、支持 ESC 退出（仅对内嵌全屏生效）
  useEffect(() => {
    if (!localFullscreen) return;
    if (typeof document === "undefined") return;

    const prevOverflow = document.body.style.overflow;
    document.body.style.overflow = "hidden";
    // 进入全屏后尽量把焦点放到思维导图容器上，保证快捷键立即生效
    queueMicrotask(() => {
      try {
        wrapperRef.current?.focus?.({ preventScroll: true });
      } catch {
        // ignore
      }
    });

    const onKeyDown = (e: KeyboardEvent) => {
      if (e.key !== "Escape") return;
      if (fullscreen) return; // 外部传入的 fullscreen 不在这里处理
      e.preventDefault();
      e.stopPropagation();
      exitLocalFullscreen();
    };

    window.addEventListener("keydown", onKeyDown, true);
    return () => {
      window.removeEventListener("keydown", onKeyDown, true);
      document.body.style.overflow = prevOverflow;
    };
  }, [localFullscreen, fullscreen, exitLocalFullscreen]);

  // “真全屏”：使用 Fullscreen API 隐藏浏览器/桌面窗口的外层 UI（Electron/Web 都可用）
  useEffect(() => {
    if (typeof document === "undefined") return;

    const onFsChange = () => {
      const active = Boolean(document.fullscreenElement);
      setFullscreenApiActive(active);
      // 用户按 ESC 退出浏览器全屏时，同步退出沉浸式全屏界面
      if (!active && localFullscreen) {
        setLocalFullscreen(false);
      }
    };

    document.addEventListener("fullscreenchange", onFsChange);
    return () => {
      document.removeEventListener("fullscreenchange", onFsChange);
    };
  }, [localFullscreen]);

  // 让 Enter/Tab/复制/粘贴在“思维导图块”内生效（避免 BlockNote 抢走快捷键）
  useLayoutEffect(() => {
    const onKeyDownCapture = (e: KeyboardEvent) => {
      if (!hotkeyScopeRef.current) return;
      const mm = mindmapRef.current;
      if (!mm || !mindmapReadyRef.current) return;

      const target = e.target as HTMLElement | null;
      const editClasses = mm.editNodeClassList;
      const wrapper = wrapperRef.current;
      const isInWrapper = Boolean(wrapper && target && wrapper.contains(target));
      if (target && isInWrapper) {
        const tag = target.tagName;
        if (tag === "INPUT" || tag === "TEXTAREA") return;
        // contenteditable 输入中不拦截：避免破坏节点文本编辑/粘贴
        if (target.isContentEditable) return;
        const editableAncestor = target.closest?.('[contenteditable="true"]');
        if (editableAncestor) return;
        if (editClasses && target.classList) {
          for (const cls of editClasses) {
            if (cls && target.classList.contains(cls)) return;
          }
        }
      }

      const isMod = e.ctrlKey || e.metaKey;
      const key = e.key;
      const lower = key.toLowerCase();
      const stop = () => {
        e.preventDefault();
        e.stopPropagation();
        e.stopImmediatePropagation();
      };

      const pickTargetNode = (inst: MindMapInstance) => {
        const renderer = inst.renderer;
        const active = renderer.activeNodeList ?? [];
        const last = renderer.lastActiveNodeList ?? [];
        const root = renderer.root ?? renderer.renderTree?._node ?? null;
        return active[0] ?? last[0] ?? root ?? null;
      };

      if (key === "Enter" && !e.shiftKey && !e.altKey && !isMod) {
        stop();
        const inst = ensureActiveBefore(mm);
        if (!inst) return;
        const node = pickTargetNode(inst);
        if (!node) return;
        inst.execCommand?.("SET_NODE_ACTIVE", node, true);
        inst.execCommand?.("INSERT_NODE", false, [node]);
        return;
      }

      if (key === "Tab" && !e.shiftKey && !e.altKey && !isMod) {
        stop();
        const inst = ensureActiveBefore(mm);
        if (!inst) return;
        const node = pickTargetNode(inst);
        if (!node) return;
        inst.execCommand?.("SET_NODE_ACTIVE", node, true);
        inst.execCommand?.("INSERT_CHILD_NODE", false, [node]);
        return;
      }

      if (isMod && lower === "c") {
        stop();
        const inst = ensureActiveBefore(mm);
        if (!inst) return;
        inst.renderer.copy?.();
        return;
      }

      if (isMod && lower === "v") {
        stop();
        const inst = ensureActiveBefore(mm);
        if (!inst) return;
        const renderer = inst.renderer;
        const copyData = renderer.beingCopyData ?? null;
        if (copyData) {
          inst.execCommand?.("PASTE_NODE", copyData);
          return;
        }
        renderer.paste?.();
      }
    };

    window.addEventListener("keydown", onKeyDownCapture, true);
    return () => {
      window.removeEventListener("keydown", onKeyDownCapture, true);      
    };
  }, []);

  const persistData = useCallback(
    (data: unknown) => {
      if (!editor) return;
      window.localStorage.setItem(autosaveKey, JSON.stringify(data));     
      editor.updateBlock(block, { props: { ...block.props, data } });     
      if (docId) {
        // 同步到本地文件 + Supabase（弱依赖）
        fetch(`/api/mindmap/${docId}`, {
          method: "POST",
          headers: { "Content-Type": "application/json" },
          body: JSON.stringify({ data }),
        })
          .then((resp) => {
            if (resp.ok) {
              emitAssetsChanged(docId, {
                id: `mindmap-${docId}`,
                document_id: docId,
                asset_type: "mindmap",
                file_name: "mindmap.json",
                file_url: `/documents/${docId}`,
              });
            }
          })
          .catch((err) => console.warn("思维导图同步失败", err));
      }
    },
    [autosaveKey, block, docId, editor],
  );

  const debouncedPersist = useDebouncedCallback((data: unknown) => {
    persistData(data);
  }, 800);

  const initialSyncDone = useRef(false);
  useEffect(() => {
    if (!docId || !mindmap || initialSyncDone.current) return;
    initialSyncDone.current = true;
    const data = mindmap.getData?.() ?? initialDataRef.current ?? defaultMindmapData;
    (async () => {
      try {
        const resp = await fetch(`/api/mindmap/${docId}`, {
          method: "POST",
          headers: { "Content-Type": "application/json" },
          body: JSON.stringify({ data }),
        });
        if (!resp.ok) {
          console.warn(
            "初次创建思维导图文件失败",
            resp.status,
            await resp.text().catch(() => ""),
          );
        }
      } catch (err) {
        console.warn("初次创建思维导图文件失败", err);
      } finally {
        // 即便远端失败，也通知侧边栏刷新，保证本地文件树及时更新
        emitAssetsChanged(docId, {
          id: `mindmap-${docId}`,
          document_id: docId,
          asset_type: "mindmap",
          file_name: "mindmap.json",
          file_url: `/documents/${docId}`,
        });
      }
    })();
  }, [docId, mindmap, initialDataRef]);

  // 初始选中根节点，后续不强制抢焦点，允许用户自由选择
  useEffect(() => {
    if (!mindmap || activeNodes.length > 0) return;
    const rootNode = mindmap.renderer?.root ?? mindmap.renderer?.renderTree?._node;
    if (rootNode) {
      setActiveNodes([rootNode as MindMapNode]);
      mindmap.renderer.activeNodeList = [rootNode];
      mindmap.renderer.lastActiveNodeList = [rootNode];
      mindmap.emit?.("node_active", rootNode, [rootNode]);
    }
  }, [mindmap, activeNodes.length]);

  // mindmap 实例一旦就绪，立刻向侧边栏广播，确保文件树即时显示 mindmap.json
  useEffect(() => {
    if (!docId || !mindmap) return;
    emitAssetsChanged(docId, {
      id: `mindmap-${docId}`,
      document_id: docId,
      asset_type: "mindmap",
      file_name: "mindmap.json",
      file_url: `/documents/${docId}`,
    });
  }, [docId, mindmap]);

  useEffect(() => {
    let destroyed = false;
    let createdInstance: MindMapInstance | null = null;
    (async () => {
      if (!containerRef.current) return;
      const hostContainer = containerRef.current;
      const [
        { default: MindMap },
        { default: Painter },
        { default: AssociativeLine },
        { default: OuterFrame },
        { default: Exporter },
        { default: Formula },
        { default: RichText },
        { default: MiniMapPlugin },
        { default: Select },
        { default: Drag },
        { default: KeyboardNavigation },
        { default: NodeImgAdjust },
        { default: Scrollbar },
        { default: RainbowLines },
        { default: Watermark },
        { default: TouchEvent },
        { default: Cooperate },
        { default: Demonstrate },
        { default: MindMapLayoutPro },
        { default: NodeBase64ImageStorage },
        { default: ExportPDF },
        { default: ExportXMind },
        MindMapNodeModule,
      ] = await Promise.all([
        import("simple-mind-map"),
        import("simple-mind-map/src/plugins/Painter.js"),
        import("simple-mind-map/src/plugins/AssociativeLine.js"),
        import("simple-mind-map/src/plugins/OuterFrame.js"),
        import("simple-mind-map/src/plugins/Export.js"),
        import("simple-mind-map/src/plugins/Formula.js"),
        import("simple-mind-map/src/plugins/RichText.js"),
        import("simple-mind-map/src/plugins/MiniMap.js"),
        import("simple-mind-map/src/plugins/Select.js"),
        import("simple-mind-map/src/plugins/Drag.js"),
        import("simple-mind-map/src/plugins/KeyboardNavigation.js"),
        import("simple-mind-map/src/plugins/NodeImgAdjust.js"),
        import("simple-mind-map/src/plugins/Scrollbar.js"),
        import("simple-mind-map/src/plugins/RainbowLines.js"),
        import("simple-mind-map/src/plugins/Watermark.js"),
        import("simple-mind-map/src/plugins/TouchEvent.js"),
        import("simple-mind-map/src/plugins/Cooperate.js"),
        import("simple-mind-map/src/plugins/Demonstrate.js"),
        import("simple-mind-map/src/plugins/MindMapLayoutPro.js"),
        import("simple-mind-map/src/plugins/NodeBase64ImageStorage.js"),        
        import("simple-mind-map/src/plugins/ExportPDF.js"),
        import("simple-mind-map/src/plugins/ExportXMind.js"),
        import("simple-mind-map/src/core/render/node/MindMapNode.js"),     
      ]);

      // React StrictMode（开发环境）会触发 effect 的“挂载-卸载-再挂载”流程：
      // 若异步加载完成时已经卸载，不应再创建实例，否则可能留下残余 DOM，导致出现多个 root。
      if (destroyed) return;

      // 兜底修补 simple-mind-map 在创建富文本节点时 host/缓存未就绪导致的空指针
      type MindMapNodeCtorLike = {
        prototype: {
          __wolaiRichtextPatched?: boolean;
          createRichTextNode?: (...args: unknown[]) => unknown;
        };
      };
      const MindMapNodeCtor = (MindMapNodeModule as unknown as { default?: MindMapNodeCtorLike })?.default;
      if (
        MindMapNodeCtor &&
        !MindMapNodeCtor.prototype.__wolaiRichtextPatched
      ) {
        const originalCreate = MindMapNodeCtor.prototype.createRichTextNode;
        MindMapNodeCtor.prototype.__wolaiRichtextPatched = true;
        MindMapNodeCtor.prototype.createRichTextNode = function patched(...args: unknown[]) {
          type RichtextThis = {
            mindMap?: { el?: HTMLElement | null; commonCaches?: Record<string, unknown> } | null;
          };
          const self = this as unknown as RichtextThis;
          // host 兜底：优先使用实例容器，否则退回 body
          const host: HTMLElement = (self?.mindMap?.el as HTMLElement | null) ?? document.body;
          if (!self.mindMap) {
            self.mindMap = { el: host, commonCaches: {} };
          }
          const el = self.mindMap.el;
          if (!el || typeof el.appendChild !== "function") {
            self.mindMap.el = host;
          }
          const caches = self.mindMap.commonCaches ?? (self.mindMap.commonCaches = {});
          const measureKey = "measureRichtextNodeTextSizeEl";
          if (!caches[measureKey]) {
            const measureDiv = document.createElement("div");
            measureDiv.style.position = "fixed";
            measureDiv.style.left = "-999999px";
            (self.mindMap.el ?? document.body).appendChild(measureDiv);
            caches[measureKey] = measureDiv;
          }
          if (typeof originalCreate === "function") {
            return originalCreate.apply(this, args);
          }
          return null;
        };
      }

      const plugins = [
        { name: "Painter", plugin: Painter },
        { name: "AssociativeLine", plugin: AssociativeLine },
        { name: "OuterFrame", plugin: OuterFrame },
        { name: "Export", plugin: Exporter },
        { name: "Formula", plugin: Formula },
        { name: "RichText", plugin: RichText },
        { name: "MiniMap", plugin: MiniMapPlugin },
        { name: "Select", plugin: Select },
        { name: "Drag", plugin: Drag },
        { name: "KeyboardNavigation", plugin: KeyboardNavigation },
        { name: "NodeImgAdjust", plugin: NodeImgAdjust },
        { name: "Scrollbar", plugin: Scrollbar },
        { name: "RainbowLines", plugin: RainbowLines },
        { name: "Watermark", plugin: Watermark },
        { name: "TouchEvent", plugin: TouchEvent },
        { name: "Cooperate", plugin: Cooperate },
        { name: "Demonstrate", plugin: Demonstrate },
        { name: "MindMapLayoutPro", plugin: MindMapLayoutPro },
        { name: "NodeBase64ImageStorage", plugin: NodeBase64ImageStorage },
        { name: "ExportPDF", plugin: ExportPDF },
        { name: "ExportXMind", plugin: ExportXMind },
      ];

      plugins.forEach(({ name, plugin }) => {
        if (!plugin) {
          console.warn(`思维导图插件加载失败：${name}`);
          return;
        }
        const MindMapCtor = MindMap as unknown as {
          hasPlugin?: (p: unknown) => number;
          usePlugin?: (p: unknown) => void;
        };
        const hasPlugin = MindMapCtor.hasPlugin;
        const notRegistered =
          typeof hasPlugin === "function" ? hasPlugin(plugin) === -1 : true;
        const registerPlugin = MindMapCtor.usePlugin;
        if (notRegistered && typeof registerPlugin === "function") {
          console.log(`注册插件: ${name}`);
          registerPlugin(plugin);
        }
      });

      const { nodeIconList, mergerIconList } = await loadIconModules();

      await patchSvgRbox();

      if (destroyed) return;

      const hostEl = hostContainer;
      // 防止同一容器残留旧实例的 svg/dom（StrictMode 或异常 destroy 场景）
      try {
        hostEl.replaceChildren();
      } catch {
        hostEl.innerHTML = "";
      }

      type MindMapConstructor = new (options: {
        el: HTMLElement;
        data: unknown;
        theme: string;
        layout: string;
        mousewheelAction: string;
        enableFreeDrag: boolean;
        enableCtrlKeyNodeSelection: boolean;
        fit: boolean;
        useLeftKeySelectionRightKeyDrag: boolean;
        createNewNodeBehavior: string;
        iconList: unknown[];
      }) => MindMapInstance;
      const MindMapCtor = MindMap as unknown as MindMapConstructor;

      const instance = new MindMapCtor({
        el: hostEl,
        data: initialDataRef.current,
        theme: "classic",
        layout: "logicalStructure",
        mousewheelAction: "zoom",
        enableFreeDrag: true,
        enableCtrlKeyNodeSelection: true,
        fit: true,
        useLeftKeySelectionRightKeyDrag: true,
        // 新建节点默认激活，编辑由我们手动触发，避免初次插入时定位到 (0,0)
        createNewNodeBehavior: "activeOnly",
        // 传入扩展图标表，和官方一致
        iconList: mergerIconList([
          ...nodeIconList,
          ...(iconConfig as unknown[]),
        ]),
      });
      createdInstance = instance;
      mindmapRef.current = instance;
      // 尽早标记为可用：主页面通过 `/mind` 插入后，若依赖 render_end/尺寸探测，可能导致按钮永远不可用
      //（render_end 可能早于监听注册触发，或容器尺寸长时间为 0）。命令本身不需要等待 render_end。
      mindmapReadyRef.current = true;
      // 确保测量节点的缓存容器始终挂在有效 DOM 上，避免 appendChild 空指针     
      type MindMapWithCaches = MindMapInstance & { commonCaches?: Record<string, unknown> };
      const instanceWithCaches = instance as MindMapWithCaches;
      if (!instanceWithCaches.commonCaches) {
        instanceWithCaches.commonCaches = {};
      }
      const measureKey = "measureRichtextNodeTextSizeEl";
      if (!instanceWithCaches.commonCaches[measureKey]) {
        const measureDiv = document.createElement("div");
        measureDiv.style.position = "fixed";
        measureDiv.style.left = "-999999px";
        instanceWithCaches.commonCaches[measureKey] = measureDiv;
        hostEl.appendChild(measureDiv);
      }
      instance.setMode?.("edit");
    const centerAndFit = (retry = 0) => {
        try {
          const hostRect = hostEl.getBoundingClientRect?.();
          if (!hostRect || hostRect.width < 10 || hostRect.height < 10) {       
            if (retry < 5) {
              window.setTimeout(() => centerAndFit(retry + 1), 80);
            }
            return;
          }
          // 只要容器尺寸已可用，就允许工具栏命令执行；render_end 可能早于监听注册触发
          //（尤其在首次构造时同步渲染），因此不要完全依赖 render_end 来标记就绪。
          mindmapReadyRef.current = true;
          const renderer = instance.renderer;
          const rootNode = renderer?.root ?? renderer?.renderTree?._node;
          if (rootNode && renderer?.setRootNodeCenter) {
            renderer.setRootNodeCenter();
          }
          instance.view?.fit?.();
        } catch (error) {
          if (retry < 3) {
            window.setTimeout(() => centerAndFit(retry + 1), 50);
          } else {
            console.warn("思维导图初始居中失败，已跳过", error);
          }
        }
      };
      window.requestAnimationFrame(() => centerAndFit());
      window.setTimeout(() => centerAndFit(), 80);

      if (typeof window !== "undefined") {
        // 便于开发阶段在控制台直接调试实例
        window.__mindmapInstance = instance;
      }

      setMindmap(instance);

      instance.on?.("back_forward", (index: number, len: number) => {
        setCanBack(index > 0);
        setCanForward(index < len - 1);
      });
      // 渲染完成后再二次居中，并确保根节点被标记为激活
      const handleRenderEnd = () => {
        const renderer = instance.renderer;
        const root = getRootNode(instance);
        if (root) {
          if (renderer?.clearActiveNodeList) renderer.clearActiveNodeList();
          if (renderer?.addNodeToActiveList) renderer.addNodeToActiveList(root, true);
          if (renderer?.emitNodeActiveEvent) renderer.emitNodeActiveEvent(root);
        }
        mindmapReadyRef.current = true;
        centerAndFit();
      };

      // 注意：部分版本的 simple-mind-map 中 `once` 回调的触发顺序可能早于 `on`，
      // 若在 once 里 off 监听，可能导致 handleRenderEnd 永远不执行，进而使工具栏按钮不可用。
      // 这里统一用 on + 手动自销毁，确保首个 render_end 一定会执行 handleRenderEnd。
      let renderHandled = false;
      const onRenderEndOnce = () => {
        if (renderHandled) return;
        renderHandled = true;
        try {
          handleRenderEnd();
        } finally {
          instance.off?.("render_end", onRenderEndOnce);
        }
      };
      instance.on?.("render_end", onRenderEndOnce);
      instance.on?.("node_active", (_node: unknown, list: unknown[]) => {
        if (!list || list.length === 0) return;
        setActiveNodes((list || []) as MindMapNode[]);
      });
      instance.on?.("node_click", (node: unknown) => {
        // 确保点击即可激活（调用节点自身的 active 逻辑，保持和官方一致）
        const renderer = instance.renderer;
        const nodeWithActive = node as unknown as { active?: () => void };
        if (typeof nodeWithActive.active === "function") {
          nodeWithActive.active();
        } else {
          // 兜底：手动维护激活列表
          if (renderer?.clearActiveNodeList) renderer.clearActiveNodeList();
          if (renderer?.addNodeToActiveList) renderer.addNodeToActiveList(node, true);
          if (renderer?.emitNodeActiveEvent) renderer.emitNodeActiveEvent(node);
        }
        const list = renderer?.activeNodeList ?? [];
        setActiveNodes(
          list.length === 0 && node
            ? [node as MindMapNode]
            : (list as MindMapNode[]),
        );
      });
      instance.on?.("painter_start", () => setPainterMode(true));
      instance.on?.("painter_end", () => setPainterMode(false));
      instance.on?.("data_change", (data: unknown) => {
        if (!applyingRemoteRef.current) {
          hasLocalEditsRef.current = true;
        }
        debouncedPersist(data);
      });
      const renderer = instance.renderer;
      const rootNode = getRootNode(instance);
      if (rootNode) {
        // 初始化时主动标记根节点为选中，确保后续插入子节点有合法的父节点
        if (renderer?.clearActiveNodeList) renderer.clearActiveNodeList();
        if (renderer?.addNodeToActiveList) renderer.addNodeToActiveList(rootNode, true);
        if (renderer?.emitNodeActiveEvent) renderer.emitNodeActiveEvent(rootNode);
        setActiveNodes([rootNode as MindMapNode]);
      }
      // 若首次渲染时 root 仍未就绪，设置兜底延迟激活
      window.setTimeout(() => {
        const root = getRootNode(instance);
        if (!root) return;
        if (renderer?.clearActiveNodeList) renderer.clearActiveNodeList();
        if (renderer?.addNodeToActiveList) renderer.addNodeToActiveList(root, true);
        if (renderer?.emitNodeActiveEvent) renderer.emitNodeActiveEvent(root);
        setActiveNodes([root as MindMapNode]);
      }, 120);

      if (destroyed) {
        instance.destroy();
      }
    })();

    return () => {
      destroyed = true;
      try {
        // 切换“内嵌/全屏”会导致实例重建：这里尽量在销毁前同步一次数据，避免丢失最后一次编辑
        if (createdInstance && typeof window !== "undefined") {
          const data =
            createdInstance.getData?.(true) ?? createdInstance.getData?.();
          if (data) {
            try {
              window.localStorage.setItem(autosaveKey, JSON.stringify(data));
            } catch {
              // ignore
            }
            try {
              editor?.updateBlock(block, { props: { ...block.props, data } });
            } catch {
              // ignore
            }
            if (docId) {
              fetch(`/api/mindmap/${docId}`, {
                method: "POST",
                headers: { "Content-Type": "application/json" },
                body: JSON.stringify({ data }),
              }).catch(() => {
                // ignore
              });
            }
          }
        }
        createdInstance?.destroy();
      } catch {
        // ignore
      }
      if (typeof window !== "undefined") {
        if (window.__mindmapInstance === createdInstance) {
          window.__mindmapInstance = null;
        }
      }
      setMindmap(null);
      mindmapRef.current = null;
      mindmapReadyRef.current = false;
    };
    // eslint-disable-next-line react-hooks/exhaustive-deps
  }, [block.id, effectiveFullscreen]);

  const getInstanceCandidate = () =>
    mindmap ??
    (typeof window !== "undefined"
      ? ((window as unknown as { __mindmapInstance?: MindMapInstance | null })
          .__mindmapInstance ?? null)
      : null);

  const runWhenReady = (
    fn: (mm: MindMapInstance) => void,
    retry = 0,
  ) => {
    const instance = getInstanceCandidate();
    if (mindmapReadyRef.current && instance) {
      fn(instance);
      return;
    }
    if (retry < 60) {
      window.setTimeout(() => runWhenReady(fn, retry + 1), 50);
    }
  };
  const execWithReflow = (runner: (mm: MindMapInstance) => void) => {
    const attempt = (retry = 0) => {
      const instance = getInstanceCandidate();
      if (!instance || !mindmapReadyRef.current) {
        if (retry < 60) window.setTimeout(() => attempt(retry + 1), 80);
        return;
      }
      const mm = ensureActiveBefore(instance);
      if (!mm) return;
      runner(mm);
    };
    attempt();
  };

  const pickActiveOrRoot = (mm: MindMapInstance) => {
    const list = mm.renderer?.activeNodeList;
    const last = mm.renderer?.lastActiveNodeList;
    return (list && list[0]) || (last && last[0]) || getRootNode(mm) || null;
  };

  const getNodeUid = (node: unknown): string | null => {
    const maybeNode = node as unknown as { getData?: (key: string) => unknown; uid?: unknown };
    const raw = typeof maybeNode.getData === "function" ? maybeNode.getData("uid") : maybeNode.uid;
    return typeof raw === "string" && raw ? raw : null;
  };

  const handleUndo = () => runWhenReady((mm) => mm.execCommand("BACK"));
  const handleRedo = () => runWhenReady((mm) => mm.execCommand("FORWARD"));
  const handlePainter = () => {
    const instance = getInstanceCandidate();
    if (!instance?.painter) {
      window.alert("格式刷插件未就绪");
      return;
    }
    instance.painter.startPainter();
  };
  const handleSibling = () =>
    execWithReflow((mm) => {
      const target = pickActiveOrRoot(mm);
      if (!target) return;
      const targetUid = getNodeUid(target);
      mm.execCommand?.("SET_NODE_ACTIVE", target, true);
      // openEdit=false：保持与 createNewNodeBehavior=activeOnly 的策略一致，避免强制进入编辑模式带来的时序问题
      mm.execCommand?.("INSERT_NODE", false, [target]);

      // 兜底：某些情况下插入后 activeNodeList 会短暂为空，导致下一步“同级/删除”无目标。
      // 这里尝试在下一帧将新插入的节点设为激活（插入同级节点时，新节点位于 target 后）。
      window.setTimeout(() => {
        try {
          const renderer = mm.renderer;
          if ((renderer.activeNodeList?.length ?? 0) > 0) return;
          if (!targetUid || typeof renderer.findNodeByUid !== "function") return;
          const currentTarget = renderer.findNodeByUid(targetUid) as unknown as {
            parent?: { children?: unknown[] } | null;
          };
          const parent = currentTarget?.parent ?? null;
          const siblings = (parent?.children ?? []) as unknown[];
          const idx = siblings.indexOf(currentTarget as unknown);
          const inserted = idx >= 0 ? siblings[idx + 1] : null;
          if (!inserted) return;
          renderer?.clearActiveNodeList?.();
          renderer?.addNodeToActiveList?.(inserted, true);
          renderer.lastActiveNodeList = [inserted];
          renderer?.emitNodeActiveEvent?.(inserted);
          mm.execCommand?.("SET_NODE_ACTIVE", inserted, true);
        } catch {
          /* ignore */
        }
      }, 0);
    });
  const handleChild = () =>
    execWithReflow((mm) => {
      const parent = pickActiveOrRoot(mm);
      if (!parent) return;
      const parentUid = getNodeUid(parent);
      mm.execCommand?.("SET_NODE_ACTIVE", parent, true);
      // openEdit=false：同上，避免插入瞬间强制进入编辑导致异常
      mm.execCommand?.("INSERT_CHILD_NODE", false, [parent]);

      // 兜底：确保新插入的子节点成为激活节点，保证随后点击“同级节点/删除节点”可用
      window.setTimeout(() => {
        try {
          const renderer = mm.renderer;
          if ((renderer.activeNodeList?.length ?? 0) > 0) return;
          if (!parentUid || typeof renderer.findNodeByUid !== "function") return;
          const currentParent = renderer.findNodeByUid(parentUid) as unknown as {
            children?: unknown[];
          };
          const children = (currentParent?.children ?? []) as unknown[];
          const inserted = children.length > 0 ? children[children.length - 1] : null;
          if (!inserted) return;
          renderer?.clearActiveNodeList?.();
          renderer?.addNodeToActiveList?.(inserted, true);
          renderer.lastActiveNodeList = [inserted];
          renderer?.emitNodeActiveEvent?.(inserted);
          mm.execCommand?.("SET_NODE_ACTIVE", inserted, true);
        } catch {
          /* ignore */
        }
      }, 0);
    });
  const handleDelete = () => {
    runWhenReady((instance) => {
      const mm = ensureActiveBefore(instance);
      if (!mm) return;
      window.setTimeout(() => mm.execCommand("REMOVE_NODE"), 0);
    });
  };
  const handleSummary = () => {
    runWhenReady((instance) => {
      const mm = ensureActiveBefore(instance);
      if (mm) {
        window.setTimeout(() => mm.execCommand("ADD_GENERALIZATION"), 0);
      }
    });
  };
  const handleAssociativeLine = () => {
    runWhenReady((instance) => {
      const mm = ensureActiveBefore(instance);
      if (mm) {
        window.setTimeout(() => mm.execCommand("ADD_ASSOCIATIVE_LINE"), 0);     
      }
    });
  };
  const handleOuterFrame = () => {
    runWhenReady((instance) => {
      const mm = ensureActiveBefore(instance);
      if (mm) {
        window.setTimeout(() => mm.execCommand("ADD_OUTER_FRAME"), 0);
      }
    });
  };

  const [showImageModal, setShowImageModal] = useState(false);
  const [imageUrl, setImageUrl] = useState("");
  const [imageWidth, setImageWidth] = useState<number | string>(260);
  const [imageHeight, setImageHeight] = useState<number | string>(200);
  const [imageTitle, setImageTitle] = useState("");
  const [imagePosition, setImagePosition] = useState<string>("top");
  const fileInputForImage = useRef<HTMLInputElement | null>(null);

  // 图片预览（双击节点图片）
  const [showImageViewer, setShowImageViewer] = useState(false);
  const [viewerSrc, setViewerSrc] = useState("");
  const [viewerMeta, setViewerMeta] = useState<{ title?: string; width?: number; height?: number }>({});
  const imgToolbarRef = useRef<HTMLDivElement | null>(null);
  const [imgToolbarState, setImgToolbarState] = useState({
    show: false,
    x: 0,
    y: 0,
    placement: "top" as "top" | "bottom" | "left" | "right",
  });
  const imgToolbarHover = useRef(false);

  const handleImage = () => {
    const first = mindmap?.renderer?.activeNodeList?.[0] as MindMapNode | undefined;
    if (first?.getStyle) {
      const placement = first.getStyle("imgPlacement", false) as string;        
      if (placement) setImagePosition(placement);
    }
    setShowImageModal(true);
  };

  // 预览：监听 mindmap 事件
  useEffect(() => {
    if (!mindmap) return;
    const handler = (node: MindMapNode, e?: Event) => {
      e?.stopPropagation?.();
      e?.preventDefault?.();
      const srcRaw =
        node?.nodeData?.data?.image ||
        node?.getData?.("image") ||
        node?.data?.image;
      const src = typeof srcRaw === "string" ? srcRaw : "";
      if (src) {
        setViewerSrc(src);
        const sizeRaw = node?.getData?.("imageSize");
        const size =
          sizeRaw && typeof sizeRaw === "object"
            ? (sizeRaw as { width?: unknown; height?: unknown })
            : ({} as { width?: unknown; height?: unknown });
        const title = node?.getData?.("imageTitle") || "";
        setViewerMeta({
          title: typeof title === "string" ? title : "",
          width: Number(size.width) || undefined,
          height: Number(size.height) || undefined,
        });
        setShowImageViewer(true);
      }
    };
    const onActive = () => {
      const list = (mindmap.renderer?.activeNodeList || []) as MindMapNode[];
      const has = list.some((n) => !!n?.getData?.("image"));
      if (!has) setImgToolbarState((s) => ({ ...s, show: false }));
    };
    const showToolbarOnClick = (node: MindMapNode, _svgImg: unknown, evt: Event | undefined) => {
      const target = evt?.target as Element | undefined;
      const bbox = target?.getBoundingClientRect?.();
      const placement = (node?.getStyle?.("imgPlacement") || "top") as "top" | "bottom" | "left" | "right";
      if (!bbox) return;
      setImagePosition(placement);
      setImgToolbarState({
        show: true,
        x: bbox.left,
        y: bbox.top,
        placement,
      });
    };
    const hideToolbar = () => {
      if (!imgToolbarHover.current) {
        setImgToolbarState((s) => ({ ...s, show: false }));
      }
    };
    mindmap.on?.("node_img_dblclick", handler);
    mindmap.on?.("node_active", onActive);
    mindmap.on?.("node_img_click", showToolbarOnClick);
    mindmap.on?.("draw_click", hideToolbar);
    return () => {
      mindmap.off?.("node_img_dblclick", handler);
      mindmap.off?.("node_active", onActive);
      mindmap.off?.("node_img_click", showToolbarOnClick);
      mindmap.off?.("draw_click", hideToolbar);
    };
  }, [mindmap]);

  // 悬浮图片位置工具条
  const renderImgToolbar = () => {
    if (!imgToolbarState.show) return null;
    const { x, y, placement } = imgToolbarState;
    const setPlacement = (p: "top" | "bottom" | "left" | "right") => {
      setImagePosition(p);
      applyToActiveNodes(mindmap, (node) =>
        mindmap?.execCommand?.("SET_NODE_STYLES", node, { imgPlacement: p }),   
      );
      setImgToolbarState((s) => ({ ...s, placement: p }));
    };
    const btn = (
      p: typeof placement,
      Icon: React.ComponentType<{ className?: string }>,
      title: string,
    ) => (
      <button
        key={p}
        type="button"
        className={`flex h-8 w-8 items-center justify-center rounded border bg-white/90 text-gray-700 shadow ${
          placement === p ? "border-blue-500 text-blue-600" : "border-gray-200"
        }`}
        onMouseDown={(e) => {
          e.stopPropagation();
          e.preventDefault();
          setPlacement(p);
        }}
        title={title}
      >
        <Icon className="h-4 w-4" />
      </button>
    );
    return (
      <div
        ref={imgToolbarRef}
        className="pointer-events-auto fixed z-[9999] flex gap-1 rounded-lg bg-black/50 p-2 backdrop-blur"
        style={{ left: x, top: y - 42 }}
        onMouseEnter={() => {
          imgToolbarHover.current = true;
          setImgToolbarState((s) => ({ ...s, show: true }));
        }}
        onMouseLeave={() => {
          imgToolbarHover.current = false;
          setImgToolbarState((s) => ({ ...s, show: false }));
        }}
      >
        {btn("top", ArrowUp, "顶部")}
        {btn("bottom", ArrowDown, "底部")}
        {btn("left", ArrowLeft, "靠左")}
        {btn("right", ArrowRight, "靠右")}
      </div>
    );
  };

  const handleIcon = () => {
    setActiveSidebar("icons");
  };

  const handleLink = () => {
    const href = createSimplePrompt("请输入超链接");
    if (!href) return;
    applyToActiveNodes(mindmap, (node) => mindmap?.execCommand("SET_NODE_HYPERLINK", node, href, href));
  };

  const handleNote = () => {
    setShowNoteModal(true);
  };

  const handleTag = () => {
    const tagsRaw = createSimplePrompt("请输入标签，使用逗号分隔", "重点,待验证");
    if (!tagsRaw) return;
    const tags = tagsRaw.split(",").map((item) => item.trim()).filter(Boolean);
    applyToActiveNodes(mindmap, (node) => mindmap?.execCommand("SET_NODE_TAG", node, tags));
  };

  const handleFormula = () => {
    setActiveSidebar("formula");
  };

  const handleNoteConfirm = () => {
    const note = noteContent.trim();
    setShowNoteModal(false);
    if (!note) return;
    applyToActiveNodes(mindmap, (node) => mindmap?.execCommand("SET_NODE_NOTE", node, note));
  };

  const handleImageConfirm = () => {
    const url = imageUrl.trim();
    const width = Number(imageWidth) || 260;
    const height = Number(imageHeight) || 200;
    if (!url) {
      window.alert("请输入图片链接");
      return;
    }
    setShowImageModal(false);
    applyToActiveNodes(mindmap, (node) =>
      mindmap?.execCommand("SET_NODE_IMAGE", node, {
        url,
        width,
        height,
        title: imageTitle || "",
        position: imagePosition || "top",
      }),
    );
    applyToActiveNodes(mindmap, (node) =>
      mindmap?.execCommand?.("SET_NODE_STYLES", node, {
        imgPlacement: imagePosition || "top",
      }),
    );
  };

  const handleAttachment = () => {
    const url = createSimplePrompt("附件链接（http/https）");
    if (!url) return;
    const name = createSimplePrompt("附件名称（可选）", "附件");
    applyToActiveNodes(mindmap, (node) => mindmap?.execCommand("SET_NODE_ATTACHMENT", node, url, name ?? ""));
  };

  const handleAiPlaceholder = () => {
    window.alert("AI 能力占位：后续接入大模型生成/优化节点内容。");
  };

  const handleImport = async (event: React.ChangeEvent<HTMLInputElement>) => {  
    const file = event.target.files?.[0];
    if (!file) return;
    const reset = () => {
      event.target.value = "";
    };
    const ext = (file.name.split(".").pop() || "").toLowerCase();
    try {
      // JSON / smm
      if (ext === "json" || ext === "smm") {
        const text = await file.text();
        const data = JSON.parse(text);
        mindmap?.setData(data);
        mindmap?.command.clearHistory();
        debouncedPersist(data);
        return;
      }

      // XMind
      if (ext === "xmind") {
        const xmindParser = await import("simple-mind-map/src/parse/xmind.js");
        const blob = new Blob([await file.arrayBuffer()]);
        const data = await xmindParser.default.parseXmindFile(blob, (content) => {
          const list = content;
          if (list.length > 1) {
            window.alert("检测到 XMind 多画布，自动导入第一个画布。");    
          }
          return list.length > 0 ? list[0] : content;
        });
        mindmap?.setData(data);
        mindmap?.command.clearHistory();
        debouncedPersist(data);
        return;
      }

      // Markdown
      if (ext === "md" || ext === "markdown") {
        const { transformMarkdownTo } = await import(
          "simple-mind-map/src/parse/markdownTo.js"
        );
        const text = await file.text();
        const data = transformMarkdownTo(text) as MindMapData;
        if (!data.data) {
          data.data = { text: file.name.replace(/\.(md|markdown)$/i, "") || "中心主题" };
        }
        mindmap?.setData(data);
        mindmap?.command.clearHistory();
        debouncedPersist(data);
        return;
      }

      window.alert("暂不支持该文件类型，请选择 .json / .smm / .xmind / .md");
    } catch (error) {
      console.error(error);
      window.alert("导入失败：文件格式或内容错误");
    } finally {
      reset();
    }
  };

  const handleNew = () => {
    if (!window.confirm("确定要新建空白导图吗？当前未保存的修改将被覆盖。")) return;
    mindmap?.setData(defaultMindmapData);
    mindmap?.command.clearHistory();
    debouncedPersist(defaultMindmapData);
  };

  const handleOpenDirectory = () => {
    const saved = window.localStorage.getItem(autosaveKey);
    if (!saved) {
      window.alert("暂无本地目录记录，可使用“导入”载入文件。");
      return;
    }
    try {
      const data = JSON.parse(saved);
      mindmap?.setData(data);
      window.alert("已从本地目录恢复最新自动保存版本。");
    } catch {
      window.alert("本地目录数据损坏，建议重新导入。");
    }
  };

  const handleExportJson = () => {
    const data = mindmap?.getData?.(true) ?? block.props.data ?? defaultMindmapData;
    downloadJson(data, "mindmap");
  };

  const handleExport = async (type: string, name = "mindmap") => {
    try {
      await mindmap?.doExport?.export(type, true, name);
    } catch (error) {
      console.error(error);
      window.alert(`导出 ${type.toUpperCase()} 失败，请稍后再试`);
    }
  };

  const handleExportPng = () => handleExport("png");
  const handleExportSvg = () => handleExport("svg");
  const handleExportPdf = () => handleExport("pdf");
  const handleExportMd = () => handleExport("md");
  const handleExportTxt = () => handleExport("txt");
  const handleExportXmind = () => handleExport("xmind");

  const handleSaveAs = () => handleExportJson();

  const handleDeleteMindmap = useCallback(async () => {
    if (!docId) {
      editor.removeBlocks([block.id]);
      return;
    }
    const confirmed = window.confirm("确认删除当前思维导图？此操作会移除文件并清空侧边栏记录。");
    if (!confirmed) return;
    const resp = await fetch(`/api/mindmap/${docId}`, { method: "DELETE" });
    if (!resp.ok) {
      const payload = await resp.json().catch(() => ({}));
      window.alert(payload?.error ?? "删除思维导图失败");
      return;
    }
    try {
      window.localStorage.removeItem(autosaveKey);
    } catch {
      // ignore
    }
    emitAssetsChanged(docId, undefined, undefined, true);
    editor.removeBlocks([block.id]);
  }, [autosaveKey, block.id, docId, editor]);

  const toolbarProps = {
    canBack,
    canForward,
    painterMode,
    onUndo: handleUndo,
    onRedo: handleRedo,
    onPainter: handlePainter,
    onSibling: handleSibling,
    onChild: handleChild,
    onDelete: handleDelete,
    onImage: handleImage,
    onIcon: handleIcon,
    onLink: handleLink,
    onNote: handleNote,
    onTag: handleTag,
    onSummary: handleSummary,
    onAssociativeLine: handleAssociativeLine,
    onFormula: handleFormula,
    onAttachment: handleAttachment,
    onOuterFrame: handleOuterFrame,
    onAi: handleAiPlaceholder,
    onImport: handleImport,
    onNew: handleNew,
    onOpenDirectory: handleOpenDirectory,
    onSaveAs: handleSaveAs,
    onDeleteMindmap: handleDeleteMindmap,
    onExportJson: handleExportJson,
    onExportPng: handleExportPng,
    onExportSvg: handleExportSvg,
    onExportPdf: handleExportPdf,
    onExportMd: handleExportMd,
    onExportTxt: handleExportTxt,
    onExportXmind: handleExportXmind,
    fileInputRef,
  };

  const noteModal = showNoteModal ? (
    <div className="fixed inset-0 z-50 flex items-center justify-center bg-black/40 p-4">
      <div className="w-full max-w-4xl rounded-xl bg-white shadow-2xl">
        <div className="flex items-center justify-between border-b px-4 py-3">  
          <span className="text-lg font-semibold text-gray-800">备注</span>     
          <button className="text-gray-400 hover:text-gray-600" onClick={() => setShowNoteModal(false)}>
            ✕
          </button>
        </div>
        <div className="px-4 py-3">
          <textarea
            className="h-64 w-full resize-none rounded-md border border-gray-200 p-3 text-sm focus:border-blue-400 focus:outline-none"
            placeholder="支持富文本/Markdown，内容将写入节点备注"
            value={noteContent}
            onChange={(e) => setNoteContent(e.target.value)}
          />
        </div>
        <div className="flex items-center justify-end gap-3 border-t px-4 py-3">
          <button
            className="rounded-md border border-gray-200 px-4 py-2 text-sm text-gray-600 hover:bg-gray-50"
            onClick={() => setShowNoteModal(false)}
          >
            取消
          </button>
          <button
            className="rounded-md bg-blue-500 px-4 py-2 text-sm text-white hover:bg-blue-600"
            onClick={handleNoteConfirm}
          >
            确定
          </button>
        </div>
      </div>
    </div>
  ) : null;

  const imageViewer = showImageViewer ? (
    <div className="fixed inset-0 z-50 flex items-center justify-center bg-black/70 p-4" onClick={() => setShowImageViewer(false)}>
      <div className="relative max-h-full max-w-5xl text-white">
        <button
          className="absolute -top-3 -right-3 rounded-full bg-white/90 px-2 py-1 text-sm text-gray-700 shadow"
          onClick={(e) => {
            e.stopPropagation();
            setShowImageViewer(false);
          }}
        >
          关闭
        </button>
        <div className="mb-2 flex items-center justify-between gap-4 px-1 text-xs text-gray-200">
          <span className="truncate">{viewerMeta.title || "图片预览"}</span>    
          <span className="shrink-0">{viewerMeta.width && viewerMeta.height ? `${viewerMeta.width} × ${viewerMeta.height}px` : ""}</span>
        </div>
        {/* eslint-disable-next-line @next/next/no-img-element */}
        <img
          src={viewerSrc}
          alt={viewerMeta.title || "预览"}
          className="max-h-[80vh] max-w-[80vw] rounded-lg shadow-2xl object-contain bg-white"
          onClick={(e) => e.stopPropagation()}
        />
      </div>
    </div>
  ) : null;

  const imgToolbar = renderImgToolbar();

  const imageModal = showImageModal ? (
    <div className="fixed inset-0 z-50 flex items-center justify-center bg-black/40 p-4">
      <div className="w-full max-w-3xl rounded-xl bg-white shadow-2xl">
        <div className="flex items-center justify-between border-b px-4 py-3">  
          <span className="text-lg font-semibold text-gray-800">插入图片</span> 
          <button className="text-gray-400 hover:text-gray-600" onClick={() => setShowImageModal(false)}>
            ✕
          </button>
        </div>
          <div className="space-y-4 px-4 py-3">
          <div className="space-y-2">
            <Label className="text-xs font-semibold text-gray-600">方式一：上传图片</Label>
            <div className="flex items-center gap-2">
              <button
                className="rounded-md border border-gray-200 px-3 py-2 text-sm text-gray-700 hover:bg-gray-50"
                onClick={() => fileInputForImage.current?.click()}
              >
                选择文件
              </button>
              <span className="text-xs text-gray-400 truncate">
                {imageUrl.startsWith("data:") ? "已选择本地图片" : "未选择文件"}
              </span>
              <input
                ref={fileInputForImage}
                type="file"
                accept="image/*"
                className="hidden"
                onChange={async (e) => {
                  const file = e.target.files?.[0];
                  if (!file) return;
                  const reader = new FileReader();
                  reader.onload = async () => {
                    const dataUrl = String(reader.result);
                    setImageUrl(dataUrl);
                    const size = await getImageSizeSafe(dataUrl);
                    if (size) {
                      setImageWidth(size.width);
                      setImageHeight(size.height);
                    }
                  };
                  reader.readAsDataURL(file);
                }}
              />
            </div>
          </div>
          <div className="space-y-2">
            <Label className="text-xs font-semibold text-gray-600">方式二：图片地址</Label>
            <Input
              value={imageUrl}
              onChange={(e) => setImageUrl(e.target.value)}
              placeholder="https://example.com/image.png"
            />
          </div>
          <div className="space-y-2">
            <Label className="text-xs font-semibold text-gray-600">快捷位置</Label>
            <div className="flex gap-2">
              <button
                type="button"
                className={`flex-1 rounded border px-2 py-2 text-sm ${imagePosition === "top" ? "border-blue-500 bg-blue-50 text-blue-600" : "border-gray-200"}`}
                onClick={() => setImagePosition("top")}
                title="顶部"
              >
                <ArrowUp className="mx-auto h-4 w-4" />
              </button>
              <button
                type="button"
                className={`flex-1 rounded border px-2 py-2 text-sm ${imagePosition === "bottom" ? "border-blue-500 bg-blue-50 text-blue-600" : "border-gray-200"}`}
                onClick={() => setImagePosition("bottom")}
                title="底部"
              >
                <ArrowDown className="mx-auto h-4 w-4" />
              </button>
              <button
                type="button"
                className={`flex-1 rounded border px-2 py-2 text-sm ${imagePosition === "left" ? "border-blue-500 bg-blue-50 text-blue-600" : "border-gray-200"}`}
                onClick={() => setImagePosition("left")}
                title="靠左"
              >
                <ArrowLeft className="mx-auto h-4 w-4" />
              </button>
              <button
                type="button"
                className={`flex-1 rounded border px-2 py-2 text-sm ${imagePosition === "right" ? "border-blue-500 bg-blue-50 text-blue-600" : "border-gray-200"}`}
                onClick={() => setImagePosition("right")}
                title="靠右"
              >
                <ArrowRight className="mx-auto h-4 w-4" />
              </button>
            </div>
          </div>
          <div className="grid grid-cols-2 gap-3">
            <div className="space-y-1">
              <Label className="text-xs text-gray-500">宽度（px）</Label>
              <Input
                type="number"
                value={imageWidth}
                onChange={(e) => setImageWidth(e.target.value)}
                min={10}
                placeholder="260"
              />
            </div>
            <div className="space-y-1">
              <Label className="text-xs text-gray-500">高度（px）</Label>
              <Input
                type="number"
                value={imageHeight}
                onChange={(e) => setImageHeight(e.target.value)}
                min={10}
                placeholder="200"
              />
            </div>
          </div>
          <div className="space-y-1">
            <Label className="text-xs text-gray-500">标题（可选）</Label>
            <Input
              value={imageTitle}
              onChange={(e) => setImageTitle(e.target.value)}
              placeholder="图片标题"
            />
          </div>
          <div className="space-y-1">
            <Label className="text-xs text-gray-500">位置</Label>
            <select
              className="h-9 w-full rounded-md border border-gray-200 px-3 text-sm"
              value={imagePosition}
              onChange={(e) => setImagePosition(e.target.value)}
            >
              <option value="center">居中</option>
              <option value="top">顶部</option>
              <option value="bottom">底部</option>
              <option value="left">靠左</option>
              <option value="right">靠右</option>
            </select>
          </div>
          <p className="text-xs text-gray-400">可选择本地图片或粘贴 URL，默认尺寸 260×200。</p>
        </div>
        <div className="flex items-center justify-end gap-3 border-t px-4 py-3">  
          <button
            className="rounded-md border border-gray-200 px-4 py-2 text-sm text-gray-600 hover:bg-gray-50"
            onClick={() => setShowImageModal(false)}
          >
            取消
          </button>
          <button
            className="rounded-md bg-blue-500 px-4 py-2 text-sm text-white hover:bg-blue-600"
            onClick={handleImageConfirm}
          >
            插入
          </button>
        </div>
      </div>
    </div>
  ) : null;

  if (effectiveFullscreen) {
    const fullscreenView = (
      <div
        ref={wrapperRef}
        data-testid="mindmap-fullscreen"
        tabIndex={0}
        className="fixed inset-0 z-[9999] overflow-hidden bg-white"       
      >
        <div className="fixed left-0 right-0 top-0 z-30 flex h-12 items-center justify-between border-b border-gray-200 bg-white/90 px-4 backdrop-blur">
          <div className="text-sm font-medium text-gray-700">
            思维导图编辑{fullscreenApiActive ? "（全屏）" : ""}
          </div>
          <button
            type="button"
            title="退出全屏"
            className="rounded p-2 text-gray-500 hover:bg-gray-100 hover:text-gray-700"
            onClick={exitLocalFullscreen}
          >
            <span className="text-lg leading-none">×</span>
          </button>
        </div>

        <div className="fixed left-1/2 top-14 z-20 flex w-fit -translate-x-1/2 transform items-center justify-center gap-4">
          <MindmapToolbar {...toolbarProps} />
        </div>
        <div className="relative h-full w-full pt-12">
          <div
            ref={containerRef}
            className="h-full w-full"
            data-testid="mindmap-canvas"
            contentEditable={false}
          />

          <MindmapSidebarTrigger
            activeSidebar={activeSidebar}
            onSelect={setActiveSidebar}
          />
          <MindmapSidebar
            mindmap={mindmap}
            activeNodes={activeNodes}
            activeTab={activeSidebar}
            onClose={() => setActiveSidebar(null)}
          />
          <MindmapNavigator
            mindmap={mindmap}
            fullscreen={effectiveFullscreen}
            toggleFullscreen={
              fullscreen
                ? undefined
                : () => {
                    if (localFullscreen) exitLocalFullscreen();
                    else enterLocalFullscreen();
                  }
            }
            onToggleMiniMap={() => setShowMiniMap(!showMiniMap)}
            miniMapOpen={showMiniMap}
          />
          <MindmapCount mindmap={mindmap} />
          <MindmapMiniMap mindmap={mindmap} show={showMiniMap} />
          {imgToolbar}
          {noteModal}
          {imageModal}
          {imageViewer}
        </div>
      </div>
    );

    if (typeof document === "undefined") return null;
    return createPortal(fullscreenView, document.body);
  }

  return (
    <div
      ref={wrapperRef}
      data-testid="mindmap-embed"
      tabIndex={0}
      className="not-prose my-4 w-full overflow-hidden rounded-xl border border-gray-200 bg-[#f8fafc] shadow-sm"
    >
      <div className="flex flex-col gap-3 border-b border-gray-100 bg-white px-4 py-3">
        <div className="w-full">
          <MindmapToolbar {...toolbarProps} />
        </div>
      </div>
      <div
        data-testid="mindmap-stage"
        className="relative h-[520px] w-full overflow-hidden bg-white"    
        onPointerDown={() => {
          hotkeyScopeRef.current = true;
          // 点击后把焦点落到容器上，避免快捷键被编辑器抢走
          try {
            wrapperRef.current?.focus?.({ preventScroll: true });
          } catch {
            // ignore
          }
        }}
        onMouseDown={() => {
          hotkeyScopeRef.current = true;
          // 兼容：某些环境下 pointer 事件不触发
          try {
            wrapperRef.current?.focus?.({ preventScroll: true });
          } catch {
            // ignore
          }
        }}
        onDoubleClick={() => {
          if (recentNodeDblclickRef.current) return;
          enterLocalFullscreen();
        }}
      >
        <div
          ref={containerRef}
          className="h-full w-full"
          data-testid="mindmap-canvas"
          contentEditable={false}
        />
        
        <MindmapSidebarTrigger
            activeSidebar={activeSidebar}
            onSelect={setActiveSidebar}
        />
        <MindmapSidebar
            mindmap={mindmap}
            activeNodes={activeNodes}
            activeTab={activeSidebar}
            onClose={() => setActiveSidebar(null)}
        />
        <MindmapNavigator
            mindmap={mindmap}
            fullscreen={effectiveFullscreen}
            toggleFullscreen={
              fullscreen
                ? undefined
                : () => {
                    if (localFullscreen) exitLocalFullscreen();
                    else enterLocalFullscreen();
                  }
            }
            onToggleMiniMap={() => setShowMiniMap(!showMiniMap)}
            miniMapOpen={showMiniMap}
        />
        <MindmapCount mindmap={mindmap} />
        <MindmapMiniMap mindmap={mindmap} show={showMiniMap} />
        {imgToolbar}
        {noteModal}
        {imageModal}
        {imageViewer}
      </div>
    </div>
  );
};

export { MindmapBlockView };

type MindmapBlockViewProps = Parameters<typeof MindmapBlockView>[0];

export const mindmapBlock = createReactBlockSpec(
  {
    type: "mindmap",
    propSchema: {
      docId: { default: "" },
      data: { default: defaultMindmapData },
    },
    content: "none",
  } as unknown as BlockConfig<"mindmap", PropSchema, "none">,
  {
    render: (props) => (
      <MindmapBlockView {...(props as unknown as MindmapBlockViewProps)} />
    ),
  },
);