/
compile.ts
65 lines (52 loc) · 1.74 KB
/
compile.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import { transferSetupPosition } from './transform'
const exportDefaultRE = /export\s+default/
const exportDefaultClassRE = /(?:(?:^|\n|;)\s*)export\s+default\s+class\s+([\w$]+)/
let index = 1
let compiler: typeof import('@vue/compiler-sfc')
function requireCompiler() {
if (!compiler) {
try {
// Vue 3.2.13+ ships the SFC compiler directly under the `vue` package
compiler = require('vue/compiler-sfc')
} catch (e) {
try {
compiler = require('@vue/compiler-sfc')
} catch (e) {
throw new Error('@vue/compiler-sfc is not present in the dependency tree.\n')
}
}
}
return compiler
}
export function compileVueCode(code: string) {
const { parse, compileScript, rewriteDefault } = requireCompiler()
const { descriptor } = parse(code)
const { script, scriptSetup } = descriptor
let content: string | null = null
let ext: string | null = null
if (script || scriptSetup) {
if (scriptSetup) {
const compiled = compileScript(descriptor, {
id: `${index++}`
})
const classMatch = compiled.content.match(exportDefaultClassRE)
if (classMatch) {
content =
compiled.content.replace(exportDefaultClassRE, `\nclass $1`) +
`\nconst _sfc_main = ${classMatch[1]}`
if (exportDefaultRE.test(content)) {
content = rewriteDefault(compiled.content, `_sfc_main`)
}
} else {
content = rewriteDefault(compiled.content, `_sfc_main`)
}
content = transferSetupPosition(content)
content += '\nexport default _sfc_main\n'
ext = scriptSetup.lang || 'js'
} else if (script && script.content) {
content = script.content
ext = script.lang || 'js'
}
}
return { content, ext }
}