Merge pull request #3 from Fimba-Code/fix/pluging-module-export

Fixed parser

Author: brunokiafuka

.DS_Store

@@ -0,0 +1,3 @@
+{
+  "plugins": ["./index.js"]
+}

.babelrc

@@ -0,0 +1,6 @@
+function add(a, b) {
+    $log;
+    const result = a + b;
+    return result;
+}
+add(2, 3);

example/index.js

@@ -0,0 +1,29 @@
+
+module.exports = function (babel) {
+  const { types: t } = babel;
+
+  let hasLogID = false;
+
+  return {
+    name: "log-transform",
+    visitor: {
+      ReturnStatement: (path) => {
+        const identifierName = path.node.argument.name;
+        if (path.node.argument.type === "Identifier" && hasLogID) {
+          path.replaceWithMultiple([
+            t.identifier(
+              `console.log('Final Result 😛 ==> ', ${identifierName})`
+            ),
+            t.identifier(`return ${identifierName}`),
+          ]);
+        }
+      },
+      Identifier: (path) => {
+        if (path.isIdentifier({ name: "$log" })) {
+          hasLogID = true;
+          path.remove();
+        }
+      },
+    },
+  };
+}