You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
4.0 KiB
86 lines
4.0 KiB
// https://github.com/vueuse/vueuse/blob/main/packages/.vitepress/plugins/markdownTransform.ts
|
|
|
|
import path from "path";
|
|
import fs from "fs-extra";
|
|
import fg from "fast-glob";
|
|
import grayMatter from "gray-matter";
|
|
|
|
export function MarkdownTransform(): any {
|
|
return {
|
|
name: 'niu-tools-md-transform',
|
|
enforce: 'pre',
|
|
async transform(code, id) {
|
|
if (!id.match(/\.md\b/))
|
|
return null
|
|
const [pkg, _name, i] = id.split('/').slice(-3)
|
|
const oneDir = path.parse(id).dir.endsWith("docs") ? path.parse(path.parse(id).dir).dir : path.parse(id).dir
|
|
const oneName = oneDir.split("/").slice(-1)
|
|
const allFiles = fg.sync('**/*.ts', { cwd: oneDir, ignore: ["**/*.test.ts", "docs", "test", "**/__tests__/**/*"] })
|
|
if (_name !== "packages" && i === "index.md") {
|
|
const matter = grayMatter(fs.readFileSync(id, "utf8"));
|
|
const { data, content } = matter
|
|
|
|
if (data.category) {
|
|
code = grayMatter.stringify(`${data.category ? `分类：\`${data.category}\`` : ''} \n\n ` + content, data)
|
|
}
|
|
|
|
let source = ""
|
|
|
|
let rawcodeArray = (Array.from(code.matchAll(/<\!--code\:(.*?)\:code-->/g) ?? []) as any).map(([_, name]) => name)
|
|
|
|
|
|
|
|
for (let i = 0; i < allFiles.length; i++) {
|
|
const file = allFiles[i];
|
|
const p = path.resolve(oneDir, file)
|
|
let name = `${oneName}/${file}`
|
|
let str = ''
|
|
if (fs.pathExistsSync(p)) {
|
|
let rawcode = fs.readFileSync(p, "utf8")
|
|
if (rawcodeArray.length) {
|
|
for (let i = 0; i < rawcodeArray.length; i++) {
|
|
const symbol = rawcodeArray[i];
|
|
let startLen = `//${symbol}===== Start`.length
|
|
let startIndex = rawcode.indexOf(`//${symbol}===== Start`)
|
|
let endIndex = rawcode.indexOf(`//${symbol}===== End`)
|
|
// console.log(`<\!--code\:${symbol}\:code-->`);
|
|
if (startIndex !== -1 && endIndex !== -1) {
|
|
code = code.replace(`<\!--code\:${symbol}\:code-->`, `:::: details ${symbol}源码\n\`\`\`ts` + rawcode.slice(startIndex + startLen, endIndex) + "\`\`\`\n ::::")
|
|
}
|
|
}
|
|
}
|
|
str = `::: details ${name.replace(/\\/g, "/")}源码 \n\`\`\`ts\ ${rawcode}\n\`\`\` \n:::\n`
|
|
}
|
|
code = code.replace('$' + name + '$', str)
|
|
source += str
|
|
}
|
|
if (source) code += `\n ## 源码 \n :::: details 查看源码 \n ${source} \n :::: \n`
|
|
}
|
|
// if (!i.startsWith("index.md")) {
|
|
// const frontmatterEnds = code.indexOf('---\n\n')
|
|
// const firstHeader = code.search(/\n#{2,6}\s.+/)
|
|
// const sliceIndex = firstHeader < 0 ? frontmatterEnds < 0 ? 0 : frontmatterEnds + 4 : firstHeader
|
|
|
|
// code = code.slice(0, sliceIndex) + "\n# " + _name + "\n" + code.slice(sliceIndex)
|
|
// console.log(code, frontmatterEnds);
|
|
|
|
// code = code
|
|
// .replace(/(# \w+?)\n/, `$1\n\n<div>aaa</div>\n`)
|
|
// .replace(/## (Components?(?:\sUsage)?)/i, '## $1\n<LearnMoreComponents />\n\n')
|
|
// .replace(/## (Directives?(?:\sUsage)?)/i, '## $1\n<LearnMoreDirectives />\n\n')
|
|
// code += `\n# 哈哈`
|
|
// }
|
|
return code
|
|
},
|
|
}
|
|
}
|
|
|
|
// [...`fsd
|
|
// /**-----start:main-----**/
|
|
// sadada
|
|
// /**-----end:mai1n-----**/
|
|
// sadasdad
|
|
// /**-----start:test-----**/
|
|
// sadada
|
|
// /**-----end:test-----**/
|
|
// sadasdad`.matchAll(/\/\*\*-----start:(.*?)-----\*\*\/\n(.*?)\n\/\*\*-----end:(.*?)-----\*\*\//g)]
|