codemod script to remove useEffect if it's not being used

removeUnusedUseEffectImport.js

module.exports = function (file, api) {
  const j = api.jscodeshift;
  const { CallExpression, Identifier } = j;

  return getReactImportStatementNode()
    .forEach((node, index) => {
      node.value.specifiers.forEach((specifier, index) => {
        if (specifier.local.name === "useEffect") {
          if (!shouldKeepUseEffectImport()) {
            delete node.value.specifiers[index];
          }
        }
      });
    })
    .toSource();

  function shouldKeepUseEffectImport() {
    return (
      j(file.source)
        .find(j.CallExpression)
        .filter((node, index) => {
          return node.value.callee.name === "useEffect";
        })
        .size() > 0
    );
  }

  function getReactImportStatementNode() {
    return j(file.source)
      .find(j.ImportDeclaration) // where it's from 'react'
      .filter((node, index) => {
        return node.value.source.value === "react";
      });
  }
};