56 lines
1.5 KiB
TypeScript
56 lines
1.5 KiB
TypeScript
import { Root as HTMLRoot } from 'hast'
|
|
import { toString } from "hast-util-to-string"
|
|
import { QuartzTransformerPlugin } from "../types"
|
|
|
|
export interface Options {
|
|
descriptionLength: number
|
|
}
|
|
|
|
const defaultOptions: Options = {
|
|
descriptionLength: 150
|
|
}
|
|
|
|
const escapeHTML = (unsafe: string) => {
|
|
return unsafe.replaceAll('&', '&').replaceAll('<', '<').replaceAll('>', '>').replaceAll('"', '"').replaceAll("'", ''');
|
|
}
|
|
|
|
export const Description: QuartzTransformerPlugin<Partial<Options> | undefined> = (userOpts) => {
|
|
const opts = { ...defaultOptions, ...userOpts }
|
|
return {
|
|
name: "Description",
|
|
htmlPlugins() {
|
|
return [
|
|
() => {
|
|
return async (tree: HTMLRoot, file) => {
|
|
const frontMatterDescription = file.data.frontmatter?.description
|
|
const text = escapeHTML(toString(tree))
|
|
|
|
const desc = frontMatterDescription ?? text
|
|
const sentences = desc.replace(/\s+/g, ' ').split('.')
|
|
let finalDesc = ""
|
|
let sentenceIdx = 0
|
|
const len = opts.descriptionLength
|
|
while (finalDesc.length < len) {
|
|
const sentence = sentences[sentenceIdx]
|
|
if (!sentence) break
|
|
finalDesc += sentence + '.'
|
|
sentenceIdx++
|
|
}
|
|
|
|
file.data.description = finalDesc
|
|
file.data.text = text
|
|
}
|
|
}
|
|
]
|
|
}
|
|
}
|
|
}
|
|
|
|
declare module 'vfile' {
|
|
interface DataMap {
|
|
description: string
|
|
text: string
|
|
}
|
|
}
|
|
|