source.config.ts

import { defineDocs, defineConfig } from "fumadocs-mdx/config";
import remarkTabsSyntax from "./plugins/remark-tabs-syntax";
import remarkCodeGroupToTabs from "./plugins/remark-codegroup-to-tabs";
import remarkCodeLanguage from "./plugins/remark-code-language";
import remarkImagePaths from "./plugins/remark-image-paths";
import remarkLinkPaths from "./plugins/remark-link-paths";
import remarkFollowExport from "./plugins/remark-follow-export";
import remarkDirective from "remark-directive";
import { remarkInclude } from "fumadocs-mdx/config";
import remarkSdkFilter from "./plugins/remark-sdk-filter";
import remarkMarkdownImageMap from "./plugins/remark-markdown-image-map";
import remarkVisibility from "./plugins/remark-visibility";
import type { LLMsOptions } from "fumadocs-core/mdx-plugins";

const isDevelopment = process.env.NODE_ENV === "development";
const markdownHeadingHandler: NonNullable<LLMsOptions["handlers"]>["heading"] = (
  node,
  _parent,
  state,
  info,
) => {
  const depth = Math.min(Math.max(node.depth, 1), 6);
  return `${"#".repeat(depth)} ${state.containerPhrasing(node, info)}`;
};

const processedMarkdownOptions: LLMsOptions = {
  handlers: {
    heading: markdownHeadingHandler,
  },
  mdxAsPlaceholder: [
    "Accordion",
    "AccordionGroup",
    "Callout",
    "Card",
    "CardGroup",
    "CodeGroup",
    "Frame",
    "Info",
    "Mermaid",
    "Note",
    "Step",
    "Steps",
    "Tab",
    "Tabs",
    "Tip",
    "Visibility",
    "Warning",
  ],
};

const markdownImageMapPassthroughPlugin = {
  "index-file": {
    serverOptions(options: any) {
      options.doc ??= {};
      options.doc.passthroughs ??= [];
      options.doc.passthroughs.push("markdownImageMap");
    },
  },
};

export const docs = defineDocs({
  dir: "content/docs",
  docs: {
    async: isDevelopment,
    postprocess: {
      includeProcessedMarkdown: processedMarkdownOptions,
    },
  },
});

export default defineConfig({
  plugins: [markdownImageMapPassthroughPlugin],
  mdxOptions: {
    // Shiki highlighting is one of the most expensive parts of the MDX pipeline.
    // Keep it in builds, but skip it in local dev so first-page SSR doesn't have
    // to highlight hundreds of code-heavy docs up front.
    rehypeCodeOptions: isDevelopment ? false : undefined,
    remarkPlugins: (existing) => [
      remarkImagePaths,
      remarkMarkdownImageMap,
      remarkLinkPaths,
      remarkFollowExport,
      // Strip agent-only content before Fumadocs captures structured search data.
      remarkVisibility,
      ...existing,
      remarkInclude,
      remarkDirective,
      // Keep the transform idempotent for content changed by later local plugins.
      remarkVisibility,
      remarkTabsSyntax,
      remarkCodeLanguage,
      remarkCodeGroupToTabs,
      remarkSdkFilter,
    ],
  },
});