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

package/src/concepts/OverriddenLogic__.ts

@@ -1,3 +1,10 @@
+import {
+  wrapForEachToGenerator as wrapForEach,
+  wrapIteratorToGenerator as wrapIterator,
+  runGeneratorSync as runSync,
+  returnOrigin,
+} from '../utils/time-slicing';
+import type { EmbeddedTSFileGenerator } from '../utils';
 import { FrontendLibrary } from './FrontendLibrary__';
 import {
   TranslatorState,
@@ -6,7 +13,11 @@ import {
   indent,
   createCompilerState,
   embeddedTSEmitter,
+  withSourceMapGenerator,
+  withGenerator,
+  TranslatorGenerator,
 } from '../translator';
+import { waitOqlQueryComponentChildrenFinish } from '../common';
 import Namespace from './Namespace__';
 import CallQueryComponent from './CallQueryComponent__';
 import ForEachStatement from './ForEachStatement__';
@@ -1122,9 +1133,12 @@ export class OverriddenLogic 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;
@@ -1175,9 +1189,12 @@ export class OverriddenLogic 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;
@@ -1228,9 +1245,12 @@ export class OverriddenLogic 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;
@@ -1281,9 +1301,12 @@ export class OverriddenLogic 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;
@@ -1334,9 +1357,12 @@ export class OverriddenLogic 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;
@@ -1384,9 +1410,12 @@ export class OverriddenLogic 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;
@@ -1643,199 +1672,181 @@ export class OverriddenLogic extends BaseNode {
     };
   }
 
-  @withSourceMap()
-  toEmbeddedTSInProcess(state?: TranslatorState, prependCode?: string, noName?: boolean, prefix?: boolean): string {
+  @withSourceMapGenerator
+  *toEmbeddedTSInProcess(
+    state = createCompilerState(),
+    prependCode?: string,
+    noName?: boolean,
+    prefix?: boolean,
+  ): TranslatorGenerator {
+    const self = this;
     let code = '';
-    if (!prefix) code += `${noName ? '' : `${this.name}:`} function(`;
-    else code += `function ${this.name}(`; // 加空格会导致无法查询到引用
-    this.params.forEach((param, index) => {
-      code += param.toEmbeddedTS(shiftState(state, code, { inline: true }));
-      if (index !== this.params.length - 1) code += ', ';
-    });
+    if (!prefix) code += `${noName ? '' : `${self.name}:`} function(`;
+    else code += `function ${self.name}(`; // 加空格会导致无法查询到引用
+    yield* wrapForEach(
+      // 加空格会导致无法查询到引用
+      self.params,
+      function* warpForEachGenerator(param, index) {
+        code += yield* param.toEmbeddedTS(shiftState(state, code, { inline: true }));
+        if (index !== self.params.length - 1) code += ', ';
+      },
+    );
     code += `) {\n`;
     code += prependCode || '';
     code += '\n';
-    this.variables.forEach((variable) => {
+    yield* wrapForEach(self.variables, function* warpForEachGenerator(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';
     });
-
-    this.returns.forEach((ret) => {
+    yield* wrapForEach(self.returns, function* warpForEachGenerator(ret) {
       code += `${indent((state?.tabSize || 0) + 1)}let `;
-      code += ret.toEmbeddedTS(shiftState(state, code, { inline: true }));
+      code += yield* ret.toEmbeddedTS(shiftState(state, code, { inline: true }));
       code += ';\n';
     });
-
-    if (!this.body.length && this.returns.length) {
-      code += `${indent((state?.tabSize || 0) + 1)}return ${this.returns[0].name};\n`;
+    if (!self.body.length && self.returns.length) {
+      code += `${indent((state?.tabSize || 0) + 1)}return ${self.returns[0].name};\n`;
     } else {
-      if ((this.parentNode as any).type === 'ExclusiveGateway') {
-        if (this.body.length <= 1) {
+      if ((self.parentNode as any).type === 'ExclusiveGateway') {
+        if (self.body.length <= 1) {
           code += indent((state?.tabSize || 0) + 1);
           code += 'const __LogicEmpty: any = null;';
         }
       } else if (
-        this.body.length <= 2 &&
-        !(this.parentNode instanceof Namespace) &&
-        !(this.parentNode instanceof Process)
+        self.body.length <= 2 &&
+        !(self.parentNode instanceof Namespace) &&
+        !(self.parentNode instanceof Process)
       ) {
         code += indent((state?.tabSize || 0) + 1);
         code += `${indent(state?.tabSize || 0 + 4)}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('');
         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 + 4)}}`;
     code += '\n';
     return code;
   }
 
   toEmbeddedTSNoExport(state?: TranslatorState): string {
-    return this.toEmbeddedTS(state, 'noExport');
+    return runSync(this.toEmbeddedTS(state, 'noExport'));
   }
 
-  @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) {
-      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}`;
     }
-    if (this.typeParams?.length) {
+    if (self.typeParams?.length) {
       code += '<';
-      this.typeParams.forEach((typeParam, typeParamIndex) => {
-        code += typeParam.toEmbeddedTS(shiftState(state, code, { inline: true }));
-        if (typeParamIndex !== this.typeParams.length - 1) code += ', ';
+      yield* wrapForEach(self.typeParams, function* warpForEachGenerator(typeParam, typeParamIndex) {
+        code += yield* typeParam.toEmbeddedTS(shiftState(state, code, { inline: true }));
+        if (typeParamIndex !== self.typeParams.length - 1) code += ', ';
       });
       code += '>';
     }
     // 有参数长度
-    if (this.params.length) {
+    if (self.params.length) {
       code += '(';
-      this.params.forEach((param, index) => {
-        code += param.toEmbeddedTS(shiftState(state, code));
-        if (index !== this.params.length - 1) code += ', ';
+      yield* wrapForEach(self.params, function* warpForEachGenerator(param, index) {
+        code += yield* param.toEmbeddedTS(shiftState(state, code));
+        if (index !== self.params.length - 1) code += ', ';
       });
       code += ')';
     } else {
       // 没有参数的
       code += '()';
     }
-
     // 需要类型推导的局部变量/返回值需要调整申明顺序
     const advanceMap: Map<Return | Variable, Assignment | BatchAssignment> = new Map();
     const needGetSourceMapOqlList: Promise<any>[] = [];
-    this.traverseChildren((el) => {
+    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,
-              },
-            },
-          ],
-        });
-      });
-    }
     // 兼容 return 没有类型情况
-    if (this.returns.length && this.returns[0].typeAnnotation) {
+    if (self.returns.length && self.returns[0].typeAnnotation) {
       code += ': ';
-      code += this.returns[0].typeAnnotation?.toEmbeddedTS(shiftState(state, code, { inline: true }));
+      code += yield* self.returns[0].typeAnnotation?.toEmbeddedTS(shiftState(state, code, { inline: true })) ?? returnOrigin('');
     }
     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';
-      });
-
-    this.variables.forEach((variable) => {
+      }
+    }
+    yield* wrapForEach(self.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';
       }
     });
-
-    this.returns.forEach((ret) => {
+    yield* wrapForEach(self.returns, function* warpForEachGenerator(ret) {
       if (!advanceMap.get(ret)) {
         code += `${indent((state?.tabSize || 0) + 1)}let `;
-        code += ret.toEmbeddedTS(shiftState(state, code, { inline: true }));
+        code += yield* ret.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
@@ -1847,11 +1858,11 @@ export class OverriddenLogic extends BaseNode {
               !(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);
             }
           }
         });
@@ -1861,14 +1872,14 @@ export class OverriddenLogic extends BaseNode {
       ) {
         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
@@ -1877,7 +1888,7 @@ export class OverriddenLogic extends BaseNode {
           !(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 = '';
@@ -1892,34 +1903,31 @@ export class OverriddenLogic extends BaseNode {
           }
           rightCode +=
             rightNode && rightNode.toEmbeddedTS
-              ? rightNode.toEmbeddedTS(shiftState(state, code, { inline: true }))
+              ? yield* rightNode.toEmbeddedTS(shiftState(state, code, { inline: true }))
               : '__RIGHT__';
-          code += this.getRightCode(rightCode, assignment, rightNode);
+          code += yield* self.getRightCode(rightCode, assignment, rightNode);
         }
       }
       code += ';\n';
     });
-
-    if (!this.body.length && this.returns.length) {
-      code += `${indent((state?.tabSize || 0) + 1)}return ${this.returns[0].name};\n`;
+    if (!self.body.length && self.returns.length) {
+      code += `${indent((state?.tabSize || 0) + 1)}return ${self.returns[0].name};\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('');
         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 += `,`;
@@ -1928,7 +1936,7 @@ export class OverriddenLogic extends BaseNode {
     return code;
   }
 
-  getRightCode(code: string, assignment: Assignment | BatchAssignment, rightNode: any): string {
+  *getRightCode(code: string, assignment: Assignment | BatchAssignment, rightNode: any) {
     code = code.replaceAll('\n', ' ');
     if (assignment instanceof Assignment && !assignment.right) return code;
 
@@ -1948,12 +1956,12 @@ export class OverriddenLogic extends BaseNode {
     // ForEachStatement 内部变量需要替换处理，如 item、index
     const forEachStatement = assignment.getAncestor('ForEachStatement') as ForEachStatement;
     if (forEachStatement) {
-      code = this.handleForEachCode(code, forEachStatement);
+      code = yield* this.handleForEachCode(code, forEachStatement);
     }
     // Match 需要针对分支缩小类型范围
     const matchCase = assignment.getAncestor('MatchCase') as MatchCase;
     if (matchCase) {
-      code = this.handleMatchCaseCode(code, matchCase);
+      code = yield* this.handleMatchCaseCode(code, matchCase);
     }
     return code;
   }
@@ -1976,37 +1984,39 @@ export class OverriddenLogic extends BaseNode {
     });
   }
 
-  handleForEachCode(code: string, forEachStatement: ForEachStatement): string {
+  *handleForEachCode(code: string, forEachStatement: ForEachStatement): Generator<any, string, any> {
     const { item, index, each } = forEachStatement;
     if (code.includes(item.name)) {
-      const eachTSCode = each ? each.toEmbeddedTS() : '__IDENTIFIER__';
+      const eachTSCode = each ? yield* each.toEmbeddedTS() : '__IDENTIFIER__';
       code = this.handleReplaceCode(
         code,
         item.name,
         eachTSCode === '__IDENTIFIER__' ? eachTSCode : `${eachTSCode}.__get(0)`,
       );
     }
-    if (code.includes(index.name)) code = this.handleReplaceCode(code, index.name, 'new nasl.core.Integer(0)');
+    if (code.includes(index.name)) {
+      code = this.handleReplaceCode(code, index.name, 'new nasl.core.Integer(0)');
+    }
 
     const parentForEachStatement = forEachStatement.getAncestor('ForEachStatement') as ForEachStatement;
     if (parentForEachStatement) {
-      this.handleForEachCode(code, parentForEachStatement);
+      code = yield* this.handleForEachCode(code, parentForEachStatement);
     }
     return code;
   }
 
-  handleMatchCaseCode(code: string, matchCase: MatchCase): string {
+  *handleMatchCaseCode(code: string, matchCase: MatchCase): Generator<any, string, any> {
     const match = matchCase.getAncestor('Match') as Match;
-    const expressionTsCode = match.expression.toEmbeddedTS();
+    const expressionTsCode = yield* match.expression.toEmbeddedTS();
     if (code.includes(expressionTsCode)) {
       let typeCode = '';
-      matchCase.patterns.forEach((pattern, index) => {
+      yield* wrapForEach(matchCase.patterns, function* wrapForEach(pattern, index) {
         if (pattern instanceof TypeAnnotation) {
           if (index === 0) {
-            typeCode += pattern.toEmbeddedTS();
+            typeCode += yield* pattern.toEmbeddedTS();
           } else {
             typeCode += '|';
-            typeCode += pattern.toEmbeddedTS();
+            typeCode += yield* pattern.toEmbeddedTS();
           }
         }
       });
@@ -2016,7 +2026,7 @@ export class OverriddenLogic extends BaseNode {
     }
     const parentMatchCase = match.getAncestor('MatchCase') as MatchCase;
     if (parentMatchCase) {
-      code = this.handleMatchCaseCode(code, parentMatchCase);
+      code = yield* this.handleMatchCaseCode(code, parentMatchCase);
     }
     return code;
   }
@@ -2096,12 +2106,12 @@ export class OverriddenLogic extends BaseNode {
     return `/embedded/${this.app.name}${_path}/overriddenLogics/${name}.ts`;
   }
 
-  toEmbeddedTSFile() {
+  *toEmbeddedTSFile(): EmbeddedTSFileGenerator {
     let code = `namespace ${this.getTsNamespace()} {\n`;
 
     const state = createCompilerState(code, { tabSize: 1 });
     try {
-      code += this.toEmbeddedTS(state);
+      code += yield* this.toEmbeddedTS(state);
     } catch (err) {
       console.log(err, '有问题翻译失败');
       code += '';