feat: collect calls before optimizing

README.md

@@ -179,3 +179,11 @@ const x = 'foo bar';
 const y = foo === 'a' ? 'classA' : 'classB';
 const z = (foo === 'a' ? 'classA ' : 'classB ') + (bar === 'c' ? 'classC' : 'classD');
 ```
+
+---
+
+| Name           | Type      | Default value |
+| -------------- | --------- | ------------- |
+| `collectCalls` | `boolean` | `false`       |
+
+Writes all function calls, before they are optimized, to a file. Used to help test the plugin on repositories.

src/index.js

@@ -8,9 +8,11 @@ import combineArguments from './visitors/combineArguments';
 import createConditionalExpression from './visitors/createConditionalExpression';
 import removeUnnecessaryCalls from './visitors/removeUnnecessaryCalls';
 import createObjectKeyLookups from './visitors/createObjectKeyLookups';
+import collectCalls from './visitors/collectCalls';
 
 const visitors = [
   findFunctionNames,
+  collectCalls,
   extractObjectProperties,
   propTypes,
   stripLiterals,

src/options.js

@@ -3,6 +3,7 @@ export function getOptions(options) {
     libraries: ['clsx', 'classnames'],
     functionNames: [],
     removeUnnecessaryCalls: true,
+    collectCalls: false,
   };
   return { ...DefaultSettings, ...options };
 }

src/visitors/collectCalls.js

@@ -0,0 +1,32 @@
+import * as t from '@babel/types';
+import generate from '@babel/generator';
+import fs from 'fs';
+import osPath from 'path';
+
+let stream = null;
+let count = 0;
+
+const visitor = {
+  CallExpression(path) {
+    const c = path.node.callee;
+    if (!t.isIdentifier(c) || !this.options.functionNames.includes(c.name)) {
+      return;
+    }
+
+    stream.write(`const x${++count} = ${generate(path.node).code};\n\n`);
+  },
+};
+
+export default (path, options) => {
+  if (!options.collectCalls) {
+    return;
+  }
+
+  if (stream === null) {
+    const filePath = osPath.join(__dirname, 'log.js');
+    stream = fs.createWriteStream(filePath, { flags: 'w' });
+    console.log('Writing calls to ' + filePath);
+  }
+
+  path.traverse(visitor, { options });
+};