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

package/out/concepts/OverriddenLogic__.js

@@ -34,15 +34,16 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 var OverriddenLogic_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.OverriddenLogic = void 0;
+const time_slicing_1 = require("../utils/time-slicing");
 const FrontendLibrary__1 = require("./FrontendLibrary__");
 const translator_1 = require("../translator");
+const common_1 = require("../common");
 const Namespace__1 = __importDefault(require("./Namespace__"));
 const CallQueryComponent__1 = __importDefault(require("./CallQueryComponent__"));
 const Assignment__1 = __importDefault(require("./Assignment__"));
 const Identifier__1 = __importDefault(require("./Identifier__"));
 const BindEvent__1 = __importDefault(require("./BindEvent__"));
 const BatchAssignment__1 = __importDefault(require("./BatchAssignment__"));
-const OqlQueryComponent__1 = __importDefault(require("./OqlQueryComponent__"));
 const TypeAnnotation__1 = __importDefault(require("./TypeAnnotation__"));
 const CallInterface__1 = __importDefault(require("./CallInterface__"));
 const CallLogic__1 = __importDefault(require("./CallLogic__"));
@@ -594,9 +595,12 @@ class OverriddenLogic extends BaseNode_1.default {
                 ~index && this[parentKey].splice(index, 1);
                 params.index = index;
             }
-            else if (t[parentKey] === n) {
-                params.index = 0;
-                this[parentKey] = undefined;
+            else {
+                const q = t[parentKey];
+                if ((q.__v_raw || q) === n) {
+                    params.index = 0;
+                    this[parentKey] = undefined;
+                }
             }
         }
         return params;
@@ -635,9 +639,12 @@ class OverriddenLogic extends BaseNode_1.default {
                 ~index && this[parentKey].splice(index, 1);
                 params.index = index;
             }
-            else if (t[parentKey] === n) {
-                params.index = 0;
-                this[parentKey] = undefined;
+            else {
+                const q = t[parentKey];
+                if ((q.__v_raw || q) === n) {
+                    params.index = 0;
+                    this[parentKey] = undefined;
+                }
             }
         }
         return params;
@@ -676,9 +683,12 @@ class OverriddenLogic extends BaseNode_1.default {
                 ~index && this[parentKey].splice(index, 1);
                 params.index = index;
             }
-            else if (t[parentKey] === n) {
-                params.index = 0;
-                this[parentKey] = undefined;
+            else {
+                const q = t[parentKey];
+                if ((q.__v_raw || q) === n) {
+                    params.index = 0;
+                    this[parentKey] = undefined;
+                }
             }
         }
         return params;
@@ -717,9 +727,12 @@ class OverriddenLogic extends BaseNode_1.default {
                 ~index && this[parentKey].splice(index, 1);
                 params.index = index;
             }
-            else if (t[parentKey] === n) {
-                params.index = 0;
-                this[parentKey] = undefined;
+            else {
+                const q = t[parentKey];
+                if ((q.__v_raw || q) === n) {
+                    params.index = 0;
+                    this[parentKey] = undefined;
+                }
             }
         }
         return params;
@@ -758,9 +771,12 @@ class OverriddenLogic extends BaseNode_1.default {
                 ~index && this[parentKey].splice(index, 1);
                 params.index = index;
             }
-            else if (t[parentKey] === n) {
-                params.index = 0;
-                this[parentKey] = undefined;
+            else {
+                const q = t[parentKey];
+                if ((q.__v_raw || q) === n) {
+                    params.index = 0;
+                    this[parentKey] = undefined;
+                }
             }
         }
         return params;
@@ -794,9 +810,12 @@ class OverriddenLogic extends BaseNode_1.default {
                 ~index && this[parentKey].splice(index, 1);
                 params.index = index;
             }
-            else if (t[parentKey] === n) {
-                params.index = 0;
-                this[parentKey] = undefined;
+            else {
+                const q = t[parentKey];
+                if ((q.__v_raw || q) === n) {
+                    params.index = 0;
+                    this[parentKey] = undefined;
+                }
             }
         }
         return params;
@@ -1033,54 +1052,57 @@ class OverriddenLogic extends BaseNode_1.default {
             },
         };
     }
-    toEmbeddedTSInProcess(state, prependCode, noName, prefix) {
+    *toEmbeddedTSInProcess(state = (0, translator_1.createCompilerState)(), prependCode, noName, prefix) {
+        const self = this;
         let code = '';
         if (!prefix)
-            code += `${noName ? '' : `${this.name}:`} function(`;
+            code += `${noName ? '' : `${self.name}:`} function(`;
         else
-            code += `function ${this.name}(`; // 加空格会导致无法查询到引用
-        this.params.forEach((param, index) => {
-            code += param.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
-            if (index !== this.params.length - 1)
+            code += `function ${self.name}(`; // 加空格会导致无法查询到引用
+        yield* (0, time_slicing_1.wrapForEachToGenerator)(
+        // 加空格会导致无法查询到引用
+        self.params, function* warpForEachGenerator(param, index) {
+            code += yield* param.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
+            if (index !== self.params.length - 1)
                 code += ', ';
         });
         code += `) {\n`;
         code += prependCode || '';
         code += '\n';
-        this.variables.forEach((variable) => {
+        yield* (0, time_slicing_1.wrapForEachToGenerator)(self.variables, function* warpForEachGenerator(variable) {
             code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}let `;
-            code += variable.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
+            code += yield* variable.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
             code += ';\n';
         });
-        this.returns.forEach((ret) => {
+        yield* (0, time_slicing_1.wrapForEachToGenerator)(self.returns, function* warpForEachGenerator(ret) {
             code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}let `;
-            code += ret.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
+            code += yield* ret.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
             code += ';\n';
         });
-        if (!this.body.length && this.returns.length) {
-            code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}return ${this.returns[0].name};\n`;
+        if (!self.body.length && self.returns.length) {
+            code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}return ${self.returns[0].name};\n`;
         }
         else {
-            if (this.parentNode.type === 'ExclusiveGateway') {
-                if (this.body.length <= 1) {
+            if (self.parentNode.type === 'ExclusiveGateway') {
+                if (self.body.length <= 1) {
                     code += (0, translator_1.indent)((state?.tabSize || 0) + 1);
                     code += 'const __LogicEmpty: any = null;';
                 }
             }
-            else if (this.body.length <= 2 &&
-                !(this.parentNode instanceof Namespace__1.default) &&
-                !(this.parentNode instanceof Process__1.default)) {
+            else if (self.body.length <= 2 &&
+                !(self.parentNode instanceof Namespace__1.default) &&
+                !(self.parentNode instanceof Process__1.default)) {
                 code += (0, translator_1.indent)((state?.tabSize || 0) + 1);
                 code += `${(0, translator_1.indent)(state?.tabSize || 0 + 4)}const __LogicEmpty: any = null;`;
             }
-            this.body.forEach((item) => {
-                code += item.toEmbeddedTS && item.toEmbeddedTS((0, translator_1.shiftState)(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
+            yield* (0, time_slicing_1.wrapForEachToGenerator)(self.body, function* warpForEachGenerator(item) {
+                code += yield* item.toEmbeddedTS?.((0, translator_1.shiftState)(state, code, { tabSize: (state?.tabSize || 0) + 1 })) ?? (0, time_slicing_1.returnOrigin)('');
                 code += ';\n';
             });
         }
         // 在body渲染完成之后把其余的节点放到下面
-        this.playground.forEach((item) => {
-            code += item.toEmbeddedTS((0, translator_1.shiftState)(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
+        yield* (0, time_slicing_1.wrapForEachToGenerator)(self.playground, function* warpForEachGenerator(item) {
+            code += yield* item.toEmbeddedTS((0, translator_1.shiftState)(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
             code += '\n';
         });
         code += `${(0, translator_1.indent)(state?.tabSize || 0 + 4)}}`;
@@ -1088,35 +1110,38 @@ class OverriddenLogic extends BaseNode_1.default {
         return code;
     }
     toEmbeddedTSNoExport(state) {
-        return this.toEmbeddedTS(state, 'noExport');
+        return (0, time_slicing_1.runGeneratorSync)(this.toEmbeddedTS(state, 'noExport'));
     }
-    toEmbeddedTS(state, prefix) {
+    *toEmbeddedTS(state = (0, translator_1.createCompilerState)(), prefix) {
+        const self = this;
+        // 等待 oql 查询返回内容
+        yield (0, common_1.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((0, translator_1.shiftState)(state, code, { inline: true }));
-                if (typeParamIndex !== this.typeParams.length - 1)
+            yield* (0, time_slicing_1.wrapForEachToGenerator)(self.typeParams, function* warpForEachGenerator(typeParam, typeParamIndex) {
+                code += yield* typeParam.toEmbeddedTS((0, translator_1.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((0, translator_1.shiftState)(state, code));
-                if (index !== this.params.length - 1)
+            yield* (0, time_slicing_1.wrapForEachToGenerator)(self.params, function* warpForEachGenerator(param, index) {
+                code += yield* param.toEmbeddedTS((0, translator_1.shiftState)(state, code));
+                if (index !== self.params.length - 1)
                     code += ', ';
             });
             code += ')';
@@ -1128,93 +1153,72 @@ class OverriddenLogic extends BaseNode_1.default {
         // 需要类型推导的局部变量/返回值需要调整申明顺序
         const advanceMap = new Map();
         const needGetSourceMapOqlList = [];
-        this.traverseChildren((el) => {
+        yield* self.traverseChildrenGenerator(function* traverseChildrenGenerator(el) {
             if (el && (el instanceof BatchAssignment__1.default || (el instanceof Assignment__1.default && el.left?.name))) {
                 if (el instanceof Assignment__1.default) {
-                    const advanceVar = this.variables?.find((variable) => !variable.typeAnnotation && el.left?.name === variable.name);
+                    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__1.default) {
-                    el.assignmentLines.forEach(({ leftIndex }) => {
+                    yield* (0, time_slicing_1.wrapForEachToGenerator)(el.assignmentLines, function* warpForEachGenerator({ leftIndex }) {
                         const leftCode = leftIndex.length === 1
-                            ? el.left.expression.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }))
-                            : el.left.members[leftIndex[1]]?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
-                        const advanceVar = this.variables?.find((variable) => !variable.typeAnnotation && leftCode === variable.name);
+                            ? yield* el.left.expression.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }))
+                            : yield* el.left.members[leftIndex[1]]?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true })) ?? (0, time_slicing_1.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__1.default) {
-                if (!el.codeSourceMap) {
-                    needGetSourceMapOqlList.push(el.requestEmbeddedTS());
-                }
-            }
         });
-        if (needGetSourceMapOqlList.length) {
-            Promise.all(needGetSourceMapOqlList).then(() => {
-                // 触发一下更新重新生成代码
-                this.app?.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((0, translator_1.shiftState)(state, code, { inline: true }));
+            code += yield* self.returns[0].typeAnnotation?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true })) ?? (0, time_slicing_1.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 += (0, translator_1.indent)((state?.tabSize || 0) + 1);
-                code += param.toEmbeddedNameTS((0, translator_1.shiftState)(state, code, { inline: true }));
+                code += yield* param.toEmbeddedNameTS((0, translator_1.shiftState)(state, code, { inline: true }));
                 code += '\n';
-            });
-        this.variables.forEach((variable) => {
+            }
+        }
+        yield* (0, time_slicing_1.wrapForEachToGenerator)(self.variables, function* warpForEachGenerator(variable) {
             if (!advanceMap.get(variable)) {
                 code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}let `;
-                code += variable.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
+                code += yield* variable.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
                 code += ';\n';
             }
         });
-        this.returns.forEach((ret) => {
+        yield* (0, time_slicing_1.wrapForEachToGenerator)(self.returns, function* warpForEachGenerator(ret) {
             if (!advanceMap.get(ret)) {
                 code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}let `;
-                code += ret.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
+                code += yield* ret.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
                 code += ';\n';
             }
         });
-        advanceMap.forEach((assignment, node) => {
+        yield* (0, time_slicing_1.wrapIteratorToGenerator)(advanceMap.entries(), function* warpForEachGenerator([node, assignment]) {
             code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}// @ts-ignore\n`;
             code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}let `;
-            code += node.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
+            code += yield* node.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
             if (assignment instanceof BatchAssignment__1.default) {
-                assignment.assignmentLines.forEach(({ leftIndex, rightIndex }) => {
+                yield* (0, time_slicing_1.wrapForEachToGenerator)(assignment.assignmentLines, function* warpForEachGenerator({ leftIndex, rightIndex }) {
                     const leftCode = leftIndex.length === 1
-                        ? assignment.left.expression.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }))
-                        : assignment.left.members[leftIndex[1]]?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
+                        ? yield* assignment.left.expression.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }))
+                        : yield* assignment.left.members[leftIndex[1]]?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true })) ?? (0, time_slicing_1.returnOrigin)('');
                     if (leftCode === node.name) {
                         const rightNode = rightIndex.length === 1
                             ? assignment.rights[rightIndex[0]]?.expression
@@ -1223,12 +1227,12 @@ class OverriddenLogic extends BaseNode_1.default {
                         if (rightNode?.typeAnnotation &&
                             !(rightNode instanceof CallQueryComponent__1.default && rightNode.ideVersion !== '2.11')) {
                             code += ': ';
-                            code += rightNode.typeAnnotation?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true })) || 'null';
+                            code += yield* rightNode.typeAnnotation?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true })) ?? (0, time_slicing_1.returnOrigin)('null');
                         }
                         else {
                             code += ' = ';
-                            const rightCode = rightNode?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }));
-                            code += this.getRightCode(rightCode, assignment, rightNode);
+                            const rightCode = yield* rightNode?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true })) ?? (0, time_slicing_1.returnOrigin)('');
+                            code += yield* self.getRightCode(rightCode, assignment, rightNode);
                         }
                     }
                 });
@@ -1237,13 +1241,13 @@ class OverriddenLogic extends BaseNode_1.default {
                 !assignment.right?.typeAnnotation) {
                 const rightNode = assignment.right;
                 code += ' = ';
-                code += rightNode.getInitCode((0, translator_1.shiftState)(state, code, { inline: true }));
+                code += yield* rightNode.getInitCode((0, translator_1.shiftState)(state, code, { inline: true }));
             }
             else if (assignment.right?.concept === 'NewComposite' &&
                 assignment.right?.typeAnnotation?.typeKind === 'anonymousStructure') {
                 const rightNode = assignment.right;
                 code += ' = ';
-                code += rightNode.getInitCode((0, translator_1.shiftState)(state, code, { inline: true }));
+                code += yield* rightNode.getInitCode((0, translator_1.shiftState)(state, code, { inline: true }));
             }
             else {
                 const rightNode = assignment.right;
@@ -1251,7 +1255,7 @@ class OverriddenLogic extends BaseNode_1.default {
                 if (rightNode?.typeAnnotation &&
                     !(rightNode instanceof CallQueryComponent__1.default && rightNode.ideVersion !== '2.11')) {
                     code += ': ';
-                    code += rightNode.typeAnnotation?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true })) || 'null';
+                    code += yield* rightNode.typeAnnotation?.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true })) ?? (0, time_slicing_1.returnOrigin)('null');
                 }
                 else {
                     code += ' = ';
@@ -1265,30 +1269,30 @@ class OverriddenLogic extends BaseNode_1.default {
                     }
                     rightCode +=
                         rightNode && rightNode.toEmbeddedTS
-                            ? rightNode.toEmbeddedTS((0, translator_1.shiftState)(state, code, { inline: true }))
+                            ? yield* rightNode.toEmbeddedTS((0, translator_1.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 += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}return ${this.returns[0].name};\n`;
+        if (!self.body.length && self.returns.length) {
+            code += `${(0, translator_1.indent)((state?.tabSize || 0) + 1)}return ${self.returns[0].name};\n`;
         }
         else {
             // 如果长度小于2 并且不是虚拟节点就 加 报错，
-            if (this.body.length <= 2 && !(this.parentNode instanceof Namespace__1.default)) {
+            if (self.body.length <= 2 && !(self.parentNode instanceof Namespace__1.default)) {
                 code += (0, translator_1.indent)((state?.tabSize || 0) + 1);
                 code += 'const __LogicEmpty: any = null;';
             }
-            this.body.forEach((item) => {
-                code += item.toEmbeddedTS && item.toEmbeddedTS((0, translator_1.shiftState)(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
+            yield* (0, time_slicing_1.wrapForEachToGenerator)(self.body, function* warpForEachGenerator(item) {
+                code += yield* item.toEmbeddedTS?.((0, translator_1.shiftState)(state, code, { tabSize: (state?.tabSize || 0) + 1 })) ?? (0, time_slicing_1.returnOrigin)('');
                 code += ';\n';
             });
         }
         // 在body渲染完成之后把其余的节点放到下面
-        this.playground.forEach((item) => {
-            code += item.toEmbeddedTS((0, translator_1.shiftState)(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
+        yield* (0, time_slicing_1.wrapForEachToGenerator)(self.playground, function* warpForEachGenerator(item) {
+            code += yield* item.toEmbeddedTS((0, translator_1.shiftState)(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
             code += ';\n';
         });
         code += `${(0, translator_1.indent)(state?.tabSize || 0)}}`;
@@ -1298,7 +1302,7 @@ class OverriddenLogic extends BaseNode_1.default {
         code += '\n';
         return code;
     }
-    getRightCode(code, assignment, rightNode) {
+    *getRightCode(code, assignment, rightNode) {
         code = code.replaceAll('\n', ' ');
         if (assignment instanceof Assignment__1.default && !assignment.right)
             return code;
@@ -1316,12 +1320,12 @@ class OverriddenLogic extends BaseNode_1.default {
         // ForEachStatement 内部变量需要替换处理，如 item、index
         const forEachStatement = assignment.getAncestor('ForEachStatement');
         if (forEachStatement) {
-            code = this.handleForEachCode(code, forEachStatement);
+            code = yield* this.handleForEachCode(code, forEachStatement);
         }
         // Match 需要针对分支缩小类型范围
         const matchCase = assignment.getAncestor('MatchCase');
         if (matchCase) {
-            code = this.handleMatchCaseCode(code, matchCase);
+            code = yield* this.handleMatchCaseCode(code, matchCase);
         }
         return code;
     }
@@ -1351,33 +1355,34 @@ class OverriddenLogic extends BaseNode_1.default {
             return fnCode;
         });
     }
-    handleForEachCode(code, forEachStatement) {
+    *handleForEachCode(code, forEachStatement) {
         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))
+        if (code.includes(index.name)) {
             code = this.handleReplaceCode(code, index.name, 'new nasl.core.Integer(0)');
+        }
         const parentForEachStatement = forEachStatement.getAncestor('ForEachStatement');
         if (parentForEachStatement) {
-            this.handleForEachCode(code, parentForEachStatement);
+            code = yield* this.handleForEachCode(code, parentForEachStatement);
         }
         return code;
     }
-    handleMatchCaseCode(code, matchCase) {
+    *handleMatchCaseCode(code, matchCase) {
         const match = matchCase.getAncestor('Match');
-        const expressionTsCode = match.expression.toEmbeddedTS();
+        const expressionTsCode = yield* match.expression.toEmbeddedTS();
         if (code.includes(expressionTsCode)) {
             let typeCode = '';
-            matchCase.patterns.forEach((pattern, index) => {
+            yield* (0, time_slicing_1.wrapForEachToGenerator)(matchCase.patterns, function* wrapForEach(pattern, index) {
                 if (pattern instanceof TypeAnnotation__1.default) {
                     if (index === 0) {
-                        typeCode += pattern.toEmbeddedTS();
+                        typeCode += yield* pattern.toEmbeddedTS();
                     }
                     else {
                         typeCode += '|';
-                        typeCode += pattern.toEmbeddedTS();
+                        typeCode += yield* pattern.toEmbeddedTS();
                     }
                 }
             });
@@ -1387,7 +1392,7 @@ class OverriddenLogic extends BaseNode_1.default {
         }
         const parentMatchCase = match.getAncestor('MatchCase');
         if (parentMatchCase) {
-            code = this.handleMatchCaseCode(code, parentMatchCase);
+            code = yield* this.handleMatchCaseCode(code, parentMatchCase);
         }
         return code;
     }
@@ -1459,11 +1464,11 @@ class OverriddenLogic extends BaseNode_1.default {
         }
         return `/embedded/${this.app.name}${_path}/overriddenLogics/${name}.ts`;
     }
-    toEmbeddedTSFile() {
+    *toEmbeddedTSFile() {
         let code = `namespace ${this.getTsNamespace()} {\n`;
         const state = (0, translator_1.createCompilerState)(code, { tabSize: 1 });
         try {
-            code += this.toEmbeddedTS(state);
+            code += yield* this.toEmbeddedTS(state);
         }
         catch (err) {
             console.log(err, '有问题翻译失败');
@@ -1647,10 +1652,10 @@ __decorate([
     (0, decorators_1.property)('LogicItem')
 ], OverriddenLogic.prototype, "playground", void 0);
 __decorate([
-    (0, translator_1.withSourceMap)()
+    translator_1.withSourceMapGenerator
 ], OverriddenLogic.prototype, "toEmbeddedTSInProcess", null);
 __decorate([
-    (0, translator_1.withSourceMap)()
+    translator_1.withSourceMapGenerator
 ], OverriddenLogic.prototype, "toEmbeddedTS", null);
 OverriddenLogic = OverriddenLogic_1 = __decorate([
     decorators_1.className,