@lcap/nasl 3.3.3-beta.6 → 3.3.3-beta.7

package/src/concepts/Attribute__.ts

@@ -189,9 +189,12 @@ export class Attribute extends BaseNode {
         const index = (t as any)[parentKey].findIndex((el) => (el.__v_raw || el) === n);
         ~index && (this as any)[parentKey].splice(index, 1);
         params.index = index;
-      } else if ((t as any)[parentKey] === n) {
-        params.index = 0;
-        (this as any)[parentKey] = undefined;
+      } else {
+        const q = (t as any)[parentKey];
+        if ((q.__v_raw || q) === n) {
+          params.index = 0;
+          (this as any)[parentKey] = undefined;
+        }
       }
     }
     return params;

package/src/concepts/AuthLogicForCallInterface__.ts

@@ -1,3 +1,8 @@
+import {
+  wrapForEachToGenerator as wrapForEach,
+  wrapIteratorToGenerator as wrapIterator,
+  returnOrigin,
+} from '../utils/time-slicing';
 import {
   TranslatorState,
   shiftState,
@@ -5,7 +10,11 @@ import {
   indent,
   createCompilerState,
   embeddedTSEmitter,
+  withSourceMapGenerator,
+  withGenerator,
+  TranslatorGenerator,
 } from '../translator';
+import { waitOqlQueryComponentChildrenFinish } from '../common';
 import Namespace from './Namespace__';
 import CallQueryComponent from './CallQueryComponent__';
 import Assignment from './Assignment__';
@@ -264,9 +273,12 @@ export class AuthLogicForCallInterface extends AuthLogic {
         const index = (t as any)[parentKey].findIndex((el) => (el.__v_raw || el) === n);
         ~index && (this as any)[parentKey].splice(index, 1);
         params.index = index;
-      } else if ((t as any)[parentKey] === n) {
-        params.index = 0;
-        (this as any)[parentKey] = undefined;
+      } else {
+        const q = (t as any)[parentKey];
+        if ((q.__v_raw || q) === n) {
+          params.index = 0;
+          (this as any)[parentKey] = undefined;
+        }
       }
     }
     return params;
@@ -277,20 +289,23 @@ export class AuthLogicForCallInterface extends AuthLogic {
   // 自动生成的代码已结束。下面可以手动编写。
   //================================================================================
 
-  @withSourceMap()
-  toEmbeddedTS(state?: TranslatorState, prefix?: string): string {
+  @withSourceMapGenerator
+  *toEmbeddedTS(state = createCompilerState(), prefix?: string): TranslatorGenerator {
+    const self = this;
+    // 等待 oql 查询返回内容
+    yield waitOqlQueryComponentChildrenFinish(self);
     // 如果传了不要前缀那就为空， 要不用传进来的，要是都没有就默认导出
     let code = '';
     if (!prefix) {
       // T、U分别用来定义 api 入参和返回值
-      code += `export function ${this.name}`;
+      code += `export function ${self.name}`;
     } else if (prefix === 'noExport') {
-      code += `function ${this.name}`;
+      code += `function ${self.name}`;
     } else if (prefix === 'inObject') {
-      code += `${this.name}`;
+      code += `${self.name}`;
     }
     const typeParams = [
-      ...(this.typeParams || []),
+      ...(self.typeParams || []),
       new TypeParam({
         name: 'ApiParams',
       }),
@@ -300,8 +315,8 @@ export class AuthLogicForCallInterface extends AuthLogic {
     ];
     if (typeParams?.length) {
       code += '<';
-      typeParams.forEach((typeParam, typeParamIndex) => {
-        code += typeParam.toEmbeddedTS(shiftState(state, code, { inline: true }));
+      yield* wrapForEach(typeParams, function* warpForEachGenerator(typeParam, typeParamIndex) {
+        code += yield* typeParam.toEmbeddedTS(shiftState(state, code, { inline: true }));
         if (typeParamIndex !== typeParams.length - 1) {
           code += ', ';
         }
@@ -309,10 +324,10 @@ export class AuthLogicForCallInterface extends AuthLogic {
       code += '>';
     }
     // 有参数长度 && 鉴权参数
-    if (this.authParams.length) {
+    if (self.authParams.length) {
       // const params = [...this.authParams, ...this.params];
-      const params = [...this.authParams];
-      params.forEach((param, index) => {
+      const params = [...self.authParams];
+      yield* wrapForEach(params, function* warpForEachGenerator(param, index) {
         if (index === 0) {
           code += '({';
         }
@@ -323,11 +338,11 @@ export class AuthLogicForCallInterface extends AuthLogic {
           code += ', api}:';
         }
       });
-      params.forEach((param, index) => {
+      yield* wrapForEach(params, function* warpForEachGenerator(param, index) {
         if (index === 0) {
           code += '{';
         }
-        code += param.toEmbeddedTS(shiftState(state, code), !param.required);
+        code += yield* param.toEmbeddedTS(shiftState(state, code), !param.required);
         if (index !== params.length - 1) {
           code += ', ';
         } else {
@@ -339,79 +354,53 @@ export class AuthLogicForCallInterface extends AuthLogic {
       // 没有参数的
       code += '({api}: {api?: ApiReturn}): ApiReturn';
     }
-
     // 需要类型推导的局部变量/返回值需要调整申明顺序
     const advanceMap: Map<Return | Variable, Assignment | BatchAssignment> = new Map();
-    const needGetSourceMapOqlList: Promise<any>[] = [];
-    this.traverseChildren((el) => {
+    // const needGetSourceMapOqlList: Promise<any>[] = [];
+    yield* self.traverseChildrenGenerator(function* traverseChildrenGenerator(el) {
       if (el && (el instanceof BatchAssignment || (el instanceof Assignment && el.left?.name))) {
         if (el instanceof Assignment) {
-          const advanceVar = this.variables?.find(
+          const advanceVar = self.variables?.find(
             (variable) => !variable.typeAnnotation && el.left?.name === variable.name,
           );
           if (advanceVar && !advanceMap.get(advanceVar)) {
             advanceMap.set(advanceVar, el);
           }
-          const advanceRn = this.returns?.find((ret) => !ret.typeAnnotation && el.left?.name === ret.name);
+          const advanceRn = self.returns?.find((ret) => !ret.typeAnnotation && el.left?.name === ret.name);
           if (advanceRn && !advanceMap.get(advanceRn)) {
             advanceMap.set(advanceRn, el);
           }
         } else if (el instanceof BatchAssignment) {
-          el.assignmentLines.forEach(({ leftIndex }) => {
+          yield* wrapForEach(el.assignmentLines, function* warpForEachGenerator({ leftIndex }) {
             const leftCode =
               leftIndex.length === 1
-                ? el.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))
-                : el.left.members[leftIndex[1]]?.toEmbeddedTS(shiftState(state, code, { inline: true }));
-            const advanceVar = this.variables?.find(
+                ? yield* el.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))
+                : yield* el.left.members[leftIndex[1]]?.toEmbeddedTS(shiftState(state, code, { inline: true })) ?? returnOrigin('');
+            const advanceVar = self.variables?.find(
               (variable) => !variable.typeAnnotation && leftCode === variable.name,
             );
             if (advanceVar && !advanceMap.get(advanceVar)) {
               advanceMap.set(advanceVar, el);
             }
-            const advanceRn = this.returns?.find((ret) => !ret.typeAnnotation && leftCode === ret.name);
+            const advanceRn = self.returns?.find((ret) => !ret.typeAnnotation && leftCode === ret.name);
             if (advanceRn && !advanceMap.get(advanceRn)) {
               advanceMap.set(advanceRn, el);
             }
           });
         }
       }
-      // 如果有sql查询组件就需要把他们存起来
-      if (el instanceof OqlQueryComponent) {
-        if (!el.codeSourceMap) {
-          needGetSourceMapOqlList.push(el.requestEmbeddedTS());
-        }
-      }
     });
-
-    if (needGetSourceMapOqlList.length) {
-      Promise.all(needGetSourceMapOqlList).then(() => {
-        // 触发一下更新重新生成代码
-        (this.app as any)?.naslServer?.embeddedTSEmitter.emit('change', {
-          value: [
-            {
-              originEvent: {
-                action: 'update',
-                target: this,
-              },
-            },
-          ],
-        });
-      });
-    }
-
     code += '{\n';
-
     // 用于虚拟节点的查找引用
-    this.virtualParams &&
-      Array.isArray(this.virtualParams) &&
-      this.virtualParams.forEach((param) => {
+    if (Array.isArray(self.virtualParams)) {
+      for (const param of self.virtualParams) {
         code += indent((state?.tabSize || 0) + 1);
-        code += param.toEmbeddedNameTS(shiftState(state, code, { inline: true }));
+        code += yield* param.toEmbeddedNameTS(shiftState(state, code, { inline: true }));
         code += '\n';
-      });
-
+      }
+    }
     const variables = [
-      ...this.variables,
+      ...self.variables,
       new Variable({
         // 将默认参数加为变量 用于参数选择校验
         name: 'request',
@@ -431,24 +420,23 @@ export class AuthLogicForCallInterface extends AuthLogic {
         }),
       }),
     ];
-    variables.forEach((variable) => {
+    yield* wrapForEach(variables, function* warpForEachGenerator(variable) {
       if (!advanceMap.get(variable)) {
         code += `${indent((state?.tabSize || 0) + 1)}let `;
-        code += variable.toEmbeddedTS(shiftState(state, code, { inline: true }));
+        code += yield* variable.toEmbeddedTS(shiftState(state, code, { inline: true }));
         code += ';\n';
       }
     });
-
-    advanceMap.forEach((assignment, node) => {
+    yield* wrapIterator(advanceMap.entries(), function* warpForEachGenerator([node, assignment]) {
       code += `${indent((state?.tabSize || 0) + 1)}// @ts-ignore\n`;
       code += `${indent((state?.tabSize || 0) + 1)}let `;
-      code += node.toEmbeddedTS(shiftState(state, code, { inline: true }));
+      code += yield* node.toEmbeddedTS(shiftState(state, code, { inline: true }));
       if (assignment instanceof BatchAssignment) {
-        assignment.assignmentLines.forEach(({ leftIndex, rightIndex }) => {
+        yield* wrapForEach(assignment.assignmentLines, function* warpForEachGenerator({ leftIndex, rightIndex }) {
           const leftCode =
             leftIndex.length === 1
-              ? assignment.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))
-              : assignment.left.members[leftIndex[1]]?.toEmbeddedTS(shiftState(state, code, { inline: true }));
+              ? yield* assignment.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))
+              : yield* assignment.left.members[leftIndex[1]]?.toEmbeddedTS(shiftState(state, code, { inline: true })) ?? returnOrigin('');
           if (leftCode === node.name) {
             const rightNode: any =
               rightIndex.length === 1
@@ -460,11 +448,11 @@ export class AuthLogicForCallInterface extends AuthLogic {
               !(rightNode instanceof CallQueryComponent && rightNode.ideVersion !== '2.11')
             ) {
               code += ': ';
-              code += rightNode.typeAnnotation?.toEmbeddedTS(shiftState(state, code, { inline: true })) || 'null';
+              code += yield* rightNode?.typeAnnotation?.toEmbeddedTS(shiftState(state, code, { inline: true })) ?? returnOrigin('null');
             } else {
               code += ' = ';
-              const rightCode = rightNode?.toEmbeddedTS(shiftState(state, code, { inline: true }));
-              code += this.getRightCode(rightCode, assignment, rightNode);
+              const rightCode = yield* rightNode?.toEmbeddedTS(shiftState(state, code, { inline: true })) ?? returnOrigin('');;
+              code += yield* self.getRightCode(rightCode, assignment, rightNode);
             }
           }
         });
@@ -474,14 +462,14 @@ export class AuthLogicForCallInterface extends AuthLogic {
       ) {
         const rightNode: any = assignment.right;
         code += ' = ';
-        code += rightNode.getInitCode(shiftState(state, code, { inline: true }));
+        code += yield* rightNode.getInitCode(shiftState(state, code, { inline: true }));
       } else if (
         assignment.right?.concept === 'NewComposite' &&
         (assignment.right as any)?.typeAnnotation?.typeKind === 'anonymousStructure'
       ) {
         const rightNode: any = assignment.right;
         code += ' = ';
-        code += rightNode.getInitCode(shiftState(state, code, { inline: true }));
+        code += yield* rightNode.getInitCode(shiftState(state, code, { inline: true }));
       } else {
         const rightNode: any = assignment.right;
         // 当右边有明确类型时，直接使用其类型。旧版本数据查询不能直接使用其 typeAnnotation
@@ -490,7 +478,7 @@ export class AuthLogicForCallInterface extends AuthLogic {
           !(rightNode instanceof CallQueryComponent && rightNode.ideVersion !== '2.11')
         ) {
           code += ': ';
-          code += rightNode.typeAnnotation?.toEmbeddedTS(shiftState(state, code, { inline: true })) || 'null';
+          code += yield* rightNode.typeAnnotation?.toEmbeddedTS(shiftState(state, code, { inline: true })) ?? returnOrigin('null');
         } else {
           code += ' = ';
           let rightCode = '';
@@ -503,39 +491,33 @@ export class AuthLogicForCallInterface extends AuthLogic {
           ) {
             rightCode += 'new ';
           }
-          rightCode +=
-            rightNode && rightNode.toEmbeddedTS
-              ? rightNode.toEmbeddedTS(shiftState(state, code, { inline: true }))
-              : '__RIGHT__';
-          code += this.getRightCode(rightCode, assignment, rightNode);
+          rightCode += yield* rightNode?.toEmbeddedTS?.(shiftState(state, code, { inline: true })) ?? returnOrigin('__RIGHT__');
+          code += yield* self.getRightCode(rightCode, assignment, rightNode);
         }
       }
       code += ';\n';
     });
-
-    if (!this.body.length) {
+    if (!self.body.length) {
       // 必有返回值，且返回值为 api 对应的返回参数
-      code += `${indent((state?.tabSize || 0) + 1)}let ${this.name}Result: ApiReturn; \n ${indent(
+      code += `${indent((state?.tabSize || 0) + 1)}let ${self.name}Result: ApiReturn; \n ${indent(
         (state?.tabSize || 0) + 1,
-      )}return ${this.name}Result;\n`;
+      )}return ${self.name}Result;\n`;
     } else {
       // 如果长度小于2 并且不是虚拟节点就 加 报错，
-      if (this.body.length <= 2 && !(this.parentNode instanceof Namespace)) {
+      if (self.body.length <= 2 && !(self.parentNode instanceof Namespace)) {
         code += indent((state?.tabSize || 0) + 1);
         code += 'const __LogicEmpty: any = null;';
       }
-      this.body.forEach((item) => {
-        code += item.toEmbeddedTS && item.toEmbeddedTS(shiftState(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
+      yield* wrapForEach(self.body, function* warpForEachGenerator(item) {
+        code += yield* item.toEmbeddedTS?.(shiftState(state, code, { tabSize: (state?.tabSize || 0) + 1 })) ?? returnOrigin('') ?? returnOrigin('');
         code += ';\n';
       });
     }
-
     // 在body渲染完成之后把其余的节点放到下面
-    this.playground.forEach((item) => {
-      code += item.toEmbeddedTS(shiftState(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
+    yield* wrapForEach(self.playground, function* warpForEachGenerator(item) {
+      code += yield* item.toEmbeddedTS(shiftState(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
       code += ';\n';
     });
-
     code += `${indent(state?.tabSize || 0)}}`;
     if (prefix === 'inObject') {
       code += `,`;

package/src/concepts/BackendVariable__.ts

@@ -1,5 +1,15 @@
+import type { EmbeddedTSFileGenerator } from '../utils';
 import { dataTypesMap } from './basics/types';
-import { TranslatorState, shiftState, withSourceMap, indent, createCompilerState } from '../translator';
+import {
+  TranslatorState,
+  shiftState,
+  withSourceMap,
+  indent,
+  createCompilerState,
+  withSourceMapGenerator,
+  withGenerator,
+  TranslatorGenerator,
+} from '../translator';
 import { ElementToVueOptions } from './ViewElement__';
 import EntityProperty from './EntityProperty__';
 import StructureProperty from './StructureProperty__';
@@ -233,9 +243,12 @@ export class BackendVariable extends BaseNode {
         const index = (t as any)[parentKey].findIndex((el) => (el.__v_raw || el) === n);
         ~index && (this as any)[parentKey].splice(index, 1);
         params.index = index;
-      } else if ((t as any)[parentKey] === n) {
-        params.index = 0;
-        (this as any)[parentKey] = undefined;
+      } else {
+        const q = (t as any)[parentKey];
+        if ((q.__v_raw || q) === n) {
+          params.index = 0;
+          (this as any)[parentKey] = undefined;
+        }
       }
     }
     return params;
@@ -313,18 +326,19 @@ export class BackendVariable extends BaseNode {
     return code;
   }
 
-  @withSourceMap()
-  toEmbeddedTS(state = createCompilerState()): string {
-    let code = this.name;
-    if (this.typeAnnotation) {
+  @withSourceMapGenerator
+  *toEmbeddedTS(state = createCompilerState()): TranslatorGenerator {
+    const self = this;
+    let code = self.name;
+    if (self.typeAnnotation) {
       code += ': ';
-      code += this.typeAnnotation.toEmbeddedTS(shiftState(state, code));
+      code += yield* self.typeAnnotation.toEmbeddedTS(shiftState(state, code));
       // toTS暂时只需要翻译引用数据类型的，因为值都是字符串
       // 就不在这里赋值了，这里单纯就为了快速查找
       // 并且屏蔽了一些异常的错误
-      if (this.typeAnnotation?.typeKind !== 'primitive' && this.defaultValue) {
-        if (this.isEnum()) {
-          code += `= ${this.typeAnnotation.typeKey}['${this.defaultValue}']`;
+      if (self.typeAnnotation?.typeKind !== 'primitive' && self.defaultValue) {
+        if (self.isEnum()) {
+          code += `= ${self.typeAnnotation.typeKey}['${self.defaultValue}']`;
         }
       }
     }
@@ -363,12 +377,12 @@ export class BackendVariable extends BaseNode {
     throw new Error('无法获取命名空间，请设置 parentNode！');
   }
 
-  toEmbeddedTSFile() {
+  *toEmbeddedTSFile(): EmbeddedTSFileGenerator {
     let code = `namespace ${this.getNamespace()} {\n`;
     code += `${indent(1)}export let `;
     const state = createCompilerState(code, { tabSize: 1 });
     try {
-      code += `${this.toEmbeddedTS(state)}\n`;
+      code += `${yield* this.toEmbeddedTS(state)}\n`;
     } catch (err) {
       code += '\n';
     }
@@ -381,12 +395,13 @@ export class BackendVariable extends BaseNode {
     };
   }
 
-  @withSourceMap()
-  toEmbeddedTSInProcess(state?: TranslatorState): string {
-    let code = this.name;
-    if (this.typeAnnotation) {
+  @withSourceMapGenerator
+  *toEmbeddedTSInProcess(state = createCompilerState()): TranslatorGenerator {
+    const self = this;
+    let code = self.name;
+    if (self.typeAnnotation) {
       code += ': undefined as ';
-      code += this.typeAnnotation.toEmbeddedTS(shiftState(state, code));
+      code += yield* self.typeAnnotation.toEmbeddedTS(shiftState(state, code));
     }
     // if (this.defaultValue) {
     //     code += `: ${this.defaultValue}`;

package/src/concepts/Backend__.ts

@@ -1,5 +1,21 @@
+import type { EmbeddedTSFileGenerator } from '../utils';
 import Namespace from './Namespace__';
-import { TranslatorState, shiftState, withSourceMap, indent, createCompilerState } from '../translator';
+import {
+  TranslatorState,
+  shiftState,
+  withSourceMap,
+  indent,
+  createCompilerState,
+  withSourceMapGenerator,
+  withGenerator,
+  TranslatorGenerator,
+} from '../translator';
+
+interface TsFileData {
+  code: string;
+  filePath: string;
+  sourceMap: any;
+}
 
 //================================================================================
 // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
@@ -92,7 +108,7 @@ export class Backend extends BaseNode {
     const excludedSet = new Set(excludedList);
     return ((this.variables as BackendVariable[]) || [])
       .filter((item) => !excludedSet.has(item))
-      .map((item) => item.name);
+      .map((item) => item?.name);
   }
   getBackendVariableUniqueName(name = 'backendVariable1') {
     return utils.unique(name, this.getVarExistingNames(), undefined, true);
@@ -284,12 +300,16 @@ export class Backend extends BaseNode {
       // @ts-ignore
       const n = backendVariable.__v_raw || backendVariable;
       if (Array.isArray((t as any)[parentKey])) {
-        const index = (t as any)[parentKey].indexOf(n);
+        // @ts-ignore
+        const index = (t as any)[parentKey].findIndex((el) => (el.__v_raw || el) === n);
         ~index && (this as any)[parentKey].splice(index, 1);
         params.index = index;
-      } else if ((t as any)[parentKey] === n) {
-        params.index = 0;
-        (this as any)[parentKey] = undefined;
+      } else {
+        const q = (t as any)[parentKey];
+        if ((q.__v_raw || q) === n) {
+          params.index = 0;
+          (this as any)[parentKey] = undefined;
+        }
       }
     }
     return params;
@@ -320,11 +340,12 @@ export class Backend extends BaseNode {
     return `/embedded/${this.rootNode?.name || this.parentNode.name}/Backend/backend.ts`;
   }
 
-  @withSourceMap()
-  toEmbeddedTS(state = createCompilerState()): string {
-    let code = `export namespace ${this.name} {\n`;
+  @withSourceMapGenerator
+  *toEmbeddedTS(state = createCompilerState()): TranslatorGenerator {
+    const self = this;
+    let code = `export namespace ${self.name} {\n`;
     code += 'const name = ';
-    code += `"${this.name}";\n`;
+    code += `"${self.name}";\n`;
     code += '}\n';
     return code;
   }
@@ -334,12 +355,12 @@ export class Backend extends BaseNode {
   //     const finalPath = path && path !== '/' ? path : '';
   //     return finalPath;
   // }
-  toEmbeddedTSFile() {
+  *toEmbeddedTSFile(): EmbeddedTSFileGenerator {
     let code = `namespace ${this.getNamespace()} {\n`;
 
     const state = createCompilerState(code, { tabSize: 1 });
     try {
-      code += this.toEmbeddedTS(state);
+      code += yield* this.toEmbeddedTS(state);
     } catch (err) {
       code += '';
       console.log(err);

package/src/concepts/BatchAssignment__.ts

@@ -1,4 +1,13 @@
-import { TranslatorState, shiftState, withSourceMap, createCompilerState } from '../translator';
+import { wrapForEachToGenerator as wrapForEach, returnOrigin } from '../utils/time-slicing';
+import {
+  TranslatorState,
+  shiftState,
+  withSourceMap,
+  createCompilerState,
+  withSourceMapGenerator,
+  withGenerator,
+  TranslatorGenerator,
+} from '../translator';
 import { ElementToVueOptions } from './ViewElement__';
 import MemberExpression from './MemberExpression__';
 import Assignment from './Assignment__';
@@ -507,9 +516,12 @@ export class BatchAssignment extends LogicItem {
         const index = (t as any)[parentKey].findIndex((el) => (el.__v_raw || el) === n);
         ~index && (this as any)[parentKey].splice(index, 1);
         params.index = index;
-      } else if ((t as any)[parentKey] === n) {
-        params.index = 0;
-        (this as any)[parentKey] = undefined;
+      } else {
+        const q = (t as any)[parentKey];
+        if ((q.__v_raw || q) === n) {
+          params.index = 0;
+          (this as any)[parentKey] = undefined;
+        }
       }
     }
     return params;
@@ -560,9 +572,12 @@ export class BatchAssignment extends LogicItem {
         const index = (t as any)[parentKey].findIndex((el) => (el.__v_raw || el) === n);
         ~index && (this as any)[parentKey].splice(index, 1);
         params.index = index;
-      } else if ((t as any)[parentKey] === n) {
-        params.index = 0;
-        (this as any)[parentKey] = undefined;
+      } else {
+        const q = (t as any)[parentKey];
+        if ((q.__v_raw || q) === n) {
+          params.index = 0;
+          (this as any)[parentKey] = undefined;
+        }
       }
     }
     return params;
@@ -1009,47 +1024,50 @@ export class BatchAssignment extends LogicItem {
     return code;
   }
 
-  @withSourceMap()
-  toEmbeddedTS(state = createCompilerState()): string {
+  @withSourceMapGenerator
+  *toEmbeddedTS(state = createCompilerState()): TranslatorGenerator {
+    const self = this;
     let code = '// 以下是批量赋值的翻译代码\n';
     // 提供sourcemap，获取类型标注和上下文可选变量
-    code += this.left.expression
-      ? `${this.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))}`
+    code += self.left.expression
+      ? `${yield* self.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))}`
       : '';
     code += ';\n';
-    this.left.members.forEach((member) => {
-      code += member.toEmbeddedTS(shiftState(state, code, { inline: true }));
+    yield* wrapForEach(self.left.members, function* warpForEachGenerator(member) {
+      code += yield* member.toEmbeddedTS(shiftState(state, code, { inline: true }));
       code += '; ';
     });
     code += '\n';
-    this.rights.forEach((right) => {
-      code += `${right.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))}`;
+    yield* wrapForEach(self.rights, function* warpForEachGenerator(right) {
+      code += `${yield* right.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))}`;
       code += ';\n';
-      right.members.forEach((member) => {
-        code += member.toEmbeddedTS(shiftState(state, code, { inline: true }));
+      yield* wrapForEach(right.members, function* warpForEachGenerator(member) {
+        code += yield* member.toEmbeddedTS(shiftState(state, code, { inline: true }));
         code += '; ';
       });
       code += right.members.length ? '\n' : '';
     });
     // 根据连线状态，拆分成多个单赋值
-    this.assignmentLines.forEach(({ leftIndex, rightIndex }) => {
+    yield* wrapForEach(self.assignmentLines, function* warpForEachGenerator({ leftIndex, rightIndex }) {
       code += 'nasl.core.assign(';
       const leftValue =
         leftIndex.length === 1
-          ? this.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))
-          : this.left.members[leftIndex[1]]?.toEmbeddedTS(shiftState(state, code, { inline: true }));
+          ? yield* self.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))
+          : yield* self.left.members[leftIndex[1]]?.toEmbeddedTS(shiftState(state, code, { inline: true })) ?? returnOrigin('');
       code += `${leftValue}`;
       code += ', ';
       const rightValue =
         rightIndex.length === 1
-          ? this.rights[rightIndex[0]]?.expression?.toEmbeddedTS(shiftState(state, code, { inline: true }))
-          : this.rights[rightIndex[0]]?.members[rightIndex[1]]?.toEmbeddedTS(shiftState(state, code, { inline: true }));
+          ? yield* self.rights[rightIndex[0]]?.expression?.toEmbeddedTS(shiftState(state, code, { inline: true })) ?? returnOrigin('')
+          : yield* self.rights[rightIndex[0]]?.members[rightIndex[1]]?.toEmbeddedTS(
+              shiftState(state, code, { inline: true }),
+            ) ?? returnOrigin('');
       code += `${rightValue}`;
       code += ')';
       code += ';';
       code += '\n';
     });
-    if (!this.assignmentLines.length) code += `__assignmentLineEmpty__\n`;
+    if (!self.assignmentLines.length) code += `__assignmentLineEmpty__\n`;
     code += '// 批量赋值的翻译代码结束---------\n';
     return code;
   }