react-code-locator 0.1.14 → 0.1.16

package/dist/unplugin.js

@@ -0,0 +1,278 @@
+// src/unplugin.ts
+import { createUnplugin } from "unplugin";
+
+// src/core/transform.ts
+import { parse } from "acorn";
+import { generate } from "astring";
+import { walk } from "estree-walker";
+
+// src/constants.ts
+var SOURCE_PROP = "__componentSourceLoc";
+var JSX_SOURCE_PROP = "$componentSourceLoc";
+var JSX_SOURCE_REGISTRY_SYMBOL = "react-code-locator.jsxSourceRegistry";
+
+// src/core/transform.ts
+function toRelativeSource(filename, loc, projectRoot) {
+  const root = projectRoot || process.cwd();
+  const relativePath = filename.startsWith(root) ? filename.slice(root.length + 1) : filename;
+  return `${relativePath}:${loc.line}:${loc.column + 1}`;
+}
+function isComponentName(name) {
+  return /^[A-Z]/.test(name);
+}
+function isReactElementFactoryCall(node) {
+  const callee = node.callee;
+  if (callee.type === "Identifier") {
+    const name = callee.name;
+    return [
+      "jsx",
+      "jsxs",
+      "jsxDEV",
+      "_jsx",
+      "_jsxs",
+      "_jsxDEV",
+      "createElement"
+    ].includes(name);
+  }
+  if (callee.type === "MemberExpression") {
+    const obj = callee.object;
+    const prop = callee.property;
+    if (obj.type === "Identifier" && obj.name === "React" && prop.type === "Identifier" && prop.name === "createElement") {
+      return true;
+    }
+  }
+  return false;
+}
+function isSupportedComponentInit(node) {
+  if (!node) return false;
+  if (node.type === "ArrowFunctionExpression" || node.type === "FunctionExpression") {
+    return true;
+  }
+  if (node.type !== "CallExpression") return false;
+  const callee = node.callee;
+  if (callee.type === "Identifier") {
+    return ["memo", "forwardRef"].includes(callee.name);
+  }
+  if (callee.type === "MemberExpression") {
+    const obj = callee.object;
+    const prop = callee.property;
+    if (obj.type === "Identifier" && obj.name === "React" && prop.type === "Identifier") {
+      return ["memo", "forwardRef"].includes(prop.name);
+    }
+  }
+  return false;
+}
+function createSourceAssignment(name, sourceValue) {
+  return {
+    type: "ExpressionStatement",
+    expression: {
+      type: "AssignmentExpression",
+      operator: "=",
+      left: {
+        type: "MemberExpression",
+        object: { type: "Identifier", name },
+        property: { type: "Identifier", name: SOURCE_PROP },
+        computed: false
+      },
+      right: { type: "Literal", value: sourceValue }
+    }
+  };
+}
+function createMarkElementHelper() {
+  const code = `
+function _markReactElementSource(element, source) {
+  const registryKey = Symbol.for("${JSX_SOURCE_REGISTRY_SYMBOL}");
+  let registry = globalThis[registryKey];
+  if (!(registry instanceof WeakMap)) {
+    registry = globalThis[registryKey] = new WeakMap();
+  }
+  if (element && typeof element === "object" && typeof element.props === "object") {
+    registry.set(element.props, source);
+  }
+  return element;
+}
+`;
+  return parse(code, { ecmaVersion: "latest" }).body[0];
+}
+function wrapWithMarkElement(node, sourceValue) {
+  return {
+    type: "CallExpression",
+    callee: { type: "Identifier", name: "_markReactElementSource" },
+    arguments: [node, { type: "Literal", value: sourceValue }],
+    optional: false
+  };
+}
+function transformSource(code, options) {
+  const {
+    filename,
+    projectRoot = process.cwd(),
+    injectJsxSource = true,
+    injectComponentSource = true
+  } = options;
+  let ast;
+  try {
+    ast = parse(code, {
+      ecmaVersion: "latest",
+      sourceType: "module",
+      ecmaFeatures: { jsx: true },
+      locations: true,
+      ranges: true
+    });
+  } catch (err) {
+    return null;
+  }
+  let modified = false;
+  let needsHelper = false;
+  const seenComponents = /* @__PURE__ */ new Set();
+  const assignments = [];
+  walk(ast, {
+    enter(node, parent, key, index) {
+      if (injectJsxSource && node.type === "CallExpression") {
+        if (isReactElementFactoryCall(node)) {
+          const loc = node.loc;
+          if (loc) {
+            const sourceValue = toRelativeSource(filename, loc.start, projectRoot);
+            const wrapped = wrapWithMarkElement(node, sourceValue);
+            this.replace(wrapped);
+            needsHelper = true;
+            modified = true;
+          }
+        }
+      }
+      if (injectJsxSource && node.type === "JSXElement") {
+        const opening = node.openingElement;
+        const loc = opening.loc;
+        if (loc) {
+          const sourceValue = toRelativeSource(filename, loc.start, projectRoot);
+          const name = opening.name;
+          if (name.type === "JSXIdentifier") {
+            if (name.name[0] === name.name[0].toLowerCase()) {
+            } else {
+              const sourceAttr = {
+                type: "JSXAttribute",
+                name: { type: "JSXIdentifier", name: JSX_SOURCE_PROP },
+                value: { type: "Literal", value: sourceValue }
+              };
+              opening.attributes.push(sourceAttr);
+              modified = true;
+            }
+          }
+        }
+      }
+      if (injectComponentSource && node.type === "FunctionDeclaration") {
+        const name = node.id?.name;
+        if (name && isComponentName(name) && !seenComponents.has(name)) {
+          const loc = node.loc;
+          if (loc) {
+            const sourceValue = toRelativeSource(filename, loc.start, projectRoot);
+            const assignment = createSourceAssignment(name, sourceValue);
+            assignments.push({ node, parent, assignment });
+            seenComponents.add(name);
+            modified = true;
+          }
+        }
+      }
+      if (injectComponentSource && node.type === "VariableDeclarator") {
+        const id = node.id;
+        if (id.type === "Identifier" && isComponentName(id.name) && !seenComponents.has(id.name)) {
+          const init = node.init;
+          if (init && isSupportedComponentInit(init)) {
+            const loc = node.loc || init.loc;
+            if (loc) {
+              const sourceValue = toRelativeSource(filename, loc.start, projectRoot);
+              const assignment = createSourceAssignment(id.name, sourceValue);
+              assignments.push({ node, parent, assignment });
+              seenComponents.add(id.name);
+              modified = true;
+            }
+          }
+        }
+      }
+    }
+  });
+  if (!modified) {
+    return null;
+  }
+  if (needsHelper) {
+    const helper = createMarkElementHelper();
+    const exists = ast.body.some(
+      (n) => n.type === "FunctionDeclaration" && n.id?.name === "_markReactElementSource"
+    );
+    if (!exists) {
+      ast.body.unshift(helper);
+    }
+  }
+  for (const { node, parent, assignment } of assignments.reverse()) {
+    if (parent && parent.type === "Program") {
+      const index = ast.body.indexOf(node);
+      if (index !== -1) {
+        ast.body.splice(index + 1, 0, assignment);
+      }
+    } else if (parent && parent.type === "ExportNamedDeclaration") {
+      const exportParent = ast.body.find(
+        (n) => n.type === "ExportNamedDeclaration" && n.declaration === parent
+      );
+      if (exportParent) {
+        const index = ast.body.indexOf(exportParent);
+        ast.body.splice(index + 1, 0, assignment);
+      }
+    } else if (parent && parent.type === "ExportDefaultDeclaration") {
+      const index = ast.body.indexOf(parent);
+      ast.body.splice(index + 1, 0, assignment);
+    }
+  }
+  const result = generate(ast, { indent: "  " });
+  return { code: result };
+}
+
+// src/unplugin.ts
+var DEFAULT_INCLUDE = /\.([jt]sx)$/;
+var DEFAULT_EXCLUDE = /node_modules/;
+function shouldTransform(id, include, exclude) {
+  const includePatterns = Array.isArray(include) ? include : [include];
+  const excludePatterns = Array.isArray(exclude) ? exclude : [exclude];
+  if (excludePatterns.some((pattern) => pattern.test(id))) {
+    return false;
+  }
+  return includePatterns.some((pattern) => pattern.test(id));
+}
+var unplugin = createUnplugin((options = {}) => {
+  const {
+    include = DEFAULT_INCLUDE,
+    exclude = DEFAULT_EXCLUDE,
+    projectRoot = process.cwd(),
+    injectComponentSource = true,
+    injectJsxSource = true
+  } = options;
+  return {
+    name: "react-code-locator",
+    transform(code, id) {
+      if (!shouldTransform(id, include, exclude)) {
+        return null;
+      }
+      const result = transformSource(code, {
+        filename: id,
+        projectRoot,
+        injectComponentSource,
+        injectJsxSource
+      });
+      return result;
+    }
+  };
+});
+var vitePlugin = unplugin.vite;
+var webpackPlugin = unplugin.webpack;
+var rollupPlugin = unplugin.rollup;
+var esbuildPlugin = unplugin.esbuild;
+var rspackPlugin = unplugin.rspack;
+var unplugin_default = unplugin;
+export {
+  unplugin_default as default,
+  esbuildPlugin,
+  rollupPlugin,
+  rspackPlugin,
+  unplugin,
+  vitePlugin,
+  webpackPlugin
+};
+//# sourceMappingURL=unplugin.js.map

package/dist/unplugin.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/unplugin.ts","../src/core/transform.ts","../src/constants.ts"],"sourcesContent":["/**\n * Unplugin-based universal adapter for all build tools\n * Supports: Vite, Webpack, Rollup, esbuild, Rspack\n * \n * Zero-dependency transform using acorn instead of Babel\n */\n\nimport { createUnplugin, type UnpluginInstance, type UnpluginOptions } from \"unplugin\";\nimport { transformSource, type TransformOptions } from \"./core/transform\";\n\nexport interface ReactCodeLocatorOptions extends Omit<TransformOptions, 'filename'> {\n  /** \n   * Enable source transform for component definitions\n   * @default true\n   */\n  injectComponentSource?: boolean;\n  \n  /** \n   * Enable source transform for JSX call sites\n   * @default true\n   */\n  injectJsxSource?: boolean;\n  \n  /**\n   * Project root for relative path calculation\n   * @default process.cwd()\n   */\n  projectRoot?: string;\n  \n  /**\n   * Include filter for file paths\n   * @default /\\.[jt]sx$/\n   */\n  include?: RegExp | RegExp[];\n  \n  /**\n   * Exclude filter for file paths\n   * @default /node_modules/\n   */\n  exclude?: RegExp | RegExp[];\n}\n\nexport type { TransformOptions };\n\nconst DEFAULT_INCLUDE = /\\.([jt]sx)$/;\nconst DEFAULT_EXCLUDE = /node_modules/;\n\nfunction shouldTransform(id: string, include: RegExp | RegExp[], exclude: RegExp | RegExp[]): boolean {\n  const includePatterns = Array.isArray(include) ? include : [include];\n  const excludePatterns = Array.isArray(exclude) ? exclude : [exclude];\n  \n  // Check exclude first\n  if (excludePatterns.some(pattern => pattern.test(id))) {\n    return false;\n  }\n  \n  // Then check include\n  return includePatterns.some(pattern => pattern.test(id));\n}\n\nexport const unplugin: UnpluginInstance<ReactCodeLocatorOptions | undefined, false> = \n  createUnplugin((options = {}) => {\n    const {\n      include = DEFAULT_INCLUDE,\n      exclude = DEFAULT_EXCLUDE,\n      projectRoot = process.cwd(),\n      injectComponentSource = true,\n      injectJsxSource = true,\n    } = options;\n\n    return {\n      name: \"react-code-locator\",\n      \n      transform(code, id) {\n        // Skip if doesn't match patterns\n        if (!shouldTransform(id, include, exclude)) {\n          return null;\n        }\n\n        // Perform transform using acorn (zero-dependency)\n        const result = transformSource(code, {\n          filename: id,\n          projectRoot,\n          injectComponentSource,\n          injectJsxSource,\n        });\n\n        return result;\n      },\n    } as UnpluginOptions;\n  });\n\n// Export individual build tool adapters\nexport const vitePlugin = unplugin.vite;\nexport const webpackPlugin = unplugin.webpack;\nexport const rollupPlugin = unplugin.rollup;\nexport const esbuildPlugin = unplugin.esbuild;\nexport const rspackPlugin = unplugin.rspack;\n\n// Default export for direct usage\nexport default unplugin;\n","/**\n * Zero-dependency source transform using acorn\n * No Babel required - pure JavaScript AST manipulation\n */\n\nimport { parse, type Node as AcornNode } from \"acorn\";\nimport { generate } from \"astring\";\nimport { walk } from \"estree-walker\";\nimport type {\n  Node,\n  Program,\n  FunctionDeclaration,\n  FunctionExpression,\n  ArrowFunctionExpression,\n  VariableDeclarator,\n  CallExpression,\n  MemberExpression,\n  Identifier,\n  ExpressionStatement,\n  AssignmentExpression,\n} from \"estree\";\nimport { SOURCE_PROP, JSX_SOURCE_PROP, JSX_SOURCE_REGISTRY_SYMBOL } from \"../constants\";\n\nexport interface TransformOptions {\n  filename: string;\n  projectRoot?: string;\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n}\n\ninterface Location {\n  line: number;\n  column: number;\n}\n\nfunction toRelativeSource(filename: string, loc: Location, projectRoot?: string): string {\n  const root = projectRoot || process.cwd();\n  const relativePath = filename.startsWith(root) \n    ? filename.slice(root.length + 1) \n    : filename;\n  return `${relativePath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction isComponentName(name: string): boolean {\n  return /^[A-Z]/.test(name);\n}\n\nfunction isReactElementFactoryCall(node: CallExpression): boolean {\n  const callee = node.callee;\n  \n  if (callee.type === \"Identifier\") {\n    const name = callee.name;\n    return [\n      \"jsx\", \"jsxs\", \"jsxDEV\",\n      \"_jsx\", \"_jsxs\", \"_jsxDEV\",\n      \"createElement\"\n    ].includes(name);\n  }\n  \n  // React.createElement\n  if (callee.type === \"MemberExpression\") {\n    const obj = callee.object;\n    const prop = callee.property;\n    if (obj.type === \"Identifier\" && obj.name === \"React\" && \n        prop.type === \"Identifier\" && prop.name === \"createElement\") {\n      return true;\n    }\n  }\n  \n  return false;\n}\n\nfunction isSupportedComponentInit(\n  node: Node | null\n): boolean {\n  if (!node) return false;\n  \n  if (node.type === \"ArrowFunctionExpression\" || \n      node.type === \"FunctionExpression\") {\n    return true;\n  }\n  \n  if (node.type !== \"CallExpression\") return false;\n  \n  const callee = node.callee;\n  if (callee.type === \"Identifier\") {\n    return [\"memo\", \"forwardRef\"].includes(callee.name);\n  }\n  \n  // React.memo, React.forwardRef\n  if (callee.type === \"MemberExpression\") {\n    const obj = callee.object;\n    const prop = callee.property;\n    if (obj.type === \"Identifier\" && obj.name === \"React\" && \n        prop.type === \"Identifier\") {\n      return [\"memo\", \"forwardRef\"].includes(prop.name);\n    }\n  }\n  \n  return false;\n}\n\nfunction createSourceAssignment(name: string, sourceValue: string): ExpressionStatement {\n  return {\n    type: \"ExpressionStatement\",\n    expression: {\n      type: \"AssignmentExpression\",\n      operator: \"=\",\n      left: {\n        type: \"MemberExpression\",\n        object: { type: \"Identifier\", name },\n        property: { type: \"Identifier\", name: SOURCE_PROP },\n        computed: false,\n      } as MemberExpression,\n      right: { type: \"Literal\", value: sourceValue },\n    } as AssignmentExpression,\n  };\n}\n\nfunction createMarkElementHelper(): FunctionDeclaration {\n  const code = `\nfunction _markReactElementSource(element, source) {\n  const registryKey = Symbol.for(\"${JSX_SOURCE_REGISTRY_SYMBOL}\");\n  let registry = globalThis[registryKey];\n  if (!(registry instanceof WeakMap)) {\n    registry = globalThis[registryKey] = new WeakMap();\n  }\n  if (element && typeof element === \"object\" && typeof element.props === \"object\") {\n    registry.set(element.props, source);\n  }\n  return element;\n}\n`;\n  return parse(code, { ecmaVersion: \"latest\" }).body[0] as FunctionDeclaration;\n}\n\nfunction wrapWithMarkElement(\n  node: Node,\n  sourceValue: string\n): CallExpression {\n  return {\n    type: \"CallExpression\",\n    callee: { type: \"Identifier\", name: \"_markReactElementSource\" },\n    arguments: [node as any, { type: \"Literal\", value: sourceValue }],\n    optional: false,\n  };\n}\n\nexport function transformSource(\n  code: string,\n  options: TransformOptions\n): { code: string; map?: any } | null {\n  const {\n    filename,\n    projectRoot = process.cwd(),\n    injectJsxSource = true,\n    injectComponentSource = true,\n  } = options;\n\n  // Parse with acorn\n  let ast: Program;\n  try {\n    ast = parse(code, {\n      ecmaVersion: \"latest\",\n      sourceType: \"module\",\n      ecmaFeatures: { jsx: true },\n      locations: true,\n      ranges: true,\n    } as any) as Program;\n  } catch (err) {\n    // Parse error - return null to skip this file\n    return null;\n  }\n\n  let modified = false;\n  let needsHelper = false;\n  const seenComponents = new Set<string>();\n  const assignments: Array<{ node: Node; parent: Node | null; assignment: ExpressionStatement }> = [];\n\n  walk(ast as any, {\n    enter(node: any, parent: any, key: any, index: any) {\n      // Inject JSX source for element factory calls\n      if (injectJsxSource && node.type === \"CallExpression\") {\n        if (isReactElementFactoryCall(node)) {\n          const loc = node.loc;\n          if (loc) {\n            const sourceValue = toRelativeSource(filename, loc.start, projectRoot);\n            const wrapped = wrapWithMarkElement(node, sourceValue);\n            \n            // Replace the node\n            this.replace(wrapped);\n            needsHelper = true;\n            modified = true;\n          }\n        }\n      }\n\n      // Inject JSX source for JSX elements\n      if (injectJsxSource && node.type === \"JSXElement\") {\n        const opening = node.openingElement;\n        const loc = opening.loc;\n        \n        if (loc) {\n          const sourceValue = toRelativeSource(filename, loc.start, projectRoot);\n          \n          // Check if intrinsic element (lowercase) or component (uppercase)\n          const name = opening.name;\n          if (name.type === \"JSXIdentifier\") {\n            if (name.name[0] === name.name[0].toLowerCase()) {\n              // Intrinsic element - use _markReactElementSource\n              // This is handled by the CallExpression visitor for jsx()\n              // But for JSX syntax directly, we need different handling\n            } else {\n              // Component - add $componentSourceLoc prop\n              const sourceAttr = {\n                type: \"JSXAttribute\",\n                name: { type: \"JSXIdentifier\", name: JSX_SOURCE_PROP },\n                value: { type: \"Literal\", value: sourceValue },\n              };\n              opening.attributes.push(sourceAttr);\n              modified = true;\n            }\n          }\n        }\n      }\n\n      // Inject component source for function declarations\n      if (injectComponentSource && node.type === \"FunctionDeclaration\") {\n        const name = node.id?.name;\n        if (name && isComponentName(name) && !seenComponents.has(name)) {\n          const loc = node.loc;\n          if (loc) {\n            const sourceValue = toRelativeSource(filename, loc.start, projectRoot);\n            const assignment = createSourceAssignment(name, sourceValue);\n            assignments.push({ node, parent, assignment });\n            seenComponents.add(name);\n            modified = true;\n          }\n        }\n      }\n\n      // Inject component source for variable declarations\n      if (injectComponentSource && node.type === \"VariableDeclarator\") {\n        const id = node.id;\n        if (id.type === \"Identifier\" && isComponentName(id.name) && !seenComponents.has(id.name)) {\n          const init = node.init;\n          if (init && isSupportedComponentInit(init)) {\n            const loc = node.loc || init.loc;\n            if (loc) {\n              const sourceValue = toRelativeSource(filename, loc.start, projectRoot);\n              const assignment = createSourceAssignment(id.name, sourceValue);\n              assignments.push({ node, parent, assignment });\n              seenComponents.add(id.name);\n              modified = true;\n            }\n          }\n        }\n      }\n    },\n  });\n\n  if (!modified) {\n    return null;\n  }\n\n  // Insert helper function if needed\n  if (needsHelper) {\n    const helper = createMarkElementHelper();\n    // Check if already exists\n    const exists = ast.body.some(\n      (n) => n.type === \"FunctionDeclaration\" && \n             n.id?.name === \"_markReactElementSource\"\n    );\n    if (!exists) {\n      ast.body.unshift(helper);\n    }\n  }\n\n  // Insert component source assignments\n  // We need to insert after the declaration, so we process in reverse order\n  for (const { node, parent, assignment } of assignments.reverse()) {\n    if (parent && parent.type === \"Program\") {\n      const index = ast.body.indexOf(node as any);\n      if (index !== -1) {\n        ast.body.splice(index + 1, 0, assignment as any);\n      }\n    } else if (parent && parent.type === \"ExportNamedDeclaration\") {\n      // For export const Component = ...\n      // Insert after the export statement\n      const exportParent = (ast as any).body.find((n: any) => \n        n.type === \"ExportNamedDeclaration\" && n.declaration === parent\n      );\n      if (exportParent) {\n        const index = ast.body.indexOf(exportParent);\n        ast.body.splice(index + 1, 0, assignment as any);\n      }\n    } else if (parent && parent.type === \"ExportDefaultDeclaration\") {\n      // For export default function Component() ...\n      const index = ast.body.indexOf(parent as any);\n      ast.body.splice(index + 1, 0, assignment as any);\n    }\n  }\n\n  // Generate code\n  const result = generate(ast as any, { indent: \"  \" });\n\n  return { code: result };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\nexport const JSX_SOURCE_PROP = \"$componentSourceLoc\";\nexport const JSX_SOURCE_REGISTRY_SYMBOL = \"react-code-locator.jsxSourceRegistry\";\n"],"mappings":";AAOA,SAAS,sBAAmE;;;ACF5E,SAAS,aAAqC;AAC9C,SAAS,gBAAgB;AACzB,SAAS,YAAY;;;ACPd,IAAM,cAAc;AACpB,IAAM,kBAAkB;AACxB,IAAM,6BAA6B;;;ADiC1C,SAAS,iBAAiB,UAAkB,KAAe,aAA8B;AACvF,QAAM,OAAO,eAAe,QAAQ,IAAI;AACxC,QAAM,eAAe,SAAS,WAAW,IAAI,IACzC,SAAS,MAAM,KAAK,SAAS,CAAC,IAC9B;AACJ,SAAO,GAAG,YAAY,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACtD;AAEA,SAAS,gBAAgB,MAAuB;AAC9C,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,0BAA0B,MAA+B;AAChE,QAAM,SAAS,KAAK;AAEpB,MAAI,OAAO,SAAS,cAAc;AAChC,UAAM,OAAO,OAAO;AACpB,WAAO;AAAA,MACL;AAAA,MAAO;AAAA,MAAQ;AAAA,MACf;AAAA,MAAQ;AAAA,MAAS;AAAA,MACjB;AAAA,IACF,EAAE,SAAS,IAAI;AAAA,EACjB;AAGA,MAAI,OAAO,SAAS,oBAAoB;AACtC,UAAM,MAAM,OAAO;AACnB,UAAM,OAAO,OAAO;AACpB,QAAI,IAAI,SAAS,gBAAgB,IAAI,SAAS,WAC1C,KAAK,SAAS,gBAAgB,KAAK,SAAS,iBAAiB;AAC/D,aAAO;AAAA,IACT;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,yBACP,MACS;AACT,MAAI,CAAC,KAAM,QAAO;AAElB,MAAI,KAAK,SAAS,6BACd,KAAK,SAAS,sBAAsB;AACtC,WAAO;AAAA,EACT;AAEA,MAAI,KAAK,SAAS,iBAAkB,QAAO;AAE3C,QAAM,SAAS,KAAK;AACpB,MAAI,OAAO,SAAS,cAAc;AAChC,WAAO,CAAC,QAAQ,YAAY,EAAE,SAAS,OAAO,IAAI;AAAA,EACpD;AAGA,MAAI,OAAO,SAAS,oBAAoB;AACtC,UAAM,MAAM,OAAO;AACnB,UAAM,OAAO,OAAO;AACpB,QAAI,IAAI,SAAS,gBAAgB,IAAI,SAAS,WAC1C,KAAK,SAAS,cAAc;AAC9B,aAAO,CAAC,QAAQ,YAAY,EAAE,SAAS,KAAK,IAAI;AAAA,IAClD;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,uBAAuB,MAAc,aAA0C;AACtF,SAAO;AAAA,IACL,MAAM;AAAA,IACN,YAAY;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,MAAM;AAAA,QACJ,MAAM;AAAA,QACN,QAAQ,EAAE,MAAM,cAAc,KAAK;AAAA,QACnC,UAAU,EAAE,MAAM,cAAc,MAAM,YAAY;AAAA,QAClD,UAAU;AAAA,MACZ;AAAA,MACA,OAAO,EAAE,MAAM,WAAW,OAAO,YAAY;AAAA,IAC/C;AAAA,EACF;AACF;AAEA,SAAS,0BAA+C;AACtD,QAAM,OAAO;AAAA;AAAA,oCAEqB,0BAA0B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAW5D,SAAO,MAAM,MAAM,EAAE,aAAa,SAAS,CAAC,EAAE,KAAK,CAAC;AACtD;AAEA,SAAS,oBACP,MACA,aACgB;AAChB,SAAO;AAAA,IACL,MAAM;AAAA,IACN,QAAQ,EAAE,MAAM,cAAc,MAAM,0BAA0B;AAAA,IAC9D,WAAW,CAAC,MAAa,EAAE,MAAM,WAAW,OAAO,YAAY,CAAC;AAAA,IAChE,UAAU;AAAA,EACZ;AACF;AAEO,SAAS,gBACd,MACA,SACoC;AACpC,QAAM;AAAA,IACJ;AAAA,IACA,cAAc,QAAQ,IAAI;AAAA,IAC1B,kBAAkB;AAAA,IAClB,wBAAwB;AAAA,EAC1B,IAAI;AAGJ,MAAI;AACJ,MAAI;AACF,UAAM,MAAM,MAAM;AAAA,MAChB,aAAa;AAAA,MACb,YAAY;AAAA,MACZ,cAAc,EAAE,KAAK,KAAK;AAAA,MAC1B,WAAW;AAAA,MACX,QAAQ;AAAA,IACV,CAAQ;AAAA,EACV,SAAS,KAAK;AAEZ,WAAO;AAAA,EACT;AAEA,MAAI,WAAW;AACf,MAAI,cAAc;AAClB,QAAM,iBAAiB,oBAAI,IAAY;AACvC,QAAM,cAA2F,CAAC;AAElG,OAAK,KAAY;AAAA,IACf,MAAM,MAAW,QAAa,KAAU,OAAY;AAElD,UAAI,mBAAmB,KAAK,SAAS,kBAAkB;AACrD,YAAI,0BAA0B,IAAI,GAAG;AACnC,gBAAM,MAAM,KAAK;AACjB,cAAI,KAAK;AACP,kBAAM,cAAc,iBAAiB,UAAU,IAAI,OAAO,WAAW;AACrE,kBAAM,UAAU,oBAAoB,MAAM,WAAW;AAGrD,iBAAK,QAAQ,OAAO;AACpB,0BAAc;AACd,uBAAW;AAAA,UACb;AAAA,QACF;AAAA,MACF;AAGA,UAAI,mBAAmB,KAAK,SAAS,cAAc;AACjD,cAAM,UAAU,KAAK;AACrB,cAAM,MAAM,QAAQ;AAEpB,YAAI,KAAK;AACP,gBAAM,cAAc,iBAAiB,UAAU,IAAI,OAAO,WAAW;AAGrE,gBAAM,OAAO,QAAQ;AACrB,cAAI,KAAK,SAAS,iBAAiB;AACjC,gBAAI,KAAK,KAAK,CAAC,MAAM,KAAK,KAAK,CAAC,EAAE,YAAY,GAAG;AAAA,YAIjD,OAAO;AAEL,oBAAM,aAAa;AAAA,gBACjB,MAAM;AAAA,gBACN,MAAM,EAAE,MAAM,iBAAiB,MAAM,gBAAgB;AAAA,gBACrD,OAAO,EAAE,MAAM,WAAW,OAAO,YAAY;AAAA,cAC/C;AACA,sBAAQ,WAAW,KAAK,UAAU;AAClC,yBAAW;AAAA,YACb;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAGA,UAAI,yBAAyB,KAAK,SAAS,uBAAuB;AAChE,cAAM,OAAO,KAAK,IAAI;AACtB,YAAI,QAAQ,gBAAgB,IAAI,KAAK,CAAC,eAAe,IAAI,IAAI,GAAG;AAC9D,gBAAM,MAAM,KAAK;AACjB,cAAI,KAAK;AACP,kBAAM,cAAc,iBAAiB,UAAU,IAAI,OAAO,WAAW;AACrE,kBAAM,aAAa,uBAAuB,MAAM,WAAW;AAC3D,wBAAY,KAAK,EAAE,MAAM,QAAQ,WAAW,CAAC;AAC7C,2BAAe,IAAI,IAAI;AACvB,uBAAW;AAAA,UACb;AAAA,QACF;AAAA,MACF;AAGA,UAAI,yBAAyB,KAAK,SAAS,sBAAsB;AAC/D,cAAM,KAAK,KAAK;AAChB,YAAI,GAAG,SAAS,gBAAgB,gBAAgB,GAAG,IAAI,KAAK,CAAC,eAAe,IAAI,GAAG,IAAI,GAAG;AACxF,gBAAM,OAAO,KAAK;AAClB,cAAI,QAAQ,yBAAyB,IAAI,GAAG;AAC1C,kBAAM,MAAM,KAAK,OAAO,KAAK;AAC7B,gBAAI,KAAK;AACP,oBAAM,cAAc,iBAAiB,UAAU,IAAI,OAAO,WAAW;AACrE,oBAAM,aAAa,uBAAuB,GAAG,MAAM,WAAW;AAC9D,0BAAY,KAAK,EAAE,MAAM,QAAQ,WAAW,CAAC;AAC7C,6BAAe,IAAI,GAAG,IAAI;AAC1B,yBAAW;AAAA,YACb;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,MAAI,CAAC,UAAU;AACb,WAAO;AAAA,EACT;AAGA,MAAI,aAAa;AACf,UAAM,SAAS,wBAAwB;AAEvC,UAAM,SAAS,IAAI,KAAK;AAAA,MACtB,CAAC,MAAM,EAAE,SAAS,yBACX,EAAE,IAAI,SAAS;AAAA,IACxB;AACA,QAAI,CAAC,QAAQ;AACX,UAAI,KAAK,QAAQ,MAAM;AAAA,IACzB;AAAA,EACF;AAIA,aAAW,EAAE,MAAM,QAAQ,WAAW,KAAK,YAAY,QAAQ,GAAG;AAChE,QAAI,UAAU,OAAO,SAAS,WAAW;AACvC,YAAM,QAAQ,IAAI,KAAK,QAAQ,IAAW;AAC1C,UAAI,UAAU,IAAI;AAChB,YAAI,KAAK,OAAO,QAAQ,GAAG,GAAG,UAAiB;AAAA,MACjD;AAAA,IACF,WAAW,UAAU,OAAO,SAAS,0BAA0B;AAG7D,YAAM,eAAgB,IAAY,KAAK;AAAA,QAAK,CAAC,MAC3C,EAAE,SAAS,4BAA4B,EAAE,gBAAgB;AAAA,MAC3D;AACA,UAAI,cAAc;AAChB,cAAM,QAAQ,IAAI,KAAK,QAAQ,YAAY;AAC3C,YAAI,KAAK,OAAO,QAAQ,GAAG,GAAG,UAAiB;AAAA,MACjD;AAAA,IACF,WAAW,UAAU,OAAO,SAAS,4BAA4B;AAE/D,YAAM,QAAQ,IAAI,KAAK,QAAQ,MAAa;AAC5C,UAAI,KAAK,OAAO,QAAQ,GAAG,GAAG,UAAiB;AAAA,IACjD;AAAA,EACF;AAGA,QAAM,SAAS,SAAS,KAAY,EAAE,QAAQ,KAAK,CAAC;AAEpD,SAAO,EAAE,MAAM,OAAO;AACxB;;;ADvQA,IAAM,kBAAkB;AACxB,IAAM,kBAAkB;AAExB,SAAS,gBAAgB,IAAY,SAA4B,SAAqC;AACpG,QAAM,kBAAkB,MAAM,QAAQ,OAAO,IAAI,UAAU,CAAC,OAAO;AACnE,QAAM,kBAAkB,MAAM,QAAQ,OAAO,IAAI,UAAU,CAAC,OAAO;AAGnE,MAAI,gBAAgB,KAAK,aAAW,QAAQ,KAAK,EAAE,CAAC,GAAG;AACrD,WAAO;AAAA,EACT;AAGA,SAAO,gBAAgB,KAAK,aAAW,QAAQ,KAAK,EAAE,CAAC;AACzD;AAEO,IAAM,WACX,eAAe,CAAC,UAAU,CAAC,MAAM;AAC/B,QAAM;AAAA,IACJ,UAAU;AAAA,IACV,UAAU;AAAA,IACV,cAAc,QAAQ,IAAI;AAAA,IAC1B,wBAAwB;AAAA,IACxB,kBAAkB;AAAA,EACpB,IAAI;AAEJ,SAAO;AAAA,IACL,MAAM;AAAA,IAEN,UAAU,MAAM,IAAI;AAElB,UAAI,CAAC,gBAAgB,IAAI,SAAS,OAAO,GAAG;AAC1C,eAAO;AAAA,MACT;AAGA,YAAM,SAAS,gBAAgB,MAAM;AAAA,QACnC,UAAU;AAAA,QACV;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAED,aAAO;AAAA,IACT;AAAA,EACF;AACF,CAAC;AAGI,IAAM,aAAa,SAAS;AAC5B,IAAM,gBAAgB,SAAS;AAC/B,IAAM,eAAe,SAAS;AAC9B,IAAM,gBAAgB,SAAS;AAC/B,IAAM,eAAe,SAAS;AAGrC,IAAO,mBAAQ;","names":[]}

package/dist/vite.cjs

@@ -42,7 +42,6 @@ var import_core = require("@babel/core");
 
 // src/constants.ts
 var SOURCE_PROP = "__componentSourceLoc";
-var JSX_SOURCE_PROP = "$componentSourceLoc";
 var JSX_SOURCE_REGISTRY_SYMBOL = "react-code-locator.jsxSourceRegistry";
 
 // src/sourceMetadata.ts
@@ -82,43 +81,11 @@ function toRelativeSource(filename, loc, projectRoot) {
   const relPath = root && normalizedFilename.startsWith(`${root}/`) ? normalizedFilename.slice(root.length + 1) : root ? computeRelativePath(root, normalizedFilename) : normalizedFilename;
   return `${relPath}:${loc.line}:${loc.column + 1}`;
 }
-function isProjectLocalFile(filename, projectRoot) {
-  if (!filename) {
-    return false;
-  }
-  const root = normalizeProjectRoot(projectRoot);
-  const normalizedFilename = normalizeSlashes(filename);
-  if (!root) {
-    return !normalizedFilename.startsWith("../") && !normalizedFilename.startsWith("/") && !/^[A-Za-z]:\//.test(normalizedFilename);
-  }
-  if (normalizedFilename.startsWith(`${root}/`) || normalizedFilename === root) {
-    return true;
-  }
-  const relativePath = computeRelativePath(root, normalizedFilename);
-  return !relativePath.startsWith("../");
-}
-function isExternalToProjectRoot(filename, projectRoot) {
-  return !isProjectLocalFile(filename, projectRoot);
-}
 
 // src/babelInjectComponentSource.ts
-var SOURCE_PROP_LOCAL = "_componentSourceLoc";
-var SOURCE_PROPS_REST = "__reactCodeLocatorProps";
 function isComponentName(name) {
   return /^[A-Z]/.test(name);
 }
-function isCustomComponentTag(name) {
-  if (import_core.types.isJSXIdentifier(name)) {
-    return isComponentName(name.name);
-  }
-  if (import_core.types.isJSXMemberExpression(name)) {
-    return true;
-  }
-  return false;
-}
-function isIntrinsicElementTag(name) {
-  return import_core.types.isJSXIdentifier(name) && /^[a-z]/.test(name.name);
-}
 function isElementFactoryIdentifier(name) {
   return name === "jsx" || name === "jsxs" || name === "jsxDEV" || name === "_jsx" || name === "_jsxs" || name === "_jsxDEV" || name === "createElement";
 }
@@ -129,29 +96,6 @@ function isReactElementFactoryCall(pathNode) {
   }
   return import_core.types.isMemberExpression(callee) && import_core.types.isIdentifier(callee.object, { name: "React" }) && import_core.types.isIdentifier(callee.property, { name: "createElement" });
 }
-function getRootJsxIdentifierName(name) {
-  if (import_core.types.isJSXIdentifier(name)) {
-    return name.name;
-  }
-  if (import_core.types.isJSXMemberExpression(name)) {
-    return getRootJsxIdentifierName(name.object);
-  }
-  return null;
-}
-function isStyledModuleImport(binding) {
-  if (!binding) {
-    return false;
-  }
-  if (!binding.path.isImportSpecifier() && !binding.path.isImportDefaultSpecifier() && !binding.path.isImportNamespaceSpecifier()) {
-    return false;
-  }
-  const source = binding.path.parentPath.isImportDeclaration() ? binding.path.parentPath.node.source.value : null;
-  if (typeof source !== "string") {
-    return false;
-  }
-  const normalized = source.replace(/\\/g, "/");
-  return normalized === "./styled" || normalized === "../styled" || normalized.endsWith("/styled");
-}
 function isSupportedComponentInit(node) {
   if (!node) {
     return false;
@@ -167,98 +111,6 @@ function isSupportedComponentInit(node) {
   }
   return import_core.types.isMemberExpression(node.callee) && import_core.types.isIdentifier(node.callee.object, { name: "React" }) && import_core.types.isIdentifier(node.callee.property) && (node.callee.property.name === "memo" || node.callee.property.name === "forwardRef");
 }
-function hasSourcePropBinding(pattern) {
-  return pattern.properties.some((property) => {
-    if (!import_core.types.isObjectProperty(property)) {
-      return false;
-    }
-    return import_core.types.isIdentifier(property.key) && property.key.name === JSX_SOURCE_PROP;
-  });
-}
-function injectSourcePropBinding(pattern) {
-  if (hasSourcePropBinding(pattern)) {
-    return;
-  }
-  const sourceBinding = import_core.types.objectProperty(
-    import_core.types.identifier(JSX_SOURCE_PROP),
-    import_core.types.identifier(SOURCE_PROP_LOCAL),
-    false,
-    false
-  );
-  const restIndex = pattern.properties.findIndex(
-    (property) => import_core.types.isRestElement(property)
-  );
-  if (restIndex === -1) {
-    pattern.properties.push(sourceBinding);
-    return;
-  }
-  pattern.properties.splice(restIndex, 0, sourceBinding);
-}
-function injectSourcePropIntoIdentifierParam(node, param) {
-  if (!import_core.types.isBlockStatement(node.body)) {
-    node.body = import_core.types.blockStatement([import_core.types.returnStatement(node.body)]);
-  }
-  const alreadyInjected = node.body.body.some(
-    (statement) => import_core.types.isVariableDeclaration(statement) && statement.declarations.some(
-      (declaration) => import_core.types.isIdentifier(declaration.id) && declaration.id.name === SOURCE_PROPS_REST
-    )
-  );
-  if (alreadyInjected) {
-    return;
-  }
-  node.body.body.unshift(
-    import_core.types.variableDeclaration("const", [
-      import_core.types.variableDeclarator(
-        import_core.types.objectPattern([
-          import_core.types.objectProperty(
-            import_core.types.identifier(JSX_SOURCE_PROP),
-            import_core.types.identifier(SOURCE_PROP_LOCAL),
-            false,
-            false
-          ),
-          import_core.types.restElement(import_core.types.identifier(SOURCE_PROPS_REST))
-        ]),
-        param
-      )
-    ]),
-    import_core.types.expressionStatement(
-      import_core.types.assignmentExpression(
-        "=",
-        import_core.types.identifier(param.name),
-        import_core.types.identifier(SOURCE_PROPS_REST)
-      )
-    )
-  );
-}
-function injectSourcePropIntoFunctionParams(node) {
-  const firstParam = node.params[0];
-  if (!firstParam) {
-    return;
-  }
-  if (import_core.types.isObjectPattern(firstParam)) {
-    injectSourcePropBinding(firstParam);
-    return;
-  }
-  if (import_core.types.isIdentifier(firstParam)) {
-    injectSourcePropIntoIdentifierParam(node, firstParam);
-  }
-}
-function injectSourcePropIntoExpression(node) {
-  if (!node) {
-    return;
-  }
-  if (import_core.types.isFunctionExpression(node) || import_core.types.isArrowFunctionExpression(node)) {
-    injectSourcePropIntoFunctionParams(node);
-    return;
-  }
-  if (!import_core.types.isCallExpression(node)) {
-    return;
-  }
-  const firstArg = node.arguments[0];
-  if (firstArg && !import_core.types.isSpreadElement(firstArg) && (import_core.types.isFunctionExpression(firstArg) || import_core.types.isArrowFunctionExpression(firstArg))) {
-    injectSourcePropIntoFunctionParams(firstArg);
-  }
-}
 function getSourceValue(state, loc, projectRoot) {
   const filename = state.file?.opts?.filename;
   if (!filename || !loc) {
@@ -275,9 +127,9 @@ function buildAssignment(name, sourceValue) {
     )
   );
 }
-function buildIntrinsicSourceHelper() {
+function buildElementSourceHelper() {
   return import_core.types.functionDeclaration(
-    import_core.types.identifier("_markIntrinsicElementSource"),
+    import_core.types.identifier("_markReactElementSource"),
     [import_core.types.identifier("element"), import_core.types.identifier("source")],
     import_core.types.blockStatement([
       import_core.types.variableDeclaration("const", [
@@ -341,15 +193,15 @@ function buildIntrinsicSourceHelper() {
     ])
   );
 }
-function ensureIntrinsicSourceHelper(programPath, state) {
+function ensureElementSourceHelper(programPath, state) {
   if (state.injectedIntrinsicHelper) {
     return;
   }
   const alreadyExists = programPath.node.body.some(
-    (node) => import_core.types.isFunctionDeclaration(node) && import_core.types.isIdentifier(node.id, { name: "_markIntrinsicElementSource" })
+    (node) => import_core.types.isFunctionDeclaration(node) && import_core.types.isIdentifier(node.id, { name: "_markReactElementSource" })
   );
   if (!alreadyExists) {
-    programPath.unshiftContainer("body", buildIntrinsicSourceHelper());
+    programPath.unshiftContainer("body", buildElementSourceHelper());
   }
   state.injectedIntrinsicHelper = true;
 }
@@ -359,9 +211,6 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen, project
     if (!name || !isComponentName(name) || seen.has(name)) {
       return;
     }
-    if (declarationPath.isFunctionDeclaration()) {
-      injectSourcePropIntoFunctionParams(declarationPath.node);
-    }
     const sourceValue = getSourceValue(
       state,
       declarationPath.node.loc?.start,
@@ -388,7 +237,6 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen, project
       if (!isSupportedComponentInit(declarator.init)) {
         return [];
       }
-      injectSourcePropIntoExpression(declarator.init);
       const sourceValue = getSourceValue(
         state,
         declarator.loc?.start ?? declarator.init.loc?.start,
@@ -422,7 +270,7 @@ function babelInjectComponentSource(options = {}) {
           return;
         }
         if (pathNode.parentPath.isCallExpression() && import_core.types.isIdentifier(pathNode.parentPath.node.callee, {
-          name: "_markIntrinsicElementSource"
+          name: "_markReactElementSource"
         })) {
           return;
         }
@@ -438,9 +286,9 @@ function babelInjectComponentSource(options = {}) {
         if (!programPath || !programPath.isProgram()) {
           return;
         }
-        ensureIntrinsicSourceHelper(programPath, state);
+        ensureElementSourceHelper(programPath, state);
         pathNode.replaceWith(
-          import_core.types.callExpression(import_core.types.identifier("_markIntrinsicElementSource"), [
+          import_core.types.callExpression(import_core.types.identifier("_markReactElementSource"), [
             pathNode.node,
             import_core.types.stringLiteral(sourceValue)
           ])
@@ -452,10 +300,7 @@ function babelInjectComponentSource(options = {}) {
           if (!injectJsxSource) {
             return;
           }
-          if (!isIntrinsicElementTag(pathNode.node.openingElement.name)) {
-            return;
-          }
-          if (pathNode.parentPath.isCallExpression() && import_core.types.isIdentifier(pathNode.parentPath.node.callee, { name: "_markIntrinsicElementSource" })) {
+          if (pathNode.parentPath.isCallExpression() && import_core.types.isIdentifier(pathNode.parentPath.node.callee, { name: "_markReactElementSource" })) {
             return;
           }
           const sourceValue = getSourceValue(
@@ -470,8 +315,8 @@ function babelInjectComponentSource(options = {}) {
           if (!programPath || !programPath.isProgram()) {
             return;
           }
-          ensureIntrinsicSourceHelper(programPath, state);
-          const wrappedNode = import_core.types.callExpression(import_core.types.identifier("_markIntrinsicElementSource"), [
+          ensureElementSourceHelper(programPath, state);
+          const wrappedNode = import_core.types.callExpression(import_core.types.identifier("_markReactElementSource"), [
             pathNode.node,
             import_core.types.stringLiteral(sourceValue)
           ]);
@@ -486,37 +331,6 @@ function babelInjectComponentSource(options = {}) {
           pathNode.replaceWith(wrappedNode);
         }
       },
-      JSXOpeningElement(pathNode, state) {
-        if (!injectJsxSource) {
-          return;
-        }
-        if (!isCustomComponentTag(pathNode.node.name)) {
-          return;
-        }
-        const rootIdentifierName = getRootJsxIdentifierName(pathNode.node.name);
-        if (rootIdentifierName && isExternalToProjectRoot(state.file?.opts?.filename, projectRoot) && isStyledModuleImport(pathNode.scope.getBinding(rootIdentifierName))) {
-          return;
-        }
-        const hasSourceProp = pathNode.node.attributes.some(
-          (attr) => import_core.types.isJSXAttribute(attr) && import_core.types.isJSXIdentifier(attr.name) && attr.name.name === JSX_SOURCE_PROP
-        );
-        if (hasSourceProp) {
-          return;
-        }
-        const filename = state.file?.opts?.filename;
-        const loc = pathNode.node.loc?.start;
-        if (!filename || !loc) {
-          return;
-        }
-        pathNode.node.attributes.push(
-          import_core.types.jsxAttribute(
-            import_core.types.jsxIdentifier(JSX_SOURCE_PROP),
-            import_core.types.stringLiteral(
-              getSourceValue(state, loc, projectRoot) ?? `${filename.replace(/\\/g, "/")}:${loc.line}:${loc.column + 1}`
-            )
-          )
-        );
-      },
       Program(programPath, state) {
         if (!injectComponentSource) {
           return;