0.6 rust重构01

wolai-frontend/src/app/api/sidebar/route.ts

@@ -1,263 +1,58 @@
 import { NextResponse } from "next/server";
-import type { SidebarInitialData } from "@/components/sidebar/types";
 import { isConvexEnabled } from "@/lib/convex/enabled";
-import { api } from "@/lib/convex/api";
 import { getAuthedConvexClient } from "@/lib/convex/route";
-import { randomUUID } from "crypto";
-import type { MediaAsset } from "@/types/media";
+import {
+  buildDocumentBridgeContext,
+  buildDocumentQueryEnvelope,
+  documentBridgeErrorResponse,
+} from "@/lib/documents/bridge";
+import { loadSidebarDataFromConvex } from "@/lib/server/sidebar-data";
 
 export const dynamic = "force-dynamic";
 
-function extractMindmapImageAssetIdsFromData(input: unknown): string[] {
-  const root = (() => {
-    if (!input || typeof input !== "object") return input;
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const record = input as any;
-    // 兼容：某些导图结构为 { root: ... }
-    return record && typeof record === "object" && "root" in record ? record.root : input;
-  })();
-
-  const ids: string[] = [];
-  const seen = new Set<string>();
-
-  const push = (value: unknown) => {
-    if (typeof value !== "string") return;
-    if (!value.startsWith("asset:")) return;
-    const id = value.slice("asset:".length).trim();
-    if (!id) return;
-    if (seen.has(id)) return;
-    seen.add(id);
-    ids.push(id);
-  };
-
-  const get = (obj: unknown, key: string): unknown => {
-    if (!obj || typeof obj !== "object") return undefined;
-    return (obj as Record<string, unknown>)[key];
-  };
-
-  const walk = (node: unknown) => {
-    if (!node || typeof node !== "object") return;
-
-    const data = get(node, "data");
-    const image = get(node, "image");
-
-    // 常见：node.data.image = "asset:xxx"
-    push(get(data, "image"));
-    // 兼容：node.image = "asset:xxx"
-    push(image);
-    // 兼容：node.image.url = "asset:xxx"
-    push(get(image, "url"));
-    // 兼容：node.data.image.url = "asset:xxx"
-    push(get(get(data, "image"), "url"));
-
-    const children = get(node, "children");
-    if (Array.isArray(children)) {
-      children.forEach(walk);
-    }
-  };
-
-  walk(root);
-  return ids;
-}
-
 export async function GET(request: Request) {
   if (isConvexEnabled()) {
     const { auth, client } = await getAuthedConvexClient();
 
     const url = new URL(request.url);
     const workspaceIdParam = url.searchParams.get("workspaceId");
-
-    const bootstrap = await client.mutation(api.workspaces.ensureDefaultWorkspace, {
+    const {
+      targetWorkspaceId,
+      sidebarInitialData,
+    } = await loadSidebarDataFromConvex({
+      client,
+      userId: auth.userId,
       fallbackName: auth.email ?? auth.name ?? "我的空间",
-      workspaceIdIfCreate: randomUUID(),
+      requestedWorkspaceId: workspaceIdParam,
     });
 
-    const summaries = await client.query(api.workspaces.fetchWorkspaceSummaries, {
-    });
-
-    const workspaces = summaries.workspaces.length > 0 ? summaries.workspaces : bootstrap.workspaces;
-    const activeWorkspaceId = summaries.activeWorkspaceId || bootstrap.activeWorkspaceId;
-    const targetWorkspaceId = workspaceIdParam || activeWorkspaceId;
-
     if (!targetWorkspaceId) {
       return NextResponse.json({ error: "暂无可用工作空间" }, { status: 404 });
     }
 
     try {
-      const documents = await client.query(api.documents.listByWorkspace, {
+      const bridgeContext = await buildDocumentBridgeContext({
+        request,
         workspaceId: targetWorkspaceId,
       });
-
-      const trashedDocuments = await client.query(api.documents.listTrashedByWorkspace, {
-        workspaceId: targetWorkspaceId,
+      const envelope = buildDocumentQueryEnvelope({
+        name: "sidebar.dataset.list",
+        payload: { workspaceId: targetWorkspaceId },
       });
+      if (!sidebarInitialData) {
+        return NextResponse.json({ error: "暂无可用工作空间" }, { status: 404 });
+      }
 
-      const mindmapRows = await client.query(api.mindmaps.listByWorkspace, {
-        workspaceId: targetWorkspaceId,
-        includeDeleted: true,
+      return NextResponse.json({
+        ...sidebarInitialData,
+        meta: {
+          requestId: bridgeContext.requestId,
+          traceId: bridgeContext.traceId,
+          queryName: envelope.name,
+        },
       });
-
-      const mediaAssets = await client.query(api.mediaAssets.listByWorkspace, {
-        userId: auth.userId,
-        workspaceId: targetWorkspaceId,
-        limit: 200,
-      });
-
-      const trashedMediaAssets = await client.query(api.mediaAssets.listDeletedByWorkspace, {
-        userId: auth.userId,
-        workspaceId: targetWorkspaceId,
-        limit: 2000,
-      });
-
-      const tables = await client.query(api.tables.listByWorkspaceForSearch, {
-        userId: auth.userId,
-        workspaceId: targetWorkspaceId,
-        includeArchived: true,
-        limit: 3000,
-      });
-
-      const activeMindmaps = (mindmapRows ?? []).filter((r) => !r.deleted_at);
-      const trashedMindmaps = (mindmapRows ?? []).filter((r) => !!r.deleted_at);
-
-      const mindmapDocs = Array.from(new Set(activeMindmaps.map((r) => r.document_id)));
-
-      const mindmapAssetChildren: Record<string, string[]> = {};
-      activeMindmaps.forEach((r) => {
-        const ids = extractMindmapImageAssetIdsFromData(r.data);
-        if (ids.length > 0) {
-          mindmapAssetChildren[r.mindmap_id] = ids;
-        }
-      });
-
-      const mindmapAssets: MediaAsset[] = activeMindmaps.map((r) => {
-        const isLegacy = r.mindmap_id.startsWith("legacy-");
-        return {
-          id: r.mindmap_id,
-          workspace_id: r.workspace_id ?? targetWorkspaceId,
-          document_id: r.document_id,
-          asset_type: "mindmap",
-          file_url: null,
-          thumbnail_url: null,
-          bucket: null,
-          storage_path: null,
-          file_name: isLegacy ? "mindmap.json" : `mindmap-${r.mindmap_id}.json`,
-          file_size: null,
-          mime_type: "application/json",
-          ocr_payload: undefined,
-          ocr_strategy: null,
-          ocr_text: null,
-          ocr_status: null,
-          signed_url: null,
-          created_at: r.created_at ?? "",
-          updated_at: r.updated_at ?? "",
-        };
-      });
-
-      const trashedMindmapAssets: MediaAsset[] = trashedMindmaps.map((r) => {
-        const isLegacy = r.mindmap_id.startsWith("legacy-");
-        return {
-          id: r.mindmap_id,
-          workspace_id: r.workspace_id ?? targetWorkspaceId,
-          document_id: r.document_id,
-          asset_type: "mindmap",
-          file_url: null,
-          thumbnail_url: null,
-          bucket: null,
-          storage_path: null,
-          file_name: isLegacy ? "mindmap.json" : `mindmap-${r.mindmap_id}.json`,
-          file_size: null,
-          mime_type: "application/json",
-          ocr_payload: undefined,
-          ocr_strategy: null,
-          ocr_text: null,
-          ocr_status: null,
-          deleted_at: r.deleted_at ?? null,
-          deleted_by: r.deleted_by ?? null,
-          purged_at: null,
-          signed_url: null,
-          created_at: r.created_at ?? "",
-          updated_at: r.updated_at ?? "",
-        };
-      });
-
-      const tableAssets: MediaAsset[] = (tables ?? [])
-        .filter((row) => !row.is_archived)
-        .map((row) => {
-        const base = String(row.title ?? "未命名表格").trim() || "未命名表格";
-        const fileName = base.toLowerCase().endsWith(".luckysheet") ? base : `${base}.luckysheet`;
-        return {
-          id: row.id,
-          workspace_id: row.workspace_id ?? targetWorkspaceId,
-          document_id: row.document_id,
-          asset_type: "luckysheet",
-          file_url: null,
-          thumbnail_url: null,
-          bucket: null,
-          storage_path: null,
-          file_name: fileName,
-          file_size: null,
-          mime_type: "application/json",
-          ocr_payload: undefined,
-          ocr_strategy: null,
-          ocr_text: null,
-          ocr_status: null,
-          signed_url: null,
-          created_at: row.created_at ?? "",
-          updated_at: row.updated_at ?? "",
-        };
-      });
-
-      const trashedTableAssets: MediaAsset[] = (tables ?? [])
-        .filter((row) => Boolean(row.is_archived))
-        .map((row) => {
-          const base = String(row.title ?? "未命名表格").trim() || "未命名表格";
-          const fileName = base.toLowerCase().endsWith(".luckysheet") ? base : `${base}.luckysheet`;
-          return {
-            id: row.id,
-            workspace_id: row.workspace_id ?? targetWorkspaceId,
-            document_id: row.document_id,
-            asset_type: "luckysheet",
-            file_url: null,
-            thumbnail_url: null,
-            bucket: null,
-            storage_path: null,
-            file_name: fileName,
-            file_size: null,
-            mime_type: "application/json",
-            ocr_payload: undefined,
-            ocr_strategy: null,
-            ocr_text: null,
-            ocr_status: null,
-            deleted_at: row.deleted_at ?? row.updated_at ?? null,
-            deleted_by: row.deleted_by ?? null,
-            purged_at: row.purged_at ?? null,
-            signed_url: null,
-            created_at: row.created_at ?? "",
-            updated_at: row.updated_at ?? "",
-          };
-        });
-
-      const payload: SidebarInitialData = {
-        activeWorkspaceId: targetWorkspaceId,
-        workspaces,
-        documents,
-        trashedDocuments,
-        trashedMediaAssets: (trashedMediaAssets ?? []) as MediaAsset[],
-        trashedMindmapAssets,
-        trashedTableAssets,
-        mindmapDocs,
-        mindmapAssets,
-        mindmapAssetChildren,
-        tableAssets,
-        mediaAssets: (mediaAssets ?? []) as MediaAsset[],
-      };
-
-      return NextResponse.json(payload);
     } catch (error) {
-      return NextResponse.json(
-        { error: error instanceof Error ? error.message : "拉取侧边栏数据失败" },
-        { status: 500 },
-      );
+      return documentBridgeErrorResponse(error);
     }
   }
 
@@ -268,22 +63,22 @@ export async function GET(request: Request) {
   const {
     data: { session },
   } = await supabase.auth.getSession();
-
-  if (!session) {
-    return NextResponse.json({ error: "未登录" }, { status: 401 });
-  }
-
-  const url = new URL(request.url);
-  const workspaceIdParam = url.searchParams.get("workspaceId");
-
-  await ensureDefaultWorkspace(supabase, session.user.id, session.user.email ?? "我的空间");
-  const { workspaces, activeWorkspaceId } = await fetchWorkspaceSummaries(supabase, session.user.id);
-  const targetWorkspaceId = workspaceIdParam || activeWorkspaceId;
-
-  if (!targetWorkspaceId) {
-    return NextResponse.json({ error: "暂无可用工作空间" }, { status: 404 });
-  }
-
+
+  if (!session) {
+    return NextResponse.json({ error: "未登录" }, { status: 401 });
+  }
+
+  const url = new URL(request.url);
+  const workspaceIdParam = url.searchParams.get("workspaceId");
+
+  await ensureDefaultWorkspace(supabase, session.user.id, session.user.email ?? "我的空间");
+  const { workspaces, activeWorkspaceId } = await fetchWorkspaceSummaries(supabase, session.user.id);
+  const targetWorkspaceId = workspaceIdParam || activeWorkspaceId;
+
+  if (!targetWorkspaceId) {
+    return NextResponse.json({ error: "暂无可用工作空间" }, { status: 404 });
+  }
+
   try {
     const dataset = await fetchSidebarDataset(supabase, targetWorkspaceId);
     const docIds = dataset.documents.map((d) => d.id);
@@ -360,10 +155,10 @@ export async function GET(request: Request) {
 
     return NextResponse.json(payload);
   } catch (error) {
-    return NextResponse.json(
-      { error: error instanceof Error ? error.message : "拉取侧边栏数据失败" },
-      { status: 500 },
-    );
-  }
+    return NextResponse.json(
+      { error: error instanceof Error ? error.message : "拉取侧边栏数据失败" },
+      { status: 500 },
+    );
+  }
   */
 }