feat(collectcalls): include the location of the code

merceyz · Aug 30, 2019 · 1ab1eab · 1ab1eab
1 parent 8b59c5d
commit 1ab1eab
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/src/combineVisitors.js b/src/combineVisitors.js
@@ -92,8 +92,9 @@ export default function combineVisitors(visitors) {
           });
         }
 
-        return (path, options) => {
+        return (path, options, state) => {
           path.traverse(combinedVisitors, {
+            state,
             options,
             functionCounters: path.functionCounters,
           });

diff --git a/src/index.js b/src/index.js
@@ -40,7 +40,7 @@ export default () => ({
 
       try {
         for (const visitor of visitors) {
-          visitor(path, options);
+          visitor(path, options, state);
         }
       } catch (err) {
         throw err;

diff --git a/src/visitors/collectCalls.js b/src/visitors/collectCalls.js
@@ -12,11 +12,19 @@ export default {
     }
 
     if (stream === null) {
-      const filePath = osPath.join(__dirname, 'log.js');
+      const filePath = osPath.join(
+        __dirname,
+        `log-${new Date(Date.now()).toISOString().replace(/:/g, '.')}.js`,
+      );
       stream = fs.createWriteStream(filePath, { flags: 'w' });
       console.log('Writing calls to ' + filePath);
     }
 
-    stream.write(`const x${++count} = ${generate(path.node).code};\n\n`);
+    const location = path.node.loc.start;
+    stream.write(
+      `// ${this.state.filename}:${location.line}:${location.column}\nconst x${++count} = ${
+        generate(path.node).code
+      };\n\n`,
+    );
   },
 };