Introduce slim documents

We introduce slim documents, it's a lighter version of the document
computed in GitBook Open to lightweight to document stored in cache and
transferred to client. The idea is to store in this document only what
is needed to display a document in GitBook Open.

Author: gregberge

Diff for: packages/gitbook-v2/src/lib/data/api.ts

@@ -1,3 +1,4 @@
+import { getSlimJSONDocument } from '@/lib/slim-document';
 import { trace } from '@/lib/tracing';
 import {
     type ComputedContentSource,
@@ -779,7 +780,7 @@ const getDocumentUncached = async (
                 cacheTag(...getCacheTagsFromResponse(res));
                 cacheLife('max');
             }
-            return res.data;
+            return getSlimJSONDocument(res.data);
         });
     });
 };
@@ -872,7 +873,7 @@ const getComputedDocumentUncached = async (
                     cacheTag(...getCacheTagsFromResponse(res));
                     cacheLife('max');
                 }
-                return res.data;
+                return getSlimJSONDocument(res.data);
             });
         }
     );

Diff for: packages/gitbook-v2/src/lib/data/pages.ts

@@ -1,4 +1,5 @@
-import type { JSONDocument, RevisionPageDocument, Space } from '@gitbook/api';
+import type { SlimJSONDocument } from '@/lib/slim-document';
+import type { RevisionPageDocument, Space } from '@gitbook/api';
 import { getDataOrNull } from './errors';
 import type { GitBookDataFetcher } from './types';
 
@@ -9,7 +10,7 @@ export async function getPageDocument(
     dataFetcher: GitBookDataFetcher,
     space: Space,
     page: RevisionPageDocument
-): Promise<JSONDocument | null> {
+): Promise<SlimJSONDocument | null> {
     if (page.documentId) {
         return getDataOrNull(
             dataFetcher.getDocument({ spaceId: space.id, documentId: page.documentId })

Diff for: packages/gitbook-v2/src/lib/data/types.ts

@@ -1,3 +1,4 @@
+import type { SlimJSONDocument } from '@/lib/slim-document';
 import type * as api from '@gitbook/api';
 
 export type DataFetcherErrorData = {
@@ -110,7 +111,7 @@ export interface GitBookDataFetcher {
      * Get a document by its space ID and document ID.
      */
     getDocument(params: { spaceId: string; documentId: string }): Promise<
-        DataFetcherResponse<api.JSONDocument>
+        DataFetcherResponse<SlimJSONDocument>
     >;
 
     /**
@@ -121,7 +122,7 @@ export interface GitBookDataFetcher {
         spaceId: string;
         source: api.ComputedContentSource;
         seed: string;
-    }): Promise<DataFetcherResponse<api.JSONDocument>>;
+    }): Promise<DataFetcherResponse<SlimJSONDocument>>;
 
     /**
      * Get a reusable content by its space ID, revision ID and reusable content ID.

Diff for: packages/gitbook/src/components/DocumentView/Block.tsx

@@ -1,4 +1,3 @@
-import type { DocumentBlock, JSONDocument } from '@gitbook/api';
 import React from 'react';
 
 import {
@@ -10,6 +9,7 @@ import {
 } from '@/components/primitives';
 import type { ClassValue } from '@/lib/tailwind';
 
+import type { SlimDocumentBlock, SlimJSONDocument } from '@/lib/slim-document';
 import { BlockContentRef } from './BlockContentRef';
 import { CodeBlock } from './CodeBlock';
 import { Divider } from './Divider';
@@ -34,10 +34,10 @@ import { StepperStep } from './StepperStep';
 import { Table } from './Table';
 import { Tabs } from './Tabs';
 
-export interface BlockProps<Block extends DocumentBlock> extends DocumentContextProps {
+export interface BlockProps<Block extends SlimDocumentBlock> extends DocumentContextProps {
     block: Block;
-    document: JSONDocument;
-    ancestorBlocks: DocumentBlock[];
+    document: SlimJSONDocument;
+    ancestorBlocks: SlimDocumentBlock[];
     /** If true, we estimate that the block will be outside the initial viewport */
     isEstimatedOffscreen: boolean;
     /** Class names to be passed to the underlying DOM element */
@@ -51,7 +51,7 @@ function nullIfNever(_value: never): null {
     return null;
 }
 
-export function Block<T extends DocumentBlock>(props: BlockProps<T>) {
+export function Block<T extends SlimDocumentBlock>(props: BlockProps<T>) {
     const { block, style, isEstimatedOffscreen, context } = props;
 
     const content = (() => {
@@ -132,7 +132,7 @@ export function Block<T extends DocumentBlock>(props: BlockProps<T>) {
 /**
  * Skeleton for a block while it is being loaded.
  */
-export function BlockSkeleton(props: { block: DocumentBlock; style: ClassValue }) {
+export function BlockSkeleton(props: { block: SlimDocumentBlock; style: ClassValue }) {
     const { block, style } = props;
     const id = 'meta' in block && block.meta && 'id' in block.meta ? block.meta.id : undefined;
 

Diff for: packages/gitbook/src/components/DocumentView/Blocks.tsx

@@ -1,15 +1,14 @@
-import type { DocumentBlock, JSONDocument } from '@gitbook/api';
-
 import { type ClassValue, tcls } from '@/lib/tailwind';
 
+import type { SlimDocumentBlock, SlimJSONDocument } from '@/lib/slim-document';
 import { Block } from './Block';
 import type { DocumentContextProps } from './DocumentView';
 import { isBlockOffscreen } from './utils';
 
 /**
  * Renders a list of blocks with a wrapper element.
  */
-export function Blocks<TBlock extends DocumentBlock, Tag extends React.ElementType = 'div'>(
+export function Blocks<TBlock extends SlimDocumentBlock, Tag extends React.ElementType = 'div'>(
     props: UnwrappedBlocksProps<TBlock> & {
         /** HTML tag to use for the wrapper */
         tag?: Tag;
@@ -30,15 +29,15 @@ export function Blocks<TBlock extends DocumentBlock, Tag extends React.ElementTy
     );
 }
 
-type UnwrappedBlocksProps<TBlock extends DocumentBlock> = DocumentContextProps & {
+type UnwrappedBlocksProps<TBlock extends SlimDocumentBlock> = DocumentContextProps & {
     /** Blocks to render */
     nodes: TBlock[];
 
     /** Document being rendered */
-    document: JSONDocument;
+    document: SlimJSONDocument;
 
     /** Ancestors of the blocks */
-    ancestorBlocks: DocumentBlock[];
+    ancestorBlocks: SlimDocumentBlock[];
 
     /** Style passed to all blocks */
     blockStyle?: ClassValue;
@@ -50,7 +49,9 @@ type UnwrappedBlocksProps<TBlock extends DocumentBlock> = DocumentContextProps &
 /**
  * Renders a list of blocks without a wrapper element.
  */
-export function UnwrappedBlocks<TBlock extends DocumentBlock>(props: UnwrappedBlocksProps<TBlock>) {
+export function UnwrappedBlocks<TBlock extends SlimDocumentBlock>(
+    props: UnwrappedBlocksProps<TBlock>
+) {
     const { nodes, blockStyle, isOffscreen: defaultIsOffscreen = false, ...contextProps } = props;
 
     let isOffscreen = defaultIsOffscreen;
@@ -69,7 +70,7 @@ export function UnwrappedBlocks<TBlock extends DocumentBlock>(props: UnwrappedBl
                 block={node}
                 style={[
                     'mx-auto w-full decoration-primary/6',
-                    node.data && 'fullWidth' in node.data && node.data.fullWidth
+                    'data' in node && node.data && 'fullWidth' in node.data && node.data.fullWidth
                         ? 'max-w-screen-xl'
                         : 'max-w-3xl',
                     blockStyle,

Diff for: packages/gitbook/src/components/DocumentView/Caption.tsx

@@ -3,12 +3,12 @@ import type {
     DocumentBlockEmbed,
     DocumentBlockFile,
     DocumentBlockImage,
-    JSONDocument,
 } from '@gitbook/api';
 
 import { getNodeFragmentByName, isNodeEmpty } from '@/lib/document';
 import { type ClassValue, tcls } from '@/lib/tailwind';
 
+import type { SlimJSONDocument } from '@/lib/slim-document';
 import type { DocumentContextProps } from './DocumentView';
 import { Inlines } from './Inlines';
 
@@ -18,7 +18,7 @@ import { Inlines } from './Inlines';
 export function Caption(
     props: {
         children: React.ReactNode;
-        document: JSONDocument;
+        document: SlimJSONDocument;
         style?: ClassValue;
         fit?: boolean;
         wrapperStyle?: ClassValue;

Diff for: packages/gitbook/src/components/DocumentView/CodeBlock/ClientCodeBlock.tsx

@@ -1,19 +1,19 @@
 'use client';
 
-import type { DocumentBlockCode } from '@gitbook/api';
 import { useEffect, useMemo, useRef, useState } from 'react';
 
 import { useInViewportListener } from '@/components/hooks/useInViewportListener';
 import { useScrollListener } from '@/components/hooks/useScrollListener';
+import type { SlimDocumentBlockCode } from '@/lib/slim-document';
 import { useDebounceCallback } from 'usehooks-ts';
-import type { BlockProps } from '../Block';
-import { CodeBlockRenderer } from './CodeBlockRenderer';
+import { CodeBlockRenderer, type CodeBlockRendererProps } from './CodeBlockRenderer';
 import type { HighlightLine, RenderedInline } from './highlight';
 import { plainHighlight } from './plain-highlight';
 
-type ClientBlockProps = Pick<BlockProps<DocumentBlockCode>, 'block' | 'style'> & {
+interface ClientBlockProps extends Omit<CodeBlockRendererProps, 'lines'> {
+    block: SlimDocumentBlockCode;
     inlines: RenderedInline[];
-};
+}
 
 /**
  * Render a code-block client-side by loading the highlighter asynchronously.
@@ -24,13 +24,13 @@ export function ClientCodeBlock(props: ClientBlockProps) {
     const blockRef = useRef<HTMLDivElement>(null);
     const isInViewportRef = useRef(false);
     const [isInViewport, setIsInViewport] = useState(false);
-    const plainLines = useMemo(() => plainHighlight(block, []), [block]);
+    const plainLines = useMemo(() => plainHighlight({ block, inlines: [] }), [block]);
     const [lines, setLines] = useState<null | HighlightLine[]>(null);
 
     // Preload the highlighter when the block is mounted.
     useEffect(() => {
-        import('./highlight').then(({ preloadHighlight }) => preloadHighlight(block));
-    }, [block]);
+        import('./highlight').then(({ preloadHighlight }) => preloadHighlight(block.data.syntax));
+    }, [block.data.syntax]);
 
     // When user scrolls, we need to wait for the scroll to finish before running the highlight
     const isScrollingRef = useRef(false);
@@ -78,7 +78,7 @@ export function ClientCodeBlock(props: ClientBlockProps) {
 
             if (typeof window !== 'undefined') {
                 import('./highlight').then(({ highlight }) => {
-                    highlight(block, inlines).then((lines) => {
+                    highlight({ block, inlines }).then((lines) => {
                         if (cancelled) {
                             return;
                         }
@@ -98,6 +98,13 @@ export function ClientCodeBlock(props: ClientBlockProps) {
     }, [isInViewport, block, inlines]);
 
     return (
-        <CodeBlockRenderer ref={blockRef} block={block} style={style} lines={lines ?? plainLines} />
+        <CodeBlockRenderer
+            ref={blockRef}
+            title={props.title}
+            withLineNumbers={props.withLineNumbers}
+            withWrap={props.withWrap}
+            style={style}
+            lines={lines ?? plainLines}
+        />
     );
 }

Diff for: packages/gitbook/src/components/DocumentView/CodeBlock/CodeBlock.tsx

@@ -1,8 +1,7 @@
-import type { DocumentBlockCode } from '@gitbook/api';
-
 import { getNodeFragmentByType } from '@/lib/document';
 import { isV2 } from '@/lib/v2';
 
+import type { SlimDocumentBlockCode } from '@/lib/slim-document';
 import type { BlockProps } from '../Block';
 import { Blocks } from '../Blocks';
 import { ClientCodeBlock } from './ClientCodeBlock';
@@ -12,8 +11,11 @@ import { type RenderedInline, getInlines, highlight } from './highlight';
 /**
  * Render a code block, can be client-side or server-side.
  */
-export async function CodeBlock(props: BlockProps<DocumentBlockCode>) {
+export async function CodeBlock(props: BlockProps<SlimDocumentBlockCode>) {
     const { block, document, style, isEstimatedOffscreen, context } = props;
+    const withLineNumbers = Boolean(block.data.lineNumbers) && block.nodes.length > 1;
+    const withWrap = block.data.overflow === 'wrap';
+    const title = block.data.title ?? '';
     const inlines = getInlines(block);
     const richInlines: RenderedInline[] = inlines.map((inline, index) => {
         const body = (() => {
@@ -38,9 +40,29 @@ export async function CodeBlock(props: BlockProps<DocumentBlockCode>) {
 
     if (isV2() && !isEstimatedOffscreen) {
         // In v2, we render the code block server-side
-        const lines = await highlight(block, richInlines);
-        return <CodeBlockRenderer block={block} style={style} lines={lines} />;
+        const lines = await highlight({
+            inlines: richInlines,
+            block,
+        });
+        return (
+            <CodeBlockRenderer
+                withLineNumbers={withLineNumbers}
+                withWrap={withWrap}
+                title={title}
+                style={style}
+                lines={lines}
+            />
+        );
     }
 
-    return <ClientCodeBlock block={block} style={style} inlines={richInlines} />;
+    return (
+        <ClientCodeBlock
+            block={block}
+            withLineNumbers={withLineNumbers}
+            withWrap={withWrap}
+            title={title}
+            style={style}
+            inlines={richInlines}
+        />
+    );
 }

Diff for: packages/gitbook/src/components/DocumentView/CodeBlock/CodeBlockRenderer.tsx

@@ -12,9 +12,12 @@ import type { HighlightLine, HighlightToken } from './highlight';
 import './theme.css';
 import './CodeBlockRenderer.css';
 
-type CodeBlockRendererProps = Pick<BlockProps<DocumentBlockCode>, 'block' | 'style'> & {
+export interface CodeBlockRendererProps extends Pick<BlockProps<DocumentBlockCode>, 'style'> {
     lines: HighlightLine[];
-};
+    withLineNumbers: boolean;
+    withWrap: boolean;
+    title: string;
+}
 
 /**
  * The logic of rendering a code block from lines.
@@ -23,12 +26,8 @@ export const CodeBlockRenderer = forwardRef(function CodeBlockRenderer(
     props: CodeBlockRendererProps,
     ref: React.ForwardedRef<HTMLDivElement>
 ) {
-    const { block, style, lines } = props;
-
+    const { style, lines, withLineNumbers, withWrap, title } = props;
     const id = useId();
-    const withLineNumbers = Boolean(block.data.lineNumbers) && block.nodes.length > 1;
-    const withWrap = block.data.overflow === 'wrap';
-    const title = block.data.title;
 
     return (
         <div ref={ref} className={tcls('group/codeblock grid grid-flow-col', style)}>

Diff for: packages/gitbook/src/components/DocumentView/CodeBlock/PlainCodeBlock.tsx

@@ -1,6 +1,6 @@
-import type { DocumentBlockCode, JSONDocument } from '@gitbook/api';
 import { useId } from 'react';
 
+import type { SlimDocumentBlockCode, SlimJSONDocument } from '@/lib/slim-document';
 import { CodeBlock } from './CodeBlock';
 
 /**
@@ -11,7 +11,7 @@ export function PlainCodeBlock(props: { code: string; syntax: string }) {
     const { code, syntax } = props;
     const id = useId();
 
-    const block: DocumentBlockCode = {
+    const block: SlimDocumentBlockCode = {
         key: id,
         object: 'block',
         type: 'code',
@@ -37,7 +37,7 @@ export function PlainCodeBlock(props: { code: string; syntax: string }) {
         })),
     };
 
-    const document: JSONDocument = {
+    const document: SlimJSONDocument = {
         object: 'document',
         data: {},
         nodes: [block],