zelune 1.6.1 → 1.6.3

package/src/server/apiMiddleware.ts

@@ -237,8 +237,11 @@ export function viteCanvasApiMiddleware(workspaceRoot: string): Plugin {
 
         const activeRoot = getActiveProjectRoot(workspaceRoot);
         const layoutFilePath = path.resolve(activeRoot, '.canvas-layout.json');
-        const urlObj = new URL(req.url, 'http://127.0.0.1');
-        const pathname = urlObj.pathname;
+        const qIndex = req.url.indexOf('?');
+        const pathname = qIndex === -1 ? req.url : req.url.slice(0, qIndex);
+        const searchParams = new URLSearchParams(
+          qIndex === -1 ? '' : req.url.slice(qIndex + 1)
+        );
 
         try {
           // GET /api/screens
@@ -253,9 +256,9 @@ export function viteCanvasApiMiddleware(workspaceRoot: string): Plugin {
 
           // GET /api/inspect?file=...&astPath=...&line=...
           if (req.method === 'GET' && pathname === '/api/inspect') {
-            const file = urlObj.searchParams.get('file') ?? '';
-            const astPath = urlObj.searchParams.get('astPath') ?? '';
-            const lineParam = urlObj.searchParams.get('line');
+            const file = searchParams.get('file') ?? '';
+            const astPath = searchParams.get('astPath') ?? '';
+            const lineParam = searchParams.get('line');
             const line = lineParam ? Number.parseInt(lineParam, 10) : undefined;
 
             const element = await inspectElement(workspaceRoot, { file, astPath, line });

package/src/server/astEngine.ts

@@ -2,14 +2,88 @@ import fs from 'node:fs/promises';
 import fsSync from 'node:fs';
 import path from 'node:path';
 import { parse } from '@babel/parser';
-import _traverse, { type NodePath } from '@babel/traverse';
 import _generate from '@babel/generator';
 import * as t from '@babel/types';
-import { getJsxTagName } from '../../plugins/viteRnSourceInject';
 
-const traverse = (_traverse as unknown as { default?: typeof _traverse }).default ?? _traverse;
 const generate = (_generate as unknown as { default?: typeof _generate }).default ?? _generate;
 
+export interface AstNodePath<T extends t.Node = t.Node> {
+  node: T;
+  parentPath: AstNodePath<t.Node> | null;
+  isJSXElement(): this is AstNodePath<t.JSXElement>;
+  isJSXFragment(): this is AstNodePath<t.JSXFragment>;
+  findParent(predicate: (p: AstNodePath<t.Node>) => boolean): AstNodePath<t.Node> | null;
+}
+
+type NodePath<T extends t.Node = t.Node> = AstNodePath<T>;
+
+export type AstVisitorMap = {
+  [K in t.Node['type']]?: (path: AstNodePath<Extract<t.Node, { type: K }>>) => void;
+};
+
+export function getJsxTagName(nameNode: t.JSXOpeningElement['name']): string {
+  if (t.isJSXIdentifier(nameNode)) {
+    return nameNode.name;
+  }
+  if (t.isJSXMemberExpression(nameNode)) {
+    return `${getJsxTagName(nameNode.object)}.${nameNode.property.name}`;
+  }
+  if (t.isJSXNamespacedName(nameNode)) {
+    return `${nameNode.namespace.name}:${nameNode.name.name}`;
+  }
+  return 'Unknown';
+}
+
+export function traverse(rootNode: t.Node, visitors: AstVisitorMap): void {
+  function createPath(node: t.Node, parentPath: AstNodePath<t.Node> | null): AstNodePath<t.Node> {
+    return {
+      node,
+      parentPath,
+      isJSXElement(): this is AstNodePath<t.JSXElement> {
+        return t.isJSXElement(node);
+      },
+      isJSXFragment(): this is AstNodePath<t.JSXFragment> {
+        return t.isJSXFragment(node);
+      },
+      findParent(predicate) {
+        let curr = parentPath;
+        while (curr) {
+          if (predicate(curr)) return curr;
+          curr = curr.parentPath;
+        }
+        return null;
+      },
+    };
+  }
+
+  function visit(node: unknown, parentPath: AstNodePath<t.Node> | null): void {
+    if (!node || typeof node !== 'object' || typeof (node as t.Node).type !== 'string') return;
+    const typedNode = node as t.Node;
+    const pathObj = createPath(typedNode, parentPath);
+    const handler = visitors[typedNode.type as t.Node['type']] as
+      | ((p: AstNodePath<t.Node>) => void)
+      | undefined;
+    if (typeof handler === 'function') {
+      handler(pathObj);
+    }
+    const keys = t.VISITOR_KEYS[typedNode.type];
+    if (!keys) return;
+    const rec = typedNode as unknown as Record<string, unknown>;
+    for (const key of keys) {
+      const child = rec[key];
+      if (Array.isArray(child)) {
+        for (let i = 0; i < child.length; i++) {
+          if (child[i]) visit(child[i], pathObj);
+        }
+      } else if (child && typeof child === 'object') {
+        visit(child, pathObj);
+      }
+    }
+  }
+
+  visit(rootNode, null);
+}
+
 export interface NavigationEdge {
   id: string;
   fromScreen: string;
@@ -807,10 +881,7 @@ function resolveCandidateToScreenId(
   const trimmed = rawCandidate.trim();
   if (!trimmed || trimmed.length > 80) return null;
   if (
-    trimmed.startsWith('http://') ||
-    trimmed.startsWith('https://') ||
-    trimmed.startsWith('mailto:') ||
-    trimmed.startsWith('tel:') ||
+    /^[a-z][a-z0-9+.-]*:/i.test(trimmed) ||
     trimmed.startsWith('#') ||
     /\.(png|jpg|jpeg|svg|webp|gif|json|css)$/i.test(trimmed)
   ) {

package/vite.config.ts

@@ -1,6 +1,5 @@
 import { parse } from '@babel/parser';
 import { build, defineConfig, loadEnv } from 'vite';
-import react from '@vitejs/plugin-react';
 import fs from 'node:fs';
 import path from 'node:path';
 import { fileURLToPath } from 'node:url';
@@ -22,6 +21,7 @@ const workspaceRoot = process.env.ANTIGRAVITY_WORKSPACE_ROOT
 
 const targetProjectRoot = findMobileAppRoot(rawTargetRoot);
 const mockNavShimPath = path.resolve(__dirname, 'src/runtime/MockNavigationContext.tsx');
+const reactNativeShimPath = path.resolve(__dirname, 'src/runtime/reactNativeShim.ts');
 
 function safeRealpath(targetPath: string): string {
   try {
@@ -42,7 +42,6 @@ function resolvePackageRootDir(pkgName: string): string {
 
 const resolvedReactDir = resolvePackageRootDir('react');
 const resolvedReactDomDir = resolvePackageRootDir('react-dom');
-const resolvedRnWebDir = resolvePackageRootDir('react-native-web');
 const resolvedLucideDir = resolvePackageRootDir('lucide-react');
 
 function loadPublicProjectEnv(roots: string[]): Record<string, string> {
@@ -288,7 +287,6 @@ export default defineConfig({
     viteReactNativeFallbackPlugin(targetProjectRoot, mockNavShimPath, workspaceRoot),
     viteVirtualScreensPlugin(targetProjectRoot),
     viteRnSourceInject(targetProjectRoot),
-    react(),
     viteCanvasApiMiddleware(targetProjectRoot),
   ],
   optimizeDeps: {
@@ -299,22 +297,23 @@ export default defineConfig({
       'react/jsx-dev-runtime',
       'react-dom',
       'react-dom/client',
-      'react-native-web',
       'lucide-react',
     ],
   },
   esbuild: {
     // Ensure .tsx/.ts/.jsx/.js files (including RN .js files with JSX) are transpiled cleanly
+    jsx: 'automatic',
     include: /\.(tsx|ts|jsx|js)$/,
     exclude: [],
     loader: 'tsx',
   },
   resolve: {
-    dedupe: ['react', 'react-dom', 'react-native-web'],
+    dedupe: ['react', 'react-dom'],
     alias: {
       react: resolvedReactDir,
       'react-dom': resolvedReactDomDir,
-      'react-native-web': resolvedRnWebDir,
+      'react-native': reactNativeShimPath,
+      'react-native-web': reactNativeShimPath,
       'lucide-react': resolvedLucideDir,
     },
     extensions: ['.web.tsx', '.web.ts', '.web.js', '.tsx', '.ts', '.jsx', '.js'],