Merge branch 'master' of https://github.com/jmdejno/ember-getset-util…

…-codemod
seanjohnson08 · Oct 21, 2019 · ea37647 · ea37647
2 parents d53542c + cb74866
commit ea37647
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/utils/imports.js b/utils/imports.js
@@ -0,0 +1,19 @@
+function ensureImport(j, root, name, source, local) {
+  const declaration = root.find(j.ImportDeclaration, {
+    source: { value: source },
+  });
+  const specifer = j.importSpecifier(j.identifier(name), local ? j.identifier(local) : null);
+  if (declaration.size()) {
+    if (!declaration.find(j.ImportSpecifier, { imported: { name } }).size()) {
+      declaration.get('specifiers').push(specifer);
+    }
+  } else {
+    const newImport = j.importDeclaration([specifer], j.literal(source));
+    root
+      .get('program')
+      .get('body')
+      .insertAt(0, newImport);
+  }
+}
+
+module.exports = { ensureImport };