react-code-locator 0.1.9 → 0.1.12

package/dist/vite.cjs

@@ -1,9 +1,7 @@
 "use strict";
-var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
-var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -17,80 +15,146 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
-  // If the importer is in node compatibility mode or this is not an ESM
-  // file that has been converted to a CommonJS file using a Babel-
-  // compatible transform (i.e. "__esModule" has not been set), then set
-  // "default" to the CommonJS "module.exports" for node compatibility.
-  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
-  mod
-));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/vite.ts
 var vite_exports = {};
 __export(vite_exports, {
   babelInjectComponentSource: () => babelInjectComponentSource,
-  elementLocatorReact: () => elementLocatorReact,
-  reactComponentJump: () => elementLocatorReact
+  createViteClientInjector: () => createViteClientInjector,
+  createViteSourceAdapter: () => createViteSourceAdapter,
+  reactComponentJump: () => createViteClientInjector,
+  viteSourceAdapter: () => viteSourceAdapter,
+  viteSourceTransformPlugin: () => viteSourceTransformPlugin
 });
 module.exports = __toCommonJS(vite_exports);
 
-// src/elementLocatorReact.ts
-var VIRTUAL_CLIENT_MODULE_ID = "virtual:react-code-locator/client";
-var RESOLVED_VIRTUAL_CLIENT_MODULE_ID = `\0${VIRTUAL_CLIENT_MODULE_ID}`;
-function createClientInjector(locatorOptions = {}) {
-  const serialized = JSON.stringify(locatorOptions);
-  return {
-    name: "element-locator-client-injector",
-    apply: "serve",
-    resolveId(id) {
-      if (id === VIRTUAL_CLIENT_MODULE_ID) {
-        return RESOLVED_VIRTUAL_CLIENT_MODULE_ID;
-      }
-      return null;
-    },
-    load(id) {
-      if (id !== RESOLVED_VIRTUAL_CLIENT_MODULE_ID) {
-        return null;
-      }
-      return `
-        import { enableReactComponentJump } from "react-code-locator/client";
-
-        enableReactComponentJump(${serialized});
-      `;
-    },
-    transformIndexHtml() {
-      return [
-        {
-          tag: "script",
-          attrs: {
-            type: "module",
-            src: `/@id/__x00__${VIRTUAL_CLIENT_MODULE_ID}`
-          },
-          injectTo: "head"
-        }
-      ];
-    }
-  };
-}
-function elementLocatorReact(options = {}) {
-  const { command = "serve", locator = {}, injectClient = true } = options;
-  const isServe = command === "serve";
-  return [isServe && injectClient ? createClientInjector(locator) : null].filter(Boolean);
+// src/sourceAdapter.ts
+function defineSourceAdapter(descriptor) {
+  return descriptor;
 }
 
+// src/sourceTransform.ts
+var import_core2 = require("@babel/core");
+
 // src/babelInjectComponentSource.ts
-var import_node_path = __toESM(require("path"), 1);
 var import_core = require("@babel/core");
 
 // src/constants.ts
 var SOURCE_PROP = "__componentSourceLoc";
+var JSX_SOURCE_PROP = "$componentSourceLoc";
+var JSX_SOURCE_REGISTRY_SYMBOL = "react-code-locator.jsxSourceRegistry";
+
+// src/sourceMetadata.ts
+function normalizeSlashes(value) {
+  return value.replace(/\\/g, "/");
+}
+function trimTrailingSlash(value) {
+  return value.replace(/\/+$/, "");
+}
+function splitPathSegments(value) {
+  return normalizeSlashes(value).split("/").filter(Boolean);
+}
+function computeRelativePath(fromPath, toPath) {
+  const fromSegments = splitPathSegments(fromPath);
+  const toSegments = splitPathSegments(toPath);
+  let sharedIndex = 0;
+  while (sharedIndex < fromSegments.length && sharedIndex < toSegments.length && fromSegments[sharedIndex] === toSegments[sharedIndex]) {
+    sharedIndex += 1;
+  }
+  const upSegments = new Array(Math.max(0, fromSegments.length - sharedIndex)).fill("..");
+  const downSegments = toSegments.slice(sharedIndex);
+  const relativeSegments = [...upSegments, ...downSegments];
+  return relativeSegments.length > 0 ? relativeSegments.join("/") : ".";
+}
+function normalizeProjectRoot(projectRoot) {
+  if (projectRoot) {
+    return trimTrailingSlash(normalizeSlashes(projectRoot));
+  }
+  if (typeof process !== "undefined" && typeof process.cwd === "function") {
+    return trimTrailingSlash(normalizeSlashes(process.cwd()));
+  }
+  return "";
+}
+function toRelativeSource(filename, loc, projectRoot) {
+  if (!filename || !loc) {
+    return null;
+  }
+  const root = normalizeProjectRoot(projectRoot);
+  const normalizedFilename = normalizeSlashes(filename);
+  const relPath = root && normalizedFilename.startsWith(`${root}/`) ? normalizedFilename.slice(root.length + 1) : root ? computeRelativePath(root, normalizedFilename) : normalizedFilename;
+  return `${relPath}:${loc.line}:${loc.column + 1}`;
+}
+function isProjectLocalFile(filename, projectRoot) {
+  if (!filename) {
+    return false;
+  }
+  const root = normalizeProjectRoot(projectRoot);
+  const normalizedFilename = normalizeSlashes(filename);
+  if (!root) {
+    return !normalizedFilename.startsWith("../") && !normalizedFilename.startsWith("/") && !/^[A-Za-z]:\//.test(normalizedFilename);
+  }
+  if (normalizedFilename.startsWith(`${root}/`) || normalizedFilename === root) {
+    return true;
+  }
+  const relativePath = computeRelativePath(root, normalizedFilename);
+  return !relativePath.startsWith("../");
+}
+function isExternalToProjectRoot(filename, projectRoot) {
+  return !isProjectLocalFile(filename, projectRoot);
+}
 
 // src/babelInjectComponentSource.ts
+var SOURCE_PROP_LOCAL = "_componentSourceLoc";
+var SOURCE_PROPS_REST = "__reactCodeLocatorProps";
 function isComponentName(name) {
   return /^[A-Z]/.test(name);
 }
+function isCustomComponentTag(name) {
+  if (import_core.types.isJSXIdentifier(name)) {
+    return isComponentName(name.name);
+  }
+  if (import_core.types.isJSXMemberExpression(name)) {
+    return true;
+  }
+  return false;
+}
+function isIntrinsicElementTag(name) {
+  return import_core.types.isJSXIdentifier(name) && /^[a-z]/.test(name.name);
+}
+function isElementFactoryIdentifier(name) {
+  return name === "jsx" || name === "jsxs" || name === "jsxDEV" || name === "_jsx" || name === "_jsxs" || name === "_jsxDEV" || name === "createElement";
+}
+function isReactElementFactoryCall(pathNode) {
+  const callee = pathNode.node.callee;
+  if (import_core.types.isIdentifier(callee)) {
+    return isElementFactoryIdentifier(callee.name);
+  }
+  return import_core.types.isMemberExpression(callee) && import_core.types.isIdentifier(callee.object, { name: "React" }) && import_core.types.isIdentifier(callee.property, { name: "createElement" });
+}
+function getRootJsxIdentifierName(name) {
+  if (import_core.types.isJSXIdentifier(name)) {
+    return name.name;
+  }
+  if (import_core.types.isJSXMemberExpression(name)) {
+    return getRootJsxIdentifierName(name.object);
+  }
+  return null;
+}
+function isStyledModuleImport(binding) {
+  if (!binding) {
+    return false;
+  }
+  if (!binding.path.isImportSpecifier() && !binding.path.isImportDefaultSpecifier() && !binding.path.isImportNamespaceSpecifier()) {
+    return false;
+  }
+  const source = binding.path.parentPath.isImportDeclaration() ? binding.path.parentPath.node.source.value : null;
+  if (typeof source !== "string") {
+    return false;
+  }
+  const normalized = source.replace(/\\/g, "/");
+  return normalized === "./styled" || normalized === "../styled" || normalized.endsWith("/styled");
+}
 function isSupportedComponentInit(node) {
   if (!node) {
     return false;
@@ -106,13 +170,104 @@ function isSupportedComponentInit(node) {
   }
   return import_core.types.isMemberExpression(node.callee) && import_core.types.isIdentifier(node.callee.object, { name: "React" }) && import_core.types.isIdentifier(node.callee.property) && (node.callee.property.name === "memo" || node.callee.property.name === "forwardRef");
 }
-function getSourceValue(state, loc) {
+function hasSourcePropBinding(pattern) {
+  return pattern.properties.some((property) => {
+    if (!import_core.types.isObjectProperty(property)) {
+      return false;
+    }
+    return import_core.types.isIdentifier(property.key) && property.key.name === JSX_SOURCE_PROP;
+  });
+}
+function injectSourcePropBinding(pattern) {
+  if (hasSourcePropBinding(pattern)) {
+    return;
+  }
+  const sourceBinding = import_core.types.objectProperty(
+    import_core.types.identifier(JSX_SOURCE_PROP),
+    import_core.types.identifier(SOURCE_PROP_LOCAL),
+    false,
+    false
+  );
+  const restIndex = pattern.properties.findIndex(
+    (property) => import_core.types.isRestElement(property)
+  );
+  if (restIndex === -1) {
+    pattern.properties.push(sourceBinding);
+    return;
+  }
+  pattern.properties.splice(restIndex, 0, sourceBinding);
+}
+function injectSourcePropIntoIdentifierParam(node, param) {
+  if (!import_core.types.isBlockStatement(node.body)) {
+    node.body = import_core.types.blockStatement([import_core.types.returnStatement(node.body)]);
+  }
+  const alreadyInjected = node.body.body.some(
+    (statement) => import_core.types.isVariableDeclaration(statement) && statement.declarations.some(
+      (declaration) => import_core.types.isIdentifier(declaration.id) && declaration.id.name === SOURCE_PROPS_REST
+    )
+  );
+  if (alreadyInjected) {
+    return;
+  }
+  node.body.body.unshift(
+    import_core.types.variableDeclaration("const", [
+      import_core.types.variableDeclarator(
+        import_core.types.objectPattern([
+          import_core.types.objectProperty(
+            import_core.types.identifier(JSX_SOURCE_PROP),
+            import_core.types.identifier(SOURCE_PROP_LOCAL),
+            false,
+            false
+          ),
+          import_core.types.restElement(import_core.types.identifier(SOURCE_PROPS_REST))
+        ]),
+        param
+      )
+    ]),
+    import_core.types.expressionStatement(
+      import_core.types.assignmentExpression(
+        "=",
+        import_core.types.identifier(param.name),
+        import_core.types.identifier(SOURCE_PROPS_REST)
+      )
+    )
+  );
+}
+function injectSourcePropIntoFunctionParams(node) {
+  const firstParam = node.params[0];
+  if (!firstParam) {
+    return;
+  }
+  if (import_core.types.isObjectPattern(firstParam)) {
+    injectSourcePropBinding(firstParam);
+    return;
+  }
+  if (import_core.types.isIdentifier(firstParam)) {
+    injectSourcePropIntoIdentifierParam(node, firstParam);
+  }
+}
+function injectSourcePropIntoExpression(node) {
+  if (!node) {
+    return;
+  }
+  if (import_core.types.isFunctionExpression(node) || import_core.types.isArrowFunctionExpression(node)) {
+    injectSourcePropIntoFunctionParams(node);
+    return;
+  }
+  if (!import_core.types.isCallExpression(node)) {
+    return;
+  }
+  const firstArg = node.arguments[0];
+  if (firstArg && !import_core.types.isSpreadElement(firstArg) && (import_core.types.isFunctionExpression(firstArg) || import_core.types.isArrowFunctionExpression(firstArg))) {
+    injectSourcePropIntoFunctionParams(firstArg);
+  }
+}
+function getSourceValue(state, loc, projectRoot) {
   const filename = state.file?.opts?.filename;
   if (!filename || !loc) {
     return null;
   }
-  const relPath = import_node_path.default.relative(process.cwd(), filename).replace(/\\/g, "/");
-  return `${relPath}:${loc.line}:${loc.column + 1}`;
+  return toRelativeSource(filename, loc, projectRoot);
 }
 function buildAssignment(name, sourceValue) {
   return import_core.types.expressionStatement(
@@ -123,13 +278,98 @@ function buildAssignment(name, sourceValue) {
     )
   );
 }
-function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
+function buildIntrinsicSourceHelper() {
+  return import_core.types.functionDeclaration(
+    import_core.types.identifier("_markIntrinsicElementSource"),
+    [import_core.types.identifier("element"), import_core.types.identifier("source")],
+    import_core.types.blockStatement([
+      import_core.types.variableDeclaration("const", [
+        import_core.types.variableDeclarator(
+          import_core.types.identifier("registryKey"),
+          import_core.types.callExpression(
+            import_core.types.memberExpression(import_core.types.identifier("Symbol"), import_core.types.identifier("for")),
+            [import_core.types.stringLiteral(JSX_SOURCE_REGISTRY_SYMBOL)]
+          )
+        )
+      ]),
+      import_core.types.variableDeclaration("let", [
+        import_core.types.variableDeclarator(
+          import_core.types.identifier("registry"),
+          import_core.types.memberExpression(import_core.types.identifier("globalThis"), import_core.types.identifier("registryKey"), true)
+        )
+      ]),
+      import_core.types.ifStatement(
+        import_core.types.unaryExpression(
+          "!",
+          import_core.types.binaryExpression("instanceof", import_core.types.identifier("registry"), import_core.types.identifier("WeakMap"))
+        ),
+        import_core.types.blockStatement([
+          import_core.types.expressionStatement(
+            import_core.types.assignmentExpression(
+              "=",
+              import_core.types.identifier("registry"),
+              import_core.types.assignmentExpression(
+                "=",
+                import_core.types.memberExpression(import_core.types.identifier("globalThis"), import_core.types.identifier("registryKey"), true),
+                import_core.types.newExpression(import_core.types.identifier("WeakMap"), [])
+              )
+            )
+          )
+        ])
+      ),
+      import_core.types.ifStatement(
+        import_core.types.logicalExpression(
+          "&&",
+          import_core.types.identifier("element"),
+          import_core.types.logicalExpression(
+            "&&",
+            import_core.types.binaryExpression("===", import_core.types.unaryExpression("typeof", import_core.types.identifier("element")), import_core.types.stringLiteral("object")),
+            import_core.types.binaryExpression(
+              "===",
+              import_core.types.unaryExpression("typeof", import_core.types.memberExpression(import_core.types.identifier("element"), import_core.types.identifier("props"))),
+              import_core.types.stringLiteral("object")
+            )
+          )
+        ),
+        import_core.types.blockStatement([
+          import_core.types.expressionStatement(
+            import_core.types.callExpression(
+              import_core.types.memberExpression(import_core.types.identifier("registry"), import_core.types.identifier("set")),
+              [import_core.types.memberExpression(import_core.types.identifier("element"), import_core.types.identifier("props")), import_core.types.identifier("source")]
+            )
+          )
+        ])
+      ),
+      import_core.types.returnStatement(import_core.types.identifier("element"))
+    ])
+  );
+}
+function ensureIntrinsicSourceHelper(programPath, state) {
+  if (state.injectedIntrinsicHelper) {
+    return;
+  }
+  const alreadyExists = programPath.node.body.some(
+    (node) => import_core.types.isFunctionDeclaration(node) && import_core.types.isIdentifier(node.id, { name: "_markIntrinsicElementSource" })
+  );
+  if (!alreadyExists) {
+    programPath.unshiftContainer("body", buildIntrinsicSourceHelper());
+  }
+  state.injectedIntrinsicHelper = true;
+}
+function visitDeclaration(declarationPath, insertAfterPath, state, seen, projectRoot) {
   if (declarationPath.isFunctionDeclaration() || declarationPath.isClassDeclaration()) {
     const name = declarationPath.node.id?.name;
     if (!name || !isComponentName(name) || seen.has(name)) {
       return;
     }
-    const sourceValue = getSourceValue(state, declarationPath.node.loc?.start);
+    if (declarationPath.isFunctionDeclaration()) {
+      injectSourcePropIntoFunctionParams(declarationPath.node);
+    }
+    const sourceValue = getSourceValue(
+      state,
+      declarationPath.node.loc?.start,
+      projectRoot
+    );
     if (!sourceValue) {
       return;
     }
@@ -140,38 +380,128 @@ function visitDeclaration(declarationPath, insertAfterPath, state, seen) {
   if (!declarationPath.isVariableDeclaration()) {
     return;
   }
-  const assignments = declarationPath.node.declarations.flatMap((declarator) => {
-    if (!import_core.types.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {
-      return [];
-    }
-    if (!declarator.init) {
-      return [];
-    }
-    if (!isSupportedComponentInit(declarator.init)) {
-      return [];
-    }
-    const sourceValue = getSourceValue(state, declarator.loc?.start ?? declarator.init.loc?.start);
-    if (!sourceValue) {
-      return [];
+  const assignments = declarationPath.node.declarations.flatMap(
+    (declarator) => {
+      if (!import_core.types.isIdentifier(declarator.id) || !isComponentName(declarator.id.name) || seen.has(declarator.id.name)) {
+        return [];
+      }
+      if (!declarator.init) {
+        return [];
+      }
+      if (!isSupportedComponentInit(declarator.init)) {
+        return [];
+      }
+      injectSourcePropIntoExpression(declarator.init);
+      const sourceValue = getSourceValue(
+        state,
+        declarator.loc?.start ?? declarator.init.loc?.start,
+        projectRoot
+      );
+      if (!sourceValue) {
+        return [];
+      }
+      seen.add(declarator.id.name);
+      return [buildAssignment(declarator.id.name, sourceValue)];
     }
-    seen.add(declarator.id.name);
-    return [buildAssignment(declarator.id.name, sourceValue)];
-  });
+  );
   if (assignments.length > 0) {
     insertAfterPath.insertAfter(assignments);
   }
 }
 function babelInjectComponentSource(options = {}) {
-  const { injectJsxSource = false, injectComponentSource = true } = options;
+  const {
+    injectJsxSource = true,
+    injectComponentSource = true,
+    projectRoot
+  } = options;
   return {
     name: "babel-inject-component-source",
     visitor: {
+      CallExpression(pathNode, state) {
+        if (!injectJsxSource) {
+          return;
+        }
+        if (!isReactElementFactoryCall(pathNode)) {
+          return;
+        }
+        if (pathNode.parentPath.isCallExpression() && import_core.types.isIdentifier(pathNode.parentPath.node.callee, {
+          name: "_markIntrinsicElementSource"
+        })) {
+          return;
+        }
+        const sourceValue = getSourceValue(
+          state,
+          pathNode.node.loc?.start,
+          projectRoot
+        );
+        if (!sourceValue) {
+          return;
+        }
+        const programPath = pathNode.findParent((parent) => parent.isProgram());
+        if (!programPath || !programPath.isProgram()) {
+          return;
+        }
+        ensureIntrinsicSourceHelper(programPath, state);
+        pathNode.replaceWith(
+          import_core.types.callExpression(import_core.types.identifier("_markIntrinsicElementSource"), [
+            pathNode.node,
+            import_core.types.stringLiteral(sourceValue)
+          ])
+        );
+        pathNode.skip();
+      },
+      JSXElement: {
+        exit(pathNode, state) {
+          if (!injectJsxSource) {
+            return;
+          }
+          if (!isIntrinsicElementTag(pathNode.node.openingElement.name)) {
+            return;
+          }
+          if (pathNode.parentPath.isCallExpression() && import_core.types.isIdentifier(pathNode.parentPath.node.callee, { name: "_markIntrinsicElementSource" })) {
+            return;
+          }
+          const sourceValue = getSourceValue(
+            state,
+            pathNode.node.openingElement.loc?.start,
+            projectRoot
+          );
+          if (!sourceValue) {
+            return;
+          }
+          const programPath = pathNode.findParent((parent) => parent.isProgram());
+          if (!programPath || !programPath.isProgram()) {
+            return;
+          }
+          ensureIntrinsicSourceHelper(programPath, state);
+          const wrappedNode = import_core.types.callExpression(import_core.types.identifier("_markIntrinsicElementSource"), [
+            pathNode.node,
+            import_core.types.stringLiteral(sourceValue)
+          ]);
+          if (pathNode.parentPath.isJSXElement() || pathNode.parentPath.isJSXFragment()) {
+            pathNode.replaceWith(import_core.types.jsxExpressionContainer(wrappedNode));
+            return;
+          }
+          if (pathNode.parentPath.isJSXExpressionContainer()) {
+            pathNode.parentPath.replaceWith(import_core.types.jsxExpressionContainer(wrappedNode));
+            return;
+          }
+          pathNode.replaceWith(wrappedNode);
+        }
+      },
       JSXOpeningElement(pathNode, state) {
         if (!injectJsxSource) {
           return;
         }
+        if (!isCustomComponentTag(pathNode.node.name)) {
+          return;
+        }
+        const rootIdentifierName = getRootJsxIdentifierName(pathNode.node.name);
+        if (rootIdentifierName && isExternalToProjectRoot(state.file?.opts?.filename, projectRoot) && isStyledModuleImport(pathNode.scope.getBinding(rootIdentifierName))) {
+          return;
+        }
         const hasSourceProp = pathNode.node.attributes.some(
-          (attr) => import_core.types.isJSXAttribute(attr) && import_core.types.isJSXIdentifier(attr.name) && attr.name.name === SOURCE_PROP
+          (attr) => import_core.types.isJSXAttribute(attr) && import_core.types.isJSXIdentifier(attr.name) && attr.name.name === JSX_SOURCE_PROP
         );
         if (hasSourceProp) {
           return;
@@ -183,8 +513,10 @@ function babelInjectComponentSource(options = {}) {
         }
         pathNode.node.attributes.push(
           import_core.types.jsxAttribute(
-            import_core.types.jsxIdentifier(SOURCE_PROP),
-            import_core.types.stringLiteral(getSourceValue(state, loc) ?? `${filename.replace(/\\/g, "/")}:${loc.line}:${loc.column + 1}`)
+            import_core.types.jsxIdentifier(JSX_SOURCE_PROP),
+            import_core.types.stringLiteral(
+              getSourceValue(state, loc, projectRoot) ?? `${filename.replace(/\\/g, "/")}:${loc.line}:${loc.column + 1}`
+            )
           )
         );
       },
@@ -197,20 +529,129 @@ function babelInjectComponentSource(options = {}) {
           if (childPath.isExportNamedDeclaration() || childPath.isExportDefaultDeclaration()) {
             const declarationPath = childPath.get("declaration");
             if (!Array.isArray(declarationPath) && declarationPath.node) {
-              visitDeclaration(declarationPath, childPath, state, seen);
+              visitDeclaration(declarationPath, childPath, state, seen, projectRoot);
             }
             continue;
           }
-          visitDeclaration(childPath, childPath, state, seen);
+          visitDeclaration(childPath, childPath, state, seen, projectRoot);
+        }
+      }
+    }
+  };
+}
+
+// src/sourceTransform.ts
+async function transformSourceWithLocator(code, options) {
+  const { filename, sourceMaps = true, ...pluginOptions } = options;
+  const result = await (0, import_core2.transformAsync)(code, {
+    filename,
+    babelrc: false,
+    configFile: false,
+    sourceMaps,
+    parserOpts: {
+      sourceType: "module",
+      plugins: ["jsx", "typescript"]
+    },
+    generatorOpts: {
+      retainLines: true
+    },
+    plugins: [[babelInjectComponentSource, pluginOptions]]
+  });
+  return {
+    code: result?.code ?? code,
+    map: result?.map ?? null
+  };
+}
+
+// src/viteClientInjector.ts
+var VIRTUAL_CLIENT_MODULE_ID = "virtual:react-code-locator/client";
+var RESOLVED_VIRTUAL_CLIENT_MODULE_ID = `\0${VIRTUAL_CLIENT_MODULE_ID}`;
+function createClientInjector(locatorOptions = {}) {
+  const serialized = JSON.stringify(locatorOptions);
+  return {
+    name: "react-code-locator-client-injector",
+    apply: "serve",
+    resolveId(id) {
+      if (id === VIRTUAL_CLIENT_MODULE_ID) {
+        return RESOLVED_VIRTUAL_CLIENT_MODULE_ID;
+      }
+      return null;
+    },
+    load(id) {
+      if (id !== RESOLVED_VIRTUAL_CLIENT_MODULE_ID) {
+        return null;
+      }
+      return `
+        import { enableReactComponentJump } from "react-code-locator/client";
+
+        enableReactComponentJump(${serialized});
+      `;
+    },
+    transformIndexHtml() {
+      return [
+        {
+          tag: "script",
+          attrs: {
+            type: "module",
+            src: `/@id/__x00__${VIRTUAL_CLIENT_MODULE_ID}`
+          },
+          injectTo: "head"
         }
+      ];
+    }
+  };
+}
+function createViteClientInjector(options = {}) {
+  const { command = "serve", locator = {}, injectClient = true } = options;
+  const isServe = command === "serve";
+  return [isServe && injectClient ? createClientInjector(locator) : null].filter(Boolean);
+}
+
+// src/vite.ts
+function shouldTransformSource(id) {
+  if (id.includes("/node_modules/") || id.startsWith("\0")) {
+    return false;
+  }
+  return /\.[mc]?[jt]sx?$/.test(id);
+}
+function viteSourceTransformPlugin(options = {}) {
+  return {
+    name: "react-code-locator-source-transform",
+    enforce: "pre",
+    async transform(code, id) {
+      if (!shouldTransformSource(id)) {
+        return null;
       }
+      return transformSourceWithLocator(code, {
+        filename: id,
+        ...options
+      });
     }
   };
 }
+function createViteSourceAdapter(options = {}) {
+  const { babel = {}, ...viteOptions } = options;
+  const plugins = [
+    viteSourceTransformPlugin(babel),
+    ...createViteClientInjector(viteOptions)
+  ].filter(Boolean);
+  return defineSourceAdapter({
+    kind: "vite",
+    name: "react-code-locator/vite",
+    options,
+    config: {
+      plugins
+    }
+  });
+}
+var viteSourceAdapter = createViteSourceAdapter();
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   babelInjectComponentSource,
-  elementLocatorReact,
-  reactComponentJump
+  createViteClientInjector,
+  createViteSourceAdapter,
+  reactComponentJump,
+  viteSourceAdapter,
+  viteSourceTransformPlugin
 });
 //# sourceMappingURL=vite.cjs.map