mnote/wolai-frontend/src/lib/server/sidebar-data.ts

import { randomUUID } from "crypto";
import type { ConvexHttpClient } from "convex/browser";
import type { FunctionReference } from "convex/server";
import type { SidebarInitialData } from "@/components/sidebar/types";
import type { DocumentRecord } from "@/lib/documents";
import { api } from "@/lib/convex/api";
import {
  mapSidebarDatasetListQueryResultToInitialData,
  type SidebarDatasetListQueryResult,
} from "@/lib/sidebar-data";
import type { WorkspaceSummary } from "@/lib/workspaces";

type LoadSidebarDataFromConvexInput = {
  client: ConvexHttpClient;
  fallbackName: string;
  requestedWorkspaceId?: string | null;
};

type LoadSidebarDataFromConvexResult = {
  workspaces: WorkspaceSummary[];
  activeWorkspaceId: string;
  targetWorkspaceId: string | null;
  sidebarDataset: SidebarDatasetListQueryResult | null;
  sidebarInitialData: SidebarInitialData | null;
  documents: DocumentRecord[];
};

const sidebarDatasetListQuery = ((api as unknown as Record<string, unknown>).sidebar as
  | Record<string, unknown>
  | undefined)?.datasetList as FunctionReference<
  "query",
  "public",
  { workspaceId: string },
  SidebarDatasetListQueryResult
>;

export async function loadSidebarDataFromConvex(
  input: LoadSidebarDataFromConvexInput,
): Promise<LoadSidebarDataFromConvexResult> {
  const bootstrap = await input.client.mutation(api.workspaces.ensureDefaultWorkspace, {
    fallbackName: input.fallbackName,
    workspaceIdIfCreate: randomUUID(),
  });

  const workspaces = bootstrap.workspaces;
  const activeWorkspaceId = bootstrap.activeWorkspaceId;
  const targetWorkspaceId = input.requestedWorkspaceId?.trim() || activeWorkspaceId || null;

  if (!targetWorkspaceId) {
    return {
      workspaces,
      activeWorkspaceId,
      targetWorkspaceId: null,
      sidebarDataset: null,
      sidebarInitialData: null,
      documents: [],
    };
  }

  const sidebarDataset = (await input.client.query(sidebarDatasetListQuery, {
    workspaceId: targetWorkspaceId,
  })) as SidebarDatasetListQueryResult;
  const normalizedDocuments = (sidebarDataset.documents ?? []) as DocumentRecord[];

  return {
    workspaces: sidebarDataset.workspaces ?? workspaces,
    activeWorkspaceId,
    targetWorkspaceId,
    sidebarDataset,
    sidebarInitialData: mapSidebarDatasetListQueryResultToInitialData(sidebarDataset),
    documents: normalizedDocuments,
  };
}