import { ImportDeclaration } from "estree-jsx";

// TODO: Swap the params so that packageName has a default value of pf/react-core and can be optional
export function pfPackageMatches(
  packageName: string,
  nodeSrc: ImportDeclaration["source"]["value"]
) {
  if (typeof nodeSrc !== "string") {
    return false;
  }

  const parts = packageName.split("/");
  const regex = new RegExp(
    "^" +
      parts[0] +
      "/" +
      parts[1] +
      "(/dist/(esm|js|dynamic))?" +
      (parts[2] ? "/" + parts[2] : "") +
      (["react-component-groups", "react-tokens", "react-user-feedback"].includes(parts[1])
        ? `(/.*)?$`
        : `(/(components|helpers${
            parts[1] === "react-icons" ? "|icons" : ""
          })/.*)?$`)
  );
  return regex.test(nodeSrc);
}