102 lines
2.8 KiB
TypeScript
102 lines
2.8 KiB
TypeScript
import { createHash } from 'node:crypto'
|
|
import type { App, Page } from 'vuepress/core'
|
|
import { colors, logger } from 'vuepress/utils'
|
|
import type { BlogPostData, BlogPostDataItem } from '../shared/index.js'
|
|
import type { PluginOption } from './plugin.js'
|
|
|
|
const HMR_CODE = `
|
|
if (import.meta.webpackHot) {
|
|
import.meta.webpackHot.accept()
|
|
if (__VUE_HMR_RUNTIME__.updateBlogData) {
|
|
__VUE_HMR_RUNTIME__.updateBlogData(blogPostData)
|
|
}
|
|
}
|
|
|
|
if (import.meta.hot) {
|
|
import.meta.hot.accept(({ blogPostData }) => {
|
|
__VUE_HMR_RUNTIME__.updateBlogData(blogPostData)
|
|
})
|
|
}
|
|
`
|
|
|
|
const headingRe = /<h(\d)[^>]*>.*?<\/h\1>/gi
|
|
|
|
const EXCERPT_SPLIT = '<!-- more -->'
|
|
let contentHash: string | undefined
|
|
|
|
export async function preparedBlogData(app: App, pageFilter: (id: string) => boolean, options: PluginOption): Promise<void> {
|
|
const start = performance.now()
|
|
|
|
let pages = app.pages.filter((page) => {
|
|
return page.filePathRelative && pageFilter(page.filePathRelative)
|
|
})
|
|
if (options.pageFilter)
|
|
pages = pages.filter(options.pageFilter)
|
|
|
|
if (options.sortBy) {
|
|
pages = pages.sort((prev, next) => {
|
|
if (options.sortBy === 'createTime') {
|
|
return getTimestamp(prev.frontmatter.createTime as Date)
|
|
< getTimestamp(next.frontmatter.createTime as Date)
|
|
? 1
|
|
: -1
|
|
}
|
|
else {
|
|
return typeof options.sortBy === 'function'
|
|
&& options.sortBy(prev, next)
|
|
? 1
|
|
: -1
|
|
}
|
|
})
|
|
}
|
|
|
|
const blogData: BlogPostData = pages.map((page: Page) => {
|
|
let extended: Partial<BlogPostDataItem> = {}
|
|
if (typeof options.extendBlogData === 'function')
|
|
extended = options.extendBlogData(page)
|
|
|
|
const data = {
|
|
path: page.path,
|
|
title: page.title,
|
|
...extended,
|
|
}
|
|
|
|
if (options.excerpt && page.contentRendered.includes(EXCERPT_SPLIT)) {
|
|
const contents = page.contentRendered.split(EXCERPT_SPLIT)
|
|
let excerpt = contents[0]
|
|
// 删除摘要中的标题
|
|
excerpt = excerpt.replace(headingRe, '')
|
|
data.excerpt = excerpt
|
|
}
|
|
|
|
return data as BlogPostDataItem
|
|
})
|
|
|
|
let content = `\
|
|
export const blogPostData = JSON.parse(${JSON.stringify(
|
|
JSON.stringify(blogData),
|
|
)});
|
|
`
|
|
|
|
// inject HMR code
|
|
if (app.env.isDev)
|
|
content += HMR_CODE
|
|
|
|
const currentHash = hash(content)
|
|
if (!contentHash || contentHash !== currentHash) {
|
|
contentHash = currentHash
|
|
await app.writeTemp('internal/blogData.js', content)
|
|
}
|
|
|
|
if (app.env.isDebug)
|
|
logger.info(`\n[${colors.green('@vuepress-plume/plugin-blog-data')}] prepare blog data time spent: ${(performance.now() - start).toFixed(2)}ms`)
|
|
}
|
|
|
|
function getTimestamp(time: Date): number {
|
|
return new Date(time).getTime()
|
|
}
|
|
|
|
function hash(content: string): string {
|
|
return createHash('md5').update(content).digest('hex')
|
|
}
|