contentlayer.config.ts

import remarkGfm from "remark-gfm"
import rehypeAutolinkHeadings from "rehype-autolink-headings"
import remarkFrontmatter from "remark-frontmatter"
import rehypeSlug from "rehype-slug"
import rehypePresetMinify from "rehype-preset-minify"
import {defineDocumentType, defineNestedType, makeSource} from "contentlayer2/source-files"
import removeMarkdown from "remove-markdown"
import {HashIcon} from "lucide-react"
import {renderToStaticMarkup} from "react-dom/server"
import {createElement} from "react"
import {fromHtmlIsomorphic} from "hast-util-from-html-isomorphic"
import remarkImagesProcessor, {generateFileHash} from "./src/lib/remark-image-processor"
import rehypePrismAll from "rehype-prism-plus"
import path from "path"
import fs from "node:fs"
import * as process from "node:process"
import {pinyin} from "pinyin-pro"

const ASSETS_DIR = path.join(process.cwd(), "public", "assets")

export const Post = defineDocumentType(() => ({
  name: "Post",
  filePathPattern: "blog/*/post*.md",
  contentType: "mdx",
  fields: {
    title: {type: "string", required: true},
    date: {type: "date", required: true},
    // @ts-ignore suppress type error
    updated: {type: "date", default: null},
    tags: {type: "list", of: {type: "string"}, default: []}
  },
  computedFields: {
    excerpt: {type: "string", resolve: post => extractExcerpt(post.body.raw, 200)},
    updated: {type: "date", resolve: (post) => post.updated ?? post.date},
    slug: {type: "string", resolve: (post) => post._raw.sourceFileDir.split("/").pop()},
    locale: {type: "string", resolve: (post) => post._raw.sourceFileName.split(".")[1]}
  }
}))

export const Activity = defineDocumentType(() => ({
  name: "Activity",
  filePathPattern: "activity/**/record*.md",
  contentType: "mdx",
  fields: {
    title: {type: "string", required: true},
    category: {type: "enum", options: ["book", "movie"], required: true},
    status: {type: "enum", options: ["todo", "doing", "done"], required: true},
    rating: {type: "number"},
    date: {type: "date", required: true},
    douban: {
      type: "nested", of: defineNestedType(() => ({
        name: "Douban",
        fields: {
          id: {type: "string"},
          title: {type: "string"},
          subtitle: {type: "string"},
          year: {type: "string"},
          intro: {type: "string"},
          rating: {type: "number"},
          cover: {type: "string"},
          link: {type: "string"},
          history: {
            type: "list", of: defineNestedType(() => ({
              name: "History",
              fields: {
                comment: {type: "string", required: true},
                rating: {type: "number"},
                status: {type: "enum", options: ["todo", "doing", "done"], required: true},
                date: {type: "date", required: true}
              }
            }))
          }
        }
      }))
    }
  },
  computedFields: {
    slug: {type: "string", resolve: (act) => slug(act._raw.sourceFileDir.split("/").pop()!)},
    locale: {type: "string", resolve: (post) => post._raw.sourceFileName.split(".")[1]},
    cover: {
      type: "string", resolve: (act) => {
        const dir = path.join("data", act._raw.sourceFileDir)
        for (const file of fs.readdirSync(dir)) {
          if (file.startsWith("cover.")) {
            const originalPath = path.join(dir, file)
            const fileExt = path.extname(originalPath)
            const fileNameHash = generateFileHash(originalPath)
            const newFileName = `${fileNameHash}${fileExt}`
            const newPath = path.join(ASSETS_DIR, newFileName)
            return "/" + path.relative(path.join(process.cwd(), "public"), newPath)
          }
        }
      }
    }
  }
}))

export default makeSource({
  contentDirPath: "data",
  contentDirExclude: ["locales"],
  documentTypes: [Post, Activity],
  onExtraFieldData: "ignore",
  mdx: {
    remarkPlugins: [remarkGfm, remarkFrontmatter, [remarkImagesProcessor, {publicDir: ASSETS_DIR}]],
    rehypePlugins: [
      rehypeSlug,
      [rehypeAutolinkHeadings, {
        headingProperties: {className: ["heading-anchor"]},
        content: fromHtmlIsomorphic(renderToStaticMarkup(createElement(HashIcon)), {fragment: true})
      }],
      [rehypePrismAll, {defaultLanguage: "js", ignoreMissing: true}],
      rehypePresetMinify
    ]
  }
})

function extractExcerpt(markdown: string, wordLimit: number) {
  const str = removeMarkdown(markdown)
  const arr = [...str]
  return arr.slice(0, wordLimit).concat(arr.length > wordLimit ? "..." : "").join("")
}

function slug(title: string): string {
  return pinyin(title, {nonZh: "consecutive", toneType: "none"})
    .replace(/[^\w\s-]/g, "").replace(/[\s_-]+/g, "-")
}