source.config.ts

import {
  defineConfig,
  defineDocs,
  frontmatterSchema,
  metaSchema,
} from 'fumadocs-mdx/config';
import fs from 'node:fs';
import path from 'node:path';
import { bundledLanguages } from 'shiki';

const shikiGrammarDir = path.join(process.cwd(), 'lib/shiki');
const betterscriptGrammar = JSON.parse(
  fs.readFileSync(path.join(shikiGrammarDir, 'betterscript.tmLanguage.json'), 'utf8'),
);
const ebnfGrammar = JSON.parse(
  fs.readFileSync(path.join(shikiGrammarDir, 'ebnf.tmLanguage.json'), 'utf8'),
);

// You can customise Zod schemas for frontmatter and `meta.json` here
// see https://fumadocs.dev/docs/mdx/collections
export const docs = defineDocs({
  docs: {
    schema: frontmatterSchema,
    postprocess: {
      includeProcessedMarkdown: true,
    },
  },
  meta: {
    schema: metaSchema,
  },
});

export default defineConfig({
  mdxOptions: {
    remarkCodeTabOptions: {
      parseMdx: true,
    },
    // Enable custom Shiki language for betterscript
    rehypeCodeOptions: {
      themes: {
        light: 'github-light',
        dark: 'github-dark',
      },
      langs: [
        ...Object.keys(bundledLanguages),
        {
          // Minimal TextMate grammar to keep ```ebnf``` fences rendering without errors
          ...ebnfGrammar,
        },
        {
          // Shiki custom language object, loaded from local TextMate grammar
          aliases: ['bs'],
          name: 'betterscript',
          scopeName: 'source.betterscript',
          // Merge the grammar JSON
          ...betterscriptGrammar,
        },
      ],
      langAlias: {
        bs: 'betterscript',
        rescript: 'purescript',
        ReScript: 'purescript',
        res: 'purescript',
        text: 'ebnf',
      },
    },
  },
});