rizaldy.club/quartz/plugins/transformers/description.ts

59 lines
1.3 KiB
TypeScript
Raw Normal View History

2023-05-30 22:02:20 +07:00
import { PluggableList } from "unified"
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
}
export class Description extends QuartzTransformerPlugin {
name = "Description"
opts: Options
2023-06-10 13:06:02 +07:00
constructor(opts?: Partial<Options>) {
2023-05-30 22:02:20 +07:00
super()
this.opts = { ...defaultOptions, ...opts }
}
markdownPlugins(): PluggableList {
return []
}
htmlPlugins(): PluggableList {
return [
() => {
return async (tree: HTMLRoot, file) => {
const frontMatterDescription = file.data.frontmatter?.description
2023-06-08 12:27:32 +07:00
const text = toString(tree)
const desc = frontMatterDescription ?? text
2023-05-30 22:02:20 +07:00
const sentences = desc.replace(/\s+/g, ' ').split('.')
let finalDesc = ""
let sentenceIdx = 0
const len = this.opts.descriptionLength
while (finalDesc.length < len) {
finalDesc += sentences[sentenceIdx] + '.'
sentenceIdx++
}
file.data.description = finalDesc
2023-06-08 12:27:32 +07:00
file.data.text = text
2023-05-30 22:02:20 +07:00
}
}
]
}
}
declare module 'vfile' {
interface DataMap {
description: string
2023-06-08 12:27:32 +07:00
text: string
2023-05-30 22:02:20 +07:00
}
}