import matter from "gray-matter" import remarkFrontmatter from "remark-frontmatter" import { QuartzTransformerPlugin } from "../types" import yaml from "js-yaml" import toml from "toml" import { slugTag } from "../../util/path" import { QuartzPluginData } from "../vfile" import { i18n } from "../../i18n" export interface Options { delims: string | string[] language: "yaml" | "toml" oneLineTagDelim: string } const defaultOptions: Options = { delims: "---", language: "yaml", oneLineTagDelim: ",", } function coalesceAliases(data: { [key: string]: any }, aliases: string[]) { for (const alias of aliases) { if (data[alias] !== undefined && data[alias] !== null) return data[alias] } } function coerceToArray(input: string | string[]): string[] | undefined { if (input === undefined || input === null) return undefined // coerce to array if (!Array.isArray(input)) { input = input .toString() .split(",") .map((tag: string) => tag.trim()) } // remove all non-strings return input .filter((tag: unknown) => typeof tag === "string" || typeof tag === "number") .map((tag: string | number) => tag.toString()) } export const FrontMatter: QuartzTransformerPlugin | undefined> = (userOpts) => { const opts = { ...defaultOptions, ...userOpts } return { name: "FrontMatter", markdownPlugins({ cfg }) { return [ [remarkFrontmatter, ["yaml", "toml"]], () => { return (_, file) => { const { data } = matter(Buffer.from(file.value), { ...opts, engines: { yaml: (s) => yaml.load(s, { schema: yaml.JSON_SCHEMA }) as object, toml: (s) => toml.parse(s) as object, }, }) if (data.title) { data.title = data.title.toString() } else if (data.title === null || data.title === undefined) { data.title = file.stem ?? i18n(cfg.configuration.locale).propertyDefaults.title } const tags = coerceToArray(coalesceAliases(data, ["tags", "tag"])) if (tags) data.tags = [...new Set(tags.map((tag: string) => slugTag(tag)))] const aliases = coerceToArray(coalesceAliases(data, ["aliases", "alias"])) if (aliases) data.aliases = aliases const cssclasses = coerceToArray(coalesceAliases(data, ["cssclasses", "cssclass"])) if (cssclasses) data.cssclasses = cssclasses // fill in frontmatter file.data.frontmatter = data as QuartzPluginData["frontmatter"] } }, ] }, } } declare module "vfile" { interface DataMap { frontmatter: { [key: string]: unknown } & { title: string } & Partial<{ tags: string[] aliases: string[] description: string publish: boolean draft: boolean enableToc: string cssclasses: string[] }> } }