0.3.6 修复shitf选择问题

wolai-frontend/src/components/sidebar/sidebar.tsx

@@ -46,7 +46,7 @@ import { useCurrentDocumentStore } from "@/store/current-document";
 import { FileTree } from "@/components/sidebar/file-tree";
 import { buildVisibleRows } from "@/lib/file-tree/rows";
 import type { FileTreeSelectionState } from "@/lib/file-tree/selection";  
-import { reduceFileTreeSelection } from "@/lib/file-tree/selection";      
+import { normalizeFileTreeSelectionForVisibleRows, reduceFileTreeSelection } from "@/lib/file-tree/selection";      
 import type { FileTreeRow } from "@/lib/file-tree/types";
 import { buildParentById, filterTopLevelDocIds, inferDropTargetDocId, isInvalidDocDrop } from "@/lib/file-tree/dnd";
 import { isRealFileAsset } from "@/lib/file-tree/asset";
@@ -260,10 +260,6 @@ function SidebarContent({ initialData, sidebarQuery }: SidebarContentProps) {
     setTableAssets(sidebarData.tableAssets ?? []);
   }, [sidebarData.tableAssets]);
 
-  useEffect(() => {
-    setFileTreeSelection({ selectedRowIds: new Set(), anchorRowId: null, focusedRowId: null });
-  }, [mediaAssets, mindmapAssets, tableAssets, sidebarData.documents]);
-
   useEffect(() => {
     setOpen(false);
   }, [activeId, setOpen]);
@@ -566,6 +562,10 @@ function SidebarContent({ initialData, sidebarQuery }: SidebarContentProps) {
   const fileTreeVisibleRowIds = useMemo(() => fileTreeRows.map((row) => row.rowId), [fileTreeRows]);
   const fileTreeRowById = useMemo(() => new Map(fileTreeRows.map((row) => [row.rowId, row])), [fileTreeRows]);
 
+  useEffect(() => {
+    setFileTreeSelection((prev) => normalizeFileTreeSelectionForVisibleRows(prev, fileTreeVisibleRowIds));
+  }, [fileTreeVisibleRowIds]);
+
   const docParentById = useMemo(
     () =>
       buildParentById(
@@ -1703,6 +1703,61 @@ function SidebarContent({ initialData, sidebarQuery }: SidebarContentProps) {
     [activeId, refreshTree, router],
   );
 
+  const handleDeleteFromContextMenuNode = useCallback(
+    async (node: DocumentNode) => {
+      if (viewMode === "filesystem") {
+        await handleDeleteFileTreeSelection();
+        return;
+      }
+
+      const ok = window.confirm("确认删除该页面到垃圾桶吗？");
+      if (!ok) return;
+
+      try {
+        await handleDelete(node.id);
+      } catch (error) {
+        window.alert(error instanceof Error ? error.message : "删除失败");
+      }
+    },
+    [handleDelete, handleDeleteFileTreeSelection, viewMode],
+  );
+
+  const handleDeleteFromAssetContextMenu = useCallback(
+    async (assetIds: string[], assetHint?: MediaAsset) => {
+      const uniqueAssetIds = Array.from(new Set(assetIds));
+      if (uniqueAssetIds.length === 0) return;
+
+      const assets = uniqueAssetIds
+        .map((id) => mediaAssets.find((item) => item.id === id) ?? mindmapAssets.find((item) => item.id === id) ?? tableAssets.find((item) => item.id === id))
+        .filter(Boolean) as MediaAsset[];
+      if (assetHint && !assets.find((item) => item.id === assetHint.id)) {
+        assets.unshift(assetHint);
+      }
+
+      const mindmapCount = assets.filter((item) => item.asset_type === "mindmap").length;
+      const tableCount = assets.filter((item) => item.asset_type === "luckysheet").length;
+      const fileCount = assets.filter((item) => item.asset_type !== "mindmap" && item.asset_type !== "luckysheet").length;
+      const unknownCount = Math.max(0, uniqueAssetIds.length - mindmapCount - tableCount - fileCount);
+
+      const parts: string[] = [];
+      if (fileCount > 0) parts.push(`${fileCount} 个附件（删除，10 分钟内可撤销）`);
+      if (mindmapCount > 0) parts.push(`${mindmapCount} 个思维导图（删除）`);
+      if (tableCount > 0) parts.push(`${tableCount} 个在线表格（删除）`);
+      if (unknownCount > 0) parts.push(`${unknownCount} 个对象（删除）`);
+
+      const ok = window.confirm(`确认删除选中的 ${parts.join(" + ")} 吗？`);
+      if (!ok) return;
+
+      try {
+        await handleDeleteAssets(uniqueAssetIds, assetHint);
+        setFileTreeSelection((prev) => reduceFileTreeSelection(prev, { type: "clear" }));
+      } catch (error) {
+        window.alert(error instanceof Error ? error.message : "删除失败");
+      }
+    },
+    [handleDeleteAssets, mediaAssets, mindmapAssets, tableAssets],
+  );
+
   const handleConvertToChild = useCallback(
     async (documentId: string) => {
       const flat = flattenedPrivate;
@@ -2373,7 +2428,7 @@ function SidebarContent({ initialData, sidebarQuery }: SidebarContentProps) {
           onRename={() => void handleRename(contextMenu.node.id, contextMenu.node.title)}
           onCreateChild={() => void handleCreate(contextMenu.node.id)}
           onConvertChild={() => void handleConvertToChild(contextMenu.node.id)}
-          onDelete={() => void handleDeleteFileTreeSelection()}
+          onDelete={() => void handleDeleteFromContextMenuNode(contextMenu.node)}
         />
       )}
       {shareTarget && (
@@ -2449,7 +2504,7 @@ function SidebarContent({ initialData, sidebarQuery }: SidebarContentProps) {
           onCopyPath={handleCopyAssetPath}
           onRename={handleRenameAsset}
           onMove={handleMoveAsset}
-          onDelete={() => void handleDeleteFileTreeSelection()}
+          onDelete={(assetIds) => void handleDeleteFromAssetContextMenu(assetIds, assetMenu.asset)}
           onDownload={handleDownloadAsset}
         />
       )}