@vitarx/plugin-vite 0.0.1-beta.6 → 0.0.1-beta.8

package/dist/context.d.ts

@@ -3,16 +3,9 @@
  * 定义编译转换过程中的上下文数据结构
  * @module context
  */
-import type { File } from '@babel/types';
-import type { CompileOptions } from './transform.js';
+import type { CallExpression, File } from '@babel/types';
 import type { CompilerWarning } from './error.js';
-import { UI_API_NAMES } from './constants/index.js';
-/**
- * UI 相关的 API 名称列表
- * 这些 API 用于创建视图，在 HMR 时需要识别
- */
-export declare const UI_API_NAMES_CONST: readonly ["createView", "branch", "dynamic", "access", "withDirectives"];
-export type UIApiName = (typeof UI_API_NAMES)[number];
+import type { CompileOptions } from './types.js';
 /**
  * 导入信息
  * 追踪需要注入的运行时 API
@@ -87,6 +80,12 @@ export interface TransformContext {
     refVariables: Set<string>;
     /** UI API 别名集合（用于 HMR 代码分离识别） */
     uiApiAliases: Set<string>;
+    /** builder 函数的本地别名（用于识别纯构建组件） */
+    builderAlias: string | null;
+    /** 已处理的 AST 节点追踪集合 */
+    processedNodes: WeakSet<any>;
+    /** 已添加 PURE 注释的节点追踪集合 */
+    pureCommentedNodes: WeakSet<CallExpression>;
     /** 编译警告列表 */
     warnings: CompilerWarning[];
 }
@@ -105,12 +104,6 @@ export declare function createContext(code: string, id: string, options: Compile
  * @param name - 导入名称
  */
 export declare function markImport(ctx: TransformContext, name: keyof ImportInfo): void;
-/**
- * 记录 UI API 别名
- * @param ctx - 转换上下文
- * @param alias - API 别名
- */
-export declare function markUiApiAlias(ctx: TransformContext, alias: string): void;
 /**
  * 添加编译警告
  * @param ctx - 转换上下文

package/dist/context.js

@@ -1,9 +1,3 @@
-import { UI_API_NAMES } from './constants/index.js';
-/**
- * UI 相关的 API 名称列表
- * 这些 API 用于创建视图，在 HMR 时需要识别
- */
-export const UI_API_NAMES_CONST = UI_API_NAMES;
 /**
  * 创建转换上下文
  * @param code - 源代码
@@ -50,6 +44,9 @@ export function createContext(code, id, options, ast) {
         },
         refVariables: new Set(),
         uiApiAliases: new Set(),
+        builderAlias: null,
+        processedNodes: new WeakSet(),
+        pureCommentedNodes: new WeakSet(),
         warnings: []
     };
 }
@@ -61,14 +58,6 @@ export function createContext(code, id, options, ast) {
 export function markImport(ctx, name) {
     ctx.imports[name] = true;
 }
-/**
- * 记录 UI API 别名
- * @param ctx - 转换上下文
- * @param alias - API 别名
- */
-export function markUiApiAlias(ctx, alias) {
-    ctx.uiApiAliases.add(alias);
-}
 /**
  * 添加编译警告
  * @param ctx - 转换上下文

package/dist/hmr-client/index.d.ts

@@ -45,8 +45,9 @@ export default class HMRManager {
      * 注册节点
      *
      * @param view - 组件视图节点
+     * @param component - 组件构造函数
      */
-    register(view: ComponentView): void;
+    register(view: ComponentView, component?: Component): void;
     /**
      * 获取模块id
      *

package/dist/hmr-client/index.js

@@ -80,11 +80,12 @@ class HMRManager {
      * 注册节点
      *
      * @param view - 组件视图节点
+     * @param component - 组件构造函数
      */
-    register(view) {
+    register(view, component) {
         if (!view)
             return;
-        const component = view.component;
+        component ?? (component = view.component);
         const id = this.getId(component);
         if (__classPrivateFieldGet(this, _HMRManager_idMapToView, "f").has(id)) {
             __classPrivateFieldGet(this, _HMRManager_idMapToView, "f").get(id).add(view);
@@ -124,7 +125,7 @@ class HMRManager {
                     if (id)
                         __classPrivateFieldGet(this, _HMRManager_idMapToComponent, "f").set(id, newComponent);
                     // 更新节点视图
-                    if (node.active && node.isMounted) {
+                    if ((node.isActive || node.active) && node.isMounted) {
                         processUpdate(node, newComponent);
                     }
                 }

package/dist/hmr-client/utils.js

@@ -10,7 +10,7 @@ import { UI_API_NAMES } from '../constants/index.js';
  * UI 相关的运行时 API 名称
  * 这些 API 调用代表 UI 描述代码
  */
-const UI_APIS = new Set([...UI_API_NAMES, 'jsxDEV']);
+const UI_APIS = new Set([...UI_API_NAMES, 'jsxDEV', 'h']);
 /**
  * 判断标识符名称是否为 UI API
  * 支持原始名称和带后缀的名称（如 jsxDEV$1）

package/dist/index.js

@@ -1,7 +1,7 @@
 import path from 'node:path';
 import { fileURLToPath } from 'node:url';
 import { version } from 'vite';
-import { transform } from './transform.js';
+import { shouldTransform, transform } from './transform.js';
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 const IS_V8 = version.startsWith('8');
 let viteTransform;
@@ -32,7 +32,6 @@ export default function vitarx(options) {
     let viteConfig;
     return {
         name: 'vite-plugin-vitarx',
-        enforce: 'pre',
         config(config, env) {
             isDEV = env.command === 'serve' && !env.isPreview;
             const configSSR = !!config.build?.ssr;
@@ -59,15 +58,17 @@ export default function vitarx(options) {
                 ssr: isSSR,
                 hmr: isDEV && !isSSR,
                 runtimeModule: 'vitarx',
-                sourceMap: false,
+                sourceMap: true,
                 transformClassNameToClass: options?.transformClassNameToClass ?? false
             };
         },
         async transform(code, id) {
+            if (!shouldTransform(id))
+                return null;
             const result = await transform(code, id, compileOptions);
             if (!result)
                 return null;
-            return await viteTransform(result.code, id, undefined, this.getCombinedSourcemap(), viteConfig);
+            return viteTransform(result.code, id, undefined, result.map || undefined, viteConfig);
         }
     };
 }

package/dist/passes/directives/processDirectives.js

@@ -27,7 +27,7 @@ export function processDirectives(viewCall, directives, ctx) {
     }
     markImport(ctx, 'withDirectives');
     const withDirectivesAlias = getAlias(ctx.vitarxAliases, 'withDirectives');
-    return addPureComment(createWithDirectivesCall(viewCall, directiveArray, withDirectivesAlias));
+    return addPureComment(createWithDirectivesCall(viewCall, directiveArray, withDirectivesAlias), ctx);
 }
 /**
  * 构建指令值对象

package/dist/passes/hmr/inject-imports.d.ts

@@ -0,0 +1,19 @@
+/**
+ * HMR 导入注入模块
+ * 负责注入 HMR 所需的导入语句
+ * @module passes/hmr/inject-imports
+ */
+import * as t from '@babel/types';
+/** getComponentView 的内部别名，避免与用户代码冲突 */
+export declare const GET_COMPONENT_VIEW_ALIAS = "__$VITARX_GET_COMPONENT_VIEW$__";
+/**
+ * 注入 HMR 客户端导入
+ * @param program - AST Program 节点
+ */
+export declare function injectHMRImport(program: t.Program): void;
+/**
+ * 注入 getComponentView 导入
+ * 使用唯一的别名避免与用户代码冲突
+ * @param program - AST Program 节点
+ */
+export declare function injectGetComponentViewImport(program: t.Program): void;

package/dist/passes/hmr/inject-imports.js

@@ -0,0 +1,26 @@
+/**
+ * HMR 导入注入模块
+ * 负责注入 HMR 所需的导入语句
+ * @module passes/hmr/inject-imports
+ */
+import * as t from '@babel/types';
+import { HMR } from '../../constants/index.js';
+/** getComponentView 的内部别名，避免与用户代码冲突 */
+export const GET_COMPONENT_VIEW_ALIAS = '__$VITARX_GET_COMPONENT_VIEW$__';
+/**
+ * 注入 HMR 客户端导入
+ * @param program - AST Program 节点
+ */
+export function injectHMRImport(program) {
+    const importDecl = t.importDeclaration([t.importDefaultSpecifier(t.identifier(HMR.manager))], t.stringLiteral('@vitarx/vite-plugin/hmr-client'));
+    program.body.unshift(importDecl);
+}
+/**
+ * 注入 getComponentView 导入
+ * 使用唯一的别名避免与用户代码冲突
+ * @param program - AST Program 节点
+ */
+export function injectGetComponentViewImport(program) {
+    const importDecl = t.importDeclaration([t.importSpecifier(t.identifier(GET_COMPONENT_VIEW_ALIAS), t.identifier('getComponentView'))], t.stringLiteral('vitarx'));
+    program.body.unshift(importDecl);
+}

package/dist/passes/hmr/inject-state.d.ts

@@ -0,0 +1,29 @@
+/**
+ * HMR 状态恢复模块
+ * 负责为组件函数注入状态保存和恢复代码
+ * @module passes/hmr/inject-state
+ */
+import * as t from '@babel/types';
+/**
+ * 创建组件函数体内的 HMR 注册代码
+ * @param variableNames - 需要追踪的变量名列表
+ * @returns HMR 注册语句数组
+ */
+export declare function createHMRRegistrationStatements(variableNames: string[]): t.Statement[];
+/**
+ * 创建包装组件的 HMR 注册代码
+ * 只注册节点，不保存状态，传入组件函数自身作为第二个参数
+ * @param componentName - 组件函数名称
+ * @returns HMR 注册语句数组
+ */
+export declare function createWrappedComponentHMRStatements(componentName: string): t.Statement[];
+/**
+ * 从函数体中收集局部变量名
+ * @param functionBody - 函数体语句块
+ * @returns 变量名数组
+ */
+export declare function collectLocalVariableNames(functionBody: t.BlockStatement): string[];
+/**
+ * 为函数体内的变量声明注入状态恢复代码
+ */
+export declare function injectStatePreservation(functionBody: t.BlockStatement): void;

package/dist/passes/hmr/inject-state.js

@@ -0,0 +1,101 @@
+/**
+ * HMR 状态恢复模块
+ * 负责为组件函数注入状态保存和恢复代码
+ * @module passes/hmr/inject-state
+ */
+import * as t from '@babel/types';
+import { HMR } from '../../constants/index.js';
+import { collectPatternBindings } from '../../utils/index.js';
+import { GET_COMPONENT_VIEW_ALIAS } from './inject-imports.js';
+/**
+ * 创建组件函数体内的 HMR 注册代码
+ * @param variableNames - 需要追踪的变量名列表
+ * @returns HMR 注册语句数组
+ */
+export function createHMRRegistrationStatements(variableNames) {
+    const statements = [];
+    statements.push(t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier(HMR.view), t.callExpression(t.identifier(GET_COMPONENT_VIEW_ALIAS), []))
+    ]));
+    statements.push(t.expressionStatement(t.callExpression(t.memberExpression(t.memberExpression(t.identifier(HMR.manager), t.identifier('instance')), t.identifier('register')), [t.identifier(HMR.view)])));
+    const stateProperties = variableNames.map(name => t.objectMethod('get', t.identifier(name), [], t.blockStatement([t.returnStatement(t.identifier(name))])));
+    statements.push(t.expressionStatement(t.logicalExpression('&&', t.identifier(HMR.view), t.callExpression(t.memberExpression(t.callExpression(t.memberExpression(t.identifier('Promise'), t.identifier('resolve')), []), t.identifier('then')), [
+        t.arrowFunctionExpression([], t.blockStatement([
+            t.expressionStatement(t.assignmentExpression('=', t.memberExpression(t.identifier(HMR.view), t.identifier(HMR.state)), t.objectExpression(stateProperties)))
+        ]))
+    ]))));
+    return statements;
+}
+/**
+ * 创建包装组件的 HMR 注册代码
+ * 只注册节点，不保存状态，传入组件函数自身作为第二个参数
+ * @param componentName - 组件函数名称
+ * @returns HMR 注册语句数组
+ */
+export function createWrappedComponentHMRStatements(componentName) {
+    const statements = [];
+    statements.push(t.variableDeclaration('const', [
+        t.variableDeclarator(t.identifier(HMR.view), t.callExpression(t.identifier(GET_COMPONENT_VIEW_ALIAS), []))
+    ]));
+    statements.push(t.expressionStatement(t.callExpression(t.memberExpression(t.memberExpression(t.identifier(HMR.manager), t.identifier('instance')), t.identifier('register')), [t.identifier(HMR.view), t.identifier(componentName)])));
+    return statements;
+}
+/**
+ * 从函数体中收集局部变量名
+ * @param functionBody - 函数体语句块
+ * @returns 变量名数组
+ */
+export function collectLocalVariableNames(functionBody) {
+    const variableNames = new Set();
+    for (const stmt of functionBody.body) {
+        if (stmt.type === 'VariableDeclaration') {
+            for (const decl of stmt.declarations) {
+                if (decl.id.type !== 'VoidPattern') {
+                    collectPatternBindings(decl.id, variableNames);
+                }
+            }
+        }
+    }
+    return Array.from(variableNames);
+}
+/**
+ * 判断表达式是否为函数类型
+ */
+function isFunctionExpression(expr) {
+    if (!expr)
+        return false;
+    return (expr.type === 'ArrowFunctionExpression' ||
+        expr.type === 'FunctionExpression' ||
+        expr.type === 'ClassExpression');
+}
+/**
+ * 创建状态恢复表达式
+ * 格式：__$VITARX_HMR$__.instance.memo(__$VITARX_HMR_VIEW_NODE$__, '变量名') ?? 原始初始值
+ */
+function createMemoExpression(variableName, originalInit) {
+    return t.logicalExpression('??', t.callExpression(t.memberExpression(t.memberExpression(t.identifier(HMR.manager), t.identifier('instance')), t.identifier('memo')), [t.identifier(HMR.view), t.stringLiteral(variableName)]), originalInit);
+}
+/**
+ * 为单个变量声明注入状态恢复代码
+ */
+function injectStatePreservationForDeclaration(decl) {
+    if (decl.id.type !== 'Identifier')
+        return;
+    if (!decl.init)
+        return;
+    if (isFunctionExpression(decl.init))
+        return;
+    decl.init = createMemoExpression(decl.id.name, decl.init);
+}
+/**
+ * 为函数体内的变量声明注入状态恢复代码
+ */
+export function injectStatePreservation(functionBody) {
+    for (const stmt of functionBody.body) {
+        if (stmt.type === 'VariableDeclaration') {
+            for (const decl of stmt.declarations) {
+                injectStatePreservationForDeclaration(decl);
+            }
+        }
+    }
+}

package/dist/passes/hmr/inject.d.ts

@@ -4,7 +4,7 @@
  * @module passes/hmr
  */
 import * as t from '@babel/types';
-import { type ComponentInfo } from '../../utils/index.js';
+import type { ComponentInfo } from '../../utils/index.js';
 /**
  * 注入 HMR 支持
  * 主入口函数，为所有组件注入完整的 HMR 支持

package/dist/passes/hmr/inject.js

@@ -5,133 +5,28 @@
  */
 import * as t from '@babel/types';
 import { HMR } from '../../constants/index.js';
-import { collectPatternBindings } from '../../utils/index.js';
-/** getComponentView 的内部别名，避免与用户代码冲突 */
-const GET_COMPONENT_VIEW_ALIAS = '__$VITARX_GET_COMPONENT_VIEW$__';
-/**
- * 注入 HMR 客户端导入
- * @param program - AST Program 节点
- */
-function injectHMRImport(program) {
-    const importDecl = t.importDeclaration([t.importDefaultSpecifier(t.identifier(HMR.manager))], t.stringLiteral('@vitarx/vite-plugin/hmr-client'));
-    program.body.unshift(importDecl);
-}
-/**
- * 注入 getComponentView 导入
- * 使用唯一的别名避免与用户代码冲突
- * @param program - AST Program 节点
- */
-function injectGetComponentViewImport(program) {
-    // 创建独立的 import 语句，使用不会冲突的别名
-    const importDecl = t.importDeclaration([t.importSpecifier(t.identifier(GET_COMPONENT_VIEW_ALIAS), t.identifier('getComponentView'))], t.stringLiteral('vitarx'));
-    program.body.unshift(importDecl);
-}
-/**
- * 创建组件函数体内的 HMR 注册代码
- * @param variableNames - 需要追踪的变量名列表
- * @returns HMR 注册语句数组
- */
-function createHMRRegistrationStatements(variableNames) {
-    const statements = [];
-    // 声明并获取当前视图实例
-    statements.push(t.variableDeclaration('const', [
-        t.variableDeclarator(t.identifier(HMR.view), t.callExpression(t.identifier(GET_COMPONENT_VIEW_ALIAS), []))
-    ]));
-    // 注册到 HMR 管理器
-    statements.push(t.expressionStatement(t.callExpression(t.memberExpression(t.memberExpression(t.identifier(HMR.manager), t.identifier('instance')), t.identifier('register')), [t.identifier(HMR.view)])));
-    // 创建状态追踪对象
-    const stateProperties = variableNames.map(name => t.objectMethod('get', t.identifier(name), [], t.blockStatement([t.returnStatement(t.identifier(name))])));
-    // 异步设置状态
-    statements.push(t.expressionStatement(t.logicalExpression('&&', t.identifier(HMR.view), t.callExpression(t.memberExpression(t.callExpression(t.memberExpression(t.identifier('Promise'), t.identifier('resolve')), []), t.identifier('then')), [
-        t.arrowFunctionExpression([], t.blockStatement([
-            t.expressionStatement(t.assignmentExpression('=', t.memberExpression(t.identifier(HMR.view), t.identifier(HMR.state)), t.objectExpression(stateProperties)))
-        ]))
-    ]))));
-    return statements;
-}
-/**
- * 从函数体中收集局部变量名
- * @param functionBody - 函数体语句块
- * @returns 变量名数组
- */
-function collectLocalVariableNames(functionBody) {
-    const variableNames = new Set();
-    for (const stmt of functionBody.body) {
-        if (stmt.type === 'VariableDeclaration') {
-            for (const decl of stmt.declarations) {
-                if (decl.id.type !== 'VoidPattern') {
-                    collectPatternBindings(decl.id, variableNames);
-                }
-            }
-        }
-    }
-    return Array.from(variableNames);
-}
-/**
- * 判断表达式是否为函数类型
- * 包括：箭头函数、函数表达式、类表达式
- */
-function isFunctionExpression(expr) {
-    if (!expr)
-        return false;
-    return (expr.type === 'ArrowFunctionExpression' ||
-        expr.type === 'FunctionExpression' ||
-        expr.type === 'ClassExpression');
-}
-/**
- * 创建状态恢复表达式
- * 格式：__$VITARX_HMR$__.instance.memo(__$VITARX_HMR_VIEW_NODE$__, '变量名') ?? 原始初始值
- */
-function createMemoExpression(variableName, originalInit) {
-    return t.logicalExpression('??', t.callExpression(t.memberExpression(t.memberExpression(t.identifier(HMR.manager), t.identifier('instance')), t.identifier('memo')), [t.identifier(HMR.view), t.stringLiteral(variableName)]), originalInit);
-}
-/**
- * 为单个变量声明注入状态恢复代码
- * 仅处理标识符类型的变量声明，且初始值不是函数
- */
-function injectStatePreservationForDeclaration(decl) {
-    // 只处理标识符类型的变量声明
-    if (decl.id.type !== 'Identifier')
-        return;
-    // 没有初始值，不需要处理
-    if (!decl.init)
-        return;
-    // 初始值是函数，不需要状态恢复
-    if (isFunctionExpression(decl.init))
-        return;
-    // 注入状态恢复代码
-    decl.init = createMemoExpression(decl.id.name, decl.init);
-}
-/**
- * 为函数体内的变量声明注入状态恢复代码
- */
-function injectStatePreservation(functionBody) {
-    for (const stmt of functionBody.body) {
-        if (stmt.type === 'VariableDeclaration') {
-            for (const decl of stmt.declarations) {
-                injectStatePreservationForDeclaration(decl);
-            }
-        }
-    }
-}
+import { injectHMRImport, injectGetComponentViewImport } from './inject-imports.js';
+import { createHMRRegistrationStatements, createWrappedComponentHMRStatements, collectLocalVariableNames, injectStatePreservation } from './inject-state.js';
 /**
  * 为组件函数注入 HMR 注册代码
  * @param func - 函数声明/表达式/箭头函数
  * @param variableNames - 需要追踪的变量名列表
+ * @param isWrapped - 是否为包装组件
+ * @param componentName - 组件名称（包装组件需要）
  */
-function injectHMRIntoFunction(func, variableNames) {
-    // 处理箭头函数没有函数体的情况（表达式体）
+function injectHMRIntoFunction(func, variableNames, isWrapped = false, componentName) {
     if (func.type === 'ArrowFunctionExpression' && func.body.type !== 'BlockStatement') {
-        // 将表达式体转换为块语句
         const returnStmt = t.returnStatement(func.body);
         func.body = t.blockStatement([returnStmt]);
     }
     if (!func.body || func.body.type !== 'BlockStatement')
         return;
-    // 注入状态恢复代码
-    injectStatePreservation(func.body);
-    // 注入 HMR 注册代码
-    const statements = createHMRRegistrationStatements(variableNames);
+    if (!isWrapped) {
+        injectStatePreservation(func.body);
+    }
+    const statements = isWrapped
+        ? createWrappedComponentHMRStatements(componentName)
+        : createHMRRegistrationStatements(variableNames);
     func.body.body.unshift(...statements);
 }
 /**
@@ -171,19 +66,14 @@ function generateComponentId(filename, componentName) {
 export function injectHMRSupport(program, components, filename) {
     if (components.length === 0)
         return;
-    // 注入必要的导入
     injectHMRImport(program);
     injectGetComponentViewImport(program);
-    // 为每个组件函数注入 HMR 代码
-    for (const { node } of components) {
-        // 收集局部变量名（如果有函数体）
+    for (const { node, isWrapped, name } of components) {
         const variableNames = node.body?.type === 'BlockStatement' ? collectLocalVariableNames(node.body) : [];
-        injectHMRIntoFunction(node, variableNames);
+        injectHMRIntoFunction(node, variableNames, isWrapped, name);
     }
-    // 为每个组件绑定 ID
     for (const { name } of components) {
         program.body.push(createBindIdStatement(name, generateComponentId(filename, name)));
     }
-    // 注入 import.meta.hot.accept
     program.body.push(createHotAcceptStatement());
 }

package/dist/passes/jsx/processChildren.js

@@ -22,7 +22,7 @@ function handleLogicalExpression(expr, ctx, loc) {
     markImport(ctx, 'dynamic');
     const dynamicAlias = getAlias(ctx.vitarxAliases, 'dynamic');
     const locInfo = loc ? getDevLocInfo(ctx, { loc }) : null;
-    return addPureComment(createDynamicCall(expr, dynamicAlias, locInfo));
+    return addPureComment(createDynamicCall(expr, dynamicAlias, locInfo), ctx);
 }
 /**
  * 处理子节点数组

package/dist/passes/jsx/processJSXElement.js

@@ -87,7 +87,7 @@ export function transformJSXElement(node, ctx, handleVIf = false) {
         viewCall = processDirectives(viewCall, directives, ctx);
     }
     else {
-        viewCall = addPureComment(viewCall);
+        viewCall = addPureComment(viewCall, ctx);
     }
     if (node.loc) {
         viewCall.loc = node.loc;

package/dist/passes/jsx/processJSXFragment.js

@@ -1,5 +1,5 @@
 import * as t from '@babel/types';
-import { addPureComment, createCreateViewCall, filterWhitespaceChildren, getAlias, getDevLocInfo, validateMatchInSwitch } from '../../utils/index.js';
+import { addPureComment, createCreateViewCall, filterWhitespaceChildren, getAlias, getDevLocInfo, validateNoDirectMatchChild } from '../../utils/index.js';
 import { processChildren } from './processChildren.js';
 /**
  * 处理 JSX Fragment
@@ -11,7 +11,7 @@ export function processJSXFragment(path, ctx) {
     // 过滤空白子节点
     const children = filterWhitespaceChildren(node.children);
     // 校验 Match 组件
-    validateMatchInSwitch(children);
+    validateNoDirectMatchChild(children);
     // 标记需要的导入
     ctx.imports.Fragment = true;
     ctx.imports.createView = true;
@@ -20,7 +20,7 @@ export function processJSXFragment(path, ctx) {
     const locInfo = getDevLocInfo(ctx, node);
     // 无子元素
     if (children.length === 0) {
-        const viewCall = addPureComment(createCreateViewCall(t.identifier(fragmentAlias), null, locInfo, createViewAlias));
+        const viewCall = addPureComment(createCreateViewCall(t.identifier(fragmentAlias), null, locInfo, createViewAlias), ctx);
         if (node.loc)
             viewCall.loc = node.loc;
         path.replaceWith(viewCall);
@@ -30,7 +30,7 @@ export function processJSXFragment(path, ctx) {
     const processedChildren = processChildren(children, ctx);
     const childrenValue = processedChildren.length === 1 ? processedChildren[0] : t.arrayExpression(processedChildren);
     const props = t.objectExpression([t.objectProperty(t.identifier('children'), childrenValue)]);
-    const viewCall = addPureComment(createCreateViewCall(t.identifier(fragmentAlias), props, locInfo, createViewAlias));
+    const viewCall = addPureComment(createCreateViewCall(t.identifier(fragmentAlias), props, locInfo, createViewAlias), ctx);
     if (node.loc)
         viewCall.loc = node.loc;
     path.replaceWith(viewCall);

package/dist/passes/props/index.js

@@ -5,16 +5,9 @@
  */
 import * as t from '@babel/types';
 import { createError } from '../../error.js';
-import { isNativeElement } from '../../utils/index.js';
+import { getJSXElementName, isNativeElement } from '../../utils/index.js';
 import { createProperty, getAttributeValue, processAttribute, processSpreadAttribute } from './attribute.js';
 import { createVModelProps, extractVModelState } from './vmodel.js';
-/**
- * 获取 JSX 元素的名称
- */
-function getJSXElementName(node) {
-    const nameNode = node.openingElement.name;
-    return nameNode.type === 'JSXIdentifier' ? nameNode.name : null;
-}
 /**
  * 校验 class 和 className 不能同时存在
  */

package/dist/transform.d.ts

@@ -1,15 +1,16 @@
-export interface TransformResult {
-    code: string;
-    map: any;
-}
-export interface CompileOptions {
-    hmr: boolean;
-    dev: boolean;
-    ssr: boolean;
-    runtimeModule: string;
-    sourceMap: boolean | 'inline' | 'both';
-    transformClassNameToClass: boolean;
+/**
+ * 主转换模块
+ * 负责解析 JSX/TSX 代码并转换为 createView 调用
+ * @module transform
+ */
+import { type GeneratorResult } from '@babel/generator';
+import type { CompileOptions } from './types.js';
+export interface TransformResult extends GeneratorResult {
 }
+/**
+ * 检查文件是否需要转换
+ */
+export declare function shouldTransform(id: string): boolean;
 /**
  * 转换 JSX/TSX 代码
  */