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

import { NextResponse } from "next/server";
import { createSupabaseRouteClient } from "@/lib/supabase/server";
import { ensureDefaultWorkspace, fetchWorkspaceSummaries } from "@/lib/workspaces";
import { fetchSidebarDataset } from "@/lib/sidebar-tree";
import type { SidebarInitialData } from "@/components/sidebar/types";
import {
  detectLocalMindmapFiles,
  detectLocalMindmapDocs,
  detectLocalMindmapImageAssetIdsByMindmapId,
  detectLocalTrashedMindmapAssets,
} from "@/lib/mindmap-files";
import type { MediaAsset } from "@/types/media";

export const dynamic = "force-dynamic";

export async function GET(request: Request) {
  const supabase = await createSupabaseRouteClient();
  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 });
  }

  try {
    const dataset = await fetchSidebarDataset(supabase, targetWorkspaceId);
    const docIds = dataset.documents.map((d) => d.id);
    const localMindmapFiles = await detectLocalMindmapFiles(docIds);
    const mindmapAssetChildren =
      await detectLocalMindmapImageAssetIdsByMindmapId(localMindmapFiles);
    const mindmapDocs = Array.from(new Set([...(dataset.mindmapDocs ?? []), ...(await detectLocalMindmapDocs(docIds))]));
    const trashedMindmapAssets = await detectLocalTrashedMindmapAssets(targetWorkspaceId, docIds);
    const docById = new Map(dataset.documents.map((d) => [d.id, d]));
    const mindmapAssets: MediaAsset[] = localMindmapFiles.map((item) => {       
      const doc = docById.get(item.documentId);
      const workspaceId = doc?.workspace_id ?? targetWorkspaceId;
      const fileUrlBase = item.source === "legacy" ? `/mindmaps/${item.documentId}` : `/documents/${item.documentId}`;
      return {
        id: item.mindmapId,
        workspace_id: workspaceId,
        document_id: item.documentId,
        asset_type: "mindmap",
        file_url: `${fileUrlBase}/${item.fileName}`,
        thumbnail_url: null,
        bucket: null,
        storage_path: null,
        file_name: item.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: "",
        updated_at: "",
      };
    });

    const tableAssets: MediaAsset[] = (dataset.tables ?? []).map((row) => {
      const base = (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: `/tables/${row.id}/view`,
        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 payload: SidebarInitialData = {
      activeWorkspaceId: targetWorkspaceId,
      workspaces,
      documents: dataset.documents,
      trashedDocuments: dataset.trashedDocuments,
      trashedMediaAssets: dataset.trashedMediaAssets ?? [],
      trashedMindmapAssets,
      mindmapDocs,
      mindmapAssets,
      mindmapAssetChildren,
      tableAssets,
      mediaAssets: dataset.mediaAssets ?? [],
    };

    return NextResponse.json(payload);
  } catch (error) {
    return NextResponse.json(
      { error: error instanceof Error ? error.message : "拉取侧边栏数据失败" },
      { status: 500 },
    );
  }
}