A codemod that adds a unique `data-testid` to a list of jsx components that are imported from a specific path

addTestIdToJSX.js

// addUuidToButton.js
const j = require("jscodeshift");
const crypto = require("crypto");

function generateUUID(filePath, position) {
  const str = filePath + JSON.stringify(position);
  return crypto.createHash("md5").update(str).digest("hex").substr(0, 5);
}

const importPaths = ["components"];
const components = ["Button", "ToggleButton"];

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

  let isComponentImported = false;
  const importDeclarations = root.find(j.ImportDeclaration);
  importDeclarations.forEach(dec => {
    const source = dec?.value?.source;
    if (
      source?.type === "StringLiteral" &&
      importPaths.includes(source?.value)
    ) {
      isComponentImported = true;
    }
  });

  if (!isComponentImported) {
    return fileInfo.source;
  }

  components.forEach(component => {
    root
      .find(j.JSXElement, {
        openingElement: {
          name: {
            name: component,
          },
        },
      })
      .forEach(path => {
        const existingProps = path.node.openingElement.attributes;

        if (
          existingProps.some(
            prop => prop.name && prop.name.name === "data-testid",
          )
        ) {
          return;
        }

        const newUuidProp = j.jsxAttribute(
          j.jsxIdentifier("data-testid"),
          j.stringLiteral(generateUUID(fileInfo.path, path.node.loc.start)),
        );

        existingProps.push(newUuidProp);
      });
  });

  return root.toSource();
};