Niklas/shadcn-vue
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: improve crawling logic

Browse Source
This commit is contained in:
zernonia 2024-03-14 18:22:57 +08:00
parent 00edec21aa
commit 017fbc1e1f
1 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
apps/www/.vitepress/theme/components/CodeWrapper.ts
View File

@ -1,6 +1,17 @@
import { type VNode, cloneVNode, defineComponent } from 'vue'
import { type VNode, type VNodeArrayChildren, cloneVNode, defineComponent } from 'vue'
import { useConfigStore } from '@/stores/config'
function crawlSpan(children: VNodeArrayChildren, cb: (vnode: VNode) => void) {
children.forEach((childNode) => {
if (!Array.isArray(childNode) && typeof childNode === 'object') {
if (typeof childNode?.children === 'string')
cb(childNode)
else
crawlSpan(childNode?.children as VNodeArrayChildren ?? [], cb)
}
})
}
export default defineComponent(
(props, { slots }) => {
const { codeConfig } = useConfigStore()
@ -16,21 +27,14 @@ export default defineComponent(
const preVNode = [...clonedVNode?.children].find((node: VNode) => node.type === 'pre') as VNode
// @ts-expect-error cloneVNode
const codeVNode = preVNode.children?.at(0) as VNode
if (codeVNode) {
// @ts-expect-error cloneVNode
[...codeVNode.children]
.filter((node: VNode) => node.type === 'span')
.forEach((node: VNode) => {
if (node.children) {
// @ts-expect-error cloneVNode
[...node.children].forEach((childNode: VNode) => {
if (typeof childNode.children === 'string') {
childNode.children = childNode.children.replaceAll('@/components', codeConfig.value.componentsPath)
childNode.children = childNode.children.replaceAll('@/libs', codeConfig.value.utilsPath)
}
})
}
})
crawlSpan(codeVNode.children as VNodeArrayChildren, (vnode) => {
if (typeof vnode.children === 'string') {
vnode.children = vnode.children.replaceAll('@/components', codeConfig.value.componentsPath)
vnode.children = vnode.children.replaceAll('@/libs', codeConfig.value.utilsPath)
}
})
return clonedVNode
}