mnote/wolai-frontend/src/app/(app)/layout.tsx

import { redirect } from "next/navigation";
import type { ReactNode } from "react";
import { Sidebar } from "@/components/sidebar/sidebar";
import { Breadcrumb } from "@/components/breadcrumb";
import { BottomToolbar } from "@/components/bottom-toolbar";
import { MobileSidebarTrigger } from "@/components/mobile-sidebar-trigger";
import type { DocumentRecord } from "@/lib/documents";
import { createSupabaseServerClient } from "@/lib/supabase/server";
import { ensureDefaultWorkspace, fetchWorkspaceSummaries } from "@/lib/workspaces";
import { fetchSidebarDataset } from "@/lib/sidebar-tree";
import type { SidebarInitialData } from "@/components/sidebar/types";
import { SearchPalette } from "@/components/search/search-palette";
import { isConvexEnabled } from "@/lib/convex/enabled";
import { requireAuthContext } from "@/lib/auth/authContext";
import { getConvexHttpClient } from "@/lib/convex/server";
import { api } from "@/lib/convex/api";
import type { MediaAsset } from "@/types/media";

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;
    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");
    push(get(data, "image"));
    push(image);
    push(get(image, "url"));
    push(get(get(data, "image"), "url"));
    const children = get(node, "children");
    if (Array.isArray(children)) children.forEach(walk);
  };

  walk(root);
  return ids;
}

function makeId(): string {
  return typeof crypto.randomUUID === "function"
    ? crypto.randomUUID()
    : `${Date.now()}_${Math.random().toString(16).slice(2)}`;
}

export default async function AppLayout({ children }: { children: ReactNode }) {
  if (isConvexEnabled()) {
    const auth = await requireAuthContext();
    const client = getConvexHttpClient();

    const ensured = await client.mutation(api.workspaces.ensureDefaultWorkspace, {
      userId: auth.userId,
      fallbackName: auth.name ?? auth.email ?? "我的空间",
      workspaceIdIfCreate: makeId(),
    });

    const workspaces = ensured.workspaces;
    const activeWorkspaceId = ensured.activeWorkspaceId;

    let documents: DocumentRecord[] = [];
    let sidebarInitialData: SidebarInitialData | null = null;

    if (activeWorkspaceId) {
      const [docRows, trashedDocs] = await Promise.all([
        client.query(api.documents.listByWorkspace, {
          userId: auth.userId,
          workspaceId: activeWorkspaceId,
        }),
        client.query(api.documents.listTrashedByWorkspace, {
          userId: auth.userId,
          workspaceId: activeWorkspaceId,
        }),
      ]);

      documents = docRows as unknown as DocumentRecord[];

      const mindmapRows = await client.query(api.mindmaps.listByWorkspace, {
        userId: auth.userId,
        workspaceId: activeWorkspaceId,
        includeDeleted: true,
      });

      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 ?? activeWorkspaceId,
          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 ?? activeWorkspaceId,
          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 ?? "",
        };
      });

      sidebarInitialData = {
        activeWorkspaceId,
        workspaces,
        documents: docRows as unknown as DocumentRecord[],
        trashedDocuments: trashedDocs as unknown as SidebarInitialData["trashedDocuments"],
        trashedMediaAssets: [],
        trashedMindmapAssets,
        mediaAssets: [],
        mindmapDocs,
        mindmapAssets,
        mindmapAssetChildren,
      };
    }

    return (
      <div className="flex h-screen bg-white">
        {sidebarInitialData && <Sidebar initialData={sidebarInitialData} />}
        <div className="flex min-w-0 flex-1 flex-col">
          <header className="flex h-10 items-center gap-4 border-b border-[#eeeeee] px-4">
            <MobileSidebarTrigger />
            <Breadcrumb documents={documents} />
          </header>
          <main className="flex-1 overflow-hidden bg-white">{children}</main>
          <BottomToolbar />
          <SearchPalette workspaceId={sidebarInitialData?.activeWorkspaceId ?? null} />
        </div>
      </div>
    );
  }

  const supabase = await createSupabaseServerClient();
  const {
    data: { session },
  } = await supabase.auth.getSession();

  if (!session) {
    redirect("/login");
  }

  await ensureDefaultWorkspace(supabase, session.user.id, session.user.email ?? "我的空间");
  const { workspaces, activeWorkspaceId } = await fetchWorkspaceSummaries(supabase, session.user.id);

  let documents: DocumentRecord[] = [];
  let sidebarInitialData: SidebarInitialData | null = null;

  if (activeWorkspaceId) {
    const dataset = await fetchSidebarDataset(supabase, activeWorkspaceId);     
    documents = dataset.documents;
    const localMindmaps = await detectLocalMindmapDocs(
      dataset.documents.map((d) => d.id),
    );
    const mindmapDocs = Array.from(
      new Set([...(dataset.mindmapDocs ?? []), ...localMindmaps]),
    );
    const trashedMindmapAssets = await detectLocalTrashedMindmapAssets(
      activeWorkspaceId,
      dataset.documents.map((d) => d.id),
    );
    sidebarInitialData = {
      activeWorkspaceId,
      workspaces,
      documents: dataset.documents,
      trashedDocuments: dataset.trashedDocuments,
      trashedMediaAssets: dataset.trashedMediaAssets ?? [],
      trashedMindmapAssets,
      mediaAssets: dataset.mediaAssets,
      mindmapDocs,
    };
  }

  return (
    <div className="flex h-screen bg-white">
      {sidebarInitialData && <Sidebar initialData={sidebarInitialData} />}
      <div className="flex min-w-0 flex-1 flex-col">
        <header className="flex h-10 items-center gap-4 border-b border-[#eeeeee] px-4">
          <MobileSidebarTrigger />
          <Breadcrumb documents={documents} />
        </header>
        <main className="flex-1 overflow-hidden bg-white">{children}</main>
        <BottomToolbar />
        <SearchPalette workspaceId={sidebarInitialData?.activeWorkspaceId ?? null} />
      </div>
    </div>
  );
}