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.
32 lines
992 B
32 lines
992 B
9 years ago
|
'use strict';
|
||
|
|
|
||
|
|
const babel = require('babel-core');
|
||
|
|
const recast = require('recast');
|
||
|
|
const builders = recast.types.builders;
|
||
|
|
|
||
|
|
const babelrc = {
|
||
|
|
presets: ['es2015', 'react']
|
||
|
|
};
|
||
|
|
|
||
|
|
// const demo = 'import { Button } from \'antd\';' +
|
||
|
|
// 'ReactDOM.render(<div>' +
|
||
|
|
// ' <Button type="primary">主按钮</Button>' +
|
||
|
|
// ' <Button>次按钮</Button>' +
|
||
|
|
// ' <Button type="ghost">幽灵按钮</Button>' +
|
||
|
|
// '</div>,' +
|
||
|
|
// 'document.getElementById(\'components-button-demo-basic\'));';
|
||
|
|
// devil(demo);
|
||
|
|
module.exports = function devil(demo, params) {
|
||
|
|
const compiled = babel.transform(demo, babelrc).code;
|
||
|
|
|
||
|
|
const ast = recast.parse(compiled);
|
||
|
|
const astProgramBody = ast.program.body;
|
||
|
|
const lastIndex = astProgramBody.length - 1;
|
||
|
|
astProgramBody[lastIndex] = builders.returnStatement(
|
||
|
|
astProgramBody[lastIndex].expression.arguments[0]
|
||
|
|
);
|
||
|
|
|
||
|
|
const code = recast.print(ast).code;
|
||
|
|
return new Function((params || []).join(', '), code);
|
||
|
|
}
|