206 lines
8.5 KiB
JavaScript
206 lines
8.5 KiB
JavaScript
const fs = require('fs');
|
||
const { relative, resolve, join } = require('path');
|
||
const t = require('@babel/types');
|
||
const assert = require('assert');
|
||
const { fork } = require('child_process');
|
||
|
||
const Regex =
|
||
/([\\/]*[a-zA-Z0-9_-\w\W]|[\\/]*[a-zA-Z0-9_-\w\W]:)*[\\/](lib|src)([\\/]*[a-zA-Z0-9_-])*[\\/](pages|components)+[\\/]/;
|
||
|
||
|
||
const ModuleDict = {};
|
||
|
||
function parseFileModuleAndNs(cwd, filename) {
|
||
const relativePath = relative(cwd, filename);
|
||
|
||
if (relativePath.startsWith('node_modules') || relativePath.startsWith('..')) { // 在测试环境下是相对路径
|
||
const moduleRelativePath = relativePath
|
||
.replace(/\\/g, '/')
|
||
.split('/')
|
||
.slice(0, 2);
|
||
const modulePath = join(cwd, ...moduleRelativePath);
|
||
const moduleDir = moduleRelativePath[1];
|
||
|
||
let moduleName = ModuleDict[moduleDir];
|
||
if (!moduleName) {
|
||
const { name } = require(join(modulePath, 'package.json'));
|
||
ModuleDict[moduleDir] = name;
|
||
moduleName = name;
|
||
}
|
||
const rel2paths = relative(modulePath, filename)
|
||
.replace(/\\/g, '/')
|
||
.split('/');
|
||
|
||
let ns;
|
||
switch (rel2paths[1]) {
|
||
case 'pages': {
|
||
ns = `${moduleName}-p-${rel2paths.slice(2, rel2paths.length - 1).join('-')}`;
|
||
break;
|
||
}
|
||
default: {
|
||
assert(rel2paths[1] === 'components', rel2paths.join('//'));
|
||
ns = `${moduleName}-c-${rel2paths.slice(2, rel2paths.length - 1).join('-')}`;
|
||
break;
|
||
}
|
||
}
|
||
|
||
return {
|
||
moduleName,
|
||
ns,
|
||
};
|
||
}
|
||
else {
|
||
let moduleName = ModuleDict['./'];
|
||
if (!moduleName) {
|
||
const { name } = require(join(cwd, 'package.json'));
|
||
ModuleDict['./'] = name;
|
||
moduleName = name;
|
||
}
|
||
|
||
const rel2paths = relative(cwd, filename)
|
||
.replace(/\\/g, '/')
|
||
.split('/');
|
||
|
||
let ns;
|
||
switch (rel2paths[1]) {
|
||
case 'pages': {
|
||
ns = `${moduleName}-p-${rel2paths.slice(2, rel2paths.length - 1).join('-')}`;
|
||
break;
|
||
}
|
||
case 'components': {
|
||
ns = `${moduleName}-c-${rel2paths.slice(2, rel2paths.length - 1).join('-')}`;
|
||
break;
|
||
}
|
||
default: {
|
||
// 处理web/wechatMp中的数据
|
||
assert(rel2paths[1] === 'src');
|
||
const p1 = rel2paths[0];
|
||
if (p1 === 'web') {
|
||
ns = `${moduleName}-w-${rel2paths.slice(2, rel2paths.length - 1).join('-')}`;
|
||
}
|
||
else if (p1 === 'wechatMp') {
|
||
ns = `${moduleName}-wmp-${rel2paths.slice(2, rel2paths.length - 1).join('-')}`;
|
||
}
|
||
else {
|
||
assert(p1.startsWith('wechatMp'));
|
||
const iter = parseInt(p1.slice(8), 10);
|
||
ns = `${moduleName}-wmp${iter}-${rel2paths.slice(2, rel2paths.length - 1).join('-')}`;
|
||
}
|
||
break;
|
||
}
|
||
}
|
||
|
||
return {
|
||
moduleName,
|
||
ns,
|
||
};
|
||
}
|
||
}
|
||
|
||
module.exports = (babel) => {
|
||
return {
|
||
visitor: {
|
||
CallExpression(path, state) {
|
||
const { cwd, filename } = state;
|
||
const res = resolve(cwd, filename).replace(/\\/g, '/');
|
||
// this.props.t/this.t/t
|
||
// t('common:detail') 不需要处理 t('detail') 需要处理;
|
||
// t(`${common}:${cc}`) 不需要处理 t(`${common}cc`) 需要处理
|
||
// 只支持t的参数为字符串或模版字符串
|
||
if (
|
||
/(pages|components)[\w|\W]+(.tsx|.ts|.jsx|.js)$/.test(res)
|
||
) {
|
||
const { node } = path;
|
||
if (
|
||
node &&
|
||
node.callee &&
|
||
((t.isIdentifier(node.callee) &&
|
||
node.callee.name === 't') ||
|
||
(t.isMemberExpression(node.callee) &&
|
||
t.isIdentifier(node.callee.property) &&
|
||
node.callee.property.name === 't'))
|
||
) {
|
||
const { moduleName, ns } = parseFileModuleAndNs(cwd, filename);
|
||
const arguments = node.arguments;
|
||
const argu0 = arguments && arguments[0];
|
||
if (t.isStringLiteral(argu0)) {
|
||
const { value } = argu0;
|
||
if (!value.includes(':')) {
|
||
// 是自己namespace下，加上ns
|
||
if (!value.startsWith(ns)) {
|
||
arguments.splice(
|
||
0,
|
||
1,
|
||
t.stringLiteral(ns + '.' + value)
|
||
);
|
||
}
|
||
}
|
||
else if (value.includes('::')) {
|
||
// 公共namespace下，加上moduleName-l
|
||
arguments.splice(
|
||
0,
|
||
1,
|
||
t.stringLiteral((moduleName + '-l-' + value).replace('::', '.'))
|
||
);
|
||
}
|
||
else {
|
||
// 是entity namespace下，改成.就行了
|
||
arguments.splice(
|
||
0,
|
||
1,
|
||
t.stringLiteral(value.replace(':', '.'))
|
||
);
|
||
}
|
||
}
|
||
else if (t.isTemplateLiteral(argu0)) {
|
||
assert (argu0.quasis);
|
||
const namespaceQuasis = argu0.quasis.find(
|
||
ele => ele.value.raw && ele.value.raw.includes(':')
|
||
);
|
||
if (namespaceQuasis) {
|
||
if (namespaceQuasis.value.raw.includes('::')) {
|
||
// 公共ns，改成.并在头上加上moduleName-l-
|
||
namespaceQuasis.value.raw = namespaceQuasis.value.raw.replace('::', '.');
|
||
argu0.quasis[0].value.raw = moduleName + '-l-' + argu0.quasis[0].value.raw || '';
|
||
}
|
||
else {
|
||
// entity的ns，改成.
|
||
namespaceQuasis.value.raw = namespaceQuasis.value.raw.replace(':', '.');
|
||
}
|
||
}
|
||
else {
|
||
// 自身ns
|
||
if (!argu0.quasis[0].value.raw || !argu0.quasis[0].value.raw.startsWith(ns)) {
|
||
argu0.quasis[0].value.raw = ns + '.' + argu0.quasis[0].value.raw || '';
|
||
}
|
||
}
|
||
}
|
||
else if (t.isIdentifier(argu0) || t.isExpression(argu0)) {
|
||
// 是变量或表达式，一定是自己的namespace下的
|
||
arguments.splice(
|
||
0,
|
||
1,
|
||
t.templateLiteral(
|
||
[
|
||
t.templateElement({
|
||
raw: ns + '.'
|
||
}),
|
||
t.templateElement({
|
||
raw: ''
|
||
})
|
||
],
|
||
[
|
||
argu0
|
||
]
|
||
)
|
||
);
|
||
}
|
||
else {
|
||
assert(false, 't函数调用的第一个参数只能是字符串、模板或表达式');
|
||
}
|
||
}
|
||
}
|
||
},
|
||
},
|
||
};
|
||
}; |