109 lines
4.1 KiB
JavaScript
109 lines
4.1 KiB
JavaScript
"use strict";
|
||
var __importDefault = (this && this.__importDefault) || function (mod) {
|
||
return (mod && mod.__esModule) ? mod : { "default": mod };
|
||
};
|
||
Object.defineProperty(exports, "__esModule", { value: true });
|
||
exports.rewriteImportVue = exports.rewriteImportVuePlugin = void 0;
|
||
const parser_1 = require("@babel/parser");
|
||
const es_module_lexer_1 = require("es-module-lexer");
|
||
const magic_string_1 = __importDefault(require("magic-string"));
|
||
/**
|
||
* 如下情况会生成import vue的代码：
|
||
* 1. uvue页面编译产物通过helpers方式导入的vue相关api，defineComponent、unref等
|
||
* 2. 用户代码script中的ref、reactive、computed等
|
||
* 3. onLoad等从@dcloudio/uni-app导入的api等
|
||
* 4. 用户代码script中import vue的api
|
||
*
|
||
* 如果在transform中处理，需要确保此插件在autoImport之后，否则此时2、3尚未生成import vue的代码。
|
||
* 如果在renderChunk中处理，能保证所有代码都已经生成import vue的代码，但是此时热更新会触发所有文件重新处理import vue。
|
||
*/
|
||
function rewriteImportVuePlugin() {
|
||
return {
|
||
name: 'uni:rewrite-import-vue',
|
||
enforce: 'post',
|
||
async transform(source, id) {
|
||
await es_module_lexer_1.init;
|
||
const [imports] = (0, es_module_lexer_1.parse)(source);
|
||
const vueImportRanges = imports.filter((specifier) => specifier.n === 'vue' && specifier.d === -1);
|
||
if (!vueImportRanges.length) {
|
||
return;
|
||
}
|
||
const rewritten = rewriteImportVue(source, vueImportRanges);
|
||
if (rewritten.hasChanged()) {
|
||
return {
|
||
code: rewritten.toString(),
|
||
// 必须指定hires，不然部分情况可能会无法正确映射行号。
|
||
map: rewritten.generateMap({ hires: 'boundary' }),
|
||
};
|
||
}
|
||
},
|
||
};
|
||
}
|
||
exports.rewriteImportVuePlugin = rewriteImportVuePlugin;
|
||
/**
|
||
* import { xx as yy, zz } from 'vue' =>
|
||
* const { xx: yy, zz } = globalThis.Vue
|
||
*/
|
||
function rewriteImportVue(input, importRanges) {
|
||
const s = new magic_string_1.default(input);
|
||
if (importRanges) {
|
||
importRanges.forEach(({ ss, se }) => {
|
||
const rewritten = rewriteImportVueStatement(input.slice(ss, se));
|
||
if (rewritten !== undefined) {
|
||
s.overwrite(ss, se, rewritten);
|
||
}
|
||
});
|
||
}
|
||
else {
|
||
const ast = (0, parser_1.parse)(input, {
|
||
sourceType: 'module',
|
||
}).program.body;
|
||
ast.forEach((node) => {
|
||
if (node.type === 'ImportDeclaration' && node.source.value === 'vue') {
|
||
const rewritten = createImportVueReplacement(node);
|
||
if (rewritten !== undefined) {
|
||
s.overwrite(node.start, node.end, rewritten);
|
||
}
|
||
}
|
||
});
|
||
}
|
||
return s;
|
||
}
|
||
exports.rewriteImportVue = rewriteImportVue;
|
||
function rewriteImportVueStatement(input) {
|
||
const node = (0, parser_1.parse)(input, {
|
||
sourceType: 'module',
|
||
}).program.body[0];
|
||
if (node?.type === 'ImportDeclaration' && node.source.value === 'vue') {
|
||
return createImportVueReplacement(node);
|
||
}
|
||
}
|
||
function createImportVueReplacement(node) {
|
||
const imports = [];
|
||
const declarations = [];
|
||
node.specifiers.forEach((specifier) => {
|
||
if (specifier.type === 'ImportSpecifier') {
|
||
const imported = specifier.imported.name;
|
||
const local = specifier.local.name;
|
||
if (imported === local) {
|
||
imports.push(imported);
|
||
}
|
||
else {
|
||
imports.push(`${imported}: ${local}`);
|
||
}
|
||
}
|
||
else if (specifier.type === 'ImportDefaultSpecifier') {
|
||
const local = specifier.local.name;
|
||
imports.push(`default: ${local}`);
|
||
}
|
||
else if (specifier.type === 'ImportNamespaceSpecifier') {
|
||
const local = specifier.local.name;
|
||
declarations.push(`const ${local} = globalThis.Vue`);
|
||
}
|
||
});
|
||
if (imports.length > 0) {
|
||
declarations.push(`const { ${imports.join(', ')} } = globalThis.Vue`);
|
||
}
|
||
return declarations.join(';');
|
||
}
|