javascript - 带有 jscodeshift 的 Codemod - 从导入中删除逗号

Question

我正在尝试编写一个小型 codemod 来重构一些代码。考虑一下我有这样的东西:

import { mod1, mod2, mod3 } from 'package1'
import localMod from 'package2'

我想将其更改为:

import { mod1, mod3 } from 'package1'
import * as mod2 from 'new-package'
import localMod from 'package2'

作为第一步，我尝试删除 mod2从我成功完成的第一行导入，但我无法删除 mod1 之后的逗号.

到目前为止，我的代码片段如下所示:

module.exports = function transformer(file, api) {
  const j = api.jscodeshift;
  const root = j(file.source);

  const reactApolloImports = root
    .find(j.ImportDeclaration)
    .filter(nodepath => nodepath.value.source.value === "package1")
    .find(j.Identifier)
    .forEach(nodepath => {
      if (nodepath.name === "imported" && nodepath.node.name === "mod2") {
        j(nodepath).remove();
      }
    });

  return root.toSource();
};

请帮忙。

Answer 1

不是在 j.ImportDeclaration 中搜索和删除 j.Identifier，而是通过 j.ImportSpecifier 表达式找到它们的父节点。如果删除这些，应该没问题。

如果您将 import { mod1, mod2, mod3 } from 'package1' 语句粘贴到 AST Explorer 中，则更容易理解。

代码:

module.exports = function transformer(file, api) {
  const j = api.jscodeshift;
  const root = j(file.source);

  const reactApolloImports = root
    .find(j.ImportDeclaration)
    .filter(impDecNodePath => impDecNodePath.value.source.value === "package1")
    .forEach(impDecNodePathFiltered => {
      j(impDecNodePathFiltered)
        // find ImportSpecifier here instead of Identifier
        .find(j.ImportSpecifier)
        .forEach(impSpecNodePath => {
          if (impSpecNodePath.node.imported.name === "mod2") {
            j(impSpecNodePath).remove();
          }
        });
    });

  return root.toSource();
};

更改输出:

import { mod1, mod3 } from 'package1';
import localMod from 'package2'

希望对您有所帮助。