fix(remove): fix

almin · Jan 17, 2018 · 329c5cd · 329c5cd
1 parent 49a5781
commit 329c5cd
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/scripts/remove-ChangedPayload.js b/scripts/remove-ChangedPayload.js
@@ -52,26 +52,29 @@ function isNewExpressionChangedPayload(path) {
 module.exports = function transformer(file, api) {
     const j = api.jscodeshift;
     // Remove `import { ChangedPayload } from "almin"
+    let hasAlmin = false;
+
     return (
         j(
             j(file.source)
                 .find(j.ImportSpecifier)
                 .filter(path => {
-                    return isChangedPayloadImportSpecifier(path);
+                    hasAlmin = isChangedPayloadImportSpecifier(path);
+                    return hasAlmin;
                 })
                 .remove()
                 .toSource()
         )
             // Rewrite new ChangedPayload() => { type: "ChangedPayload" }
             .find(j.NewExpression)
             .filter(path => {
-                let newExpressionChangedPayload = isNewExpressionChangedPayload(path);
-                console.log(newExpressionChangedPayload);
-                return newExpressionChangedPayload;
+                if (!hasAlmin) {
+                    return false;
+                }
+                return isNewExpressionChangedPayload(path);
             })
             .replaceWith(path => {
                 return `{ type: "ChangedPayload" }`;
-                R;
             })
             .toSource()
     );