react-code-locator 0.1.4 → 0.1.6

package/dist/client.js

@@ -46,7 +46,22 @@ function getSourceFromType(type) {
   const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];
   return typeof source === "string" ? source : null;
 }
-function resolveSourceFromFiber(fiber) {
+function getSourceFromProps(props) {
+  const source = props?.[SOURCE_PROP];
+  return typeof source === "string" ? source : null;
+}
+function resolveJsxSourceFromFiber(fiber) {
+  let current = fiber;
+  while (current) {
+    const source = getSourceFromProps(current.pendingProps) ?? getSourceFromProps(current.memoizedProps);
+    if (source) {
+      return source;
+    }
+    current = current.return ?? null;
+  }
+  return null;
+}
+function resolveComponentSourceFromFiber(fiber) {
   let current = fiber;
   while (current) {
     const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);
@@ -74,14 +89,14 @@ function locateComponentSource(target) {
   if (!fiber) {
     return null;
   }
-  const debugSource = getDebugSource(fiber);
-  if (debugSource) {
+  const jsxSource = resolveJsxSourceFromFiber(fiber) ?? getDebugSource(fiber);
+  if (jsxSource) {
     return {
-      source: debugSource,
+      source: jsxSource,
       mode: "jsx"
     };
   }
-  const componentSource = resolveSourceFromFiber(fiber);
+  const componentSource = resolveComponentSourceFromFiber(fiber);
   if (!componentSource) {
     return null;
   }
@@ -94,10 +109,10 @@ function enableReactComponentJump(options = {}) {
   const {
     triggerKey = "shift",
     onLocate = (result) => {
-      console.log(`[react-component-jump] ${result.source}`);
+      console.log(`[react-code-locator] ${result.source}`);
     },
     onError = (error) => {
-      console.error("[react-component-jump]", error);
+      console.error("[react-code-locator]", error);
     }
   } = options;
   const handler = (event) => {

package/dist/client.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/constants.ts","../src/runtime.ts"],"sourcesContent":["export const SOURCE_PROP = \"__componentSourceLoc\";\n\n","import { SOURCE_PROP } from \"./constants\";\n\nexport type TriggerKey = \"alt\" | \"meta\" | \"ctrl\" | \"shift\" | \"none\";\n\ntype ReactFiber = {\n  return?: ReactFiber | null;\n  type?: unknown;\n  elementType?: unknown;\n  _debugSource?: {\n    fileName?: string;\n    lineNumber?: number;\n    columnNumber?: number;\n  } | null;\n};\n\nexport type LocatorResult = {\n  source: string;\n  mode: \"jsx\" | \"component\";\n};\n\nexport type LocatorOptions = {\n  triggerKey?: TriggerKey;\n  onLocate?: (result: LocatorResult) => void;\n  onError?: (error: unknown) => void;\n};\n\nfunction isTriggerPressed(event: MouseEvent, triggerKey: TriggerKey) {\n  if (triggerKey === \"none\") {\n    return true;\n  }\n\n  if (triggerKey === \"alt\") {\n    return event.altKey;\n  }\n\n  if (triggerKey === \"meta\") {\n    return event.metaKey;\n  }\n\n  if (triggerKey === \"ctrl\") {\n    return event.ctrlKey;\n  }\n\n  return event.shiftKey;\n}\n\nfunction getReactFiberKey(element: Element) {\n  return Object.keys(element).find((key) => key.startsWith(\"__reactFiber$\") || key.startsWith(\"__reactInternalInstance$\"));\n}\n\nfunction getClosestReactFiber(target: Element | null) {\n  let current = target;\n\n  while (current) {\n    const fiberKey = getReactFiberKey(current);\n    if (fiberKey) {\n      return (current as unknown as Record<string, unknown>)[fiberKey] as ReactFiber;\n    }\n\n    current = current.parentElement;\n  }\n\n  return null;\n}\n\nfunction getSourceFromType(type: unknown) {\n  if (!type) {\n    return null;\n  }\n\n  if (typeof type === \"function\") {\n    const source = (type as unknown as Record<string, unknown>)[SOURCE_PROP];\n    return typeof source === \"string\" ? source : null;\n  }\n\n  if (typeof type !== \"object\") {\n    return null;\n  }\n\n  const record = type as {\n    type?: Record<string, unknown>;\n    render?: Record<string, unknown>;\n    [SOURCE_PROP]?: unknown;\n  };\n\n  const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction resolveSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction getDebugSource(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const debugSource = current._debugSource;\n    if (debugSource?.fileName && typeof debugSource.lineNumber === \"number\") {\n      return `${debugSource.fileName.replace(/\\\\/g, \"/\")}:${debugSource.lineNumber}:${debugSource.columnNumber ?? 1}`;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nexport function locateComponentSource(target: EventTarget | null): LocatorResult | null {\n  const elementTarget =\n    target instanceof Element ? target : target instanceof Node ? target.parentElement : null;\n  const fiber = getClosestReactFiber(elementTarget);\n  if (!fiber) {\n    return null;\n  }\n\n  const debugSource = getDebugSource(fiber);\n  if (debugSource) {\n    return {\n      source: debugSource,\n      mode: \"jsx\",\n    };\n  }\n\n  const componentSource = resolveSourceFromFiber(fiber);\n  if (!componentSource) {\n    return null;\n  }\n\n  return {\n    source: componentSource,\n    mode: \"component\",\n  };\n}\n\nexport function enableReactComponentJump(options: LocatorOptions = {}) {\n  const {\n    triggerKey = \"shift\",\n    onLocate = (result) => {\n      console.log(`[react-component-jump] ${result.source}`);\n    },\n    onError = (error) => {\n      console.error(\"[react-component-jump]\", error);\n    },\n  } = options;\n\n  const handler = (event: MouseEvent) => {\n    if (!isTriggerPressed(event, triggerKey)) {\n      return;\n    }\n\n    const result = locateComponentSource(event.target);\n    if (!result) {\n      onError(new Error(\"No React component source metadata found for clicked element.\"));\n      return;\n    }\n\n    event.preventDefault();\n    event.stopPropagation();\n    onLocate(result);\n  };\n\n  document.addEventListener(\"click\", handler, true);\n\n  return () => {\n    document.removeEventListener(\"click\", handler, true);\n  };\n}\n"],"mappings":";AAAO,IAAM,cAAc;;;AC0B3B,SAAS,iBAAiB,OAAmB,YAAwB;AACnE,MAAI,eAAe,QAAQ;AACzB,WAAO;AAAA,EACT;AAEA,MAAI,eAAe,OAAO;AACxB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,SAAO,MAAM;AACf;AAEA,SAAS,iBAAiB,SAAkB;AAC1C,SAAO,OAAO,KAAK,OAAO,EAAE,KAAK,CAAC,QAAQ,IAAI,WAAW,eAAe,KAAK,IAAI,WAAW,0BAA0B,CAAC;AACzH;AAEA,SAAS,qBAAqB,QAAwB;AACpD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,WAAW,iBAAiB,OAAO;AACzC,QAAI,UAAU;AACZ,aAAQ,QAA+C,QAAQ;AAAA,IACjE;AAEA,cAAU,QAAQ;AAAA,EACpB;AAEA,SAAO;AACT;AAEA,SAAS,kBAAkB,MAAe;AACxC,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,SAAS,YAAY;AAC9B,UAAMA,UAAU,KAA4C,WAAW;AACvE,WAAO,OAAOA,YAAW,WAAWA,UAAS;AAAA,EAC/C;AAEA,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO;AAAA,EACT;AAEA,QAAM,SAAS;AAMf,QAAM,SAAS,OAAO,WAAW,KAAK,OAAO,OAAO,WAAW,KAAK,OAAO,SAAS,WAAW;AAC/F,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,uBAAuB,OAA0B;AACxD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,kBAAkB,QAAQ,IAAI,KAAK,kBAAkB,QAAQ,WAAW;AACvF,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,eAAe,OAA0B;AAChD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,cAAc,QAAQ;AAC5B,QAAI,aAAa,YAAY,OAAO,YAAY,eAAe,UAAU;AACvE,aAAO,GAAG,YAAY,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,YAAY,UAAU,IAAI,YAAY,gBAAgB,CAAC;AAAA,IAC/G;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEO,SAAS,sBAAsB,QAAkD;AACtF,QAAM,gBACJ,kBAAkB,UAAU,SAAS,kBAAkB,OAAO,OAAO,gBAAgB;AACvF,QAAM,QAAQ,qBAAqB,aAAa;AAChD,MAAI,CAAC,OAAO;AACV,WAAO;AAAA,EACT;AAEA,QAAM,cAAc,eAAe,KAAK;AACxC,MAAI,aAAa;AACf,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,kBAAkB,uBAAuB,KAAK;AACpD,MAAI,CAAC,iBAAiB;AACpB,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,MAAM;AAAA,EACR;AACF;AAEO,SAAS,yBAAyB,UAA0B,CAAC,GAAG;AACrE,QAAM;AAAA,IACJ,aAAa;AAAA,IACb,WAAW,CAAC,WAAW;AACrB,cAAQ,IAAI,0BAA0B,OAAO,MAAM,EAAE;AAAA,IACvD;AAAA,IACA,UAAU,CAAC,UAAU;AACnB,cAAQ,MAAM,0BAA0B,KAAK;AAAA,IAC/C;AAAA,EACF,IAAI;AAEJ,QAAM,UAAU,CAAC,UAAsB;AACrC,QAAI,CAAC,iBAAiB,OAAO,UAAU,GAAG;AACxC;AAAA,IACF;AAEA,UAAM,SAAS,sBAAsB,MAAM,MAAM;AACjD,QAAI,CAAC,QAAQ;AACX,cAAQ,IAAI,MAAM,+DAA+D,CAAC;AAClF;AAAA,IACF;AAEA,UAAM,eAAe;AACrB,UAAM,gBAAgB;AACtB,aAAS,MAAM;AAAA,EACjB;AAEA,WAAS,iBAAiB,SAAS,SAAS,IAAI;AAEhD,SAAO,MAAM;AACX,aAAS,oBAAoB,SAAS,SAAS,IAAI;AAAA,EACrD;AACF;","names":["source"]}
+{"version":3,"sources":["../src/constants.ts","../src/runtime.ts"],"sourcesContent":["export const SOURCE_PROP = \"__componentSourceLoc\";\n\n","import { SOURCE_PROP } from \"./constants\";\n\nexport type TriggerKey = \"alt\" | \"meta\" | \"ctrl\" | \"shift\" | \"none\";\n\ntype ReactFiber = {\n  return?: ReactFiber | null;\n  type?: unknown;\n  elementType?: unknown;\n  pendingProps?: Record<string, unknown> | null;\n  memoizedProps?: Record<string, unknown> | null;\n  _debugSource?: {\n    fileName?: string;\n    lineNumber?: number;\n    columnNumber?: number;\n  } | null;\n};\n\nexport type LocatorResult = {\n  source: string;\n  mode: \"jsx\" | \"component\";\n};\n\nexport type LocatorOptions = {\n  triggerKey?: TriggerKey;\n  onLocate?: (result: LocatorResult) => void;\n  onError?: (error: unknown) => void;\n};\n\nfunction isTriggerPressed(event: MouseEvent, triggerKey: TriggerKey) {\n  if (triggerKey === \"none\") {\n    return true;\n  }\n\n  if (triggerKey === \"alt\") {\n    return event.altKey;\n  }\n\n  if (triggerKey === \"meta\") {\n    return event.metaKey;\n  }\n\n  if (triggerKey === \"ctrl\") {\n    return event.ctrlKey;\n  }\n\n  return event.shiftKey;\n}\n\nfunction getReactFiberKey(element: Element) {\n  return Object.keys(element).find((key) => key.startsWith(\"__reactFiber$\") || key.startsWith(\"__reactInternalInstance$\"));\n}\n\nfunction getClosestReactFiber(target: Element | null) {\n  let current = target;\n\n  while (current) {\n    const fiberKey = getReactFiberKey(current);\n    if (fiberKey) {\n      return (current as unknown as Record<string, unknown>)[fiberKey] as ReactFiber;\n    }\n\n    current = current.parentElement;\n  }\n\n  return null;\n}\n\nfunction getSourceFromType(type: unknown) {\n  if (!type) {\n    return null;\n  }\n\n  if (typeof type === \"function\") {\n    const source = (type as unknown as Record<string, unknown>)[SOURCE_PROP];\n    return typeof source === \"string\" ? source : null;\n  }\n\n  if (typeof type !== \"object\") {\n    return null;\n  }\n\n  const record = type as {\n    type?: Record<string, unknown>;\n    render?: Record<string, unknown>;\n    [SOURCE_PROP]?: unknown;\n  };\n\n  const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction getSourceFromProps(props: Record<string, unknown> | null | undefined) {\n  const source = props?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction resolveJsxSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromProps(current.pendingProps) ?? getSourceFromProps(current.memoizedProps);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction resolveComponentSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction getDebugSource(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const debugSource = current._debugSource;\n    if (debugSource?.fileName && typeof debugSource.lineNumber === \"number\") {\n      return `${debugSource.fileName.replace(/\\\\/g, \"/\")}:${debugSource.lineNumber}:${debugSource.columnNumber ?? 1}`;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nexport function locateComponentSource(target: EventTarget | null): LocatorResult | null {\n  const elementTarget =\n    target instanceof Element ? target : target instanceof Node ? target.parentElement : null;\n  const fiber = getClosestReactFiber(elementTarget);\n  if (!fiber) {\n    return null;\n  }\n\n  const jsxSource = resolveJsxSourceFromFiber(fiber) ?? getDebugSource(fiber);\n  if (jsxSource) {\n    return {\n      source: jsxSource,\n      mode: \"jsx\",\n    };\n  }\n\n  const componentSource = resolveComponentSourceFromFiber(fiber);\n  if (!componentSource) {\n    return null;\n  }\n\n  return {\n    source: componentSource,\n    mode: \"component\",\n  };\n}\n\nexport function enableReactComponentJump(options: LocatorOptions = {}) {\n  const {\n    triggerKey = \"shift\",\n    onLocate = (result) => {\n      console.log(`[react-code-locator] ${result.source}`);\n    },\n    onError = (error) => {\n      console.error(\"[react-code-locator]\", error);\n    },\n  } = options;\n\n  const handler = (event: MouseEvent) => {\n    if (!isTriggerPressed(event, triggerKey)) {\n      return;\n    }\n\n    const result = locateComponentSource(event.target);\n    if (!result) {\n      onError(new Error(\"No React component source metadata found for clicked element.\"));\n      return;\n    }\n\n    event.preventDefault();\n    event.stopPropagation();\n    onLocate(result);\n  };\n\n  document.addEventListener(\"click\", handler, true);\n\n  return () => {\n    document.removeEventListener(\"click\", handler, true);\n  };\n}\n"],"mappings":";AAAO,IAAM,cAAc;;;AC4B3B,SAAS,iBAAiB,OAAmB,YAAwB;AACnE,MAAI,eAAe,QAAQ;AACzB,WAAO;AAAA,EACT;AAEA,MAAI,eAAe,OAAO;AACxB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,SAAO,MAAM;AACf;AAEA,SAAS,iBAAiB,SAAkB;AAC1C,SAAO,OAAO,KAAK,OAAO,EAAE,KAAK,CAAC,QAAQ,IAAI,WAAW,eAAe,KAAK,IAAI,WAAW,0BAA0B,CAAC;AACzH;AAEA,SAAS,qBAAqB,QAAwB;AACpD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,WAAW,iBAAiB,OAAO;AACzC,QAAI,UAAU;AACZ,aAAQ,QAA+C,QAAQ;AAAA,IACjE;AAEA,cAAU,QAAQ;AAAA,EACpB;AAEA,SAAO;AACT;AAEA,SAAS,kBAAkB,MAAe;AACxC,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,SAAS,YAAY;AAC9B,UAAMA,UAAU,KAA4C,WAAW;AACvE,WAAO,OAAOA,YAAW,WAAWA,UAAS;AAAA,EAC/C;AAEA,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO;AAAA,EACT;AAEA,QAAM,SAAS;AAMf,QAAM,SAAS,OAAO,WAAW,KAAK,OAAO,OAAO,WAAW,KAAK,OAAO,SAAS,WAAW;AAC/F,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,mBAAmB,OAAmD;AAC7E,QAAM,SAAS,QAAQ,WAAW;AAClC,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,0BAA0B,OAA0B;AAC3D,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,mBAAmB,QAAQ,YAAY,KAAK,mBAAmB,QAAQ,aAAa;AACnG,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,gCAAgC,OAA0B;AACjE,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,kBAAkB,QAAQ,IAAI,KAAK,kBAAkB,QAAQ,WAAW;AACvF,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,eAAe,OAA0B;AAChD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,cAAc,QAAQ;AAC5B,QAAI,aAAa,YAAY,OAAO,YAAY,eAAe,UAAU;AACvE,aAAO,GAAG,YAAY,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,YAAY,UAAU,IAAI,YAAY,gBAAgB,CAAC;AAAA,IAC/G;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEO,SAAS,sBAAsB,QAAkD;AACtF,QAAM,gBACJ,kBAAkB,UAAU,SAAS,kBAAkB,OAAO,OAAO,gBAAgB;AACvF,QAAM,QAAQ,qBAAqB,aAAa;AAChD,MAAI,CAAC,OAAO;AACV,WAAO;AAAA,EACT;AAEA,QAAM,YAAY,0BAA0B,KAAK,KAAK,eAAe,KAAK;AAC1E,MAAI,WAAW;AACb,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,kBAAkB,gCAAgC,KAAK;AAC7D,MAAI,CAAC,iBAAiB;AACpB,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,MAAM;AAAA,EACR;AACF;AAEO,SAAS,yBAAyB,UAA0B,CAAC,GAAG;AACrE,QAAM;AAAA,IACJ,aAAa;AAAA,IACb,WAAW,CAAC,WAAW;AACrB,cAAQ,IAAI,wBAAwB,OAAO,MAAM,EAAE;AAAA,IACrD;AAAA,IACA,UAAU,CAAC,UAAU;AACnB,cAAQ,MAAM,wBAAwB,KAAK;AAAA,IAC7C;AAAA,EACF,IAAI;AAEJ,QAAM,UAAU,CAAC,UAAsB;AACrC,QAAI,CAAC,iBAAiB,OAAO,UAAU,GAAG;AACxC;AAAA,IACF;AAEA,UAAM,SAAS,sBAAsB,MAAM,MAAM;AACjD,QAAI,CAAC,QAAQ;AACX,cAAQ,IAAI,MAAM,+DAA+D,CAAC;AAClF;AAAA,IACF;AAEA,UAAM,eAAe;AACrB,UAAM,gBAAgB;AACtB,aAAS,MAAM;AAAA,EACjB;AAEA,WAAS,iBAAiB,SAAS,SAAS,IAAI;AAEhD,SAAO,MAAM;AACX,aAAS,oBAAoB,SAAS,SAAS,IAAI;AAAA,EACrD;AACF;","names":["source"]}

package/dist/index.cjs

@@ -73,7 +73,22 @@ function getSourceFromType(type) {
   const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];
   return typeof source === "string" ? source : null;
 }
-function resolveSourceFromFiber(fiber) {
+function getSourceFromProps(props) {
+  const source = props?.[SOURCE_PROP];
+  return typeof source === "string" ? source : null;
+}
+function resolveJsxSourceFromFiber(fiber) {
+  let current = fiber;
+  while (current) {
+    const source = getSourceFromProps(current.pendingProps) ?? getSourceFromProps(current.memoizedProps);
+    if (source) {
+      return source;
+    }
+    current = current.return ?? null;
+  }
+  return null;
+}
+function resolveComponentSourceFromFiber(fiber) {
   let current = fiber;
   while (current) {
     const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);
@@ -101,14 +116,14 @@ function locateComponentSource(target) {
   if (!fiber) {
     return null;
   }
-  const debugSource = getDebugSource(fiber);
-  if (debugSource) {
+  const jsxSource = resolveJsxSourceFromFiber(fiber) ?? getDebugSource(fiber);
+  if (jsxSource) {
     return {
-      source: debugSource,
+      source: jsxSource,
       mode: "jsx"
     };
   }
-  const componentSource = resolveSourceFromFiber(fiber);
+  const componentSource = resolveComponentSourceFromFiber(fiber);
   if (!componentSource) {
     return null;
   }
@@ -121,10 +136,10 @@ function enableReactComponentJump(options = {}) {
   const {
     triggerKey = "shift",
     onLocate = (result) => {
-      console.log(`[react-component-jump] ${result.source}`);
+      console.log(`[react-code-locator] ${result.source}`);
     },
     onError = (error) => {
-      console.error("[react-component-jump]", error);
+      console.error("[react-code-locator]", error);
     }
   } = options;
   const handler = (event) => {

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/constants.ts","../src/runtime.ts"],"sourcesContent":["export { enableReactComponentJump, locateComponentSource } from \"./runtime\";\nexport type { LocatorOptions, LocatorResult, TriggerKey } from \"./runtime\";\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n","import { SOURCE_PROP } from \"./constants\";\n\nexport type TriggerKey = \"alt\" | \"meta\" | \"ctrl\" | \"shift\" | \"none\";\n\ntype ReactFiber = {\n  return?: ReactFiber | null;\n  type?: unknown;\n  elementType?: unknown;\n  _debugSource?: {\n    fileName?: string;\n    lineNumber?: number;\n    columnNumber?: number;\n  } | null;\n};\n\nexport type LocatorResult = {\n  source: string;\n  mode: \"jsx\" | \"component\";\n};\n\nexport type LocatorOptions = {\n  triggerKey?: TriggerKey;\n  onLocate?: (result: LocatorResult) => void;\n  onError?: (error: unknown) => void;\n};\n\nfunction isTriggerPressed(event: MouseEvent, triggerKey: TriggerKey) {\n  if (triggerKey === \"none\") {\n    return true;\n  }\n\n  if (triggerKey === \"alt\") {\n    return event.altKey;\n  }\n\n  if (triggerKey === \"meta\") {\n    return event.metaKey;\n  }\n\n  if (triggerKey === \"ctrl\") {\n    return event.ctrlKey;\n  }\n\n  return event.shiftKey;\n}\n\nfunction getReactFiberKey(element: Element) {\n  return Object.keys(element).find((key) => key.startsWith(\"__reactFiber$\") || key.startsWith(\"__reactInternalInstance$\"));\n}\n\nfunction getClosestReactFiber(target: Element | null) {\n  let current = target;\n\n  while (current) {\n    const fiberKey = getReactFiberKey(current);\n    if (fiberKey) {\n      return (current as unknown as Record<string, unknown>)[fiberKey] as ReactFiber;\n    }\n\n    current = current.parentElement;\n  }\n\n  return null;\n}\n\nfunction getSourceFromType(type: unknown) {\n  if (!type) {\n    return null;\n  }\n\n  if (typeof type === \"function\") {\n    const source = (type as unknown as Record<string, unknown>)[SOURCE_PROP];\n    return typeof source === \"string\" ? source : null;\n  }\n\n  if (typeof type !== \"object\") {\n    return null;\n  }\n\n  const record = type as {\n    type?: Record<string, unknown>;\n    render?: Record<string, unknown>;\n    [SOURCE_PROP]?: unknown;\n  };\n\n  const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction resolveSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction getDebugSource(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const debugSource = current._debugSource;\n    if (debugSource?.fileName && typeof debugSource.lineNumber === \"number\") {\n      return `${debugSource.fileName.replace(/\\\\/g, \"/\")}:${debugSource.lineNumber}:${debugSource.columnNumber ?? 1}`;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nexport function locateComponentSource(target: EventTarget | null): LocatorResult | null {\n  const elementTarget =\n    target instanceof Element ? target : target instanceof Node ? target.parentElement : null;\n  const fiber = getClosestReactFiber(elementTarget);\n  if (!fiber) {\n    return null;\n  }\n\n  const debugSource = getDebugSource(fiber);\n  if (debugSource) {\n    return {\n      source: debugSource,\n      mode: \"jsx\",\n    };\n  }\n\n  const componentSource = resolveSourceFromFiber(fiber);\n  if (!componentSource) {\n    return null;\n  }\n\n  return {\n    source: componentSource,\n    mode: \"component\",\n  };\n}\n\nexport function enableReactComponentJump(options: LocatorOptions = {}) {\n  const {\n    triggerKey = \"shift\",\n    onLocate = (result) => {\n      console.log(`[react-component-jump] ${result.source}`);\n    },\n    onError = (error) => {\n      console.error(\"[react-component-jump]\", error);\n    },\n  } = options;\n\n  const handler = (event: MouseEvent) => {\n    if (!isTriggerPressed(event, triggerKey)) {\n      return;\n    }\n\n    const result = locateComponentSource(event.target);\n    if (!result) {\n      onError(new Error(\"No React component source metadata found for clicked element.\"));\n      return;\n    }\n\n    event.preventDefault();\n    event.stopPropagation();\n    onLocate(result);\n  };\n\n  document.addEventListener(\"click\", handler, true);\n\n  return () => {\n    document.removeEventListener(\"click\", handler, true);\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAO,IAAM,cAAc;;;AC0B3B,SAAS,iBAAiB,OAAmB,YAAwB;AACnE,MAAI,eAAe,QAAQ;AACzB,WAAO;AAAA,EACT;AAEA,MAAI,eAAe,OAAO;AACxB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,SAAO,MAAM;AACf;AAEA,SAAS,iBAAiB,SAAkB;AAC1C,SAAO,OAAO,KAAK,OAAO,EAAE,KAAK,CAAC,QAAQ,IAAI,WAAW,eAAe,KAAK,IAAI,WAAW,0BAA0B,CAAC;AACzH;AAEA,SAAS,qBAAqB,QAAwB;AACpD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,WAAW,iBAAiB,OAAO;AACzC,QAAI,UAAU;AACZ,aAAQ,QAA+C,QAAQ;AAAA,IACjE;AAEA,cAAU,QAAQ;AAAA,EACpB;AAEA,SAAO;AACT;AAEA,SAAS,kBAAkB,MAAe;AACxC,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,SAAS,YAAY;AAC9B,UAAMA,UAAU,KAA4C,WAAW;AACvE,WAAO,OAAOA,YAAW,WAAWA,UAAS;AAAA,EAC/C;AAEA,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO;AAAA,EACT;AAEA,QAAM,SAAS;AAMf,QAAM,SAAS,OAAO,WAAW,KAAK,OAAO,OAAO,WAAW,KAAK,OAAO,SAAS,WAAW;AAC/F,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,uBAAuB,OAA0B;AACxD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,kBAAkB,QAAQ,IAAI,KAAK,kBAAkB,QAAQ,WAAW;AACvF,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,eAAe,OAA0B;AAChD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,cAAc,QAAQ;AAC5B,QAAI,aAAa,YAAY,OAAO,YAAY,eAAe,UAAU;AACvE,aAAO,GAAG,YAAY,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,YAAY,UAAU,IAAI,YAAY,gBAAgB,CAAC;AAAA,IAC/G;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEO,SAAS,sBAAsB,QAAkD;AACtF,QAAM,gBACJ,kBAAkB,UAAU,SAAS,kBAAkB,OAAO,OAAO,gBAAgB;AACvF,QAAM,QAAQ,qBAAqB,aAAa;AAChD,MAAI,CAAC,OAAO;AACV,WAAO;AAAA,EACT;AAEA,QAAM,cAAc,eAAe,KAAK;AACxC,MAAI,aAAa;AACf,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,kBAAkB,uBAAuB,KAAK;AACpD,MAAI,CAAC,iBAAiB;AACpB,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,MAAM;AAAA,EACR;AACF;AAEO,SAAS,yBAAyB,UAA0B,CAAC,GAAG;AACrE,QAAM;AAAA,IACJ,aAAa;AAAA,IACb,WAAW,CAAC,WAAW;AACrB,cAAQ,IAAI,0BAA0B,OAAO,MAAM,EAAE;AAAA,IACvD;AAAA,IACA,UAAU,CAAC,UAAU;AACnB,cAAQ,MAAM,0BAA0B,KAAK;AAAA,IAC/C;AAAA,EACF,IAAI;AAEJ,QAAM,UAAU,CAAC,UAAsB;AACrC,QAAI,CAAC,iBAAiB,OAAO,UAAU,GAAG;AACxC;AAAA,IACF;AAEA,UAAM,SAAS,sBAAsB,MAAM,MAAM;AACjD,QAAI,CAAC,QAAQ;AACX,cAAQ,IAAI,MAAM,+DAA+D,CAAC;AAClF;AAAA,IACF;AAEA,UAAM,eAAe;AACrB,UAAM,gBAAgB;AACtB,aAAS,MAAM;AAAA,EACjB;AAEA,WAAS,iBAAiB,SAAS,SAAS,IAAI;AAEhD,SAAO,MAAM;AACX,aAAS,oBAAoB,SAAS,SAAS,IAAI;AAAA,EACrD;AACF;","names":["source"]}
+{"version":3,"sources":["../src/index.ts","../src/constants.ts","../src/runtime.ts"],"sourcesContent":["export { enableReactComponentJump, locateComponentSource } from \"./runtime\";\nexport type { LocatorOptions, LocatorResult, TriggerKey } from \"./runtime\";\n","export const SOURCE_PROP = \"__componentSourceLoc\";\n\n","import { SOURCE_PROP } from \"./constants\";\n\nexport type TriggerKey = \"alt\" | \"meta\" | \"ctrl\" | \"shift\" | \"none\";\n\ntype ReactFiber = {\n  return?: ReactFiber | null;\n  type?: unknown;\n  elementType?: unknown;\n  pendingProps?: Record<string, unknown> | null;\n  memoizedProps?: Record<string, unknown> | null;\n  _debugSource?: {\n    fileName?: string;\n    lineNumber?: number;\n    columnNumber?: number;\n  } | null;\n};\n\nexport type LocatorResult = {\n  source: string;\n  mode: \"jsx\" | \"component\";\n};\n\nexport type LocatorOptions = {\n  triggerKey?: TriggerKey;\n  onLocate?: (result: LocatorResult) => void;\n  onError?: (error: unknown) => void;\n};\n\nfunction isTriggerPressed(event: MouseEvent, triggerKey: TriggerKey) {\n  if (triggerKey === \"none\") {\n    return true;\n  }\n\n  if (triggerKey === \"alt\") {\n    return event.altKey;\n  }\n\n  if (triggerKey === \"meta\") {\n    return event.metaKey;\n  }\n\n  if (triggerKey === \"ctrl\") {\n    return event.ctrlKey;\n  }\n\n  return event.shiftKey;\n}\n\nfunction getReactFiberKey(element: Element) {\n  return Object.keys(element).find((key) => key.startsWith(\"__reactFiber$\") || key.startsWith(\"__reactInternalInstance$\"));\n}\n\nfunction getClosestReactFiber(target: Element | null) {\n  let current = target;\n\n  while (current) {\n    const fiberKey = getReactFiberKey(current);\n    if (fiberKey) {\n      return (current as unknown as Record<string, unknown>)[fiberKey] as ReactFiber;\n    }\n\n    current = current.parentElement;\n  }\n\n  return null;\n}\n\nfunction getSourceFromType(type: unknown) {\n  if (!type) {\n    return null;\n  }\n\n  if (typeof type === \"function\") {\n    const source = (type as unknown as Record<string, unknown>)[SOURCE_PROP];\n    return typeof source === \"string\" ? source : null;\n  }\n\n  if (typeof type !== \"object\") {\n    return null;\n  }\n\n  const record = type as {\n    type?: Record<string, unknown>;\n    render?: Record<string, unknown>;\n    [SOURCE_PROP]?: unknown;\n  };\n\n  const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction getSourceFromProps(props: Record<string, unknown> | null | undefined) {\n  const source = props?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction resolveJsxSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromProps(current.pendingProps) ?? getSourceFromProps(current.memoizedProps);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction resolveComponentSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction getDebugSource(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const debugSource = current._debugSource;\n    if (debugSource?.fileName && typeof debugSource.lineNumber === \"number\") {\n      return `${debugSource.fileName.replace(/\\\\/g, \"/\")}:${debugSource.lineNumber}:${debugSource.columnNumber ?? 1}`;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nexport function locateComponentSource(target: EventTarget | null): LocatorResult | null {\n  const elementTarget =\n    target instanceof Element ? target : target instanceof Node ? target.parentElement : null;\n  const fiber = getClosestReactFiber(elementTarget);\n  if (!fiber) {\n    return null;\n  }\n\n  const jsxSource = resolveJsxSourceFromFiber(fiber) ?? getDebugSource(fiber);\n  if (jsxSource) {\n    return {\n      source: jsxSource,\n      mode: \"jsx\",\n    };\n  }\n\n  const componentSource = resolveComponentSourceFromFiber(fiber);\n  if (!componentSource) {\n    return null;\n  }\n\n  return {\n    source: componentSource,\n    mode: \"component\",\n  };\n}\n\nexport function enableReactComponentJump(options: LocatorOptions = {}) {\n  const {\n    triggerKey = \"shift\",\n    onLocate = (result) => {\n      console.log(`[react-code-locator] ${result.source}`);\n    },\n    onError = (error) => {\n      console.error(\"[react-code-locator]\", error);\n    },\n  } = options;\n\n  const handler = (event: MouseEvent) => {\n    if (!isTriggerPressed(event, triggerKey)) {\n      return;\n    }\n\n    const result = locateComponentSource(event.target);\n    if (!result) {\n      onError(new Error(\"No React component source metadata found for clicked element.\"));\n      return;\n    }\n\n    event.preventDefault();\n    event.stopPropagation();\n    onLocate(result);\n  };\n\n  document.addEventListener(\"click\", handler, true);\n\n  return () => {\n    document.removeEventListener(\"click\", handler, true);\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAO,IAAM,cAAc;;;AC4B3B,SAAS,iBAAiB,OAAmB,YAAwB;AACnE,MAAI,eAAe,QAAQ;AACzB,WAAO;AAAA,EACT;AAEA,MAAI,eAAe,OAAO;AACxB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,SAAO,MAAM;AACf;AAEA,SAAS,iBAAiB,SAAkB;AAC1C,SAAO,OAAO,KAAK,OAAO,EAAE,KAAK,CAAC,QAAQ,IAAI,WAAW,eAAe,KAAK,IAAI,WAAW,0BAA0B,CAAC;AACzH;AAEA,SAAS,qBAAqB,QAAwB;AACpD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,WAAW,iBAAiB,OAAO;AACzC,QAAI,UAAU;AACZ,aAAQ,QAA+C,QAAQ;AAAA,IACjE;AAEA,cAAU,QAAQ;AAAA,EACpB;AAEA,SAAO;AACT;AAEA,SAAS,kBAAkB,MAAe;AACxC,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,SAAS,YAAY;AAC9B,UAAMA,UAAU,KAA4C,WAAW;AACvE,WAAO,OAAOA,YAAW,WAAWA,UAAS;AAAA,EAC/C;AAEA,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO;AAAA,EACT;AAEA,QAAM,SAAS;AAMf,QAAM,SAAS,OAAO,WAAW,KAAK,OAAO,OAAO,WAAW,KAAK,OAAO,SAAS,WAAW;AAC/F,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,mBAAmB,OAAmD;AAC7E,QAAM,SAAS,QAAQ,WAAW;AAClC,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,0BAA0B,OAA0B;AAC3D,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,mBAAmB,QAAQ,YAAY,KAAK,mBAAmB,QAAQ,aAAa;AACnG,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,gCAAgC,OAA0B;AACjE,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,kBAAkB,QAAQ,IAAI,KAAK,kBAAkB,QAAQ,WAAW;AACvF,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,eAAe,OAA0B;AAChD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,cAAc,QAAQ;AAC5B,QAAI,aAAa,YAAY,OAAO,YAAY,eAAe,UAAU;AACvE,aAAO,GAAG,YAAY,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,YAAY,UAAU,IAAI,YAAY,gBAAgB,CAAC;AAAA,IAC/G;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEO,SAAS,sBAAsB,QAAkD;AACtF,QAAM,gBACJ,kBAAkB,UAAU,SAAS,kBAAkB,OAAO,OAAO,gBAAgB;AACvF,QAAM,QAAQ,qBAAqB,aAAa;AAChD,MAAI,CAAC,OAAO;AACV,WAAO;AAAA,EACT;AAEA,QAAM,YAAY,0BAA0B,KAAK,KAAK,eAAe,KAAK;AAC1E,MAAI,WAAW;AACb,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,kBAAkB,gCAAgC,KAAK;AAC7D,MAAI,CAAC,iBAAiB;AACpB,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,MAAM;AAAA,EACR;AACF;AAEO,SAAS,yBAAyB,UAA0B,CAAC,GAAG;AACrE,QAAM;AAAA,IACJ,aAAa;AAAA,IACb,WAAW,CAAC,WAAW;AACrB,cAAQ,IAAI,wBAAwB,OAAO,MAAM,EAAE;AAAA,IACrD;AAAA,IACA,UAAU,CAAC,UAAU;AACnB,cAAQ,MAAM,wBAAwB,KAAK;AAAA,IAC7C;AAAA,EACF,IAAI;AAEJ,QAAM,UAAU,CAAC,UAAsB;AACrC,QAAI,CAAC,iBAAiB,OAAO,UAAU,GAAG;AACxC;AAAA,IACF;AAEA,UAAM,SAAS,sBAAsB,MAAM,MAAM;AACjD,QAAI,CAAC,QAAQ;AACX,cAAQ,IAAI,MAAM,+DAA+D,CAAC;AAClF;AAAA,IACF;AAEA,UAAM,eAAe;AACrB,UAAM,gBAAgB;AACtB,aAAS,MAAM;AAAA,EACjB;AAEA,WAAS,iBAAiB,SAAS,SAAS,IAAI;AAEhD,SAAO,MAAM;AACX,aAAS,oBAAoB,SAAS,SAAS,IAAI;AAAA,EACrD;AACF;","names":["source"]}

package/dist/index.js

@@ -46,7 +46,22 @@ function getSourceFromType(type) {
   const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];
   return typeof source === "string" ? source : null;
 }
-function resolveSourceFromFiber(fiber) {
+function getSourceFromProps(props) {
+  const source = props?.[SOURCE_PROP];
+  return typeof source === "string" ? source : null;
+}
+function resolveJsxSourceFromFiber(fiber) {
+  let current = fiber;
+  while (current) {
+    const source = getSourceFromProps(current.pendingProps) ?? getSourceFromProps(current.memoizedProps);
+    if (source) {
+      return source;
+    }
+    current = current.return ?? null;
+  }
+  return null;
+}
+function resolveComponentSourceFromFiber(fiber) {
   let current = fiber;
   while (current) {
     const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);
@@ -74,14 +89,14 @@ function locateComponentSource(target) {
   if (!fiber) {
     return null;
   }
-  const debugSource = getDebugSource(fiber);
-  if (debugSource) {
+  const jsxSource = resolveJsxSourceFromFiber(fiber) ?? getDebugSource(fiber);
+  if (jsxSource) {
     return {
-      source: debugSource,
+      source: jsxSource,
       mode: "jsx"
     };
   }
-  const componentSource = resolveSourceFromFiber(fiber);
+  const componentSource = resolveComponentSourceFromFiber(fiber);
   if (!componentSource) {
     return null;
   }
@@ -94,10 +109,10 @@ function enableReactComponentJump(options = {}) {
   const {
     triggerKey = "shift",
     onLocate = (result) => {
-      console.log(`[react-component-jump] ${result.source}`);
+      console.log(`[react-code-locator] ${result.source}`);
     },
     onError = (error) => {
-      console.error("[react-component-jump]", error);
+      console.error("[react-code-locator]", error);
     }
   } = options;
   const handler = (event) => {

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/constants.ts","../src/runtime.ts"],"sourcesContent":["export const SOURCE_PROP = \"__componentSourceLoc\";\n\n","import { SOURCE_PROP } from \"./constants\";\n\nexport type TriggerKey = \"alt\" | \"meta\" | \"ctrl\" | \"shift\" | \"none\";\n\ntype ReactFiber = {\n  return?: ReactFiber | null;\n  type?: unknown;\n  elementType?: unknown;\n  _debugSource?: {\n    fileName?: string;\n    lineNumber?: number;\n    columnNumber?: number;\n  } | null;\n};\n\nexport type LocatorResult = {\n  source: string;\n  mode: \"jsx\" | \"component\";\n};\n\nexport type LocatorOptions = {\n  triggerKey?: TriggerKey;\n  onLocate?: (result: LocatorResult) => void;\n  onError?: (error: unknown) => void;\n};\n\nfunction isTriggerPressed(event: MouseEvent, triggerKey: TriggerKey) {\n  if (triggerKey === \"none\") {\n    return true;\n  }\n\n  if (triggerKey === \"alt\") {\n    return event.altKey;\n  }\n\n  if (triggerKey === \"meta\") {\n    return event.metaKey;\n  }\n\n  if (triggerKey === \"ctrl\") {\n    return event.ctrlKey;\n  }\n\n  return event.shiftKey;\n}\n\nfunction getReactFiberKey(element: Element) {\n  return Object.keys(element).find((key) => key.startsWith(\"__reactFiber$\") || key.startsWith(\"__reactInternalInstance$\"));\n}\n\nfunction getClosestReactFiber(target: Element | null) {\n  let current = target;\n\n  while (current) {\n    const fiberKey = getReactFiberKey(current);\n    if (fiberKey) {\n      return (current as unknown as Record<string, unknown>)[fiberKey] as ReactFiber;\n    }\n\n    current = current.parentElement;\n  }\n\n  return null;\n}\n\nfunction getSourceFromType(type: unknown) {\n  if (!type) {\n    return null;\n  }\n\n  if (typeof type === \"function\") {\n    const source = (type as unknown as Record<string, unknown>)[SOURCE_PROP];\n    return typeof source === \"string\" ? source : null;\n  }\n\n  if (typeof type !== \"object\") {\n    return null;\n  }\n\n  const record = type as {\n    type?: Record<string, unknown>;\n    render?: Record<string, unknown>;\n    [SOURCE_PROP]?: unknown;\n  };\n\n  const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction resolveSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction getDebugSource(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const debugSource = current._debugSource;\n    if (debugSource?.fileName && typeof debugSource.lineNumber === \"number\") {\n      return `${debugSource.fileName.replace(/\\\\/g, \"/\")}:${debugSource.lineNumber}:${debugSource.columnNumber ?? 1}`;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nexport function locateComponentSource(target: EventTarget | null): LocatorResult | null {\n  const elementTarget =\n    target instanceof Element ? target : target instanceof Node ? target.parentElement : null;\n  const fiber = getClosestReactFiber(elementTarget);\n  if (!fiber) {\n    return null;\n  }\n\n  const debugSource = getDebugSource(fiber);\n  if (debugSource) {\n    return {\n      source: debugSource,\n      mode: \"jsx\",\n    };\n  }\n\n  const componentSource = resolveSourceFromFiber(fiber);\n  if (!componentSource) {\n    return null;\n  }\n\n  return {\n    source: componentSource,\n    mode: \"component\",\n  };\n}\n\nexport function enableReactComponentJump(options: LocatorOptions = {}) {\n  const {\n    triggerKey = \"shift\",\n    onLocate = (result) => {\n      console.log(`[react-component-jump] ${result.source}`);\n    },\n    onError = (error) => {\n      console.error(\"[react-component-jump]\", error);\n    },\n  } = options;\n\n  const handler = (event: MouseEvent) => {\n    if (!isTriggerPressed(event, triggerKey)) {\n      return;\n    }\n\n    const result = locateComponentSource(event.target);\n    if (!result) {\n      onError(new Error(\"No React component source metadata found for clicked element.\"));\n      return;\n    }\n\n    event.preventDefault();\n    event.stopPropagation();\n    onLocate(result);\n  };\n\n  document.addEventListener(\"click\", handler, true);\n\n  return () => {\n    document.removeEventListener(\"click\", handler, true);\n  };\n}\n"],"mappings":";AAAO,IAAM,cAAc;;;AC0B3B,SAAS,iBAAiB,OAAmB,YAAwB;AACnE,MAAI,eAAe,QAAQ;AACzB,WAAO;AAAA,EACT;AAEA,MAAI,eAAe,OAAO;AACxB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,SAAO,MAAM;AACf;AAEA,SAAS,iBAAiB,SAAkB;AAC1C,SAAO,OAAO,KAAK,OAAO,EAAE,KAAK,CAAC,QAAQ,IAAI,WAAW,eAAe,KAAK,IAAI,WAAW,0BAA0B,CAAC;AACzH;AAEA,SAAS,qBAAqB,QAAwB;AACpD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,WAAW,iBAAiB,OAAO;AACzC,QAAI,UAAU;AACZ,aAAQ,QAA+C,QAAQ;AAAA,IACjE;AAEA,cAAU,QAAQ;AAAA,EACpB;AAEA,SAAO;AACT;AAEA,SAAS,kBAAkB,MAAe;AACxC,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,SAAS,YAAY;AAC9B,UAAMA,UAAU,KAA4C,WAAW;AACvE,WAAO,OAAOA,YAAW,WAAWA,UAAS;AAAA,EAC/C;AAEA,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO;AAAA,EACT;AAEA,QAAM,SAAS;AAMf,QAAM,SAAS,OAAO,WAAW,KAAK,OAAO,OAAO,WAAW,KAAK,OAAO,SAAS,WAAW;AAC/F,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,uBAAuB,OAA0B;AACxD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,kBAAkB,QAAQ,IAAI,KAAK,kBAAkB,QAAQ,WAAW;AACvF,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,eAAe,OAA0B;AAChD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,cAAc,QAAQ;AAC5B,QAAI,aAAa,YAAY,OAAO,YAAY,eAAe,UAAU;AACvE,aAAO,GAAG,YAAY,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,YAAY,UAAU,IAAI,YAAY,gBAAgB,CAAC;AAAA,IAC/G;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEO,SAAS,sBAAsB,QAAkD;AACtF,QAAM,gBACJ,kBAAkB,UAAU,SAAS,kBAAkB,OAAO,OAAO,gBAAgB;AACvF,QAAM,QAAQ,qBAAqB,aAAa;AAChD,MAAI,CAAC,OAAO;AACV,WAAO;AAAA,EACT;AAEA,QAAM,cAAc,eAAe,KAAK;AACxC,MAAI,aAAa;AACf,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,kBAAkB,uBAAuB,KAAK;AACpD,MAAI,CAAC,iBAAiB;AACpB,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,MAAM;AAAA,EACR;AACF;AAEO,SAAS,yBAAyB,UAA0B,CAAC,GAAG;AACrE,QAAM;AAAA,IACJ,aAAa;AAAA,IACb,WAAW,CAAC,WAAW;AACrB,cAAQ,IAAI,0BAA0B,OAAO,MAAM,EAAE;AAAA,IACvD;AAAA,IACA,UAAU,CAAC,UAAU;AACnB,cAAQ,MAAM,0BAA0B,KAAK;AAAA,IAC/C;AAAA,EACF,IAAI;AAEJ,QAAM,UAAU,CAAC,UAAsB;AACrC,QAAI,CAAC,iBAAiB,OAAO,UAAU,GAAG;AACxC;AAAA,IACF;AAEA,UAAM,SAAS,sBAAsB,MAAM,MAAM;AACjD,QAAI,CAAC,QAAQ;AACX,cAAQ,IAAI,MAAM,+DAA+D,CAAC;AAClF;AAAA,IACF;AAEA,UAAM,eAAe;AACrB,UAAM,gBAAgB;AACtB,aAAS,MAAM;AAAA,EACjB;AAEA,WAAS,iBAAiB,SAAS,SAAS,IAAI;AAEhD,SAAO,MAAM;AACX,aAAS,oBAAoB,SAAS,SAAS,IAAI;AAAA,EACrD;AACF;","names":["source"]}
+{"version":3,"sources":["../src/constants.ts","../src/runtime.ts"],"sourcesContent":["export const SOURCE_PROP = \"__componentSourceLoc\";\n\n","import { SOURCE_PROP } from \"./constants\";\n\nexport type TriggerKey = \"alt\" | \"meta\" | \"ctrl\" | \"shift\" | \"none\";\n\ntype ReactFiber = {\n  return?: ReactFiber | null;\n  type?: unknown;\n  elementType?: unknown;\n  pendingProps?: Record<string, unknown> | null;\n  memoizedProps?: Record<string, unknown> | null;\n  _debugSource?: {\n    fileName?: string;\n    lineNumber?: number;\n    columnNumber?: number;\n  } | null;\n};\n\nexport type LocatorResult = {\n  source: string;\n  mode: \"jsx\" | \"component\";\n};\n\nexport type LocatorOptions = {\n  triggerKey?: TriggerKey;\n  onLocate?: (result: LocatorResult) => void;\n  onError?: (error: unknown) => void;\n};\n\nfunction isTriggerPressed(event: MouseEvent, triggerKey: TriggerKey) {\n  if (triggerKey === \"none\") {\n    return true;\n  }\n\n  if (triggerKey === \"alt\") {\n    return event.altKey;\n  }\n\n  if (triggerKey === \"meta\") {\n    return event.metaKey;\n  }\n\n  if (triggerKey === \"ctrl\") {\n    return event.ctrlKey;\n  }\n\n  return event.shiftKey;\n}\n\nfunction getReactFiberKey(element: Element) {\n  return Object.keys(element).find((key) => key.startsWith(\"__reactFiber$\") || key.startsWith(\"__reactInternalInstance$\"));\n}\n\nfunction getClosestReactFiber(target: Element | null) {\n  let current = target;\n\n  while (current) {\n    const fiberKey = getReactFiberKey(current);\n    if (fiberKey) {\n      return (current as unknown as Record<string, unknown>)[fiberKey] as ReactFiber;\n    }\n\n    current = current.parentElement;\n  }\n\n  return null;\n}\n\nfunction getSourceFromType(type: unknown) {\n  if (!type) {\n    return null;\n  }\n\n  if (typeof type === \"function\") {\n    const source = (type as unknown as Record<string, unknown>)[SOURCE_PROP];\n    return typeof source === \"string\" ? source : null;\n  }\n\n  if (typeof type !== \"object\") {\n    return null;\n  }\n\n  const record = type as {\n    type?: Record<string, unknown>;\n    render?: Record<string, unknown>;\n    [SOURCE_PROP]?: unknown;\n  };\n\n  const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction getSourceFromProps(props: Record<string, unknown> | null | undefined) {\n  const source = props?.[SOURCE_PROP];\n  return typeof source === \"string\" ? source : null;\n}\n\nfunction resolveJsxSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromProps(current.pendingProps) ?? getSourceFromProps(current.memoizedProps);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction resolveComponentSourceFromFiber(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);\n    if (source) {\n      return source;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nfunction getDebugSource(fiber: ReactFiber | null) {\n  let current = fiber;\n\n  while (current) {\n    const debugSource = current._debugSource;\n    if (debugSource?.fileName && typeof debugSource.lineNumber === \"number\") {\n      return `${debugSource.fileName.replace(/\\\\/g, \"/\")}:${debugSource.lineNumber}:${debugSource.columnNumber ?? 1}`;\n    }\n\n    current = current.return ?? null;\n  }\n\n  return null;\n}\n\nexport function locateComponentSource(target: EventTarget | null): LocatorResult | null {\n  const elementTarget =\n    target instanceof Element ? target : target instanceof Node ? target.parentElement : null;\n  const fiber = getClosestReactFiber(elementTarget);\n  if (!fiber) {\n    return null;\n  }\n\n  const jsxSource = resolveJsxSourceFromFiber(fiber) ?? getDebugSource(fiber);\n  if (jsxSource) {\n    return {\n      source: jsxSource,\n      mode: \"jsx\",\n    };\n  }\n\n  const componentSource = resolveComponentSourceFromFiber(fiber);\n  if (!componentSource) {\n    return null;\n  }\n\n  return {\n    source: componentSource,\n    mode: \"component\",\n  };\n}\n\nexport function enableReactComponentJump(options: LocatorOptions = {}) {\n  const {\n    triggerKey = \"shift\",\n    onLocate = (result) => {\n      console.log(`[react-code-locator] ${result.source}`);\n    },\n    onError = (error) => {\n      console.error(\"[react-code-locator]\", error);\n    },\n  } = options;\n\n  const handler = (event: MouseEvent) => {\n    if (!isTriggerPressed(event, triggerKey)) {\n      return;\n    }\n\n    const result = locateComponentSource(event.target);\n    if (!result) {\n      onError(new Error(\"No React component source metadata found for clicked element.\"));\n      return;\n    }\n\n    event.preventDefault();\n    event.stopPropagation();\n    onLocate(result);\n  };\n\n  document.addEventListener(\"click\", handler, true);\n\n  return () => {\n    document.removeEventListener(\"click\", handler, true);\n  };\n}\n"],"mappings":";AAAO,IAAM,cAAc;;;AC4B3B,SAAS,iBAAiB,OAAmB,YAAwB;AACnE,MAAI,eAAe,QAAQ;AACzB,WAAO;AAAA,EACT;AAEA,MAAI,eAAe,OAAO;AACxB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,eAAe,QAAQ;AACzB,WAAO,MAAM;AAAA,EACf;AAEA,SAAO,MAAM;AACf;AAEA,SAAS,iBAAiB,SAAkB;AAC1C,SAAO,OAAO,KAAK,OAAO,EAAE,KAAK,CAAC,QAAQ,IAAI,WAAW,eAAe,KAAK,IAAI,WAAW,0BAA0B,CAAC;AACzH;AAEA,SAAS,qBAAqB,QAAwB;AACpD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,WAAW,iBAAiB,OAAO;AACzC,QAAI,UAAU;AACZ,aAAQ,QAA+C,QAAQ;AAAA,IACjE;AAEA,cAAU,QAAQ;AAAA,EACpB;AAEA,SAAO;AACT;AAEA,SAAS,kBAAkB,MAAe;AACxC,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,SAAS,YAAY;AAC9B,UAAMA,UAAU,KAA4C,WAAW;AACvE,WAAO,OAAOA,YAAW,WAAWA,UAAS;AAAA,EAC/C;AAEA,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAO;AAAA,EACT;AAEA,QAAM,SAAS;AAMf,QAAM,SAAS,OAAO,WAAW,KAAK,OAAO,OAAO,WAAW,KAAK,OAAO,SAAS,WAAW;AAC/F,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,mBAAmB,OAAmD;AAC7E,QAAM,SAAS,QAAQ,WAAW;AAClC,SAAO,OAAO,WAAW,WAAW,SAAS;AAC/C;AAEA,SAAS,0BAA0B,OAA0B;AAC3D,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,mBAAmB,QAAQ,YAAY,KAAK,mBAAmB,QAAQ,aAAa;AACnG,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,gCAAgC,OAA0B;AACjE,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,SAAS,kBAAkB,QAAQ,IAAI,KAAK,kBAAkB,QAAQ,WAAW;AACvF,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEA,SAAS,eAAe,OAA0B;AAChD,MAAI,UAAU;AAEd,SAAO,SAAS;AACd,UAAM,cAAc,QAAQ;AAC5B,QAAI,aAAa,YAAY,OAAO,YAAY,eAAe,UAAU;AACvE,aAAO,GAAG,YAAY,SAAS,QAAQ,OAAO,GAAG,CAAC,IAAI,YAAY,UAAU,IAAI,YAAY,gBAAgB,CAAC;AAAA,IAC/G;AAEA,cAAU,QAAQ,UAAU;AAAA,EAC9B;AAEA,SAAO;AACT;AAEO,SAAS,sBAAsB,QAAkD;AACtF,QAAM,gBACJ,kBAAkB,UAAU,SAAS,kBAAkB,OAAO,OAAO,gBAAgB;AACvF,QAAM,QAAQ,qBAAqB,aAAa;AAChD,MAAI,CAAC,OAAO;AACV,WAAO;AAAA,EACT;AAEA,QAAM,YAAY,0BAA0B,KAAK,KAAK,eAAe,KAAK;AAC1E,MAAI,WAAW;AACb,WAAO;AAAA,MACL,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AAAA,EACF;AAEA,QAAM,kBAAkB,gCAAgC,KAAK;AAC7D,MAAI,CAAC,iBAAiB;AACpB,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,MAAM;AAAA,EACR;AACF;AAEO,SAAS,yBAAyB,UAA0B,CAAC,GAAG;AACrE,QAAM;AAAA,IACJ,aAAa;AAAA,IACb,WAAW,CAAC,WAAW;AACrB,cAAQ,IAAI,wBAAwB,OAAO,MAAM,EAAE;AAAA,IACrD;AAAA,IACA,UAAU,CAAC,UAAU;AACnB,cAAQ,MAAM,wBAAwB,KAAK;AAAA,IAC7C;AAAA,EACF,IAAI;AAEJ,QAAM,UAAU,CAAC,UAAsB;AACrC,QAAI,CAAC,iBAAiB,OAAO,UAAU,GAAG;AACxC;AAAA,IACF;AAEA,UAAM,SAAS,sBAAsB,MAAM,MAAM;AACjD,QAAI,CAAC,QAAQ;AACX,cAAQ,IAAI,MAAM,+DAA+D,CAAC;AAClF;AAAA,IACF;AAEA,UAAM,eAAe;AACrB,UAAM,gBAAgB;AACtB,aAAS,MAAM;AAAA,EACjB;AAEA,WAAS,iBAAiB,SAAS,SAAS,IAAI;AAEhD,SAAO,MAAM;AACX,aAAS,oBAAoB,SAAS,SAAS,IAAI;AAAA,EACrD;AACF;","names":["source"]}

package/dist/vite.cjs

@@ -104,13 +104,17 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
     insertAfterPath.insertAfter(assignments);
   }
 }
-function babelInjectComponentSource() {
+function babelInjectComponentSource(options = {}) {
+  const { injectJsxSource = true, injectComponentSource = true } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {
       JSXOpeningElement(pathNode, state) {
+        if (!injectJsxSource) {
+          return;
+        }
         const hasSourceProp = pathNode.node.attributes.some(
-          (attr) => import_core.types.isJSXAttribute(attr) && import_core.types.isJSXIdentifier(attr.name) && attr.name.name === "__source"
+          (attr) => import_core.types.isJSXAttribute(attr) && import_core.types.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP
         );
         if (hasSourceProp) {
           return;
@@ -122,18 +126,15 @@ function babelInjectComponentSource() {
         }
         pathNode.node.attributes.push(
           import_core.types.jsxAttribute(
-            import_core.types.jsxIdentifier("__source"),
-            import_core.types.jsxExpressionContainer(
-              import_core.types.objectExpression([
-                import_core.types.objectProperty(import_core.types.identifier("fileName"), import_core.types.stringLiteral(filename)),
-                import_core.types.objectProperty(import_core.types.identifier("lineNumber"), import_core.types.numericLiteral(loc.line)),
-                import_core.types.objectProperty(import_core.types.identifier("columnNumber"), import_core.types.numericLiteral(loc.column + 1))
-              ])
-            )
+            import_core.types.jsxIdentifier(SOURCE_PROP),
+            import_core.types.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\/g, "/")}:${loc.line}:${loc.column + 1}`)
           )
         );
       },
       Program(programPath, state) {
+        if (!injectComponentSource) {
+          return;
+        }
         const seen = /* @__PURE__ */ new Set();
         for (const childPath of programPath.get("body")) {
           if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {
@@ -153,13 +154,14 @@ function babelInjectComponentSource() {
 // 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 ?? [], babelInjectComponentSource];
+        const plugins = [...resolved2.plugins ?? [], locatorBabelPlugin];
         return {
           ...resolved2,
           plugins
@@ -172,7 +174,7 @@ function withLocatorBabel(reactOptions = {}) {
     ...reactOptions,
     babel: {
       ...resolved,
-      plugins: [...resolved.plugins ?? [], babelInjectComponentSource]
+      plugins: [...resolved.plugins ?? [], locatorBabelPlugin]
     }
   };
 }
@@ -188,7 +190,7 @@ function createClientInjector(locatorOptions = {}) {
           attrs: {
             type: "module"
           },
-          children: `import { enableReactComponentJump } from "react-component-jump/client"; enableReactComponentJump(${serialized});`,
+          children: `import { enableReactComponentJump } from "react-code-locator/client"; enableReactComponentJump(${serialized});`,
           injectTo: "head"
         }
       ];

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\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 ?? []), babelInjectComponentSource as PluginItem];\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 ?? []), babelInjectComponentSource as PluginItem],\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-component-jump/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\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(): PluginObj<BabelState> {\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === \"__source\",\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\"),\n            t.jsxExpressionContainer(\n              t.objectExpression([\n                t.objectProperty(t.identifier(\"fileName\"), t.stringLiteral(filename)),\n                t.objectProperty(t.identifier(\"lineNumber\"), t.numericLiteral(loc.line)),\n                t.objectProperty(t.identifier(\"columnNumber\"), t.numericLiteral(loc.column + 1)),\n              ]),\n            ),\n          ),\n        );\n      },\n      Program(programPath, state) {\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;;;ADY3B,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,6BAAoD;AAClE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,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,UAAU;AAAA,YAC1B,YAAAA,MAAE;AAAA,cACA,YAAAA,MAAE,iBAAiB;AAAA,gBACjB,YAAAA,MAAE,eAAe,YAAAA,MAAE,WAAW,UAAU,GAAG,YAAAA,MAAE,cAAc,QAAQ,CAAC;AAAA,gBACpE,YAAAA,MAAE,eAAe,YAAAA,MAAE,WAAW,YAAY,GAAG,YAAAA,MAAE,eAAe,IAAI,IAAI,CAAC;AAAA,gBACvE,YAAAA,MAAE,eAAe,YAAAA,MAAE,WAAW,cAAc,GAAG,YAAAA,MAAE,eAAe,IAAI,SAAS,CAAC,CAAC;AAAA,cACjF,CAAC;AAAA,YACH;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,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;;;AD5HA,SAAS,iBAAiB,eAA6B,CAAC,GAAiB;AACvE,QAAM,YAAY,aAAa;AAE/B,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,0BAAwC;AAEtF,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,0BAAwC;AAAA,IACjF;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,oGAAoG,UAAU;AAAA,UACxH,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\";\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"]}

package/dist/vite.js

@@ -67,13 +67,17 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
     insertAfterPath.insertAfter(assignments);
   }
 }
-function babelInjectComponentSource() {
+function babelInjectComponentSource(options = {}) {
+  const { injectJsxSource = true, injectComponentSource = true } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {
       JSXOpeningElement(pathNode, state) {
+        if (!injectJsxSource) {
+          return;
+        }
         const hasSourceProp = pathNode.node.attributes.some(
-          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === "__source"
+          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP
         );
         if (hasSourceProp) {
           return;
@@ -85,18 +89,15 @@ function babelInjectComponentSource() {
         }
         pathNode.node.attributes.push(
           t.jsxAttribute(
-            t.jsxIdentifier("__source"),
-            t.jsxExpressionContainer(
-              t.objectExpression([
-                t.objectProperty(t.identifier("fileName"), t.stringLiteral(filename)),
-                t.objectProperty(t.identifier("lineNumber"), t.numericLiteral(loc.line)),
-                t.objectProperty(t.identifier("columnNumber"), t.numericLiteral(loc.column + 1))
-              ])
-            )
+            t.jsxIdentifier(SOURCE_PROP),
+            t.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\/g, "/")}:${loc.line}:${loc.column + 1}`)
           )
         );
       },
       Program(programPath, state) {
+        if (!injectComponentSource) {
+          return;
+        }
         const seen = /* @__PURE__ */ new Set();
         for (const childPath of programPath.get("body")) {
           if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {
@@ -116,13 +117,14 @@ function babelInjectComponentSource() {
 // 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 ?? [], babelInjectComponentSource];
+        const plugins = [...resolved2.plugins ?? [], locatorBabelPlugin];
         return {
           ...resolved2,
           plugins
@@ -135,7 +137,7 @@ function withLocatorBabel(reactOptions = {}) {
     ...reactOptions,
     babel: {
       ...resolved,
-      plugins: [...resolved.plugins ?? [], babelInjectComponentSource]
+      plugins: [...resolved.plugins ?? [], locatorBabelPlugin]
     }
   };
 }
@@ -151,7 +153,7 @@ function createClientInjector(locatorOptions = {}) {
           attrs: {
             type: "module"
           },
-          children: `import { enableReactComponentJump } from "react-component-jump/client"; enableReactComponentJump(${serialized});`,
+          children: `import { enableReactComponentJump } from "react-code-locator/client"; enableReactComponentJump(${serialized});`,
           injectTo: "head"
         }
       ];

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\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 ?? []), babelInjectComponentSource as PluginItem];\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 ?? []), babelInjectComponentSource as PluginItem],\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-component-jump/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\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(): PluginObj<BabelState> {\n  return {\n    name: \"babel-inject-component-source\",\n    visitor: {\n      JSXOpeningElement(pathNode, state) {\n        const hasSourceProp = pathNode.node.attributes.some(\n          (attr) => t.isJSXAttribute(attr) && t.isJSXIdentifier(attr.name) && attr.name.name === \"__source\",\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\"),\n            t.jsxExpressionContainer(\n              t.objectExpression([\n                t.objectProperty(t.identifier(\"fileName\"), t.stringLiteral(filename)),\n                t.objectProperty(t.identifier(\"lineNumber\"), t.numericLiteral(loc.line)),\n                t.objectProperty(t.identifier(\"columnNumber\"), t.numericLiteral(loc.column + 1)),\n              ]),\n            ),\n          ),\n        );\n      },\n      Program(programPath, state) {\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;;;ADY3B,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,6BAAoD;AAClE,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS;AAAA,MACP,kBAAkB,UAAU,OAAO;AACjC,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,UAAU;AAAA,YAC1B,EAAE;AAAA,cACA,EAAE,iBAAiB;AAAA,gBACjB,EAAE,eAAe,EAAE,WAAW,UAAU,GAAG,EAAE,cAAc,QAAQ,CAAC;AAAA,gBACpE,EAAE,eAAe,EAAE,WAAW,YAAY,GAAG,EAAE,eAAe,IAAI,IAAI,CAAC;AAAA,gBACvE,EAAE,eAAe,EAAE,WAAW,cAAc,GAAG,EAAE,eAAe,IAAI,SAAS,CAAC,CAAC;AAAA,cACjF,CAAC;AAAA,YACH;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,MACA,QAAQ,aAAa,OAAO;AAC1B,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;;;AD5HA,SAAS,iBAAiB,eAA6B,CAAC,GAAiB;AACvE,QAAM,YAAY,aAAa;AAE/B,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,0BAAwC;AAEtF,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,0BAAwC;AAAA,IACjF;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,oGAAoG,UAAU;AAAA,UACxH,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 { 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"]}

package/dist/webpackRuntimeEntry.cjs

@@ -48,7 +48,22 @@ function getSourceFromType(type) {
   const source = record[SOURCE_PROP] ?? record.type?.[SOURCE_PROP] ?? record.render?.[SOURCE_PROP];
   return typeof source === "string" ? source : null;
 }
-function resolveSourceFromFiber(fiber) {
+function getSourceFromProps(props) {
+  const source = props?.[SOURCE_PROP];
+  return typeof source === "string" ? source : null;
+}
+function resolveJsxSourceFromFiber(fiber) {
+  let current = fiber;
+  while (current) {
+    const source = getSourceFromProps(current.pendingProps) ?? getSourceFromProps(current.memoizedProps);
+    if (source) {
+      return source;
+    }
+    current = current.return ?? null;
+  }
+  return null;
+}
+function resolveComponentSourceFromFiber(fiber) {
   let current = fiber;
   while (current) {
     const source = getSourceFromType(current.type) ?? getSourceFromType(current.elementType);
@@ -76,14 +91,14 @@ function locateComponentSource(target) {
   if (!fiber) {
     return null;
   }
-  const debugSource = getDebugSource(fiber);
-  if (debugSource) {
+  const jsxSource = resolveJsxSourceFromFiber(fiber) ?? getDebugSource(fiber);
+  if (jsxSource) {
     return {
-      source: debugSource,
+      source: jsxSource,
       mode: "jsx"
     };
   }
-  const componentSource = resolveSourceFromFiber(fiber);
+  const componentSource = resolveComponentSourceFromFiber(fiber);
   if (!componentSource) {
     return null;
   }
@@ -96,10 +111,10 @@ function enableReactComponentJump(options = {}) {
   const {
     triggerKey = "shift",
     onLocate = (result) => {
-      console.log(`[react-component-jump] ${result.source}`);
+      console.log(`[react-code-locator] ${result.source}`);
     },
     onError = (error) => {
-      console.error("[react-component-jump]", error);
+      console.error("[react-code-locator]", error);
     }
   } = options;
   const handler = (event) => {