openpanel/packages/codemod/src/helpers/separate-imports.ts

import type { Collection, ImportSpecifier, JSCodeshift } from "jscodeshift";

/**
 * Separate imports from currentLibName to nextLibName
 * import { Create, Button, Form } from "@pankod/refine-antd";
 * to
 * import { Create } from "@pankod/refine-antd";
 * import { Button, Form } from "antd";
 */
export function separateImports(payload: {
  j: JSCodeshift;
  source: Collection<any>;
  imports: string[];
  renameImports: { [key: string]: string };
  renameToDefault?: { [key: string]: string };
  otherImports: { [key: string]: string };
  currentLibName: string;
  nextLibName: string;
}) {
  const {
    j,
    source,
    imports,
    renameImports,
    renameToDefault = {},
    otherImports,
    currentLibName,
    nextLibName,
  } = payload;
  const nextLibImports: ImportSpecifier[] = [];
  const otherImportItems: ImportSpecifier[] = [];

  const refineImport = source.find(j.ImportDeclaration, {
    source: {
      value: currentLibName,
    },
  });

  refineImport.replaceWith((path) => {
    for (const item of path.node.specifiers) {
      if (renameToDefault[(item as ImportSpecifier)?.imported?.name]) {
        // means the imported element is going to be converted to default import from a package (value of renameToDefault)
        const pkg = renameToDefault[(item as ImportSpecifier)?.imported?.name];

        const localName = (item as ImportSpecifier)?.local?.name;

        // remove
        path.node.specifiers = path.node.specifiers.filter((p) => p !== item);

        // insert
        j(path)
          .at(0)
          .insertAfter(
            j.importDeclaration(
              [j.importDefaultSpecifier(j.identifier(localName))],
              j.stringLiteral(pkg),
            ),
          );
      } else {
        // also checking for imported name which is actually the correct one to check for renamed ones
        if (imports.includes((item as ImportSpecifier)?.imported?.name)) {
          nextLibImports.push(item as ImportSpecifier);
        }

        if (otherImports[(item as ImportSpecifier)?.imported?.name]) {
          otherImportItems.push(item as ImportSpecifier);
        }
      }
    }

    path.node.specifiers = path.node.specifiers
      .filter((p) => !nextLibImports.includes(p as ImportSpecifier))
      .filter((p) => !otherImportItems.includes(p as ImportSpecifier));

    return path.node;
  });

  if (nextLibImports.length > 0) {
    // rename imports
    nextLibImports.forEach((item) => {
      if (renameImports[item.imported.name]) {
        // if imported and local names are different, then means its renamed. we can continue to do our swap
        // if imported and local are same, means we're changing the imported elements name, then we need to change the local name as well
        if (item.imported.name !== item.local.name) {
          // if the renamed name is the same as local name, then we don't need to rename it
          if (item.local.name === renameImports[item.imported.name]) {
            item.imported.name = renameImports[item.imported.name];
            item.local = undefined;
          } else {
            item.imported.name = renameImports[item.imported.name];
          }
        } else if (item.imported.name === item.local.name) {
          item.imported.name = `${renameImports[item.imported.name]} as ${
            item.imported.name
          }`;
        }
      }
    });

    source
      .find(j.ImportDeclaration, {
        source: {
          value: currentLibName,
        },
      })
      .forEach((path, i) => {
        if (i === 0) {
          path.insertAfter(
            j.importDeclaration(nextLibImports, j.literal(nextLibName)),
          );
        }
      });
  }

  // add other imports
  if (otherImportItems.length > 0) {
    const otherImportPaths: { [key: string]: ImportSpecifier[] } = {};
    otherImportItems.forEach((item) => {
      // find import path
      const importPath =
        otherImports[(item as ImportSpecifier)?.imported?.name];

      if (otherImportPaths[importPath]) {
        otherImportPaths[importPath].push(item);
      } else {
        otherImportPaths[importPath] = [item];
      }
    });

    Object.keys(otherImportPaths).forEach((importPath) => {
      source
        .find(j.ImportDeclaration, {
          source: {
            value: currentLibName,
          },
        })
        .forEach((path, i) => {
          if (i === 0) {
            path.insertAfter(
              j.importDeclaration(
                otherImportPaths[importPath],
                j.literal(importPath),
              ),
            );
          }
        });
    });
  }

  // remove empty imports
  source
    .find(j.ImportDeclaration, {
      source: {
        value: currentLibName,
      },
    })
    .filter((path) => path.node.specifiers.length === 0)
    .remove();
}