35 lines
1.1 KiB
JavaScript
35 lines
1.1 KiB
JavaScript
|
import { t, visit } from "./babel.js";
|
||
|
|
function ensureImport(root, importDeclaration) {
|
||
|
|
let specifiersToFind = [...importDeclaration.specifiers];
|
||
|
|
visit(root, {
|
||
|
|
ImportDeclaration(path) {
|
||
|
|
if (path.node.source.value === importDeclaration.source.value) {
|
||
|
|
path.node.specifiers.forEach(
|
||
|
|
(specifier) => specifiersToFind.forEach((specifierToFind, i) => {
|
||
|
|
if (specifier.type !== specifierToFind.type)
|
||
|
|
return;
|
||
|
|
if (specifier.local.name === specifierToFind.local.name) {
|
||
|
|
specifiersToFind.splice(i, 1);
|
||
|
|
}
|
||
|
|
})
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
if (specifiersToFind.length === 0)
|
||
|
|
return;
|
||
|
|
visit(root, {
|
||
|
|
Program(path) {
|
||
|
|
const declaration = t.importDeclaration(specifiersToFind, importDeclaration.source);
|
||
|
|
const latestImport = path.get("body").filter((statement) => statement.isImportDeclaration()).pop();
|
||
|
|
if (latestImport)
|
||
|
|
latestImport.insertAfter(declaration);
|
||
|
|
else
|
||
|
|
path.unshiftContainer("body", declaration);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
export {
|
||
|
|
ensureImport
|
||
|
|
};
|