2021-07-28 12:11:32 +00:00
|
|
|
import { createUnplugin } from 'unplugin'
|
|
|
|
import { parseQuery, parseURL } from 'ufo'
|
2021-11-21 16:14:46 +00:00
|
|
|
import { Component } from '@nuxt/schema'
|
2021-07-28 12:11:32 +00:00
|
|
|
|
|
|
|
interface LoaderOptions {
|
|
|
|
getComponents(): Component[]
|
|
|
|
}
|
|
|
|
|
|
|
|
export const loaderPlugin = createUnplugin((options: LoaderOptions) => ({
|
|
|
|
name: 'nuxt-components-loader',
|
|
|
|
enforce: 'post',
|
|
|
|
transformInclude (id) {
|
|
|
|
const { pathname, search } = parseURL(id)
|
|
|
|
const query = parseQuery(search)
|
|
|
|
// we only transform render functions
|
|
|
|
// from `type=template` (in Webpack) and bare `.vue` file (in Vite)
|
|
|
|
return pathname.endsWith('.vue') && (query.type === 'template' || !search)
|
|
|
|
},
|
|
|
|
transform (code) {
|
|
|
|
return transform(code, options.getComponents())
|
|
|
|
}
|
|
|
|
}))
|
|
|
|
|
|
|
|
function findComponent (components: Component[], name:string) {
|
|
|
|
return components.find(({ pascalName, kebabName }) => [pascalName, kebabName].includes(name))
|
|
|
|
}
|
|
|
|
|
|
|
|
function transform (content: string, components: Component[]) {
|
|
|
|
let num = 0
|
|
|
|
let imports = ''
|
|
|
|
const map = new Map<Component, string>()
|
|
|
|
|
|
|
|
// replace `_resolveComponent("...")` to direct import
|
|
|
|
const newContent = content.replace(/ _resolveComponent\("(.*?)"\)/g, (full, name) => {
|
|
|
|
const component = findComponent(components, name)
|
|
|
|
if (component) {
|
|
|
|
const identifier = map.get(component) || `__nuxt_component_${num++}`
|
|
|
|
map.set(component, identifier)
|
|
|
|
imports += `import ${identifier} from "${component.filePath}";`
|
|
|
|
return ` ${identifier}`
|
|
|
|
}
|
|
|
|
// no matched
|
|
|
|
return full
|
|
|
|
})
|
|
|
|
|
|
|
|
return `${imports}\n${newContent}`
|
|
|
|
}
|