quartz-research-note/quartz/plugins/transformers/description.ts

52 lines
1.4 KiB
TypeScript
Raw Normal View History

2023-07-23 00:27:41 +00:00
import { Root as HTMLRoot } from "hast"
2023-05-30 15:02:20 +00:00
import { toString } from "hast-util-to-string"
import { QuartzTransformerPlugin } from "../types"
2023-09-07 04:47:59 +00:00
import { escapeHTML } from "../../util/escape"
2023-05-30 15:02:20 +00:00
export interface Options {
descriptionLength: number
}
const defaultOptions: Options = {
2023-07-23 00:27:41 +00:00
descriptionLength: 150,
2023-05-30 15:02:20 +00:00
}
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
2023-07-01 20:35:27 +00:00
const text = escapeHTML(toString(tree))
const desc = frontMatterDescription ?? text
2023-07-23 00:27:41 +00:00
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
2023-07-23 00:27:41 +00:00
finalDesc += sentence + "."
sentenceIdx++
}
file.data.description = finalDesc
file.data.text = text
2023-05-30 15:02:20 +00:00
}
2023-07-23 00:27:41 +00:00
},
]
2023-07-23 00:27:41 +00:00
},
2023-05-30 15:02:20 +00:00
}
}
2023-07-23 00:27:41 +00:00
declare module "vfile" {
2023-05-30 15:02:20 +00:00
interface DataMap {
description: string
2023-06-08 05:27:32 +00:00
text: string
2023-05-30 15:02:20 +00:00
}
}