content-collections.ts

import { postCollectionSchema } from "@/lib/post/schema";
import { defineCollection, defineConfig } from "@content-collections/core";
import { compileMDX } from "@content-collections/mdx";
import GithubSlugger from "github-slugger";
import rehypeAutolinkHeadings from "rehype-autolink-headings";
import rehypePrettyCode, { type LineElement } from "rehype-pretty-code";
import rehypeSlug from "rehype-slug";
import remarkGfm from "remark-gfm";

const posts = defineCollection({
  name: "posts",
  directory: "content/posts",
  include: "**/*.mdx",
  schema: postCollectionSchema,
  transform: async (document, context) => {
    const mdx = await compileMDX(context, document, {
      remarkPlugins: [remarkGfm],
      rehypePlugins: [
        rehypeSlug,
        [
          rehypeAutolinkHeadings,
          {
            behavior: "wrap",
            properties: {
              className: ["no-underline"],
            },
          },
        ],
        [
          rehypePrettyCode,
          {
            theme: "night-owl",
            keepBackground: true,
            onVisitLine(node: LineElement) {
              if (node.children.length === 0) {
                node.children = [{ type: "text", value: " " }];
              }

              const firstChild = node.children[0];
              if (firstChild) {
                let textValue = "";
                if (firstChild.type === "text") {
                  textValue = firstChild.value;
                } else if (
                  firstChild.type === "element" &&
                  firstChild.children?.[0]?.type === "text"
                ) {
                  textValue = firstChild.children[0].value;
                }

                if (textValue.startsWith("+")) {
                  node.properties["data-line"] = "add";
              
                } else if (textValue.startsWith("-")) {
                  node.properties["data-line"] = "remove";
          
                }
              }
            },
          },
        ],
      ],
    });

    const slugger = new GithubSlugger();
    const toc = Array.from(
      document.content.matchAll(/^(#{2,6})\s+(.+)$/gm)
    ).map((match) => {
      const level = match[1].length;
      const raw = match[2].trim();
      const title = raw
        .replace(/\[([^\]]+)\]\([^)]+\)/g, "$1")
        .replace(/`([^`]+)`/g, "$1")
        .replace(/\*\*([^*]+)\*\*/g, "$1")
        .replace(/__([^_]+)__/g, "$1")
        .replace(/\*([^*]+)\*/g, "$1")
        .replace(/_([^_]+)_/g, "$1");
      const slug = slugger.slug(title);
      return { level, title, slug };
    });

    const type =
      document._meta.directory === "archived"
        ? ("archived" as const)
        : ("active" as const);

    const { content, ...rest } = document; 
    const allDocs = (await context.collection.documents())
      .filter((d) =>
        type === "archived"
          ? d._meta.directory === "archived"
          : d._meta.directory !== "archived",
      )
      .sort(
        (a, b) =>
          new Date(b.publishedAt).getTime() - new Date(a.publishedAt).getTime(),
      );

    const idx = allDocs.findIndex(
      (d) => d._meta.filePath === document._meta.filePath,
    );
    
    const pick = (d: (typeof allDocs)[number]) => ({
      title: d.title,
      slug: d._meta.fileName.replace(/\.mdx?$/, ""),
      description: d.description,
      publishedAt: d.publishedAt,
      category: d.category,
      type,
    });

    return {
      ...rest,
      slug: document._meta.fileName.replace(/\.mdx?$/, ""),
      type,
      mdx,
      toc,
      prev: idx > 0 ? pick(allDocs[idx - 1]) : null,
      next: idx < allDocs.length - 1 ? pick(allDocs[idx + 1]) : null,
    };
  },
});

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