2017-03-04 10:46:01 -05:00

42 lines
1.2 KiB
JavaScript

import syntaxExportExtensions from "babel-plugin-syntax-export-extensions";
export default function ({ types: t }) {
function build(node, nodes, scope) {
const first = node.specifiers[0];
if (!t.isExportNamespaceSpecifier(first) && !t.isExportDefaultSpecifier(first)) return;
const specifier = node.specifiers.shift();
const uid = scope.generateUidIdentifier(specifier.exported.name);
let newSpecifier;
if (t.isExportNamespaceSpecifier(specifier)) {
newSpecifier = t.importNamespaceSpecifier(uid);
} else {
newSpecifier = t.importDefaultSpecifier(uid);
}
nodes.push(t.importDeclaration([newSpecifier], node.source));
nodes.push(t.exportNamedDeclaration(null, [t.exportSpecifier(uid, specifier.exported)]));
build(node, nodes, scope);
}
return {
inherits: syntaxExportExtensions,
visitor: {
ExportNamedDeclaration(path) {
const { node, scope } = path;
const nodes = [];
build(node, nodes, scope);
if (!nodes.length) return;
if (node.specifiers.length >= 1) {
nodes.push(node);
}
path.replaceWithMultiple(nodes);
},
},
};
}