Use cache and fix crash in TOC

src/app/[spaceId]/.gitbook/pdf/page.tsx

@@ -1,6 +1,6 @@
 import * as React from 'react';
 
-import { api } from '@/lib/api';
+import { getPageDocument, getSpace, getCurrentRevision } from '@/lib/api';
 import { resolvePageId } from '@/lib/pages';
 import { pagePDFContainerId, PageHrefContext } from '@/lib/links';
 import { DocumentView } from '@/components/DocumentView';
@@ -27,9 +27,9 @@ export default async function PDFHTMLOutput(props: {
     const { params, searchParams } = props;
     const { spaceId } = params;
 
-    const [{ data: space }, { data: revision }] = await Promise.all([
-        api().spaces.getSpaceById(spaceId),
-        api().spaces.getCurrentRevision(spaceId),
+    const [space, revision] = await Promise.all([
+        getSpace(spaceId),
+        getCurrentRevision(spaceId),
     ]);
 
     const pages = selectPages(revision, searchParams).slice(0, 4); // TODO: remove slice
@@ -75,9 +75,7 @@ async function PDFPageDocument(props: {
 }) {
     const { space, revision, page, linksContext } = props;
 
-    const {
-        data: { document },
-    } = await api().spaces.getPageInRevisionById(space.id, revision.id, page.id);
+    const document = await getPageDocument(space.id, revision.id, page.id);
 
     return (
         <div id={pagePDFContainerId(page)}>

src/app/[spaceId]/[[...pathname]]/page.tsx

@@ -3,7 +3,7 @@ import { PageHrefContext, absoluteHref, pageHref } from '@/lib/links';
 import { Metadata } from 'next';
 import { notFound, redirect } from 'next/navigation';
 import { PagePathParams, fetchPageData, getPagePath } from '../fetch';
-import { api } from '@/lib/api';
+import { getPageDocument } from '@/lib/api';
 
 /**
  * Fetch and render a page.
@@ -20,9 +20,7 @@ export default async function Page(props: { params: PagePathParams }) {
         redirect(pageHref(page, linksContext));
     }
 
-    const {
-        data: { document },
-    } = await api().spaces.getPageInRevisionById(space.id, revision.id, page.id);
+    const document = await getPageDocument(space.id, revision.id, page.id);
 
     return (
         <SpaceContent

src/app/[spaceId]/fetch.ts

@@ -1,4 +1,4 @@
-import { api } from '@/lib/api';
+import { getSpace, getCurrentRevision } from '@/lib/api';
 
 import { resolvePagePath, resolvePageId } from '@/lib/pages';
 
@@ -20,9 +20,9 @@ export interface PageIdParams extends SpaceParams {
 export async function fetchPageData(params: PagePathParams | PageIdParams) {
     const { spaceId } = params;
 
-    const [{ data: space }, { data: revision }] = await Promise.all([
-        api().spaces.getSpaceById(spaceId),
-        api().spaces.getCurrentRevision(spaceId),
+    const [space, revision] = await Promise.all([
+        getSpace(spaceId),
+        getCurrentRevision(spaceId),
     ]);
 
     const page =

src/components/PageBody/PageFooterNavigation.tsx

@@ -25,7 +25,7 @@ export function PageFooterNavigation(props: {
                     icon={IconArrowLeft}
                     label={t({ space }, 'previous_page')}
                     title={previous.title}
-                    href={pageHref(previous.path)}
+                    href={pageHref(previous)}
                     reversed
                 />
             ) : null}
@@ -34,7 +34,7 @@ export function PageFooterNavigation(props: {
                     icon={IconArrowRight}
                     label={t({ space }, 'next_page')}
                     title={next.title}
-                    href={pageHref(next.path)}
+                    href={pageHref(next)}
                 />
             ) : null}
         </div>

src/components/TableOfContents/PageDocumentItem.tsx

@@ -1,6 +1,6 @@
 import { RevisionPageDocument, RevisionPageGroup } from '@gitbook/api';
 import { tcls } from '@/lib/tailwind';
-import Link, { LinkProps } from 'next/link';
+import Link from 'next/link';
 import { PagesList } from './PagesList';
 import { pageHref } from '@/lib/links';
 import { ToggeableLinkItem } from './ToggeableLinkItem';

src/components/TableOfContents/PageLinkItem.tsx

@@ -0,0 +1,30 @@
+import { RevisionPageLink } from '@gitbook/api';
+import { tcls } from '@/lib/tailwind';
+import Link from 'next/link';
+
+export function PageLinkItem(props: {
+    page: RevisionPageLink;
+}) {
+    const { page } = props;
+
+    return (
+        <li className={tcls('flex', 'flex-col', 'mb-0.5')}>
+            <Link href={
+                page.href ?? '#todo' // TODO: Will be fixed soon as the `target`
+            }
+            className={tcls(
+                'flex',
+                'flex-row',
+                'justify-between',
+                'rounded',
+                'px-2',
+                'py-1.5',
+                'text-sm',
+                'transition-colors',
+                'duration-100',
+                'hover:bg-slate-100', 'text-slate-500', 'font-normal',
+            )}
+            >{page.title}</Link>
+        </li>
+    );
+}

src/components/TableOfContents/PagesList.tsx

@@ -2,6 +2,7 @@ import { RevisionPage, RevisionPageDocument, RevisionPageGroup } from '@gitbook/
 import { ClassValue, tcls } from '@/lib/tailwind';
 import { PageDocumentItem } from './PageDocumentItem';
 import { PageGroupItem } from './PageGroupItem';
+import { PageLinkItem } from './PageLinkItem';
 
 export function PagesList(props: {
     pages: RevisionPage[];
@@ -23,6 +24,13 @@ export function PagesList(props: {
                             ancestors={ancestors}
                         />
                     );
+                } else if (page.type === 'link') {
+                    return (
+                        <PageLinkItem
+                            key={page.id}
+                            page={page}
+                        />
+                    );
                 }
 
                 return (

src/lib/api.ts

@@ -1,7 +1,8 @@
 import 'server-only';
 
 import { headers } from 'next/headers';
-import { GitBookAPI } from '@gitbook/api';
+import { unstable_cache } from 'next/cache';
+import { GitBookAPI, JSONDocument } from '@gitbook/api';
 
 /**
  * Create an API client for the current request.
@@ -17,3 +18,36 @@ export function api(): GitBookAPI {
 
     return gitbook;
 }
+
+/**
+ * Get a space by its ID.
+ */
+export const getSpace = unstable_cache(async (spaceId: string) => {
+    const { data } = await api().spaces.getSpaceById(spaceId);
+    return data;
+}, ['api', 'spaces'], {
+    tags: ['api', 'spaces']
+});
+
+/**
+ * Get the current revision of a space
+ */
+export const getCurrentRevision = unstable_cache(async (spaceId: string) => {
+    const { data } = await api().spaces.getCurrentRevision(spaceId);
+    return data;
+}, ['api', 'revisions'], {
+    tags: ['api', 'revisions']
+});
+
+/**
+ * Get the document for a page.
+ */
+export const getPageDocument = unstable_cache(async (spaceId: string, revisionId: string, pageId: string) => {
+    const { data } = await api().spaces.getPageInRevisionById(spaceId, revisionId, pageId);
+    // @ts-ignore
+    return data.document as JSONDocument;
+}, ['api', 'documents'], {
+    tags: ['api', 'documents']
+});
+
+