lodash-es6-imports.tsx

/**
 * Converts lodash imports to use named imports.
 *
 * E.g.:
 * ```
 * import noop from 'lodash/noop';
 * ```
 * ... to ...
 * ```
 * import { noop } from 'lodash';
 * ```
 *
 * Usage:
 * ```sh
 * npx jscodeshift -t lodash-es6-imports.tsx src --parser=tsx
 * ```
 */

import type { API, FileInfo } from 'jscodeshift';

export default function transform(file: FileInfo, api: API) {
  const j = api.jscodeshift;
  const root = j(file.source);

  root.find(j.ImportDeclaration).forEach(element => {
    const importPath = element.value.source.value?.toString();

    const { node } = element;
    if (!importPath?.startsWith('lodash/')) {
      return;
    }

    const fnNameImport = importPath.split('/').pop();
    if (!fnNameImport) {
      throw new Error(`Could not parse import path: ${importPath}`);
    }

    const localImportName = node.specifiers?.[0]?.local?.name;
    if (localImportName !== fnNameImport) {
      throw new Error(
        `Import name ${localImportName} does not match function name ${fnNameImport}`
      );
    }

    node.specifiers = [j.importSpecifier(j.identifier(fnNameImport))];
    node.source = j.literal(`lodash`);
  });

  return root.toSource();
}