feat: remove unused imports

README.md

@@ -158,11 +158,12 @@ If you want the plugin to match on all functions with a specific name, no matter
 | ------------------------ | --------- | ------------- |
 | `removeUnnecessaryCalls` | `boolean` | `true`        |
 
-By default the plugin will remove unnecessary function calls, if for some reason you need to keep them, you can set this option to false.
+By default the plugin will remove unnecessary function calls and if all calls are removed, imports. If you need to keep them, you can set this option to false.
 
 Example of some unnecessary calls
 
 ```javascript
+import clsx from 'clsx';
 const x = clsx('foo', 'bar');
 const y = clsx({ classA: foo === 'a', classB: foo !== 'a' });
 const z = clsx({

src/index.js

@@ -17,6 +17,7 @@ const visitors = [
   combineArguments,
   createConditionalExpression,
   removeUnnecessaryCalls,
+  (path, options) => findFunctionNames(path, { ...options, _removeUnusedImports: true }),
 ];
 
 export default () => ({

src/visitors/findFunctionNames.js

@@ -1,24 +1,29 @@
 import * as t from '@babel/types';
 
 export default (path, options) => {
-  if (options.libraries.length === 0) {
+  if (
+    options.libraries.length === 0 ||
+    (options._removeUnusedImports && !options.removeUnnecessaryCalls)
+  ) {
     return;
   }
 
-  path.node.body.forEach(item => {
+  path.get('body').forEach(nodePath => {
+    const item = nodePath.node;
+
     // import x from 'y';
-    if (t.isImportDeclaration(item)) {
-      if (
-        options.libraries.includes(item.source.value) &&
-        item.specifiers.length === 1 &&
-        t.isImportDefaultSpecifier(item.specifiers[0])
-      ) {
-        options.functionNames.push(item.specifiers[0].local.name);
-      }
+    if (
+      t.isImportDeclaration(item) &&
+      options.libraries.includes(item.source.value) &&
+      item.specifiers.length === 1 &&
+      t.isImportDefaultSpecifier(item.specifiers[0])
+    ) {
+      addOrRemoveNode(nodePath, item.specifiers[0].local.name);
     }
     // const x = require('y');
     else if (t.isVariableDeclaration(item)) {
-      item.declarations.forEach(dec => {
+      nodePath.get('declarations').forEach(decPath => {
+        const dec = decPath.node;
         if (
           t.isVariableDeclarator(dec) &&
           t.isCallExpression(dec.init) &&
@@ -27,9 +32,22 @@ export default (path, options) => {
           t.isLiteral(dec.init.arguments[0]) &&
           options.libraries.includes(dec.init.arguments[0].value)
         ) {
-          options.functionNames.push(dec.id.name);
+          addOrRemoveNode(decPath, dec.id.name);
         }
       });
     }
+
+    function addOrRemoveNode(itemPath, name) {
+      if (!options._removeUnusedImports) {
+        options.functionNames.push(name);
+        return;
+      }
+
+      // Need to refresh references
+      itemPath.scope.crawl();
+      if (!itemPath.scope.getBinding(name).referenced) {
+        itemPath.remove();
+      }
+    }
   });
 };

test/fixtures/find-function-names/unused-import/code.js

@@ -0,0 +1,5 @@
+import clsx from 'clsx';
+const cn = require('classnames');
+
+const x = clsx('foo');
+const y = cn('bar');

test/fixtures/find-function-names/unused-import/options.json

@@ -0,0 +1,3 @@
+{
+  "removeUnnecessaryCalls": true
+}

test/fixtures/find-function-names/unused-import/output.js

@@ -0,0 +1,2 @@
+const x = 'foo';
+const y = 'bar';