Only remove export source if it belongs to Lodash. [closes #53]

lodash · May 4, 2016 · 62c4328 · 62c4328
1 parent 99eee01
commit 62c4328
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/index.js b/src/index.js
@@ -87,8 +87,8 @@ export default function({ 'types': types }) {
       const { file } = path.hub;
       const { node } = path;
       const { value: pkgId } = node.source;
-
       const pkgStore = store.get(pkgId);
+
       if (!pkgStore) {
         return;
       }
@@ -168,13 +168,13 @@ export default function({ 'types': types }) {
     ExportNamedDeclaration(path) {
       const { node } = path;
       const { file } = path.hub;
-
       const pkgId = _.get(node, 'source.value');
       const pkgStore = store.get(pkgId);
       const importBase = getImportBase(pkgStore);
 
-      node.source = null;
-
+      if (pkgStore) {
+        node.source = null;
+      }
       _.each(node.specifiers, spec => {
         const localName = spec.local.name;
         spec.local = pkgStore