import fs from 'fs'; import path from 'path'; import matter from 'gray-matter'; import LRUCache from 'lru-cache'; import type { MarkdownOptions, MarkdownParsedData, MarkdownRenderer } from './markdown/markdown'; import { createMarkdownRenderer } from './markdown/markdown'; import { deeplyParseHeader } from './utils/parseHeader'; import type { PageData, HeadConfig } from '../shared'; import slash from 'slash'; import escapeHtml from 'escape-html'; import fetchCode from './utils/fetchCode'; import tsToJs from './utils/tsToJs'; // eslint-disable-next-line @typescript-eslint/no-var-requires const debug = require('debug')('vitepress:md'); const cache = new LRUCache({ max: 1024 }); interface MarkdownCompileResult { vueSrc: string; pageData: PageData; } export function createMarkdownToVueRenderFn( root: string = process.cwd(), options: MarkdownOptions = {}, ) { const md = createMarkdownRenderer(options); return async (src: string, file: string): Promise => { const relativePath = slash(path.relative(root, file)); const cached = cache.get(src); if (cached) { debug(`[cache hit] ${relativePath}`); return cached; } const start = Date.now(); const { content, data: frontmatter } = matter(src); // eslint-disable-next-line prefer-const let { html, data } = md.render(content); // avoid env variables being replaced by vite html = html .replace(/import\.meta/g, 'import.meta') .replace(/process\.env/g, 'process.env'); // TODO validate data.links? const pageData: PageData = { title: inferTitle(frontmatter, content), description: inferDescription(frontmatter), frontmatter, headers: data.headers, relativePath, content: escapeHtml(content), html, // TODO use git timestamp? lastUpdated: Math.round(fs.statSync(file).mtimeMs), }; const newContent = data.vueCode ? await genComponentCode(md, data, pageData) : ` ${fetchCode(content, 'style')} `; debug(`[render] ${file} in ${Date.now() - start}ms.`); const result = { vueSrc: newContent?.trim(), pageData, }; cache.set(src, result); return result; }; } async function genComponentCode(md: MarkdownRenderer, data: PageData, pageData: PageData) { const { vueCode, headers = [] } = data as MarkdownParsedData; const cn = headers.find(h => h.title === 'zh-CN')?.content; const us = headers.find(h => h.title === 'en-US')?.content; let { html } = md.render(`\`\`\`vue ${vueCode?.trim()} \`\`\``); html = html .replace(/import\.meta/g, 'import.meta') .replace(/process\.env/g, 'process.env'); const template = fetchCode(vueCode, 'template'); const script = fetchCode(vueCode, 'script'); const style = fetchCode(vueCode, 'style'); const scriptContent = fetchCode(vueCode, 'scriptContent'); let jsCode = (await tsToJs(scriptContent))?.trim(); jsCode = jsCode ? `` : ''; const jsSourceCode = ` ${template} ${jsCode} ${style} `.trim(); let { html: jsVersion } = md.render(`\`\`\`vue ${jsSourceCode} \`\`\``); jsVersion = jsVersion .replace(/import\.meta/g, 'import.meta') .replace(/process\.env/g, 'process.env'); const jsfiddle = escapeHtml( JSON.stringify({ us, cn, docHtml: pageData.html.split('
')[0],
      ...pageData.frontmatter,
      relativePath: pageData.relativePath,
      // htmlCode: Buffer.from(html).toString('base64'),
      // jsVersionHtml: Buffer.from(jsVersion).toString('base64'),
      sourceCode: Buffer.from(vueCode).toString('base64'),
      jsSourceCode: Buffer.from(jsSourceCode).toString('base64'),
    }),
  );

  const newContent = `