You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
2.7 KiB
108 lines
2.7 KiB
import assert from 'assert';
|
|
import Plugin from './Plugin';
|
|
|
|
export default function ({ types }) {
|
|
let plugins = null;
|
|
|
|
// Only for test
|
|
global.__clearBabelAntdPlugin = () => {
|
|
plugins = null;
|
|
};
|
|
|
|
function applyInstance(method, args, context) {
|
|
for (const plugin of plugins) {
|
|
if (plugin[method]) {
|
|
plugin[method].apply(plugin, [...args, context]);
|
|
}
|
|
}
|
|
}
|
|
|
|
const Program = {
|
|
enter(path, { opts = {} }) {
|
|
// Init plugin instances once.
|
|
if (!plugins) {
|
|
if (Array.isArray(opts)) {
|
|
plugins = opts.map(({
|
|
libraryName,
|
|
libraryDirectory,
|
|
style,
|
|
styleLibraryDirectory,
|
|
customStyleName,
|
|
camel2DashComponentName,
|
|
camel2UnderlineComponentName,
|
|
fileName,
|
|
customName,
|
|
transformToDefaultImport,
|
|
}, index) => {
|
|
assert(libraryName, 'libraryName should be provided');
|
|
return new Plugin(
|
|
libraryName,
|
|
libraryDirectory,
|
|
style,
|
|
styleLibraryDirectory,
|
|
customStyleName,
|
|
camel2DashComponentName,
|
|
camel2UnderlineComponentName,
|
|
fileName,
|
|
customName,
|
|
transformToDefaultImport,
|
|
types,
|
|
index
|
|
);
|
|
});
|
|
} else {
|
|
assert(opts.libraryName, 'libraryName should be provided');
|
|
plugins = [
|
|
new Plugin(
|
|
opts.libraryName,
|
|
opts.libraryDirectory,
|
|
opts.style,
|
|
opts.styleLibraryDirectory,
|
|
opts.customStyleName,
|
|
opts.camel2DashComponentName,
|
|
opts.camel2UnderlineComponentName,
|
|
opts.fileName,
|
|
opts.customName,
|
|
opts.transformToDefaultImport,
|
|
types
|
|
),
|
|
];
|
|
}
|
|
}
|
|
applyInstance('ProgramEnter', arguments, this); // eslint-disable-line
|
|
},
|
|
exit() {
|
|
applyInstance('ProgramExit', arguments, this); // eslint-disable-line
|
|
},
|
|
};
|
|
|
|
const methods = [
|
|
'ImportDeclaration',
|
|
'CallExpression',
|
|
'MemberExpression',
|
|
'Property',
|
|
'VariableDeclarator',
|
|
'ArrayExpression',
|
|
'LogicalExpression',
|
|
'ConditionalExpression',
|
|
'IfStatement',
|
|
'ExpressionStatement',
|
|
'ReturnStatement',
|
|
'ExportDefaultDeclaration',
|
|
'BinaryExpression',
|
|
'NewExpression',
|
|
'ClassDeclaration',
|
|
];
|
|
|
|
const ret = {
|
|
visitor: { Program },
|
|
};
|
|
|
|
for (const method of methods) {
|
|
ret.visitor[method] = function () { // eslint-disable-line
|
|
applyInstance(method, arguments, ret.visitor); // eslint-disable-line
|
|
};
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|