react-code-locator 0.1.6 → 0.1.7

package/README.md

@@ -2,7 +2,7 @@
 
 개발 중인 React 앱에서 요소를 `Shift + Click`하면 해당 UI와 연결된 소스 위치를 찾는 패키지입니다.
 
-- JSX 태그의 `__source`를 우선 사용해서 정확한 JSX 위치를 찾습니다.
+- JSX 디버그 정보가 있으면 우선 사용해서 정확한 JSX 위치를 찾습니다.
 - JSX 정보가 없으면 컴포넌트 정의 위치로 fallback 합니다.
 - Vite, Webpack, Babel, 브라우저 런타임을 각각 분리해서 사용할 수 있습니다.
 
@@ -20,19 +20,27 @@ npm i -D /absolute/path/to/react-code-locator
 
 ## 빠른 시작
 
-Vite 환경이면 `react-code-locator/vite`만 붙이면 됩니다.
+Vite 환경에서는 기존 `@vitejs/plugin-react` 설정에 Babel 플러그인을 추가하고, `react-code-locator/vite`는 클라이언트 자동 주입만 맡깁니다.
 
 ```ts
 import { defineConfig } from "vite";
-import { reactComponentJump } from "react-code-locator/vite";
+import react from "@vitejs/plugin-react";
+import { babelInjectComponentSource, reactComponentJump } from "react-code-locator/vite";
 
 export default defineConfig(({ command }) => ({
-  plugins: reactComponentJump({
-    command,
-    locator: {
-      triggerKey: "shift",
-    },
-  }),
+  plugins: [
+    react({
+      babel: {
+        plugins: [babelInjectComponentSource],
+      },
+    }),
+    ...reactComponentJump({
+      command,
+      locator: {
+        triggerKey: "shift",
+      },
+    }),
+  ],
 }));
 ```
 
@@ -48,22 +56,31 @@ export default defineConfig(({ command }) => ({
 
 Vite + React 프로젝트용 기본 진입점입니다.
 
-- 개발 서버에서만 Babel 플러그인을 주입합니다.
-- 라이브러리 메타데이터는 `__source` 대신 `__componentSourceLoc`로 주입해서 Vite React dev transform과 충돌하지 않습니다.
+- 개발 서버에서만 브라우저 클라이언트를 자동 주입합니다.
+- HTML에는 bare import를 직접 넣지 않고 Vite 가상 모듈을 통해 클라이언트를 로드합니다.
 - 기본값으로 브라우저 클라이언트도 자동 주입합니다.
+- React 플러그인은 직접 만들지 않습니다. 기존 `@vitejs/plugin-react` 설정에 Babel 플러그인을 추가해서 사용해야 합니다.
 
 ```ts
 import { defineConfig } from "vite";
-import { reactComponentJump } from "react-code-locator/vite";
+import react from "@vitejs/plugin-react";
+import { babelInjectComponentSource, reactComponentJump } from "react-code-locator/vite";
 
 export default defineConfig(({ command }) => ({
-  plugins: reactComponentJump({
-    command,
-    locator: {
-      triggerKey: "shift",
-    },
-    injectClient: true,
-  }),
+  plugins: [
+    react({
+      babel: {
+        plugins: [babelInjectComponentSource],
+      },
+    }),
+    ...reactComponentJump({
+      command,
+      locator: {
+        triggerKey: "shift",
+      },
+      injectClient: true,
+    }),
+  ],
 }));
 ```
 
@@ -74,6 +91,7 @@ export default defineConfig(({ command }) => ({
 - `locator.onLocate(result)`: 위치를 찾았을 때 커스텀 처리
 - `locator.onError(error)`: 위치를 못 찾았을 때 커스텀 처리
 - `injectClient`: `false`로 두면 브라우저 런타임 자동 주입 비활성화
+- `babelInjectComponentSource`: 기존 `@vitejs/plugin-react`의 `babel.plugins`에 추가
 
 ### `react-code-locator/client`
 
@@ -97,13 +115,21 @@ dispose();
 
 ```ts
 import { defineConfig } from "vite";
-import { reactComponentJump } from "react-code-locator/vite";
+import react from "@vitejs/plugin-react";
+import { babelInjectComponentSource, reactComponentJump } from "react-code-locator/vite";
 
 export default defineConfig(({ command }) => ({
-  plugins: reactComponentJump({
-    command,
-    injectClient: false,
-  }),
+  plugins: [
+    react({
+      babel: {
+        plugins: [babelInjectComponentSource],
+      },
+    }),
+    ...reactComponentJump({
+      command,
+      injectClient: false,
+    }),
+  ],
 }));
 ```
 
@@ -121,11 +147,12 @@ if (import.meta.env.DEV) {
 
 Babel 플러그인만 따로 사용할 때 사용합니다.
 
-이 플러그인은 두 가지를 주입합니다.
+이 플러그인은 기본적으로 한 가지를 주입합니다.
 
-- JSX 요소에 `__componentSourceLoc`
 - React 컴포넌트 함수/클래스에 `__componentSourceLoc`
 
+옵션으로 JSX 요소에도 `__componentSourceLoc`를 주입할 수 있지만 기본값은 `false`입니다.
+
 ```js
 const { babelInjectComponentSource } = require("react-code-locator/babel");
 
@@ -144,6 +171,16 @@ export default {
 };
 ```
 
+JSX 주입이 꼭 필요하면:
+
+```ts
+import { babelInjectComponentSource } from "react-code-locator/babel";
+
+export default {
+  plugins: [[babelInjectComponentSource, { injectJsxSource: true }]],
+};
+```
+
 ### `react-code-locator/webpack`
 
 Webpack 설정에 Babel 플러그인과 런타임 엔트리를 함께 주입합니다.

package/dist/babel.cjs

@@ -45,6 +45,21 @@ var SOURCE_PROP = "__componentSourceLoc";
 function isComponentName(name) {
   return /^[A-Z]/.test(name);
 }
+function isSupportedComponentInit(node) {
+  if (!node) {
+    return false;
+  }
+  if (import_core.types.isArrowFunctionExpression(node) || import_core.types.isFunctionExpression(node)) {
+    return true;
+  }
+  if (!import_core.types.isCallExpression(node)) {
+    return false;
+  }
+  if (import_core.types.isIdentifier(node.callee) && (node.callee.name === "memo" || node.callee.name === "forwardRef")) {
+    return true;
+  }
+  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 getSourceValue(state, loc) {
   const filename = state.file?.opts?.filename;
   if (!filename || !loc) {
@@ -86,7 +101,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
     if (!declarator.init) {
       return [];
     }
-    if (!import_core.types.isArrowFunctionExpression(declarator.init) && !import_core.types.isFunctionExpression(declarator.init)) {
+    if (!isSupportedComponentInit(declarator.init)) {
       return [];
     }
     const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);
@@ -101,7 +116,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
   }
 }
 function babelInjectComponentSource(options = {}) {
-  const { injectJsxSource = true, injectComponentSource = true } = options;
+  const { injectJsxSource = false, injectComponentSource = true } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {

package/dist/babel.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/babel.ts","../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["export { babelInjectComponentSource } from \"./babelInjectComponentSource\";\nexport type { BabelInjectComponentSourceOptions } from \"./babelInjectComponentSource\";\n","import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = true, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,uBAAiB;AACjB,kBAA0D;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,iBAAAA,QAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,YAAAC,MAAE;AAAA,IACP,YAAAA,MAAE;AAAA,MACA;AAAA,MACA,YAAAA,MAAE,iBAAiB,YAAAA,MAAE,WAAW,IAAI,GAAG,YAAAA,MAAE,WAAW,WAAW,CAAC;AAAA,MAChE,YAAAA,MAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,YAAAA,MAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,YAAAA,MAAE,0BAA0B,WAAW,IAAI,KAAK,CAAC,YAAAA,MAAE,qBAAqB,WAAW,IAAI,GAAG;AAC7F,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,MAAM,wBAAwB,KAAK,IAAI;AAEjE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,YAAAA,MAAE,eAAe,IAAI,KAAK,YAAAA,MAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,YAAAA,MAAE;AAAA,YACA,YAAAA,MAAE,cAAc,WAAW;AAAA,YAC3B,YAAAA,MAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":["path","t"]}
+{"version":3,"sources":["../src/babel.ts","../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["export { babelInjectComponentSource } from \"./babelInjectComponentSource\";\nexport type { BabelInjectComponentSourceOptions } from \"./babelInjectComponentSource\";\n","import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction isSupportedComponentInit(node: t.Expression | null | undefined): boolean {\n  if (!node) {\n    return false;\n  }\n\n  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {\n    return true;\n  }\n\n  if (!t.isCallExpression(node)) {\n    return false;\n  }\n\n  if (t.isIdentifier(node.callee) && (node.callee.name === \"memo\" || node.callee.name === \"forwardRef\")) {\n    return true;\n  }\n\n  return (\n    t.isMemberExpression(node.callee) &&\n    t.isIdentifier(node.callee.object, { name: \"React\" }) &&\n    t.isIdentifier(node.callee.property) &&\n    (node.callee.property.name === \"memo\" || node.callee.property.name === \"forwardRef\")\n  );\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!isSupportedComponentInit(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = false, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,uBAAiB;AACjB,kBAA0D;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,yBAAyB,MAAgD;AAChF,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,YAAAA,MAAE,0BAA0B,IAAI,KAAK,YAAAA,MAAE,qBAAqB,IAAI,GAAG;AACrE,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,YAAAA,MAAE,iBAAiB,IAAI,GAAG;AAC7B,WAAO;AAAA,EACT;AAEA,MAAI,YAAAA,MAAE,aAAa,KAAK,MAAM,MAAM,KAAK,OAAO,SAAS,UAAU,KAAK,OAAO,SAAS,eAAe;AACrG,WAAO;AAAA,EACT;AAEA,SACE,YAAAA,MAAE,mBAAmB,KAAK,MAAM,KAChC,YAAAA,MAAE,aAAa,KAAK,OAAO,QAAQ,EAAE,MAAM,QAAQ,CAAC,KACpD,YAAAA,MAAE,aAAa,KAAK,OAAO,QAAQ,MAClC,KAAK,OAAO,SAAS,SAAS,UAAU,KAAK,OAAO,SAAS,SAAS;AAE3E;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,iBAAAC,QAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,YAAAD,MAAE;AAAA,IACP,YAAAA,MAAE;AAAA,MACA;AAAA,MACA,YAAAA,MAAE,iBAAiB,YAAAA,MAAE,WAAW,IAAI,GAAG,YAAAA,MAAE,WAAW,WAAW,CAAC;AAAA,MAChE,YAAAA,MAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,YAAAA,MAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,yBAAyB,WAAW,IAAI,GAAG;AAC9C,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,OAAO,wBAAwB,KAAK,IAAI;AAElE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,YAAAA,MAAE,eAAe,IAAI,KAAK,YAAAA,MAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,YAAAA,MAAE;AAAA,YACA,YAAAA,MAAE,cAAc,WAAW;AAAA,YAC3B,YAAAA,MAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":["t","path"]}

package/dist/babel.js

@@ -9,6 +9,21 @@ var SOURCE_PROP = "__componentSourceLoc";
 function isComponentName(name) {
   return /^[A-Z]/.test(name);
 }
+function isSupportedComponentInit(node) {
+  if (!node) {
+    return false;
+  }
+  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {
+    return true;
+  }
+  if (!t.isCallExpression(node)) {
+    return false;
+  }
+  if (t.isIdentifier(node.callee) && (node.callee.name === "memo" || node.callee.name === "forwardRef")) {
+    return true;
+  }
+  return t.isMemberExpression(node.callee) && t.isIdentifier(node.callee.object, { name: "React" }) && t.isIdentifier(node.callee.property) && (node.callee.property.name === "memo" || node.callee.property.name === "forwardRef");
+}
 function getSourceValue(state, loc) {
   const filename = state.file?.opts?.filename;
   if (!filename || !loc) {
@@ -50,7 +65,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
     if (!declarator.init) {
       return [];
     }
-    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {
+    if (!isSupportedComponentInit(declarator.init)) {
       return [];
     }
     const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);
@@ -65,7 +80,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
   }
 }
 function babelInjectComponentSource(options = {}) {
-  const { injectJsxSource = true, injectComponentSource = true } = options;
+  const { injectJsxSource = false, injectComponentSource = true } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {

package/dist/babel.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = true, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";AAAA,OAAO,UAAU;AACjB,SAAS,SAAS,SAAwC;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,KAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,EAAE;AAAA,IACP,EAAE;AAAA,MACA;AAAA,MACA,EAAE,iBAAiB,EAAE,WAAW,IAAI,GAAG,EAAE,WAAW,WAAW,CAAC;AAAA,MAChE,EAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,EAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,EAAE,0BAA0B,WAAW,IAAI,KAAK,CAAC,EAAE,qBAAqB,WAAW,IAAI,GAAG;AAC7F,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,MAAM,wBAAwB,KAAK,IAAI;AAEjE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,EAAE,eAAe,IAAI,KAAK,EAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,EAAE;AAAA,YACA,EAAE,cAAc,WAAW;AAAA,YAC3B,EAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":[]}
+{"version":3,"sources":["../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction isSupportedComponentInit(node: t.Expression | null | undefined): boolean {\n  if (!node) {\n    return false;\n  }\n\n  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {\n    return true;\n  }\n\n  if (!t.isCallExpression(node)) {\n    return false;\n  }\n\n  if (t.isIdentifier(node.callee) && (node.callee.name === \"memo\" || node.callee.name === \"forwardRef\")) {\n    return true;\n  }\n\n  return (\n    t.isMemberExpression(node.callee) &&\n    t.isIdentifier(node.callee.object, { name: \"React\" }) &&\n    t.isIdentifier(node.callee.property) &&\n    (node.callee.property.name === \"memo\" || node.callee.property.name === \"forwardRef\")\n  );\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!isSupportedComponentInit(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = false, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";AAAA,OAAO,UAAU;AACjB,SAAS,SAAS,SAAwC;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,yBAAyB,MAAgD;AAChF,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,EAAE,0BAA0B,IAAI,KAAK,EAAE,qBAAqB,IAAI,GAAG;AACrE,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,EAAE,iBAAiB,IAAI,GAAG;AAC7B,WAAO;AAAA,EACT;AAEA,MAAI,EAAE,aAAa,KAAK,MAAM,MAAM,KAAK,OAAO,SAAS,UAAU,KAAK,OAAO,SAAS,eAAe;AACrG,WAAO;AAAA,EACT;AAEA,SACE,EAAE,mBAAmB,KAAK,MAAM,KAChC,EAAE,aAAa,KAAK,OAAO,QAAQ,EAAE,MAAM,QAAQ,CAAC,KACpD,EAAE,aAAa,KAAK,OAAO,QAAQ,MAClC,KAAK,OAAO,SAAS,SAAS,UAAU,KAAK,OAAO,SAAS,SAAS;AAE3E;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,KAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,EAAE;AAAA,IACP,EAAE;AAAA,MACA;AAAA,MACA,EAAE,iBAAiB,EAAE,WAAW,IAAI,GAAG,EAAE,WAAW,WAAW,CAAC;AAAA,MAChE,EAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,EAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,yBAAyB,WAAW,IAAI,GAAG;AAC9C,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,OAAO,wBAAwB,KAAK,IAAI;AAElE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,EAAE,eAAe,IAAI,KAAK,EAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,EAAE;AAAA,YACA,EAAE,cAAc,WAAW;AAAA,YAC3B,EAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/dist/babelInjectComponentSource.cjs

@@ -43,6 +43,21 @@ var SOURCE_PROP = "__componentSourceLoc";
 function isComponentName(name) {
   return /^[A-Z]/.test(name);
 }
+function isSupportedComponentInit(node) {
+  if (!node) {
+    return false;
+  }
+  if (import_core.types.isArrowFunctionExpression(node) || import_core.types.isFunctionExpression(node)) {
+    return true;
+  }
+  if (!import_core.types.isCallExpression(node)) {
+    return false;
+  }
+  if (import_core.types.isIdentifier(node.callee) && (node.callee.name === "memo" || node.callee.name === "forwardRef")) {
+    return true;
+  }
+  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 getSourceValue(state, loc) {
   const filename = state.file?.opts?.filename;
   if (!filename || !loc) {
@@ -84,7 +99,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
     if (!declarator.init) {
       return [];
     }
-    if (!import_core.types.isArrowFunctionExpression(declarator.init) && !import_core.types.isFunctionExpression(declarator.init)) {
+    if (!isSupportedComponentInit(declarator.init)) {
       return [];
     }
     const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);
@@ -99,7 +114,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
   }
 }
 function babelInjectComponentSource(options = {}) {
-  const { injectJsxSource = true, injectComponentSource = true } = options;
+  const { injectJsxSource = false, injectComponentSource = true } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {

package/dist/babelInjectComponentSource.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = true, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,uBAAiB;AACjB,kBAA0D;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,iBAAAA,QAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,YAAAC,MAAE;AAAA,IACP,YAAAA,MAAE;AAAA,MACA;AAAA,MACA,YAAAA,MAAE,iBAAiB,YAAAA,MAAE,WAAW,IAAI,GAAG,YAAAA,MAAE,WAAW,WAAW,CAAC;AAAA,MAChE,YAAAA,MAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,YAAAA,MAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,YAAAA,MAAE,0BAA0B,WAAW,IAAI,KAAK,CAAC,YAAAA,MAAE,qBAAqB,WAAW,IAAI,GAAG;AAC7F,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,MAAM,wBAAwB,KAAK,IAAI;AAEjE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,YAAAA,MAAE,eAAe,IAAI,KAAK,YAAAA,MAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,YAAAA,MAAE;AAAA,YACA,YAAAA,MAAE,cAAc,WAAW;AAAA,YAC3B,YAAAA,MAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":["path","t"]}
+{"version":3,"sources":["../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction isSupportedComponentInit(node: t.Expression | null | undefined): boolean {\n  if (!node) {\n    return false;\n  }\n\n  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {\n    return true;\n  }\n\n  if (!t.isCallExpression(node)) {\n    return false;\n  }\n\n  if (t.isIdentifier(node.callee) && (node.callee.name === \"memo\" || node.callee.name === \"forwardRef\")) {\n    return true;\n  }\n\n  return (\n    t.isMemberExpression(node.callee) &&\n    t.isIdentifier(node.callee.object, { name: \"React\" }) &&\n    t.isIdentifier(node.callee.property) &&\n    (node.callee.property.name === \"memo\" || node.callee.property.name === \"forwardRef\")\n  );\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!isSupportedComponentInit(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = false, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,uBAAiB;AACjB,kBAA0D;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,yBAAyB,MAAgD;AAChF,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,YAAAA,MAAE,0BAA0B,IAAI,KAAK,YAAAA,MAAE,qBAAqB,IAAI,GAAG;AACrE,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,YAAAA,MAAE,iBAAiB,IAAI,GAAG;AAC7B,WAAO;AAAA,EACT;AAEA,MAAI,YAAAA,MAAE,aAAa,KAAK,MAAM,MAAM,KAAK,OAAO,SAAS,UAAU,KAAK,OAAO,SAAS,eAAe;AACrG,WAAO;AAAA,EACT;AAEA,SACE,YAAAA,MAAE,mBAAmB,KAAK,MAAM,KAChC,YAAAA,MAAE,aAAa,KAAK,OAAO,QAAQ,EAAE,MAAM,QAAQ,CAAC,KACpD,YAAAA,MAAE,aAAa,KAAK,OAAO,QAAQ,MAClC,KAAK,OAAO,SAAS,SAAS,UAAU,KAAK,OAAO,SAAS,SAAS;AAE3E;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,iBAAAC,QAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,YAAAD,MAAE;AAAA,IACP,YAAAA,MAAE;AAAA,MACA;AAAA,MACA,YAAAA,MAAE,iBAAiB,YAAAA,MAAE,WAAW,IAAI,GAAG,YAAAA,MAAE,WAAW,WAAW,CAAC;AAAA,MAChE,YAAAA,MAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,YAAAA,MAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,yBAAyB,WAAW,IAAI,GAAG;AAC9C,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,OAAO,wBAAwB,KAAK,IAAI;AAElE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,YAAAA,MAAE,eAAe,IAAI,KAAK,YAAAA,MAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,YAAAA,MAAE;AAAA,YACA,YAAAA,MAAE,cAAc,WAAW;AAAA,YAC3B,YAAAA,MAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":["t","path"]}

package/dist/babelInjectComponentSource.js

@@ -9,6 +9,21 @@ var SOURCE_PROP = "__componentSourceLoc";
 function isComponentName(name) {
   return /^[A-Z]/.test(name);
 }
+function isSupportedComponentInit(node) {
+  if (!node) {
+    return false;
+  }
+  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {
+    return true;
+  }
+  if (!t.isCallExpression(node)) {
+    return false;
+  }
+  if (t.isIdentifier(node.callee) && (node.callee.name === "memo" || node.callee.name === "forwardRef")) {
+    return true;
+  }
+  return t.isMemberExpression(node.callee) && t.isIdentifier(node.callee.object, { name: "React" }) && t.isIdentifier(node.callee.property) && (node.callee.property.name === "memo" || node.callee.property.name === "forwardRef");
+}
 function getSourceValue(state, loc) {
   const filename = state.file?.opts?.filename;
   if (!filename || !loc) {
@@ -50,7 +65,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
     if (!declarator.init) {
       return [];
     }
-    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {
+    if (!isSupportedComponentInit(declarator.init)) {
       return [];
     }
     const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);
@@ -65,7 +80,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
   }
 }
 function babelInjectComponentSource(options = {}) {
-  const { injectJsxSource = true, injectComponentSource = true } = options;
+  const { injectJsxSource = false, injectComponentSource = true } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {

package/dist/babelInjectComponentSource.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = true, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";AAAA,OAAO,UAAU;AACjB,SAAS,SAAS,SAAwC;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,KAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,EAAE;AAAA,IACP,EAAE;AAAA,MACA;AAAA,MACA,EAAE,iBAAiB,EAAE,WAAW,IAAI,GAAG,EAAE,WAAW,WAAW,CAAC;AAAA,MAChE,EAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,EAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,EAAE,0BAA0B,WAAW,IAAI,KAAK,CAAC,EAAE,qBAAqB,WAAW,IAAI,GAAG;AAC7F,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,MAAM,wBAAwB,KAAK,IAAI;AAEjE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,EAAE,eAAe,IAAI,KAAK,EAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,EAAE;AAAA,YACA,EAAE,cAAc,WAAW;AAAA,YAC3B,EAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":[]}
+{"version":3,"sources":["../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction isSupportedComponentInit(node: t.Expression | null | undefined): boolean {\n  if (!node) {\n    return false;\n  }\n\n  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {\n    return true;\n  }\n\n  if (!t.isCallExpression(node)) {\n    return false;\n  }\n\n  if (t.isIdentifier(node.callee) && (node.callee.name === \"memo\" || node.callee.name === \"forwardRef\")) {\n    return true;\n  }\n\n  return (\n    t.isMemberExpression(node.callee) &&\n    t.isIdentifier(node.callee.object, { name: \"React\" }) &&\n    t.isIdentifier(node.callee.property) &&\n    (node.callee.property.name === \"memo\" || node.callee.property.name === \"forwardRef\")\n  );\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!isSupportedComponentInit(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = false, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";AAAA,OAAO,UAAU;AACjB,SAAS,SAAS,SAAwC;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,yBAAyB,MAAgD;AAChF,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,EAAE,0BAA0B,IAAI,KAAK,EAAE,qBAAqB,IAAI,GAAG;AACrE,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,EAAE,iBAAiB,IAAI,GAAG;AAC7B,WAAO;AAAA,EACT;AAEA,MAAI,EAAE,aAAa,KAAK,MAAM,MAAM,KAAK,OAAO,SAAS,UAAU,KAAK,OAAO,SAAS,eAAe;AACrG,WAAO;AAAA,EACT;AAEA,SACE,EAAE,mBAAmB,KAAK,MAAM,KAChC,EAAE,aAAa,KAAK,OAAO,QAAQ,EAAE,MAAM,QAAQ,CAAC,KACpD,EAAE,aAAa,KAAK,OAAO,QAAQ,MAClC,KAAK,OAAO,SAAS,SAAS,UAAU,KAAK,OAAO,SAAS,SAAS;AAE3E;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,KAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,EAAE;AAAA,IACP,EAAE;AAAA,MACA;AAAA,MACA,EAAE,iBAAiB,EAAE,WAAW,IAAI,GAAG,EAAE,WAAW,WAAW,CAAC;AAAA,MAChE,EAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,EAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,yBAAyB,WAAW,IAAI,GAAG;AAC9C,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,OAAO,wBAAwB,KAAK,IAAI;AAElE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,EAAE,eAAe,IAAI,KAAK,EAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,EAAE;AAAA,YACA,EAAE,cAAc,WAAW;AAAA,YAC3B,EAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/dist/vite.cjs

@@ -30,13 +30,55 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/vite.ts
 var vite_exports = {};
 __export(vite_exports, {
+  babelInjectComponentSource: () => babelInjectComponentSource,
   elementLocatorReact: () => elementLocatorReact,
   reactComponentJump: () => elementLocatorReact
 });
 module.exports = __toCommonJS(vite_exports);
 
 // src/elementLocatorReact.ts
-var import_plugin_react = __toESM(require("@vitejs/plugin-react"), 1);
+var VIRTUAL_CLIENT_MODULE_ID = "virtual:react-code-locator/client";
+var RESOLVED_VIRTUAL_CLIENT_MODULE_ID = `\0${VIRTUAL_CLIENT_MODULE_ID}`;
+function createClientInjector(locatorOptions = {}) {
+  const serialized = JSON.stringify(locatorOptions);
+  return {
+    name: "element-locator-client-injector",
+    apply: "serve",
+    resolveId(id) {
+      if (id === VIRTUAL_CLIENT_MODULE_ID) {
+        return RESOLVED_VIRTUAL_CLIENT_MODULE_ID;
+      }
+      return null;
+    },
+    load(id) {
+      if (id !== RESOLVED_VIRTUAL_CLIENT_MODULE_ID) {
+        return null;
+      }
+      return `
+        import { enableReactComponentJump } from "react-code-locator/client";
+
+        enableReactComponentJump(${serialized});
+      `;
+    },
+    transformIndexHtml() {
+      return [
+        {
+          tag: "script",
+          attrs: {
+            type: "module"
+          },
+          children: `import "${VIRTUAL_CLIENT_MODULE_ID}";`,
+          injectTo: "head"
+        }
+      ];
+    }
+  };
+}
+function elementLocatorReact(options = {}) {
+  const { command = "serve", locator = {}, injectClient = true } = options;
+  const isServe = command === "serve";
+  return [isServe && injectClient ? createClientInjector(locator) : null].filter(Boolean);
+}
 
 // src/babelInjectComponentSource.ts
 var import_node_path = __toESM(require("path"), 1);
@@ -49,6 +91,21 @@ var SOURCE_PROP = "__componentSourceLoc";
 function isComponentName(name) {
   return /^[A-Z]/.test(name);
 }
+function isSupportedComponentInit(node) {
+  if (!node) {
+    return false;
+  }
+  if (import_core.types.isArrowFunctionExpression(node) || import_core.types.isFunctionExpression(node)) {
+    return true;
+  }
+  if (!import_core.types.isCallExpression(node)) {
+    return false;
+  }
+  if (import_core.types.isIdentifier(node.callee) && (node.callee.name === "memo" || node.callee.name === "forwardRef")) {
+    return true;
+  }
+  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 getSourceValue(state, loc) {
   const filename = state.file?.opts?.filename;
   if (!filename || !loc) {
@@ -90,7 +147,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
     if (!declarator.init) {
       return [];
     }
-    if (!import_core.types.isArrowFunctionExpression(declarator.init) && !import_core.types.isFunctionExpression(declarator.init)) {
+    if (!isSupportedComponentInit(declarator.init)) {
       return [];
     }
     const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);
@@ -105,7 +162,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
   }
 }
 function babelInjectComponentSource(options = {}) {
-  const { injectJsxSource = true, injectComponentSource = true } = options;
+  const { injectJsxSource = false, injectComponentSource = true } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {
@@ -150,63 +207,9 @@ function babelInjectComponentSource(options = {}) {
     }
   };
 }
-
-// src/elementLocatorReact.ts
-function withLocatorBabel(reactOptions = {}) {
-  const baseBabel = reactOptions.babel;
-  const locatorBabelPlugin = babelInjectComponentSource;
-  if (typeof baseBabel === "function") {
-    return {
-      ...reactOptions,
-      babel(id, options) {
-        const result = baseBabel(id, options);
-        const resolved2 = result ?? {};
-        const plugins = [...resolved2.plugins ?? [], locatorBabelPlugin];
-        return {
-          ...resolved2,
-          plugins
-        };
-      }
-    };
-  }
-  const resolved = baseBabel ?? {};
-  return {
-    ...reactOptions,
-    babel: {
-      ...resolved,
-      plugins: [...resolved.plugins ?? [], locatorBabelPlugin]
-    }
-  };
-}
-function createClientInjector(locatorOptions = {}) {
-  const serialized = JSON.stringify(locatorOptions);
-  return {
-    name: "element-locator-client-injector",
-    apply: "serve",
-    transformIndexHtml() {
-      return [
-        {
-          tag: "script",
-          attrs: {
-            type: "module"
-          },
-          children: `import { enableReactComponentJump } from "react-code-locator/client"; enableReactComponentJump(${serialized});`,
-          injectTo: "head"
-        }
-      ];
-    }
-  };
-}
-function elementLocatorReact(options = {}) {
-  const { command = "serve", react: reactOptions, locator = {}, injectClient = true } = options;
-  const isServe = command === "serve";
-  return [
-    (0, import_plugin_react.default)(isServe ? withLocatorBabel(reactOptions) : reactOptions),
-    isServe && injectClient ? createClientInjector(locator) : null
-  ].filter(Boolean);
-}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  babelInjectComponentSource,
   elementLocatorReact,
   reactComponentJump
 });

package/dist/vite.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/vite.ts","../src/elementLocatorReact.ts","../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["export { elementLocatorReact as reactComponentJump } from \"./elementLocatorReact\";\nexport { elementLocatorReact } from \"./elementLocatorReact\";\nexport type { ElementLocatorReactOptions } from \"./elementLocatorReact\";\n\n","import type { TransformOptions, PluginItem } from \"@babel/core\";\nimport react, { type Options as ReactOptions } from \"@vitejs/plugin-react\";\nimport type { Plugin, PluginOption } from \"vite\";\nimport { babelInjectComponentSource } from \"./babelInjectComponentSource\";\nimport { type LocatorOptions } from \"./runtime\";\n\nexport type ElementLocatorReactOptions = {\n  command?: \"serve\" | \"build\";\n  react?: ReactOptions;\n  locator?: LocatorOptions;\n  injectClient?: boolean;\n};\n\nfunction withLocatorBabel(reactOptions: ReactOptions = {}): ReactOptions {\n  const baseBabel = reactOptions.babel;\n  const locatorBabelPlugin: PluginItem = babelInjectComponentSource as PluginItem;\n\n  if (typeof baseBabel === \"function\") {\n    return {\n      ...reactOptions,\n      babel(id, options) {\n        const result = baseBabel(id, options);\n        const resolved = (result ?? {}) as TransformOptions;\n        const plugins = [...(resolved.plugins ?? []), locatorBabelPlugin];\n\n        return {\n          ...resolved,\n          plugins,\n        };\n      },\n    };\n  }\n\n  const resolved = (baseBabel ?? {}) as TransformOptions;\n  return {\n    ...reactOptions,\n    babel: {\n      ...resolved,\n      plugins: [...(resolved.plugins ?? []), locatorBabelPlugin],\n    },\n  };\n}\n\nfunction createClientInjector(locatorOptions: LocatorOptions = {}): Plugin {\n  const serialized = JSON.stringify(locatorOptions);\n\n  return {\n    name: \"element-locator-client-injector\",\n    apply: \"serve\",\n    transformIndexHtml() {\n      return [\n        {\n          tag: \"script\",\n          attrs: {\n            type: \"module\",\n          },\n          children: `import { enableReactComponentJump } from \"react-code-locator/client\"; enableReactComponentJump(${serialized});`,\n          injectTo: \"head\",\n        },\n      ];\n    },\n  };\n}\n\nexport function elementLocatorReact(options: ElementLocatorReactOptions = {}): PluginOption[] {\n  const { command = \"serve\", react: reactOptions, locator = {}, injectClient = true } = options;\n  const isServe = command === \"serve\";\n\n  return [\n    react(isServe ? withLocatorBabel(reactOptions) : reactOptions),\n    isServe && injectClient ? createClientInjector(locator) : null,\n  ].filter(Boolean);\n}\n","import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = true, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,0BAAoD;;;ACDpD,uBAAiB;AACjB,kBAA0D;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,iBAAAA,QAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,YAAAC,MAAE;AAAA,IACP,YAAAA,MAAE;AAAA,MACA;AAAA,MACA,YAAAA,MAAE,iBAAiB,YAAAA,MAAE,WAAW,IAAI,GAAG,YAAAA,MAAE,WAAW,WAAW,CAAC;AAAA,MAChE,YAAAA,MAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,YAAAA,MAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,YAAAA,MAAE,0BAA0B,WAAW,IAAI,KAAK,CAAC,YAAAA,MAAE,qBAAqB,WAAW,IAAI,GAAG;AAC7F,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,MAAM,wBAAwB,KAAK,IAAI;AAEjE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,YAAAA,MAAE,eAAe,IAAI,KAAK,YAAAA,MAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,YAAAA,MAAE;AAAA,YACA,YAAAA,MAAE,cAAc,WAAW;AAAA,YAC3B,YAAAA,MAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;;;ADvIA,SAAS,iBAAiB,eAA6B,CAAC,GAAiB;AACvE,QAAM,YAAY,aAAa;AAC/B,QAAM,qBAAiC;AAEvC,MAAI,OAAO,cAAc,YAAY;AACnC,WAAO;AAAA,MACL,GAAG;AAAA,MACH,MAAM,IAAI,SAAS;AACjB,cAAM,SAAS,UAAU,IAAI,OAAO;AACpC,cAAMC,YAAY,UAAU,CAAC;AAC7B,cAAM,UAAU,CAAC,GAAIA,UAAS,WAAW,CAAC,GAAI,kBAAkB;AAEhE,eAAO;AAAA,UACL,GAAGA;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,WAAY,aAAa,CAAC;AAChC,SAAO;AAAA,IACL,GAAG;AAAA,IACH,OAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS,CAAC,GAAI,SAAS,WAAW,CAAC,GAAI,kBAAkB;AAAA,IAC3D;AAAA,EACF;AACF;AAEA,SAAS,qBAAqB,iBAAiC,CAAC,GAAW;AACzE,QAAM,aAAa,KAAK,UAAU,cAAc;AAEhD,SAAO;AAAA,IACL,MAAM;AAAA,IACN,OAAO;AAAA,IACP,qBAAqB;AACnB,aAAO;AAAA,QACL;AAAA,UACE,KAAK;AAAA,UACL,OAAO;AAAA,YACL,MAAM;AAAA,UACR;AAAA,UACA,UAAU,kGAAkG,UAAU;AAAA,UACtH,UAAU;AAAA,QACZ;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,oBAAoB,UAAsC,CAAC,GAAmB;AAC5F,QAAM,EAAE,UAAU,SAAS,OAAO,cAAc,UAAU,CAAC,GAAG,eAAe,KAAK,IAAI;AACtF,QAAM,UAAU,YAAY;AAE5B,SAAO;AAAA,QACL,oBAAAC,SAAM,UAAU,iBAAiB,YAAY,IAAI,YAAY;AAAA,IAC7D,WAAW,eAAe,qBAAqB,OAAO,IAAI;AAAA,EAC5D,EAAE,OAAO,OAAO;AAClB;","names":["path","t","resolved","react"]}
+{"version":3,"sources":["../src/vite.ts","../src/elementLocatorReact.ts","../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["export { elementLocatorReact as reactComponentJump } from \"./elementLocatorReact\";\nexport { elementLocatorReact } from \"./elementLocatorReact\";\nexport type { ElementLocatorReactOptions } from \"./elementLocatorReact\";\nexport { babelInjectComponentSource } from \"./babelInjectComponentSource\";\nexport type { BabelInjectComponentSourceOptions } from \"./babelInjectComponentSource\";\n","import type { Plugin, PluginOption } from \"vite\";\nimport { type LocatorOptions } from \"./runtime\";\n\nconst VIRTUAL_CLIENT_MODULE_ID = \"virtual:react-code-locator/client\";\nconst RESOLVED_VIRTUAL_CLIENT_MODULE_ID = `\\0${VIRTUAL_CLIENT_MODULE_ID}`;\n\nexport type ElementLocatorReactOptions = {\n  command?: \"serve\" | \"build\";\n  locator?: LocatorOptions;\n  injectClient?: boolean;\n};\n\nfunction createClientInjector(locatorOptions: LocatorOptions = {}): Plugin {\n  const serialized = JSON.stringify(locatorOptions);\n\n  return {\n    name: \"element-locator-client-injector\",\n    apply: \"serve\",\n    resolveId(id) {\n      if (id === VIRTUAL_CLIENT_MODULE_ID) {\n        return RESOLVED_VIRTUAL_CLIENT_MODULE_ID;\n      }\n\n      return null;\n    },\n    load(id) {\n      if (id !== RESOLVED_VIRTUAL_CLIENT_MODULE_ID) {\n        return null;\n      }\n\n      return `\n        import { enableReactComponentJump } from \"react-code-locator/client\";\n\n        enableReactComponentJump(${serialized});\n      `;\n    },\n    transformIndexHtml() {\n      return [\n        {\n          tag: \"script\",\n          attrs: {\n            type: \"module\",\n          },\n          children: `import \"${VIRTUAL_CLIENT_MODULE_ID}\";`,\n          injectTo: \"head\",\n        },\n      ];\n    },\n  };\n}\n\nexport function elementLocatorReact(options: ElementLocatorReactOptions = {}): PluginOption[] {\n  const { command = \"serve\", locator = {}, injectClient = true } = options;\n  const isServe = command === \"serve\";\n\n  return [isServe && injectClient ? createClientInjector(locator) : null].filter(Boolean);\n}\n","import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction isSupportedComponentInit(node: t.Expression | null | undefined): boolean {\n  if (!node) {\n    return false;\n  }\n\n  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {\n    return true;\n  }\n\n  if (!t.isCallExpression(node)) {\n    return false;\n  }\n\n  if (t.isIdentifier(node.callee) && (node.callee.name === \"memo\" || node.callee.name === \"forwardRef\")) {\n    return true;\n  }\n\n  return (\n    t.isMemberExpression(node.callee) &&\n    t.isIdentifier(node.callee.object, { name: \"React\" }) &&\n    t.isIdentifier(node.callee.property) &&\n    (node.callee.property.name === \"memo\" || node.callee.property.name === \"forwardRef\")\n  );\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!isSupportedComponentInit(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = false, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACGA,IAAM,2BAA2B;AACjC,IAAM,oCAAoC,KAAK,wBAAwB;AAQvE,SAAS,qBAAqB,iBAAiC,CAAC,GAAW;AACzE,QAAM,aAAa,KAAK,UAAU,cAAc;AAEhD,SAAO;AAAA,IACL,MAAM;AAAA,IACN,OAAO;AAAA,IACP,UAAU,IAAI;AACZ,UAAI,OAAO,0BAA0B;AACnC,eAAO;AAAA,MACT;AAEA,aAAO;AAAA,IACT;AAAA,IACA,KAAK,IAAI;AACP,UAAI,OAAO,mCAAmC;AAC5C,eAAO;AAAA,MACT;AAEA,aAAO;AAAA;AAAA;AAAA,mCAGsB,UAAU;AAAA;AAAA,IAEzC;AAAA,IACA,qBAAqB;AACnB,aAAO;AAAA,QACL;AAAA,UACE,KAAK;AAAA,UACL,OAAO;AAAA,YACL,MAAM;AAAA,UACR;AAAA,UACA,UAAU,WAAW,wBAAwB;AAAA,UAC7C,UAAU;AAAA,QACZ;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,oBAAoB,UAAsC,CAAC,GAAmB;AAC5F,QAAM,EAAE,UAAU,SAAS,UAAU,CAAC,GAAG,eAAe,KAAK,IAAI;AACjE,QAAM,UAAU,YAAY;AAE5B,SAAO,CAAC,WAAW,eAAe,qBAAqB,OAAO,IAAI,IAAI,EAAE,OAAO,OAAO;AACxF;;;ACxDA,uBAAiB;AACjB,kBAA0D;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,yBAAyB,MAAgD;AAChF,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,YAAAA,MAAE,0BAA0B,IAAI,KAAK,YAAAA,MAAE,qBAAqB,IAAI,GAAG;AACrE,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,YAAAA,MAAE,iBAAiB,IAAI,GAAG;AAC7B,WAAO;AAAA,EACT;AAEA,MAAI,YAAAA,MAAE,aAAa,KAAK,MAAM,MAAM,KAAK,OAAO,SAAS,UAAU,KAAK,OAAO,SAAS,eAAe;AACrG,WAAO;AAAA,EACT;AAEA,SACE,YAAAA,MAAE,mBAAmB,KAAK,MAAM,KAChC,YAAAA,MAAE,aAAa,KAAK,OAAO,QAAQ,EAAE,MAAM,QAAQ,CAAC,KACpD,YAAAA,MAAE,aAAa,KAAK,OAAO,QAAQ,MAClC,KAAK,OAAO,SAAS,SAAS,UAAU,KAAK,OAAO,SAAS,SAAS;AAE3E;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,iBAAAC,QAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,YAAAD,MAAE;AAAA,IACP,YAAAA,MAAE;AAAA,MACA;AAAA,MACA,YAAAA,MAAE,iBAAiB,YAAAA,MAAE,WAAW,IAAI,GAAG,YAAAA,MAAE,WAAW,WAAW,CAAC;AAAA,MAChE,YAAAA,MAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,YAAAA,MAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,yBAAyB,WAAW,IAAI,GAAG;AAC9C,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,OAAO,wBAAwB,KAAK,IAAI;AAElE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,YAAAA,MAAE,eAAe,IAAI,KAAK,YAAAA,MAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,YAAAA,MAAE;AAAA,YACA,YAAAA,MAAE,cAAc,WAAW;AAAA,YAC3B,YAAAA,MAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":["t","path"]}

package/dist/vite.d.cts

@@ -1,10 +1,10 @@
-import { Options } from '@vitejs/plugin-react';
 import { PluginOption } from 'vite';
 import { LocatorOptions } from './client.cjs';
+export { BabelInjectComponentSourceOptions, babelInjectComponentSource } from './babelInjectComponentSource.cjs';
+import '@babel/core';
 
 type ElementLocatorReactOptions = {
     command?: "serve" | "build";
-    react?: Options;
     locator?: LocatorOptions;
     injectClient?: boolean;
 };

package/dist/vite.d.ts

@@ -1,10 +1,10 @@
-import { Options } from '@vitejs/plugin-react';
 import { PluginOption } from 'vite';
 import { LocatorOptions } from './client.js';
+export { BabelInjectComponentSourceOptions, babelInjectComponentSource } from './babelInjectComponentSource.js';
+import '@babel/core';
 
 type ElementLocatorReactOptions = {
     command?: "serve" | "build";
-    react?: Options;
     locator?: LocatorOptions;
     injectClient?: boolean;
 };

package/dist/vite.js

@@ -1,5 +1,46 @@
 // src/elementLocatorReact.ts
-import react from "@vitejs/plugin-react";
+var VIRTUAL_CLIENT_MODULE_ID = "virtual:react-code-locator/client";
+var RESOLVED_VIRTUAL_CLIENT_MODULE_ID = `\0${VIRTUAL_CLIENT_MODULE_ID}`;
+function createClientInjector(locatorOptions = {}) {
+  const serialized = JSON.stringify(locatorOptions);
+  return {
+    name: "element-locator-client-injector",
+    apply: "serve",
+    resolveId(id) {
+      if (id === VIRTUAL_CLIENT_MODULE_ID) {
+        return RESOLVED_VIRTUAL_CLIENT_MODULE_ID;
+      }
+      return null;
+    },
+    load(id) {
+      if (id !== RESOLVED_VIRTUAL_CLIENT_MODULE_ID) {
+        return null;
+      }
+      return `
+        import { enableReactComponentJump } from "react-code-locator/client";
+
+        enableReactComponentJump(${serialized});
+      `;
+    },
+    transformIndexHtml() {
+      return [
+        {
+          tag: "script",
+          attrs: {
+            type: "module"
+          },
+          children: `import "${VIRTUAL_CLIENT_MODULE_ID}";`,
+          injectTo: "head"
+        }
+      ];
+    }
+  };
+}
+function elementLocatorReact(options = {}) {
+  const { command = "serve", locator = {}, injectClient = true } = options;
+  const isServe = command === "serve";
+  return [isServe && injectClient ? createClientInjector(locator) : null].filter(Boolean);
+}
 
 // src/babelInjectComponentSource.ts
 import path from "path";
@@ -12,6 +53,21 @@ var SOURCE_PROP = "__componentSourceLoc";
 function isComponentName(name) {
   return /^[A-Z]/.test(name);
 }
+function isSupportedComponentInit(node) {
+  if (!node) {
+    return false;
+  }
+  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {
+    return true;
+  }
+  if (!t.isCallExpression(node)) {
+    return false;
+  }
+  if (t.isIdentifier(node.callee) && (node.callee.name === "memo" || node.callee.name === "forwardRef")) {
+    return true;
+  }
+  return t.isMemberExpression(node.callee) && t.isIdentifier(node.callee.object, { name: "React" }) && t.isIdentifier(node.callee.property) && (node.callee.property.name === "memo" || node.callee.property.name === "forwardRef");
+}
 function getSourceValue(state, loc) {
   const filename = state.file?.opts?.filename;
   if (!filename || !loc) {
@@ -53,7 +109,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
     if (!declarator.init) {
       return [];
     }
-    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {
+    if (!isSupportedComponentInit(declarator.init)) {
       return [];
     }
     const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);
@@ -68,7 +124,7 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
   }
 }
 function babelInjectComponentSource(options = {}) {
-  const { injectJsxSource = true, injectComponentSource = true } = options;
+  const { injectJsxSource = false, injectComponentSource = true } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {
@@ -113,62 +169,8 @@ function babelInjectComponentSource(options = {}) {
     }
   };
 }
-
-// src/elementLocatorReact.ts
-function withLocatorBabel(reactOptions = {}) {
-  const baseBabel = reactOptions.babel;
-  const locatorBabelPlugin = babelInjectComponentSource;
-  if (typeof baseBabel === "function") {
-    return {
-      ...reactOptions,
-      babel(id, options) {
-        const result = baseBabel(id, options);
-        const resolved2 = result ?? {};
-        const plugins = [...resolved2.plugins ?? [], locatorBabelPlugin];
-        return {
-          ...resolved2,
-          plugins
-        };
-      }
-    };
-  }
-  const resolved = baseBabel ?? {};
-  return {
-    ...reactOptions,
-    babel: {
-      ...resolved,
-      plugins: [...resolved.plugins ?? [], locatorBabelPlugin]
-    }
-  };
-}
-function createClientInjector(locatorOptions = {}) {
-  const serialized = JSON.stringify(locatorOptions);
-  return {
-    name: "element-locator-client-injector",
-    apply: "serve",
-    transformIndexHtml() {
-      return [
-        {
-          tag: "script",
-          attrs: {
-            type: "module"
-          },
-          children: `import { enableReactComponentJump } from "react-code-locator/client"; enableReactComponentJump(${serialized});`,
-          injectTo: "head"
-        }
-      ];
-    }
-  };
-}
-function elementLocatorReact(options = {}) {
-  const { command = "serve", react: reactOptions, locator = {}, injectClient = true } = options;
-  const isServe = command === "serve";
-  return [
-    react(isServe ? withLocatorBabel(reactOptions) : reactOptions),
-    isServe && injectClient ? createClientInjector(locator) : null
-  ].filter(Boolean);
-}
 export {
+  babelInjectComponentSource,
   elementLocatorReact,
   elementLocatorReact as reactComponentJump
 };

package/dist/vite.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/elementLocatorReact.ts","../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["import type { TransformOptions, PluginItem } from \"@babel/core\";\nimport react, { type Options as ReactOptions } from \"@vitejs/plugin-react\";\nimport type { Plugin, PluginOption } from \"vite\";\nimport { babelInjectComponentSource } from \"./babelInjectComponentSource\";\nimport { type LocatorOptions } from \"./runtime\";\n\nexport type ElementLocatorReactOptions = {\n  command?: \"serve\" | \"build\";\n  react?: ReactOptions;\n  locator?: LocatorOptions;\n  injectClient?: boolean;\n};\n\nfunction withLocatorBabel(reactOptions: ReactOptions = {}): ReactOptions {\n  const baseBabel = reactOptions.babel;\n  const locatorBabelPlugin: PluginItem = babelInjectComponentSource as PluginItem;\n\n  if (typeof baseBabel === \"function\") {\n    return {\n      ...reactOptions,\n      babel(id, options) {\n        const result = baseBabel(id, options);\n        const resolved = (result ?? {}) as TransformOptions;\n        const plugins = [...(resolved.plugins ?? []), locatorBabelPlugin];\n\n        return {\n          ...resolved,\n          plugins,\n        };\n      },\n    };\n  }\n\n  const resolved = (baseBabel ?? {}) as TransformOptions;\n  return {\n    ...reactOptions,\n    babel: {\n      ...resolved,\n      plugins: [...(resolved.plugins ?? []), locatorBabelPlugin],\n    },\n  };\n}\n\nfunction createClientInjector(locatorOptions: LocatorOptions = {}): Plugin {\n  const serialized = JSON.stringify(locatorOptions);\n\n  return {\n    name: \"element-locator-client-injector\",\n    apply: \"serve\",\n    transformIndexHtml() {\n      return [\n        {\n          tag: \"script\",\n          attrs: {\n            type: \"module\",\n          },\n          children: `import { enableReactComponentJump } from \"react-code-locator/client\"; enableReactComponentJump(${serialized});`,\n          injectTo: \"head\",\n        },\n      ];\n    },\n  };\n}\n\nexport function elementLocatorReact(options: ElementLocatorReactOptions = {}): PluginOption[] {\n  const { command = \"serve\", react: reactOptions, locator = {}, injectClient = true } = options;\n  const isServe = command === \"serve\";\n\n  return [\n    react(isServe ? withLocatorBabel(reactOptions) : reactOptions),\n    isServe && injectClient ? createClientInjector(locator) : null,\n  ].filter(Boolean);\n}\n","import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!t.isArrowFunctionExpression(declarator.init) && !t.isFunctionExpression(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = true, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";AACA,OAAO,WAA6C;;;ACDpD,OAAO,UAAU;AACjB,SAAS,SAAS,SAAwC;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,KAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,EAAE;AAAA,IACP,EAAE;AAAA,MACA;AAAA,MACA,EAAE,iBAAiB,EAAE,WAAW,IAAI,GAAG,EAAE,WAAW,WAAW,CAAC;AAAA,MAChE,EAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,EAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,EAAE,0BAA0B,WAAW,IAAI,KAAK,CAAC,EAAE,qBAAqB,WAAW,IAAI,GAAG;AAC7F,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,MAAM,wBAAwB,KAAK,IAAI;AAEjE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,EAAE,eAAe,IAAI,KAAK,EAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,EAAE;AAAA,YACA,EAAE,cAAc,WAAW;AAAA,YAC3B,EAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;;;ADvIA,SAAS,iBAAiB,eAA6B,CAAC,GAAiB;AACvE,QAAM,YAAY,aAAa;AAC/B,QAAM,qBAAiC;AAEvC,MAAI,OAAO,cAAc,YAAY;AACnC,WAAO;AAAA,MACL,GAAG;AAAA,MACH,MAAM,IAAI,SAAS;AACjB,cAAM,SAAS,UAAU,IAAI,OAAO;AACpC,cAAMA,YAAY,UAAU,CAAC;AAC7B,cAAM,UAAU,CAAC,GAAIA,UAAS,WAAW,CAAC,GAAI,kBAAkB;AAEhE,eAAO;AAAA,UACL,GAAGA;AAAA,UACH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,WAAY,aAAa,CAAC;AAChC,SAAO;AAAA,IACL,GAAG;AAAA,IACH,OAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS,CAAC,GAAI,SAAS,WAAW,CAAC,GAAI,kBAAkB;AAAA,IAC3D;AAAA,EACF;AACF;AAEA,SAAS,qBAAqB,iBAAiC,CAAC,GAAW;AACzE,QAAM,aAAa,KAAK,UAAU,cAAc;AAEhD,SAAO;AAAA,IACL,MAAM;AAAA,IACN,OAAO;AAAA,IACP,qBAAqB;AACnB,aAAO;AAAA,QACL;AAAA,UACE,KAAK;AAAA,UACL,OAAO;AAAA,YACL,MAAM;AAAA,UACR;AAAA,UACA,UAAU,kGAAkG,UAAU;AAAA,UACtH,UAAU;AAAA,QACZ;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,oBAAoB,UAAsC,CAAC,GAAmB;AAC5F,QAAM,EAAE,UAAU,SAAS,OAAO,cAAc,UAAU,CAAC,GAAG,eAAe,KAAK,IAAI;AACtF,QAAM,UAAU,YAAY;AAE5B,SAAO;AAAA,IACL,MAAM,UAAU,iBAAiB,YAAY,IAAI,YAAY;AAAA,IAC7D,WAAW,eAAe,qBAAqB,OAAO,IAAI;AAAA,EAC5D,EAAE,OAAO,OAAO;AAClB;","names":["resolved"]}
+{"version":3,"sources":["../src/elementLocatorReact.ts","../src/babelInjectComponentSource.ts","../src/constants.ts"],"sourcesContent":["import type { Plugin, PluginOption } from \"vite\";\nimport { type LocatorOptions } from \"./runtime\";\n\nconst VIRTUAL_CLIENT_MODULE_ID = \"virtual:react-code-locator/client\";\nconst RESOLVED_VIRTUAL_CLIENT_MODULE_ID = `\\0${VIRTUAL_CLIENT_MODULE_ID}`;\n\nexport type ElementLocatorReactOptions = {\n  command?: \"serve\" | \"build\";\n  locator?: LocatorOptions;\n  injectClient?: boolean;\n};\n\nfunction createClientInjector(locatorOptions: LocatorOptions = {}): Plugin {\n  const serialized = JSON.stringify(locatorOptions);\n\n  return {\n    name: \"element-locator-client-injector\",\n    apply: \"serve\",\n    resolveId(id) {\n      if (id === VIRTUAL_CLIENT_MODULE_ID) {\n        return RESOLVED_VIRTUAL_CLIENT_MODULE_ID;\n      }\n\n      return null;\n    },\n    load(id) {\n      if (id !== RESOLVED_VIRTUAL_CLIENT_MODULE_ID) {\n        return null;\n      }\n\n      return `\n        import { enableReactComponentJump } from \"react-code-locator/client\";\n\n        enableReactComponentJump(${serialized});\n      `;\n    },\n    transformIndexHtml() {\n      return [\n        {\n          tag: \"script\",\n          attrs: {\n            type: \"module\",\n          },\n          children: `import \"${VIRTUAL_CLIENT_MODULE_ID}\";`,\n          injectTo: \"head\",\n        },\n      ];\n    },\n  };\n}\n\nexport function elementLocatorReact(options: ElementLocatorReactOptions = {}): PluginOption[] {\n  const { command = \"serve\", locator = {}, injectClient = true } = options;\n  const isServe = command === \"serve\";\n\n  return [isServe && injectClient ? createClientInjector(locator) : null].filter(Boolean);\n}\n","import path from \"node:path\";\nimport { types as t, type NodePath, type PluginObj } from \"@babel/core\";\nimport { SOURCE_PROP } from \"./constants\";\n\nexport type BabelInjectComponentSourceOptions = {\n  injectJsxSource?: boolean;\n  injectComponentSource?: boolean;\n};\n\ntype BabelState = {\n  file?: {\n    opts?: {\n      filename?: string;\n    };\n  };\n};\n\nfunction isComponentName(name: string) {\n  return /^[A-Z]/.test(name);\n}\n\nfunction isSupportedComponentInit(node: t.Expression | null | undefined): boolean {\n  if (!node) {\n    return false;\n  }\n\n  if (t.isArrowFunctionExpression(node) || t.isFunctionExpression(node)) {\n    return true;\n  }\n\n  if (!t.isCallExpression(node)) {\n    return false;\n  }\n\n  if (t.isIdentifier(node.callee) && (node.callee.name === \"memo\" || node.callee.name === \"forwardRef\")) {\n    return true;\n  }\n\n  return (\n    t.isMemberExpression(node.callee) &&\n    t.isIdentifier(node.callee.object, { name: \"React\" }) &&\n    t.isIdentifier(node.callee.property) &&\n    (node.callee.property.name === \"memo\" || node.callee.property.name === \"forwardRef\")\n  );\n}\n\nfunction getSourceValue(state: BabelState, loc: { line: number; column: number } | null | undefined) {\n  const filename = state.file?.opts?.filename;\n  if (!filename || !loc) {\n    return null;\n  }\n\n  const relPath = path.relative(process.cwd(), filename).replace(/\\\\/g, \"/\");\n  return `${relPath}:${loc.line}:${loc.column + 1}`;\n}\n\nfunction buildAssignment(name: string, sourceValue: string) {\n  return t.expressionStatement(\n    t.assignmentExpression(\n      \"=\",\n      t.memberExpression(t.identifier(name), t.identifier(SOURCE_PROP)),\n      t.stringLiteral(sourceValue),\n    ),\n  );\n}\n\nfunction visitDeclaration(\n  declarationPath: NodePath,\n  insertAfterPath: NodePath,\n  state: BabelState,\n  seen: Set<string>,\n) {\n  if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {\n    const name = declarationPath.node.id?.name;\n    if (!name || !isComponentName(name) || seen.has(name)) {\n      return;\n    }\n\n    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);\n    if (!sourceValue) {\n      return;\n    }\n\n    seen.add(name);\n    insertAfterPath.insertAfter(buildAssignment(name, sourceValue));\n    return;\n  }\n\n  if (!declarationPath.isVariableDeclaration()) {\n    return;\n  }\n\n  const assignments = declarationPath.node.declarations.flatMap((declarator) => {\n    if (!t.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {\n      return [];\n    }\n\n    if (!declarator.init) {\n      return [];\n    }\n\n    if (!isSupportedComponentInit(declarator.init)) {\n      return [];\n    }\n\n    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);\n    if (!sourceValue) {\n      return [];\n    }\n\n    seen.add(declarator.id.name);\n    return [buildAssignment(declarator.id.name, sourceValue)];\n  });\n\n  if (assignments.length > 0) {\n    insertAfterPath.insertAfter(assignments);\n  }\n}\n\nexport function babelInjectComponentSource(\n  options: BabelInjectComponentSourceOptions = {},\n): PluginObj<BabelState> {\n  const { injectJsxSource = false, injectComponentSource = true } = options;\n\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        if (!injectJsxSource) {\n          return;\n        }\n\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP,\n        );\n        if (hasSourceProp) {\n          return;\n        }\n\n        const filename = state.file?.opts?.filename;\n        const loc = pathNode.node.loc?.start;\n        if (!filename || !loc) {\n          return;\n        }\n\n        pathNode.node.attributes.push(\n          t.jsxAttribute(\n            t.jsxIdentifier(SOURCE_PROP),\n            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\\\/g, \"/\")}:${loc.line}:${loc.column + 1}`),\n          ),\n        );\n      },\n      Program(programPath, state) {\n        if (!injectComponentSource) {\n          return;\n        }\n\n        const seen = new Set<string>();\n\n        for (const childPath of programPath.get(\"body\")) {\n          if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {\n            const declarationPath = childPath.get(\"declaration\");\n            if (!Array.isArray(declarationPath) && declarationPath.node) {\n              visitDeclaration(declarationPath, childPath, state, seen);\n            }\n            continue;\n          }\n\n          visitDeclaration(childPath, childPath, state, seen);\n        }\n      },\n    },\n  };\n}\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n"],"mappings":";AAGA,IAAM,2BAA2B;AACjC,IAAM,oCAAoC,KAAK,wBAAwB;AAQvE,SAAS,qBAAqB,iBAAiC,CAAC,GAAW;AACzE,QAAM,aAAa,KAAK,UAAU,cAAc;AAEhD,SAAO;AAAA,IACL,MAAM;AAAA,IACN,OAAO;AAAA,IACP,UAAU,IAAI;AACZ,UAAI,OAAO,0BAA0B;AACnC,eAAO;AAAA,MACT;AAEA,aAAO;AAAA,IACT;AAAA,IACA,KAAK,IAAI;AACP,UAAI,OAAO,mCAAmC;AAC5C,eAAO;AAAA,MACT;AAEA,aAAO;AAAA;AAAA;AAAA,mCAGsB,UAAU;AAAA;AAAA,IAEzC;AAAA,IACA,qBAAqB;AACnB,aAAO;AAAA,QACL;AAAA,UACE,KAAK;AAAA,UACL,OAAO;AAAA,YACL,MAAM;AAAA,UACR;AAAA,UACA,UAAU,WAAW,wBAAwB;AAAA,UAC7C,UAAU;AAAA,QACZ;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,oBAAoB,UAAsC,CAAC,GAAmB;AAC5F,QAAM,EAAE,UAAU,SAAS,UAAU,CAAC,GAAG,eAAe,KAAK,IAAI;AACjE,QAAM,UAAU,YAAY;AAE5B,SAAO,CAAC,WAAW,eAAe,qBAAqB,OAAO,IAAI,IAAI,EAAE,OAAO,OAAO;AACxF;;;ACxDA,OAAO,UAAU;AACjB,SAAS,SAAS,SAAwC;;;ACDnD,IAAM,cAAc;;;ADiB3B,SAAS,gBAAgB,MAAc;AACrC,SAAO,SAAS,KAAK,IAAI;AAC3B;AAEA,SAAS,yBAAyB,MAAgD;AAChF,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,EAAE,0BAA0B,IAAI,KAAK,EAAE,qBAAqB,IAAI,GAAG;AACrE,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,EAAE,iBAAiB,IAAI,GAAG;AAC7B,WAAO;AAAA,EACT;AAEA,MAAI,EAAE,aAAa,KAAK,MAAM,MAAM,KAAK,OAAO,SAAS,UAAU,KAAK,OAAO,SAAS,eAAe;AACrG,WAAO;AAAA,EACT;AAEA,SACE,EAAE,mBAAmB,KAAK,MAAM,KAChC,EAAE,aAAa,KAAK,OAAO,QAAQ,EAAE,MAAM,QAAQ,CAAC,KACpD,EAAE,aAAa,KAAK,OAAO,QAAQ,MAClC,KAAK,OAAO,SAAS,SAAS,UAAU,KAAK,OAAO,SAAS,SAAS;AAE3E;AAEA,SAAS,eAAe,OAAmB,KAA0D;AACnG,QAAM,WAAW,MAAM,MAAM,MAAM;AACnC,MAAI,CAAC,YAAY,CAAC,KAAK;AACrB,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,KAAK,SAAS,QAAQ,IAAI,GAAG,QAAQ,EAAE,QAAQ,OAAO,GAAG;AACzE,SAAO,GAAG,OAAO,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC;AACjD;AAEA,SAAS,gBAAgB,MAAc,aAAqB;AAC1D,SAAO,EAAE;AAAA,IACP,EAAE;AAAA,MACA;AAAA,MACA,EAAE,iBAAiB,EAAE,WAAW,IAAI,GAAG,EAAE,WAAW,WAAW,CAAC;AAAA,MAChE,EAAE,cAAc,WAAW;AAAA,IAC7B;AAAA,EACF;AACF;AAEA,SAAS,iBACP,iBACA,iBACA,OACA,MACA;AACA,MAAI,gBAAgB,sBAAsB,KAAK,gBAAgB,mBAAmB,GAAG;AACnF,UAAM,OAAO,gBAAgB,KAAK,IAAI;AACtC,QAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG;AACrD;AAAA,IACF;AAEA,UAAM,cAAc,eAAe,OAAO,gBAAgB,KAAK,KAAK,KAAK;AACzE,QAAI,CAAC,aAAa;AAChB;AAAA,IACF;AAEA,SAAK,IAAI,IAAI;AACb,oBAAgB,YAAY,gBAAgB,MAAM,WAAW,CAAC;AAC9D;AAAA,EACF;AAEA,MAAI,CAAC,gBAAgB,sBAAsB,GAAG;AAC5C;AAAA,EACF;AAEA,QAAM,cAAc,gBAAgB,KAAK,aAAa,QAAQ,CAAC,eAAe;AAC5E,QAAI,CAAC,EAAE,aAAa,WAAW,EAAE,KAAK,CAAC,gBAAgB,WAAW,GAAG,IAAI,KAAK,KAAK,IAAI,WAAW,GAAG,IAAI,GAAG;AAC1G,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,WAAW,MAAM;AACpB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,CAAC,yBAAyB,WAAW,IAAI,GAAG;AAC9C,aAAO,CAAC;AAAA,IACV;AAEA,UAAM,cAAc,eAAe,OAAO,WAAW,KAAK,SAAS,WAAW,KAAK,KAAK,KAAK;AAC7F,QAAI,CAAC,aAAa;AAChB,aAAO,CAAC;AAAA,IACV;AAEA,SAAK,IAAI,WAAW,GAAG,IAAI;AAC3B,WAAO,CAAC,gBAAgB,WAAW,GAAG,MAAM,WAAW,CAAC;AAAA,EAC1D,CAAC;AAED,MAAI,YAAY,SAAS,GAAG;AAC1B,oBAAgB,YAAY,WAAW;AAAA,EACzC;AACF;AAEO,SAAS,2BACd,UAA6C,CAAC,GACvB;AACvB,QAAM,EAAE,kBAAkB,OAAO,wBAAwB,KAAK,IAAI;AAElE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,YAAI,CAAC,iBAAiB;AACpB;AAAA,QACF;AAEA,cAAM,gBAAgB,SAAS,KAAK,WAAW;AAAA,UAC7C,CAAC,SAAS,EAAE,eAAe,IAAI,KAAK,EAAE,gBAAgB,KAAK,IAAI,KAAK,KAAK,KAAK,SAAS;AAAA,QACzF;AACA,YAAI,eAAe;AACjB;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,MAAM,MAAM;AACnC,cAAM,MAAM,SAAS,KAAK,KAAK;AAC/B,YAAI,CAAC,YAAY,CAAC,KAAK;AACrB;AAAA,QACF;AAEA,iBAAS,KAAK,WAAW;AAAA,UACvB,EAAE;AAAA,YACA,EAAE,cAAc,WAAW;AAAA,YAC3B,EAAE,cAAc,eAAe,OAAO,GAAG,KAAK,GAAG,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,SAAS,CAAC,EAAE;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,YAAI,CAAC,uBAAuB;AAC1B;AAAA,QACF;AAEA,cAAM,OAAO,oBAAI,IAAY;AAE7B,mBAAW,aAAa,YAAY,IAAI,MAAM,GAAG;AAC/C,cAAI,UAAU,yBAAyB,KAAK,UAAU,2BAA2B,GAAG;AAClF,kBAAM,kBAAkB,UAAU,IAAI,aAAa;AACnD,gBAAI,CAAC,MAAM,QAAQ,eAAe,KAAK,gBAAgB,MAAM;AAC3D,+BAAiB,iBAAiB,WAAW,OAAO,IAAI;AAAA,YAC1D;AACA;AAAA,UACF;AAEA,2BAAiB,WAAW,WAAW,OAAO,IAAI;AAAA,QACpD;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-code-locator",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "type": "module",
   "main": "./dist/index.js",
   "module": "./dist/index.js",
@@ -13,12 +13,12 @@
     },
     "./client": {
       "types": "./dist/client.d.ts",
-      "import": "./dist/client.cjs",
+      "import": "./dist/client.js",
       "require": "./dist/client.cjs"
     },
     "./babel": {
       "types": "./dist/babel.d.ts",
-      "import": "./dist/babel.cjs",
+      "import": "./dist/babel.js",
       "require": "./dist/babel.cjs"
     },
     "./vite": {
@@ -42,21 +42,16 @@
     "typecheck": "tsc --noEmit"
   },
   "peerDependencies": {
-    "@babel/core": ">=7",
-    "@vitejs/plugin-react": ">=4"
+    "@babel/core": ">=7"
   },
   "peerDependenciesMeta": {
     "@babel/core": {
       "optional": true
-    },
-    "@vitejs/plugin-react": {
-      "optional": true
     }
   },
   "devDependencies": {
     "@babel/core": "^7.26.0",
     "@types/node": "^22.13.10",
-    "@vitejs/plugin-react": "^4.3.4",
     "tsup": "^8.5.1",
     "typescript": "^5.8.2",
     "vite": "^5.4.14"