@vitarx/plugin-vite 0.0.1-beta.7 → 1.0.0-beta.0

package/README.md

@@ -222,6 +222,30 @@ import.meta.hot.accept(mod => {
 2. **被导出** - 使用 `export` 导出
 3. **包含 JSX** - 函数体内包含 JSX 语法或返回编译宏组件
 
+### HMR 代码分离
+
+HMR 在检测到组件更新时，会将组件代码分离为 **UI 代码** 和 **逻辑代码** 两部分：
+
+- **UI 代码**：`createView`、`branch`、`expr`、`accessor`、`dynamic`、`withDirectives` 等调用
+- **逻辑代码**：其余所有代码
+
+仅逻辑代码变化时，组件完全重新挂载；仅 UI 代码变化时，仅重建视图树。
+
+### 开发模式位置信息
+
+开发模式下，`createView` 调用会注入位置信息参数（用于调试），`branch`、`expr`、`accessor` 不注入：
+
+```javascript
+// dev 模式
+createView("div", { children: "hello" }, { fileName: "App.tsx", lineNumber: 5, columnNumber: 10 })
+expr(() => a && b)  // 无位置信息
+branch(() => unref(show) ? 0 : 1)  // 无位置信息
+accessor(props, "value")  // 无位置信息
+
+// 生产模式
+createView("div", { children: "hello" })  // 无位置信息
+```
+
 ## 子元素处理
 
 ### 响应式子元素
@@ -230,19 +254,55 @@ import.meta.hot.accept(mod => {
 // 标识符保持原样
 <div>{value}</div>
 
-// 成员表达式使用 access
+// 成员表达式使用 accessor
 <div>{props.value}</div>
-// 编译为: access(props, 'value')
+// 编译为: accessor(props, 'value')
 
 // 条件表达式使用 branch
 <div>{show ? 'yes' : 'no'}</div>
 // 编译为: branch(() => unref(show) ? 0 : 1, [...])
 
-// 逻辑表达式使用 dynamic
+// 逻辑表达式使用 expr
 <div>{a && b}</div>
-// 编译为: dynamic(() => a && b)
+// 编译为: expr(() => a && b)
+
+// 二元表达式使用 expr
+<div>{count + 1}</div>
+// 编译为: expr(() => count + 1)
+
+// 函数调用使用 expr
+<div>{render()}</div>
+// 编译为: expr(() => render())
 ```
 
+### expr 与数组返回值
+
+`expr` 底层由 `DynamicView` 渲染。当 `expr` 中的表达式返回数组时（如 `arr.map()`），运行时会自动检测并原样返回数组（仅渲染一次，不跟踪响应式）。推荐使用 `For` 组件实现动态数组的响应式渲染。
+
+```jsx
+// arr.map() 返回数组，expr 运行时原样返回（仅渲染一次）
+<div>{items.map(item => <span>{item}</span>)}</div>
+// 编译为: expr(() => items.map(item => createView("span", { children: item })))
+
+// 推荐：使用 For 组件实现响应式数组渲染
+<div>
+  <For each={items} key="id">{item => <span>{item}</span>}</For>
+</div>
+```
+
+### 运行时 API 说明
+
+| API              | 用途      | 编译器注入     | HMR 识别为 UI 代码 |
+|------------------|---------|-----------|---------------|
+| `createView`     | 创建视图节点  | ✅         | ✅             |
+| `branch`         | 条件分支渲染  | ✅         | ✅             |
+| `expr`           | 动态表达式包装 | ✅         | ✅             |
+| `accessor`       | 属性访问追踪  | ✅         | ✅             |
+| `dynamic`        | 动态值包装   | ❌（用户手动使用） | ✅             |
+| `withDirectives` | 指令绑定    | ✅         | ✅             |
+
+> **注意：** `expr` 是编译器插桩注入的辅助函数，用于包装逻辑/二元/调用表达式；`dynamic` 是运行时独立 API，供用户手动使用。两者在 HMR 代码分离时均被视为 UI 描述代码。
+
 ## API 参考
 
 ### 插件选项

package/dist/constants/index.d.ts

@@ -2,21 +2,11 @@
  * Vitarx 运行时 API 名称
  * 这些 API 需要从 vitarx 包导入
  */
-export declare const VITARX_APIS: {
-    readonly CREATE_VIEW: "createView";
-    readonly FRAGMENT: "Fragment";
-    readonly BRANCH: "branch";
-    readonly DYNAMIC: "dynamic";
-    readonly ACCESS: "access";
-    readonly WITH_DIRECTIVES: "withDirectives";
-    readonly UNREF: "unref";
-    readonly IS_REF: "isRef";
-};
 /**
  * UI 相关的 API 名称列表
  * 这些 API 用于创建视图，在 HMR 时需要识别
  */
-export declare const UI_API_NAMES: readonly ["createView", "branch", "dynamic", "access", "withDirectives"];
+export declare const UI_API_NAMES: readonly ["createView", "branch", "dynamic", "expr", "accessor", "withDirectives"];
 /**
  * Ref 相关 API 名称
  * 用于识别 ref 变量

package/dist/constants/index.js

@@ -2,21 +2,11 @@
  * Vitarx 运行时 API 名称
  * 这些 API 需要从 vitarx 包导入
  */
-export const VITARX_APIS = {
-    CREATE_VIEW: 'createView',
-    FRAGMENT: 'Fragment',
-    BRANCH: 'branch',
-    DYNAMIC: 'dynamic',
-    ACCESS: 'access',
-    WITH_DIRECTIVES: 'withDirectives',
-    UNREF: 'unref',
-    IS_REF: 'isRef'
-};
 /**
  * UI 相关的 API 名称列表
  * 这些 API 用于创建视图，在 HMR 时需要识别
  */
-export const UI_API_NAMES = ['createView', 'branch', 'dynamic', 'access', 'withDirectives'];
+export const UI_API_NAMES = ['createView', 'branch', 'dynamic', 'expr', 'accessor', 'withDirectives'];
 /**
  * Ref 相关 API 名称
  * 用于识别 ref 变量

package/dist/context.d.ts

@@ -3,9 +3,9 @@
  * 定义编译转换过程中的上下文数据结构
  * @module context
  */
-import type { File } from '@babel/types';
+import type { CallExpression, File } from '@babel/types';
 import type { CompilerWarning } from './error.js';
-import type { CompileOptions } from './transform.js';
+import type { CompileOptions } from './types.js';
 /**
  * 导入信息
  * 追踪需要注入的运行时 API
@@ -17,10 +17,10 @@ export interface ImportInfo {
     Fragment: boolean;
     /** branch - 条件分支函数 */
     branch: boolean;
-    /** dynamic - 动态值包装函数 */
-    dynamic: boolean;
-    /** access - 属性访问函数 */
-    access: boolean;
+    /** expr - 动态表达式包装函数 */
+    expr: boolean;
+    /** accessor - 属性访问函数 */
+    accessor: boolean;
     /** withDirectives - 指令包装函数 */
     withDirectives: boolean;
     /** unref - ref 解包函数 */
@@ -36,8 +36,8 @@ export interface VitarxImportAliases {
     createView: string | null;
     Fragment: string | null;
     branch: string | null;
-    dynamic: string | null;
-    access: string | null;
+    expr: string | null;
+    accessor: string | null;
     withDirectives: string | null;
     unref: string | null;
     isRef: string | null;
@@ -78,8 +78,12 @@ export interface TransformContext {
     refApiAliases: RefApiAliases;
     /** ref 变量名集合 */
     refVariables: Set<string>;
-    /** UI API 别名集合（用于 HMR 代码分离识别） */
-    uiApiAliases: Set<string>;
+    /** builder 函数的本地别名（用于识别纯构建组件） */
+    builderAlias: string | null;
+    /** 已处理的 AST 节点追踪集合 */
+    processedNodes: WeakSet<any>;
+    /** 已添加 PURE 注释的节点追踪集合 */
+    pureCommentedNodes: WeakSet<CallExpression>;
     /** 编译警告列表 */
     warnings: CompilerWarning[];
 }

package/dist/context.js

@@ -18,8 +18,8 @@ export function createContext(code, id, options, ast) {
             createView: false,
             Fragment: false,
             branch: false,
-            dynamic: false,
-            access: false,
+            expr: false,
+            accessor: false,
             withDirectives: false,
             unref: false,
             isRef: false
@@ -29,8 +29,8 @@ export function createContext(code, id, options, ast) {
             createView: null,
             Fragment: null,
             branch: null,
-            dynamic: null,
-            access: null,
+            expr: null,
+            accessor: null,
             withDirectives: null,
             unref: null,
             isRef: null
@@ -43,7 +43,9 @@ export function createContext(code, id, options, ast) {
             computed: null
         },
         refVariables: new Set(),
-        uiApiAliases: new Set(),
+        builderAlias: null,
+        processedNodes: new WeakSet(),
+        pureCommentedNodes: new WeakSet(),
         warnings: []
     };
 }

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);

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

@@ -11,7 +11,7 @@ export interface ChangeCode {
  * 判断两个函数组件的代码差异
  *
  * 分析策略：
- * 1. 解析函数代码，提取所有 createView/branch/dynamic 等 UI API 调用
+ * 1. 解析函数代码，提取所有 createView/branch/expr 等 UI API 调用
  * 2. 将这些调用识别为 UI 描述代码
  * 3. 其余代码识别为非 UI 代码（逻辑代码）
  * 4. 分别比较 UI 代码和非 UI 代码是否变化

package/dist/hmr-client/utils.js

@@ -13,13 +13,16 @@ import { UI_API_NAMES } from '../constants/index.js';
 const UI_APIS = new Set([...UI_API_NAMES, 'jsxDEV', 'h']);
 /**
  * 判断标识符名称是否为 UI API
- * 支持原始名称和带后缀的名称（如 jsxDEV$1）
+ * 支持原始名称和带后缀的名称（如 jsxDEV$1、branch$2、expr$1）
  */
 function isUIApi(name) {
     if (UI_APIS.has(name))
         return true;
-    // 支持带数字后缀的 createView
-    return /^jsxDEV\$\d+$/.test(name) || /^createView\$\d+$/.test(name);
+    for (const api of UI_APIS) {
+        if (name.startsWith(api + '$'))
+            return true;
+    }
+    return false;
 }
 /**
  * 从函数代码中分离 UI 代码和非 UI 代码
@@ -100,7 +103,7 @@ function normalizeCode(code) {
  * 判断两个函数组件的代码差异
  *
  * 分析策略：
- * 1. 解析函数代码，提取所有 createView/branch/dynamic 等 UI API 调用
+ * 1. 解析函数代码，提取所有 createView/branch/expr 等 UI API 调用
  * 2. 将这些调用识别为 UI 描述代码
  * 3. 其余代码识别为非 UI 代码（逻辑代码）
  * 4. 分别比较 UI 代码和非 UI 代码是否变化

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,10 +32,10 @@ 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;
-            isSSR = env.isSsrBuild === true || configSSR;
+            isSSR = env.isSsrBuild === true;
             return {
                 define: {
                     __VITARX_DEV__: JSON.stringify(isDEV),
@@ -52,7 +52,11 @@ export default function vitarx(options) {
             };
         },
         configResolved(config) {
-            viteConfig = config;
+            // 根据最终的配置结果，确定是否是 SSR 模式
+            if (!isSSR) {
+                isSSR = !!config.build?.ssr;
+                config.define['__VITARX_SSR__'] = JSON.stringify(isSSR);
+            }
             compileOptions = {
                 dev: isDEV,
                 ssr: isSSR,
@@ -63,6 +67,8 @@ export default function vitarx(options) {
             };
         },
         async transform(code, id) {
+            if (!shouldTransform(id))
+                return null;
             const result = await transform(code, id, compileOptions);
             if (!result)
                 return null;

package/dist/passes/components/ifBlock.js

@@ -1,6 +1,6 @@
 import { isJSXElement } from '@babel/types';
 import { createError } from '../../error.js';
-import { collectVIfChainInfo, createArrowFunction, createBranch, filterWhitespaceChildren, getDevLocInfo, hasDirective, removeVIfChainDirectives, validateVIfChain } from '../../utils/index.js';
+import { collectVIfChainInfo, createArrowFunction, createBranch, filterWhitespaceChildren, hasDirective, removeVIfChainDirectives, validateVIfChain } from '../../utils/index.js';
 /**
  * 处理 IfBlock 组件
  */
@@ -15,9 +15,7 @@ export function processIfBlock(path, ctx) {
     const { nodes, conditions } = collectVIfChainInfo(jsxChildren);
     nodes.forEach(node => removeVIfChainDirectives(node));
     const branches = nodes.map(node => createArrowFunction(node));
-    // 获取位置信息（仅开发环境）
-    const locInfo = getDevLocInfo(ctx, path.node);
-    const branchCall = createBranch({ conditions, branches, locInfo }, ctx);
+    const branchCall = createBranch({ conditions, branches }, ctx);
     if (path.node.loc) {
         branchCall.loc = path.node.loc;
     }

package/dist/passes/components/switch.js

@@ -1,7 +1,7 @@
 import * as t from '@babel/types';
 import { isJSXElement, isJSXText } from '@babel/types';
 import { createError } from '../../error.js';
-import { createArrowFunction, createBranch, getDevLocInfo, getJSXAttributeByName, getJSXElementName } from '../../utils/index.js';
+import { createArrowFunction, createBranch, getJSXAttributeByName, getJSXElementName } from '../../utils/index.js';
 /**
  * 处理 Switch 组件
  * @param path - JSX 元素路径
@@ -20,11 +20,9 @@ export function processSwitch(path, ctx) {
         conditions.push(t.booleanLiteral(true));
         branches.push(createArrowFunction(fallbackValue));
     }
-    // 获取位置信息（仅开发环境）
-    const locInfo = getDevLocInfo(ctx, path.node);
     // 生成 branch 调用
     // Switch 的 when 条件如果是标识符，需要 unref
-    const branchCall = createBranch({ conditions, branches, locInfo }, ctx);
+    const branchCall = createBranch({ conditions, branches }, ctx);
     if (path.node.loc) {
         branchCall.loc = path.node.loc;
     }

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/directives/vIf.js

@@ -1,6 +1,6 @@
 import * as t from '@babel/types';
 import { isJSXElement, isJSXText } from '@babel/types';
-import { collectFragmentVIfChains, createArrowFunction, createBranch, getDevLocInfo, isWhitespaceJSXText, removeVIfChainDirectives } from '../../utils/index.js';
+import { collectFragmentVIfChains, createArrowFunction, createBranch, isWhitespaceJSXText, removeVIfChainDirectives } from '../../utils/index.js';
 /**
  * 处理 Fragment 中的 v-if 链
  * @param path - JSX Fragment 路径
@@ -42,11 +42,8 @@ function processVIfChainItem(chain, children, ctx, transformJSXElement) {
             branches.push(createArrowFunction(transformedNode));
         }
     }
-    // 获取第一个节点的位置信息（仅开发环境）
-    const firstNode = nodes[0];
-    const locInfo = getDevLocInfo(ctx, firstNode);
     // 生成 branch 调用
-    const branchCall = createBranch({ conditions, branches, locInfo }, ctx);
+    const branchCall = createBranch({ conditions, branches }, ctx);
     // 替换链的第一个节点，其他节点标记为 null
     if (start >= 0) {
         children[start] = branchCall;

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), [t.booleanLiteral(true)]))
+    ]));
+    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), [t.booleanLiteral(true)]))
+    ]));
+    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 支持