content-collections.ts

import { defineCollection, defineConfig } from "@content-collections/core";
import { compileMDX } from "@content-collections/mdx";
import readingTime from "reading-time";
import rehypeAutolinkHeadings from "rehype-autolink-headings";
import rehypePrettyCode from "rehype-pretty-code";
import rehypeSlug from "rehype-slug";
import { remark } from "remark";
import remarkCapitalizeHeadings from "remark-capitalize-headings";
import remarkGfm from "remark-gfm";
import { z } from "zod";
import { capitalizationOptions } from "@/lib/capitalization";

import {
  rehypePostprocessPrettyCode,
  rehypePreprocessPrettyCode,
  rehypePrettyCodeOptions,
} from "./src/components/mdx/rehype-pretty-code";
import { type TOCNode, tocPlugin } from "./src/components/mdx/remark-toc";

const posts = defineCollection({
  name: "posts",
  directory: "content",
  include: "**/*.mdx",
  schema: z.object({
    title: z.string(),
    description: z.string(),
    date: z.string().pipe(z.coerce.date()),
    lastUpdated: z.string().pipe(z.coerce.date()).optional(),
    archived: z.boolean().default(false),
    tags: z.array(z.string()).default([]),
    content: z.string(),
  }),
  transform: async (document, context) => {
    const mdx = await compileMDX(context, document, {
      remarkPlugins: [
        [
          remarkCapitalizeHeadings,
          { replaceHeadingRegExp: capitalizationOptions },
        ],
        remarkGfm,
      ],
      rehypePlugins: [
        rehypeSlug,
        rehypeAutolinkHeadings,
        rehypePreprocessPrettyCode,
        [rehypePrettyCode, rehypePrettyCodeOptions],
        rehypePostprocessPrettyCode,
      ],
    });

    // Generate TOC separately using remark with capitalization
    const tocRoot = (
      await remark()
        .use(remarkCapitalizeHeadings, {
          replaceHeadingRegExp: capitalizationOptions,
        })
        .use(tocPlugin)
        .process(document.content)
    ).data.toc as TOCNode;

    // Calculate reading time
    const readingTimeStats = readingTime(document.content);

    // Generate clean slug without folder prefixes
    const fileName =
      document._meta.path
        .split("/")
        .pop()
        ?.replace(/\.mdx$/, "") || "";

    // Derive locale and top-level folder from path: <locale>/<folder>/...
    const [locale, folder] = document._meta.path.split("/");

    return {
      ...document,
      mdx,
      toc: JSON.stringify(tocRoot || { children: [] }),
      readingTime: readingTimeStats.text,
      slug: fileName,
      url: `/posts/${fileName}`,
      locale,
      folder,
    };
  },
});

export default defineConfig({
  collections: [posts],
});