48 lines
1.0 KiB
JavaScript
48 lines
1.0 KiB
JavaScript
import * as t from "@babel/types";
|
|
|
|
const visitor = {
|
|
Scope(path, state) {
|
|
if (state.kind === "let") path.skip();
|
|
},
|
|
|
|
Function(path) {
|
|
path.skip();
|
|
},
|
|
|
|
VariableDeclaration(path, state) {
|
|
if (state.kind && path.node.kind !== state.kind) return;
|
|
|
|
const nodes = [];
|
|
|
|
const declarations: Array<Object> = path.get("declarations");
|
|
let firstId;
|
|
|
|
for (const declar of declarations) {
|
|
firstId = declar.node.id;
|
|
|
|
if (declar.node.init) {
|
|
nodes.push(
|
|
t.expressionStatement(
|
|
t.assignmentExpression("=", declar.node.id, declar.node.init),
|
|
),
|
|
);
|
|
}
|
|
|
|
for (const name in declar.getBindingIdentifiers()) {
|
|
state.emit(t.identifier(name), name);
|
|
}
|
|
}
|
|
|
|
// for (var i in test)
|
|
if (path.parentPath.isFor({ left: path.node })) {
|
|
path.replaceWith(firstId);
|
|
} else {
|
|
path.replaceWithMultiple(nodes);
|
|
}
|
|
},
|
|
};
|
|
|
|
export default function(path, emit: Function, kind: "var" | "let" = "var") {
|
|
path.traverse(visitor, { kind, emit });
|
|
}
|