import type { Plugin } from "esbuild";

export const removeTestIdsPlugin: Plugin = {
  name: "react-remove-testids",
  setup(build) {
    build.onEnd((args) => {
      // data-testid regexp
      const regexp = /("data-testid":)(.*?)(?:(,)|(})|(\n))/gi;

      // output files with `*.js`
      const jsOutputFiles =
        args.outputFiles?.filter(
          (el) =>
            el.path.endsWith(".cjs") ||
            el.path.endsWith(".mjs") ||
            el.path.endsWith(".js"),
        ) ?? [];

      // replace data-testid in output files
      for (const jsOutputFile of jsOutputFiles) {
        const str = new TextDecoder("utf-8").decode(jsOutputFile.contents);
        const newStr = str.replace(regexp, "$4");
        jsOutputFile.contents = new TextEncoder().encode(newStr);
      }
    });
  },
};