Function ChatWithDocs

apps/web/lib/fe/components/chat-with-docs.tsx:30–261

({
  chat,
  chatMessages,
}: {
  chat: ChatResponse;
  chatMessages: ChatMessageResponse[];
})

import { Id, DocumentCollectionResponse, DocumentResponse, DocumentToCollectionResponse } from "@repo/core";

export function ChatWithDocs({
  chat,
  chatMessages,
}: {
  chat: ChatResponse;
  chatMessages: ChatMessageResponse[];
}) {
  const [previewDocumentId, setPreviewDocumentId] = useState<
    string | undefined
  >();

  // A state that doesn't need to re-render component. So it's never set -- only values are modified on
  // appropriate events.
  const [documentsCurrentPageIndexMap, _] = useState<Map<string, number>>(
    new Map(),
  );

  const pageNavigationPluginInstance = pageNavigationPlugin();
  const { jumpToPage } = pageNavigationPluginInstance;

  const documentCollectionId = Id.from<DocumentCollectionResponse>(
    chat.documentCollectionId!,
  );

  // Fetch documents
  const {
    data: collectionDocumentsResponse,
    error: fetchCollectionDocumentsError,
  } = useSWR(
    getDocumentCollectionDocumentsApiPath({
      documentCollectionId: documentCollectionId,
      pagination: {
        // TODO: Allow viewing beyond first 1024 docs in a collection! Scaling problem for later
        page: 1,
        pageSize: 1024,
      },
      ordering: {
        orderBy: "createdAt",
        order: "desc",
      },
    }),
    createFetcher<DocumentResponse[]>(),
    {
      revalidateOnFocus: false,
    },
  );

  const shouldFetchDocumentToCollections = collectionDocumentsResponse && collectionDocumentsResponse.response.length > 0;
  const {
    data: documentToCollectionsResponse,
    error: documentToCollectionsFetchError,
  } = useSWR(
    shouldFetchDocumentToCollections
      ? getDocumentToCollections({
          documentCollectionId: documentCollectionId,
          documentIds: collectionDocumentsResponse!.response.map(d => Id.from(d.id)),
        })
      : null,