gitea-yuany3721/vuepress-theme-plume
1
0
Fork 1
mirror of https://github.com/pengzhanbo/vuepress-theme-plume.git synced 2026-04-23 10:58:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

perf(plugin-md-power): improve file-tree attrs parsing (#238)

Browse Source
This commit is contained in:
pengzhanbo 2024-10-01 01:26:18 +08:00 committed by GitHub
parent 1db43196aa
commit 8a0b0e7092
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 13 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
plugins/plugin-md-power/src/node/container/fileTree.ts
View File

@ -4,6 +4,7 @@ import Token from 'markdown-it/lib/token.mjs'
import container from 'markdown-it-container'
import { removeEndingSlash, removeLeadingSlash } from 'vuepress/shared'
import { defaultFile, defaultFolder, getFileIcon } from '../fileIcons/index.js'
import { resolveAttrs } from '../utils/resolveAttrs.js'
interface FileTreeNode {
filename: string
@ -13,13 +14,16 @@ interface FileTreeNode {
empty: boolean
}
interface FileTreeAttrs {
title?: string
icon?: FileTreeIconMode
}
const type = 'file-tree'
const closeType = `container_${type}_close`
const componentName = 'FileTreeItem'
const itemOpen = 'file_tree_item_open'
const itemClose = 'file_tree_item_close'
const RE_SIMPLE_ICON = /:simple-icon\b/
const RE_COLORED_ICON = /:colored-icon\b/
export function fileTreePlugin(md: Markdown, options: FileTreeOptions = {}) {
const getIcon = (filename: string, type: 'folder' | 'file', mode?: FileTreeIconMode): string => {
@ -32,7 +36,7 @@ export function fileTreePlugin(md: Markdown, options: FileTreeOptions = {}) {
const validate = (info: string): boolean => info.trim().startsWith(type)
const render = (tokens: Token[], idx: number): string => {
const mode = getFileIconMode(tokens[idx].info)
const { attrs } = resolveAttrs<FileTreeAttrs>(tokens[idx].info.slice(type.length - 1))
if (tokens[idx].nesting === 1) {
const hasRes: number[] = [] // level stack
@ -49,7 +53,7 @@ export function fileTreePlugin(md: Markdown, options: FileTreeOptions = {}) {
hasRes.push(token.level)
const [info, inline] = result
const { filename, type, expanded, empty } = info
const icon = getIcon(filename, type, mode)
const icon = getIcon(filename, type, attrs.icon)
token.type = itemOpen
token.tag = componentName
@ -69,8 +73,7 @@ export function fileTreePlugin(md: Markdown, options: FileTreeOptions = {}) {
}
}
}
const title = resolveTitle(tokens[idx].info)
const title = attrs.title
return `<div class="vp-file-tree">${title ? `<p class="vp-file-tree-title">${title}</p>` : ''}`
}
else {
@ -81,20 +84,6 @@ export function fileTreePlugin(md: Markdown, options: FileTreeOptions = {}) {
md.use(container, type, { validate, render })
}
function getFileIconMode(info: string): FileTreeIconMode | undefined {
if (RE_SIMPLE_ICON.test(info))
return 'simple'
if (RE_COLORED_ICON.test(info))
return 'colored'
return undefined
}
function resolveTitle(info: string): string {
info = info.trim().slice(type.length).trim()
info = info.replace(RE_SIMPLE_ICON, '').replace(RE_COLORED_ICON, '')
return info.trim()
}
export function resolveTreeNodeInfo(
tokens: Token[],
current: Token,

8
plugins/plugin-md-power/src/node/utils/resolveAttrs.ts
View File

@ -1,13 +1,13 @@
const RE_ATTR_VALUE = /(?:^|\s+)(?<attr>[\w-]+)(?:=\s*(?<quote>['"])(?<value>.+?)\k<quote>)?(?:\s+|$)/
export function resolveAttrs(info: string): {
attrs: Record<string, any>
export function resolveAttrs<T extends Record<string, any> = Record<string, any>>(info: string): {
attrs: T
rawAttrs: string
} {
info = info.trim()
if (!info)
return { rawAttrs: '', attrs: {} }
return { rawAttrs: '', attrs: {} as T }
const attrs: Record<string, string | boolean> = {}
const rawAttrs = info
@ -37,5 +37,5 @@ export function resolveAttrs(info: string): {
}
})
return { attrs, rawAttrs }
return { attrs: attrs as T, rawAttrs }
}