Replace Nextra MDX Wrapper and restyle TOC

Author: hasparus

patches/nextra-theme-docs.patch

@@ -1,8 +1,8 @@
 diff --git a/dist/index.d.mts b/dist/index.d.mts
-index 71f87bcd1dde49d7c19ad49fc098e715a76c5c10..6671e29326be99861058895916185910452ced17 100644
+index 71f87bcd1dde49d7c19ad49fc098e715a76c5c10..c26af21e9bf3e8b1a2641b9d91c597575c30bfd9 100644
 --- a/dist/index.d.mts
 +++ b/dist/index.d.mts
-@@ -1421,3 +1421,19 @@ declare function ThemeSwitch({ lite, className }: ThemeSwitchProps): ReactElemen
+@@ -1421,3 +1421,25 @@ declare function ThemeSwitch({ lite, className }: ThemeSwitchProps): ReactElemen
  declare function Layout({ children, themeConfig, pageOpts }: NextraThemeLayoutProps): ReactElement;
 
  export { Bleed, Collapse, type PartialDocsThemeConfig as DocsThemeConfig, Link, LocaleSwitch, Navbar, NotFoundPage, SkipNavContent, SkipNavLink, ThemeSwitch, Layout as default, getComponents, useConfig, useMenu, useThemeConfig };
@@ -22,8 +22,14 @@ index 71f87bcd1dde49d7c19ad49fc098e715a76c5c10..6671e29326be99861058895916185910
 +export declare const useIntersectionObserver: () => IntersectionObserver | null
 +export declare const useSlugs: () => WeakMap<any, any>
 +
++export declare const Breadcrumb: (props: { activeElement: Item[] }) => ReactElement | null
++export declare const NavLinks: (props: NavLinkProps) => ReactElement | null
++export interface NavLinkProps {
++  currentIndex: number
++  flatDocsDirectories: Item[]
++}
 diff --git a/dist/index.js b/dist/index.js
-index 56201641fd965dcc5ab7c5df53e444c41293c00e..b0344053f3e0682eda3dd23e142abc5458f9a683 100644
+index 56201641fd965dcc5ab7c5df53e444c41293c00e..07147c688ae75c4c7daf082833acc71de16b36ee 100644
 --- a/dist/index.js
 +++ b/dist/index.js
 @@ -100,10 +100,10 @@ IntersectionObserverContext.displayName = "IntersectionObserver";
@@ -41,12 +47,21 @@ index 56201641fd965dcc5ab7c5df53e444c41293c00e..b0344053f3e0682eda3dd23e142abc54
  var ActiveAnchorProvider = ({
    children
  }) => {
-@@ -2502,7 +2502,7 @@ var DEFAULT_COMPONENTS = {
-         ),
-         children: [
-           /* @__PURE__ */ jsx26(
--            Sidebar,
-+            themeConfig.sidebar.component,
-             {
-               docsDirectories,
-               fullDirectories: directories,
+@@ -526,7 +526,7 @@ import NextLink2 from "next/link";
+ import { ArrowRightIcon } from "nextra/icons";
+ import { Fragment as Fragment3 } from "react";
+ import { jsx as jsx9, jsxs as jsxs3 } from "react/jsx-runtime";
+-function Breadcrumb({
++export function Breadcrumb({
+   activePath
+ }) {
+   return /* @__PURE__ */ jsx9("div", { className: "nextra-breadcrumb _mt-1.5 _flex _items-center _gap-1 _overflow-hidden _text-sm _text-gray-500 dark:_text-gray-400 contrast-more:_text-current", children: activePath.map((item, index, arr) => {
+@@ -1255,7 +1255,7 @@ var classes = {
+   ),
+   icon: cn10("_inline _h-5 _shrink-0")
+ };
+-function NavLinks({
++export function NavLinks({
+   flatDocsDirectories,
+   currentIndex
+ }) {

pnpm-lock.yaml

@@ -0,0 +1,54 @@
+/**
+ * @file BackToTop component extracted from Nextra
+ */
+
+import { clsx } from "clsx"
+import { Button } from "nextra/components"
+import { ArrowRightIcon } from "nextra/icons"
+import type { ComponentProps, ReactElement, ReactNode } from "react"
+
+const SCROLL_TO_OPTIONS = { top: 0, behavior: "smooth" } as const
+
+const scrollToTop: ComponentProps<"button">["onClick"] = event => {
+  const buttonElement = event.currentTarget
+  const tocElement = buttonElement.parentElement!.parentElement!
+
+  window.scrollTo(SCROLL_TO_OPTIONS)
+  tocElement.scrollTo(SCROLL_TO_OPTIONS)
+
+  // Fixes https://github.com/facebook/react/issues/20770
+  // Fixes https://github.com/shuding/nextra/issues/2917
+  buttonElement.disabled = true
+}
+
+export function BackToTop({
+  children,
+  className,
+  hidden,
+}: {
+  children: ReactNode
+  className?: string
+  hidden: boolean
+}): ReactElement {
+  return (
+    <Button
+      // elements with `aria-hidden: true` must not be focusable or contain focusable elements
+      aria-hidden={hidden ? "true" : undefined}
+      onClick={scrollToTop}
+      disabled={hidden}
+      className={({ disabled }) =>
+        clsx(
+          "flex items-center gap-1.5",
+          disabled ? "opacity-0" : "opacity-100",
+          className,
+        )
+      }
+    >
+      {children}
+      <ArrowRightIcon
+        height="16"
+        className="-rotate-90 rounded-full border border-current"
+      />
+    </Button>
+  )
+}

src/components/back-to-top.tsx

@@ -0,0 +1,155 @@
+import type { ReactElement, ReactNode } from "react"
+import { useMounted } from "nextra/hooks"
+import { Heading } from "nextra"
+import {
+  useConfig,
+  useThemeConfig,
+  SkipNavContent,
+  Breadcrumb,
+  NavLinks,
+} from "nextra-theme-docs"
+import { clsx } from "clsx"
+
+import { Sidebar } from "./sidebar"
+import { renderComponent } from "./utils"
+import { TableOfContents } from "./table-of-contents"
+
+const classes = {
+  toc: clsx("nextra-toc order-last max-xl:hidden w-64 shrink-0 print:hidden"),
+  main: clsx("w-full break-words"),
+}
+
+export interface NextraMdxWrapperProps {
+  toc?: Heading[]
+  children?: React.ReactNode
+}
+
+export function NextraMdxWrapper({
+  toc = [],
+  children,
+}: NextraMdxWrapperProps) {
+  const config = useConfig()
+  const {
+    activeType,
+    activeThemeContext: themeContext,
+    docsDirectories,
+    directories,
+  } = config.normalizePagesResult
+
+  const tocEl =
+    activeType === "page" ||
+    !themeContext.toc ||
+    themeContext.layout !== "default" ? (
+      themeContext.layout !== "full" &&
+      themeContext.layout !== "raw" && (
+        <nav className={classes.toc} aria-label="table of contents" />
+      )
+    ) : (
+      <nav className={classes.toc} aria-label="table of contents">
+        <TableOfContents toc={toc} filePath={config.filePath} />
+      </nav>
+    )
+  return (
+    <div
+      className={clsx(
+        "mx-auto flex",
+        themeContext.layout !== "raw" && "max-w-[90rem]",
+      )}
+    >
+      <Sidebar
+        docsDirectories={docsDirectories}
+        fullDirectories={directories}
+        toc={toc}
+        asPopover={config.hideSidebar}
+        includePlaceholder={themeContext.layout === "default"}
+      />
+      {tocEl}
+      <SkipNavContent />
+      <Body>{children}</Body>
+    </div>
+  )
+}
+
+function Body({ children }: { children: ReactNode }): ReactElement {
+  const config = useConfig()
+  const themeConfig = useThemeConfig()
+  const mounted = useMounted()
+  const {
+    activeThemeContext: themeContext,
+    activeType,
+    activeIndex,
+    flatDocsDirectories,
+    activePath,
+  } = config.normalizePagesResult
+
+  if (themeContext.layout === "raw") {
+    return <div className={classes.main}>{children}</div>
+  }
+
+  const date =
+    themeContext.timestamp && themeConfig.gitTimestamp && config.timestamp
+      ? new Date(config.timestamp)
+      : null
+
+  const gitTimestampEl =
+    // Because a user's time zone may be different from the server page
+    mounted && date ? (
+      <div className="mb-8 mt-12 block text-xs text-neu-500 ltr:text-right rtl:text-left">
+        {renderComponent(themeConfig.gitTimestamp, { timestamp: date })}
+      </div>
+    ) : (
+      <div className="mt-16" />
+    )
+
+  const content = (
+    <>
+      {renderComponent(themeContext.topContent)}
+      {children}
+      {gitTimestampEl}
+      {renderComponent(themeContext.bottomContent)}
+      {activeType !== "page" && themeContext.pagination && (
+        <NavLinks
+          flatDocsDirectories={flatDocsDirectories}
+          currentIndex={activeIndex}
+        />
+      )}
+    </>
+  )
+
+  const body = themeConfig.main ? (
+    <themeConfig.main>{content}</themeConfig.main>
+  ) : (
+    content
+  )
+
+  if (themeContext.layout === "full") {
+    return (
+      <article
+        className={clsx(
+          classes.main,
+          "nextra-content min-h-[calc(100vh-var(--nextra-navbar-height))] pl-[max(env(safe-area-inset-left),1.5rem)] pr-[max(env(safe-area-inset-right),1.5rem)]",
+        )}
+      >
+        {body}
+      </article>
+    )
+  }
+
+  return (
+    <article
+      className={clsx(
+        classes.main,
+        "nextra-content flex min-h-[calc(100vh-var(--nextra-navbar-height))] min-w-0 justify-center pb-8 pr-[calc(env(safe-area-inset-right)-1.5rem)]",
+        themeContext.typesetting === "article" &&
+          "nextra-body-typesetting-article",
+      )}
+    >
+      <main className="w-full min-w-0 max-w-6xl px-6 pt-4 md:px-12">
+        {activeType !== "page" && themeContext.breadcrumb && (
+          <Breadcrumb activePath={activePath} />
+        )}
+        {body}
+      </main>
+    </article>
+  )
+}

src/components/nextra-mdx-wrapper.tsx

@@ -1,6 +1,5 @@
 /**
  * @file sidebar module extracted from Nextra 3.3.1
- * @description
  */
 
 import cn from "clsx"
@@ -21,7 +20,7 @@ import {
   useState,
 } from "react"
 import scrollIntoView from "scroll-into-view-if-needed"
-import { renderComponent } from "../utils"
+import { renderComponent } from "./utils"
 import {
   useActiveAnchor,
   useMenu,
@@ -30,7 +29,7 @@ import {
   LocaleSwitch,
 } from "nextra-theme-docs"
 
-import { Anchor } from "../../app/conf/_design-system/anchor"
+import { Anchor } from "../app/conf/_design-system/anchor"
 
 const TreeState: Record<string, boolean> = Object.create(null)