2023-02-16 15:00:40 +00:00
|
|
|
import MagicString from 'magic-string'
|
|
|
|
import { createUnplugin } from 'unplugin'
|
|
|
|
import { stripLiteral } from 'strip-literal'
|
2023-04-19 18:17:36 +00:00
|
|
|
import { isJS, isVue } from '../../../nuxt/src/core/utils/plugins'
|
2023-02-16 15:00:40 +00:00
|
|
|
|
|
|
|
export interface PureAnnotationsOptions {
|
|
|
|
sourcemap: boolean
|
|
|
|
functions: string[]
|
|
|
|
}
|
|
|
|
|
|
|
|
export const pureAnnotationsPlugin = createUnplugin((options: PureAnnotationsOptions) => {
|
|
|
|
const FUNCTION_RE = new RegExp(`(?<!\\/\\* #__PURE__ \\*\\/ )\\b(${options.functions.join('|')})\\s*\\(`, 'g')
|
|
|
|
const FUNCTION_RE_SINGLE = new RegExp(`(?<!\\/\\* #__PURE__ \\*\\/ )\\b(${options.functions.join('|')})\\s*\\(`)
|
|
|
|
return {
|
|
|
|
name: 'nuxt:pure-annotations',
|
|
|
|
enforce: 'post',
|
|
|
|
transformInclude (id) {
|
2023-04-19 18:17:36 +00:00
|
|
|
return isVue(id, { type: ['script'] }) || isJS(id)
|
2023-02-16 15:00:40 +00:00
|
|
|
},
|
2023-04-14 17:21:08 +00:00
|
|
|
transform (code) {
|
2023-02-16 15:00:40 +00:00
|
|
|
if (!FUNCTION_RE_SINGLE.test(code)) { return }
|
|
|
|
|
|
|
|
const s = new MagicString(code)
|
|
|
|
const strippedCode = stripLiteral(code)
|
|
|
|
|
|
|
|
for (const match of strippedCode.matchAll(FUNCTION_RE)) {
|
|
|
|
s.overwrite(match.index!, match.index! + match[0].length, '/* #__PURE__ */ ' + match[0])
|
|
|
|
}
|
|
|
|
|
|
|
|
if (s.hasChanged()) {
|
|
|
|
return {
|
|
|
|
code: s.toString(),
|
|
|
|
map: options.sourcemap
|
2023-04-14 17:21:08 +00:00
|
|
|
? s.generateMap({ hires: true })
|
2023-02-16 15:00:40 +00:00
|
|
|
: undefined
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|