45 lines
929 B
JavaScript
45 lines
929 B
JavaScript
import { types as t } from "@babel/core";
|
|
|
|
export default function() {
|
|
function statementList(key, path) {
|
|
const paths: Array = path.get(key);
|
|
|
|
for (const path of paths) {
|
|
const func = path.node;
|
|
if (!path.isFunctionDeclaration()) continue;
|
|
|
|
const declar = t.variableDeclaration("let", [
|
|
t.variableDeclarator(func.id, t.toExpression(func)),
|
|
]);
|
|
|
|
// hoist it up above everything else
|
|
declar._blockHoist = 2;
|
|
|
|
// todo: name this
|
|
func.id = null;
|
|
|
|
path.replaceWith(declar);
|
|
}
|
|
}
|
|
|
|
return {
|
|
visitor: {
|
|
BlockStatement(path) {
|
|
const { node, parent } = path;
|
|
if (
|
|
t.isFunction(parent, { body: node }) ||
|
|
t.isExportDeclaration(parent)
|
|
) {
|
|
return;
|
|
}
|
|
|
|
statementList("body", path);
|
|
},
|
|
|
|
SwitchCase(path) {
|
|
statementList("consequent", path);
|
|
},
|
|
},
|
|
};
|
|
}
|