@shaper.org/vite-react-plugin 1.0.14 → 1.0.16

package/dist/index.mjs

@@ -1799,14 +1799,57 @@ function jsxSourcePlugin(options = {}) {
 					runtime: "automatic"
 				}], "@babel/preset-typescript"],
 				plugins: [function injectSourceFile() {
-					return { visitor: { JSXOpeningElement(path, state) {
-						const root = state.file.opts.root;
-						const filename = state.file.opts.filename;
-						const relativeFilename = PathUtil.relative(root, filename);
-						const { line, column } = path.node.loc.start;
-						const attr = t.jsxAttribute(t.jsxIdentifier("data-loc"), t.stringLiteral(`${relativeFilename}:${line}:${column}`));
-						path.node.attributes.push(attr);
-					} } };
+					return { visitor: {
+						Program(programPath, state) {
+							const root = state.file.opts.root;
+							const filename = state.file.opts.filename;
+							const relativeFilename = PathUtil.relative(root, filename);
+							const injected = /* @__PURE__ */ new Set();
+							function injectAtProgramEnd(name) {
+								if (injected.has(name)) return;
+								injected.add(name);
+								programPath.pushContainer("body", t.expressionStatement(t.assignmentExpression("=", t.memberExpression(t.identifier(name), t.identifier("__file")), t.stringLiteral(relativeFilename))));
+							}
+							programPath.traverse({
+								ExportNamedDeclaration(path) {
+									const decl = path.node.declaration;
+									if (t.isFunctionDeclaration(decl) && decl.id) injectAtProgramEnd(decl.id.name);
+									if (t.isClassDeclaration(decl) && decl.id) injectAtProgramEnd(decl.id.name);
+									if (t.isVariableDeclaration(decl)) {
+										for (const d of decl.declarations) if (t.isIdentifier(d.id) && (t.isArrowFunctionExpression(d.init) || t.isFunctionExpression(d.init))) injectAtProgramEnd(d.id.name);
+									}
+									for (const spec of path.node.specifiers) if (t.isExportSpecifier(spec)) injectAtProgramEnd(spec.local.name);
+								},
+								ExportDefaultDeclaration(path) {
+									const decl = path.node.declaration;
+									if (t.isFunctionDeclaration(decl) && decl.id) {
+										injectAtProgramEnd(decl.id.name);
+										return;
+									}
+									if (t.isClassDeclaration(decl) && decl.id) {
+										injectAtProgramEnd(decl.id.name);
+										return;
+									}
+									if (t.isIdentifier(decl)) {
+										injectAtProgramEnd(decl.name);
+										return;
+									}
+									if (t.isExportSpecifier?.(decl)) {
+										injectAtProgramEnd(decl.local.name);
+										return;
+									}
+								}
+							});
+						},
+						JSXOpeningElement(path, state) {
+							const root = state.file.opts.root;
+							const filename = state.file.opts.filename;
+							const relativeFilename = PathUtil.relative(root, filename);
+							const { line, column } = path.node.loc.start;
+							const attr = t.jsxAttribute(t.jsxIdentifier("data-loc"), t.stringLiteral(`${relativeFilename}:${line}:${column}`));
+							path.node.attributes.push(attr);
+						}
+					} };
 				}]
 			});
 			if (!result) return null;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shaper.org/vite-react-plugin",
-  "version": "1.0.14",
+  "version": "1.0.16",
   "description": "",
   "type": "module",
   "private": false,

package/templates/main.ts

@@ -138,12 +138,15 @@ export class ReactRouteMonitor {
         const validRoutes = event.detail.routes.filter(
             (route) => route.element !== undefined && route.path !== "*",
         );
+
         const routes = validRoutes.map((route) => ({
             name: route.path,
             path: route.path,
-            file: route.element.type().props["data-loc"],
+            file: route.element.type.__file || "src/",
         }));
 
+        console.log(routes);
+
         this.iframeClient.sendAllRoutes(routes);
     }
 
@@ -181,7 +184,7 @@ export class ReactRouteMonitor {
         const routeInfo = {
             name: toPathname,
             path: toPathname,
-            file: to.element.type().props["data-loc"],
+            file: to.element.type.__file || "src/",
         };
 
         if (toPathname === fromPathname) {
@@ -236,5 +239,7 @@ const monitor = new ReactRouteMonitor({
     },
 });
 
+
+
 monitor.start();
 monitor.triggerRoutes();