@lcap/nasl 3.3.0-beta.0 → 3.4.0-alpha.1

package/src/concepts/CallLogic__.ts

@@ -1,26 +1,26 @@
 import * as JSON5 from 'json5';
-import { TranslatorState, shiftState, withSourceMap } from '../translator';
+import { v4 as uuidv4 } from 'uuid';
+import { TranslatorState, shiftState, withSourceMap, createCompilerState } from '../translator';
 import Logic from './Logic__';
-import type Param from './Param__';
+import Param from './Param__';
 import View from './View__';
+import ViewElement, { ElementToVueOptions } from './ViewElement__';
 import { getNodeByNodeCallee, getNaslNodeByNodeCallee } from '../automate/engine/utils';
 import { findConceptKeyWordIndexFromNamespace } from '../utils';
 import { config } from '../config';
 import Identifier from './Identifier__';
-import type { ViewElement, ElementToVueOptions } from './ViewElement__';
-import type { MemberExpression } from './MemberExpression__';
-import type { TSPPosition } from '../translator';
+import type { TSPPosition } from '../translator/tsp';
 
-//================================================================================
+//= ===============================================================================
 // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
 // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-//================================================================================
+//= ===============================================================================
 import { EventPayload, Params } from '../common/EventEmitter';
-import { concept, excludedInJSON, property, className } from '../decorators';
+import { concept, excludedInJSON, property } from '../decorators';
 
 import * as utils from '../utils';
-import { v4 as uuidv4 } from 'uuid';
 import BaseNode from '../common/BaseNode';
+import classMap from '../common/classMap';
 import TypeAnnotation from './TypeAnnotation__';
 import Argument from './Argument__';
 import BindAttribute from './BindAttribute__';
@@ -29,15 +29,10 @@ import LogicItem from './LogicItem__';
 /**
  * 调用逻辑
  */
-@className
 @concept('调用逻辑')
-// @ts-ignore
 export class CallLogic extends LogicItem {
-  /** 类名 */
-  static readonly ConceptName = 'CallLogic';
-
   /**
-   * 唯一标识
+   * uuid
    */
   @property()
   uuid: string = uuidv4().replace(/-/g, '');
@@ -45,8 +40,8 @@ export class CallLogic extends LogicItem {
   /**
    * 产品概念
    */
-  // @ts-ignore
-  readonly concept!: 'CallLogic' | 'CallConnector';
+  @property()
+  concept: 'CallLogic' | 'CallConnector' = 'CallLogic';
 
   /**
    * 调用逻辑标题
@@ -58,29 +53,25 @@ export class CallLogic extends LogicItem {
    * calleeNamespace
    */
   @property()
-  // @ts-ignore
-  calleeNamespace!: string;
+  calleeNamespace: string = undefined;
 
   /**
    * calleeName
    */
   @property()
-  // @ts-ignore
-  calleeName!: string;
+  calleeName: string = undefined;
 
   /**
    * shortcut
    */
   @property()
-  // @ts-ignore
-  shortcut!: boolean;
+  shortcut: boolean = undefined;
 
   /**
    * handleError
    */
   @property()
-  // @ts-ignore
-  handleError!: boolean;
+  handleError: boolean = undefined;
 
   /**
    * 类型参数
@@ -98,25 +89,28 @@ export class CallLogic extends LogicItem {
    * 验证规则
    */
   @property('BindAttribute')
-  // @ts-ignore
-  validation!: BindAttribute;
+  validation: BindAttribute = undefined;
 
   /**
    * 是否被折叠
    */
   @property()
-  // @ts-ignore
-  isAdvancedSettingFolded!: boolean;
+  isAdvancedSettingFolded: boolean = undefined;
 
   /**
    * @param source 需要合并的部分参数
    */
   constructor(source?: Partial<CallLogic>) {
-    source = Object.assign({}, CallLogic.getDefaultOptions(), source);
+    // @ts-ignore
+    source = { ...CallLogic.getDefaultOptions(), ...source };
     super(source);
     super.subConstructor(source);
   }
 
+  getClassName() {
+    return 'CallLogic';
+  }
+
   static from(source: any, parentNode?: any, parentKey?: string): CallLogic {
     return super.from(source, parentNode, parentKey) as CallLogic;
   }
@@ -177,7 +171,7 @@ export class CallLogic extends LogicItem {
           ...options,
         },
         this,
-        'typeArguments',
+        'typeArguments'
       );
     }
     this.typeArguments.splice(index, 0, typeAnnotation);
@@ -277,7 +271,7 @@ export class CallLogic extends LogicItem {
           ...options,
         },
         this,
-        'arguments',
+        'arguments'
       );
     }
     this.arguments.splice(index, 0, argument);
@@ -355,12 +349,12 @@ export class CallLogic extends LogicItem {
   removeTypeAnnotation(typeAnnotation: TypeAnnotation): void;
 
   removeTypeAnnotation(options: TypeAnnotation) {
-    let typeAnnotation: TypeAnnotation = options;
+    const typeAnnotation: TypeAnnotation = options;
     return typeAnnotation.delete();
   }
 
   __removeTypeAnnotation(typeAnnotation: TypeAnnotation) {
-    const parentKey = typeAnnotation.parentKey;
+    const { parentKey } = typeAnnotation;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -389,12 +383,12 @@ export class CallLogic extends LogicItem {
   removeArgument(argument: Argument): void;
 
   removeArgument(options: Argument) {
-    let argument: Argument = options;
+    const argument: Argument = options;
     return argument.delete();
   }
 
   __removeLogicItem(logicItem: LogicItem) {
-    const parentKey = logicItem.parentKey;
+    const { parentKey } = logicItem;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -433,7 +427,7 @@ export class CallLogic extends LogicItem {
     if (typeof options === 'string') {
       validation = this.validation;
       if (!validation) {
-        throw new Error('找不到元素绑定属性 ' + options);
+        throw new Error(`找不到元素绑定属性 ${options}`);
       }
     } else {
       validation = options;
@@ -442,7 +436,7 @@ export class CallLogic extends LogicItem {
   }
 
   __removeBindAttribute(bindAttribute: BindAttribute) {
-    const parentKey = bindAttribute.parentKey;
+    const { parentKey } = bindAttribute;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -464,10 +458,10 @@ export class CallLogic extends LogicItem {
     return params;
   }
 
-  //================================================================================
+  //= ===============================================================================
   // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
   // 自动生成的代码已结束。下面可以手动编写。
-  //================================================================================
+  //= ===============================================================================
 
   get tsCalleeNamespace() {
     if (this.calleeNamespace) {
@@ -559,7 +553,7 @@ export class CallLogic extends LogicItem {
         keyword: param.name,
       });
       if (param.defaultExpression) {
-        const exp: any = BaseNode.from(param.defaultExpression.toJSON());
+        const exp: any = BaseNode.from({ ...param.defaultExpression.toJSON() });
         argument.setExpression(exp);
       }
     });
@@ -584,7 +578,7 @@ export class CallLogic extends LogicItem {
           keyword: param.name,
           expression: param.isFunction ? new Identifier() : param.defaultExpression,
         },
-        this,
+        this
       );
       Object.assign(argument, relationOptions);
       return argument;
@@ -618,7 +612,7 @@ export class CallLogic extends LogicItem {
         handleError,
       });
       if (parentNode.concept === 'MemberExpression') {
-        const topNode = (parentNode as MemberExpression).firstMemberExpressionNode();
+        const topNode = parentNode.firstMemberExpressionNode();
         topNode.replaceNode(this);
       }
       this.app.emit('collect:end');
@@ -640,8 +634,8 @@ export class CallLogic extends LogicItem {
         {
           ...argument,
         },
-        this,
-      ),
+        this
+      )
     );
     const relationOptions = { parentNode: this, parentKey: 'arguments' };
     /**
@@ -655,7 +649,7 @@ export class CallLogic extends LogicItem {
             {
               keyword: param.name,
             },
-            this,
+            this
           );
           Object.assign(newParam, relationOptions);
           newArgs.push(newParam);
@@ -709,15 +703,11 @@ export class CallLogic extends LogicItem {
       code += '$global.';
     }
     const completeArguments = this.completeArguments;
-
     if (this.concept === 'CallLogic' && this.getAncestor('BindAttribute')?.name === 'data-source') {
-      if (this.calleeNamespace) {
-        const currentElement = this.getAncestor('ViewElement') as ViewElement;
-        code += '(params) => {\n';
-        code += `this['__${currentElement.name}_params'] = {...params};\n`;
-        code += `return this.$logics['${
-          this.calleeNamespace ? `${this.calleeNamespace}.${this.calleeName}` : this.calleeName
-        }']`;
+        const currentElement = this.getAncestor('ViewElement') as ViewElement
+        code += '(params) => {\n'
+        code += `this['__${currentElement.name}_params'] = {...params};\n`
+        code += `return this.$logics['${this.calleeNamespace ? `${this.calleeNamespace}.${this.calleeName}` : this.calleeName}']`
         code += `({
             config: {
                 download: false,
@@ -731,41 +721,36 @@ export class CallLogic extends LogicItem {
         code += `path: {},
         body: {`;
         const logic = getNodeByNodeCallee(this.app, this.calleewholeKey, (key: string, tempPoint: any) => {
-          // 处理 NASL 中未定义的假的节点
-          if (key === tempPoint?.name) {
+            // 处理 NASL 中未定义的假的节点
+            if (key === tempPoint?.name) {
+                return {
+                    tempPoint,
+                    state: 1,
+                };
+            }
+            const t = tempPoint?.[key];
             return {
-              tempPoint,
-              state: 1,
+                tempPoint: t,
+                state: Array.isArray(t) ? 2 : 1,
             };
-          }
-          const t = tempPoint?.[key];
-          return {
-            tempPoint: t,
-            state: Array.isArray(t) ? 2 : 1,
-          };
         });
         const args = logic?.params || [];
         const argsStr = args
-          .map((arg: any, index: number) => {
-            const actualArg = completeArguments[index];
-            return {
-              keyword: arg?.name,
-              arg: actualArg,
-            };
-          })
-          .map((obj: any) => `\n${obj?.keyword}: ${obj?.arg?.toVue(options)}`)
-          .join(', ');
+            .map((arg: any, index: number) => {
+                const actualArg = completeArguments[index];
+                return {
+                    keyword: arg?.name,
+                    arg: actualArg,
+                };
+            })
+            .map((obj: any) => `\n${obj?.keyword}: ${obj?.arg?.toVue(options)}`)
+            .join(', ');
         code += argsStr;
         if (Array.isArray(this.typeArguments) && this.typeArguments.length) {
-          code += `,\n__typeArguments: ${JSON5.stringify(this.typeArguments)}`;
+            code += `,\n__typeArguments: ${JSON5.stringify(this.typeArguments)}`;
         }
         code += '\n}';
         code += '}';
-      } else {
-        // 前端逻辑的calleeNamespace一直为空字符串
-        code += this.calleeName;
-        return code;
-      }
     } else {
       code += this.calleeName;
       code += '(';
@@ -775,7 +760,6 @@ export class CallLogic extends LogicItem {
       });
     }
     code += ')}';
-
     return code;
   }
 
@@ -806,7 +790,7 @@ export class CallLogic extends LogicItem {
     }
     code += ')';
     if (this.concept === 'CallLogic' && this.getAncestor('BindAttribute')?.name === 'data-source') {
-      return this.calleeName;
+      return this.calleeName
     }
 
     return code;
@@ -880,9 +864,7 @@ export class CallLogic extends LogicItem {
       const logic = this.getCallNode();
       const args = logic?.params || [];
       const keywordEntitiesIndex = findConceptKeyWordIndexFromNamespace(this.calleeNamespace, 'entities');
-      code += `this.$logics['${
-        this.calleeNamespace ? `${this.calleeNamespace}.${this.calleeName}` : this.calleeName
-      }']`;
+      code += `this.$logics['${this.calleeNamespace ? `${this.calleeNamespace}.${this.calleeName}` : this.calleeName}']`;
       code += `({
                 config: {
                     download: false,
@@ -936,8 +918,7 @@ export class CallLogic extends LogicItem {
             code += `body: {\n`;
             const argsStr = logicArgs
               .map((arg) => {
-                const name =
-                  arg.keyword === 'body' ? (this.calleeName === 'batchUpdate' ? 'entities' : 'entity') : arg.keyword;
+                const name = arg.keyword === 'body' ? (this.calleeName === 'batchUpdate' ? 'entities' : 'entity') : arg.keyword;
                 return `${name}: ${arg.expression.toJS(shiftState(state, code))}`;
               })
               .join(',\n');
@@ -954,10 +935,7 @@ export class CallLogic extends LogicItem {
                 const name = arg.keyword === 'body' ? 'entity' : arg.keyword;
                 // filter回调函数
                 if (arg.expression.returnExpression) {
-                  const argsStr = JSON.stringify(handleMember(arg.expression.returnExpression)).replace(
-                    /"__@@(.*?)@@__"/g,
-                    '$1',
-                  );
+                  const argsStr = JSON.stringify(handleMember(arg.expression.returnExpression)).replace(/"__@@(.*?)@@__"/g, '$1');
                   return `${name}: {
                                         concept: 'Function',
                                         name: 'filter',
@@ -975,10 +953,7 @@ export class CallLogic extends LogicItem {
           }
           case 'deleteBy': {
             code += `body: `;
-            const argsStr = JSON.stringify(handleMember((logicArgs[0].expression as any).returnExpression)).replace(
-              /"__@@(.*?)@@__"/g,
-              '$1',
-            );
+            const argsStr = JSON.stringify(handleMember((logicArgs[0].expression as any).returnExpression)).replace(/"__@@(.*?)@@__"/g, '$1');
             code += `{
                             concept: 'Function',
                             name: 'filter',
@@ -1147,8 +1122,7 @@ export class CallLogic extends LogicItem {
     }
     const typeAnnotation = (this as any).typeAnnotation || this.__TypeAnnotation;
     const needGenInitFromSchema =
-      typeAnnotation &&
-      (!['primitive', 'union'].includes(typeAnnotation.typeKind) || typeAnnotation.typeName === 'DateTime');
+      typeAnnotation && (!['primitive', 'union'].includes(typeAnnotation.typeKind) || typeAnnotation.typeName === 'DateTime');
     if (needGenInitFromSchema) {
       code = `this.$genInitFromSchema('${typeAnnotation?.sortedTypeKey}', ${code})`;
     }
@@ -1171,7 +1145,7 @@ export class CallLogic extends LogicItem {
   }
 
   @withSourceMap()
-  toEmbeddedTS(state?: TranslatorState): string {
+  toEmbeddedTS(state = createCompilerState()): string {
     let code = '';
 
     const {
@@ -1206,10 +1180,7 @@ export class CallLogic extends LogicItem {
     /** 设置输入参数 */
     const setInputParam = () => {
       const paramRequired = ['showMessage', 'consoleLog'].includes(calleeName);
-      if (
-        (tsCalleeNamespace.startsWith('elements') && tsCalleeNamespace.endsWith('logics')) ||
-        tsCalleeNamespace.startsWith('nasl.')
-      ) {
+      if ((tsCalleeNamespace.startsWith('elements') && tsCalleeNamespace.endsWith('logics')) || tsCalleeNamespace.startsWith('nasl.')) {
         code += '(';
         logicArgs.forEach((arg, index) => {
           // 页面逻辑先都非必填
@@ -1218,12 +1189,8 @@ export class CallLogic extends LogicItem {
         });
         code += ')';
       } else if (completeArguments.length) {
-        if (
-          ['expression'].includes(parentKey) &&
-          this.getAncestor('BindAttribute')?.name === 'data-source' &&
-          (state as any)?.noNeedParams
-        ) {
-          return;
+        if (['expression'].includes(parentKey) && this.getAncestor('BindAttribute')?.name === 'data-source' && (state as any)?.noNeedParams) {
+          return
         }
         code += '(';
         completeArguments.forEach((arg, index) => {
@@ -1233,24 +1200,7 @@ export class CallLogic extends LogicItem {
         });
         code += ')';
       } else {
-        if (
-          ['expression'].includes(parentKey) &&
-          this.getAncestor('BindAttribute')?.name === 'data-source' &&
-          (state as any)?.noNeedParams
-        ) {
-          return;
-        }
-        if (completeArguments.length) {
-          code += '(';
-          completeArguments.forEach((arg, index) => {
-            // 取到是不是必填放入参数中
-            code += arg.toEmbeddedTS(shiftState(state, code), arg?.paramRequired);
-            if (index !== logicArgs.length - 1) code += ', ';
-          });
-          code += ')';
-        } else {
-          code += '()';
-        }
+        code += '()';
       }
     };
 
@@ -1316,6 +1266,126 @@ export class CallLogic extends LogicItem {
     return code;
   }
 
+  @withSourceMap()
+  toNaturalTS(state = createCompilerState()): string {
+    const {
+      handleError,
+      calleeKey,
+      calleeName,
+      parentKey,
+      typeArguments,
+      tsCalleeNamespace,
+      completeArguments,
+      parentNode,
+      validation,
+      useIife,
+      useValidation,
+      arguments: logicArgs,
+    } = this;
+
+    let code = calleeKey.replace('__elements', '$refs');
+    if (/^app.+?\.entities\..+?logics/.test(code))
+      code = code.replace(/^app.+?\.entities\.(.+?)\.logics/, '$1');
+    /** 设置函数范型参数 */
+    const setTypeArguments = () => {
+      if (Array.isArray(typeArguments) && typeArguments.length) {
+        // 如果有内容
+        code += `<`;
+        typeArguments.forEach((arg, i, arr) => {
+          code += arg.toNaturalTS(shiftState(state, code, { tabSize: 0 }));
+          if (i !== arr.length - 1) {
+            code += ', ';
+          }
+        });
+        code += `>`;
+      }
+    };
+    /** 设置输入参数 */
+    const setInputParam = () => {
+      const paramRequired = ['showMessage', 'consoleLog'].includes(calleeName);
+      if ((tsCalleeNamespace.startsWith('elements') && tsCalleeNamespace.endsWith('logics')) || tsCalleeNamespace.startsWith('nasl.')) {
+        code += '(';
+        logicArgs.forEach((arg, index) => {
+          // 页面逻辑先都非必填
+          code += arg.toNaturalTS(shiftState(state, code), paramRequired);
+          if (index !== logicArgs.length - 1) code += ', ';
+        });
+        code += ')';
+      } else if (completeArguments.length) {
+        code += '(';
+        completeArguments.forEach((arg, index) => {
+          // 取到是不是必填放入参数中
+          code += arg.toNaturalTS(shiftState(state, code), arg?.paramRequired);
+          if (index !== logicArgs.length - 1) code += ', ';
+        });
+        code += ')';
+      } else {
+        code += '()';
+      }
+    };
+
+    /** 设置表达式末尾分号 */
+    const setEndSemi = () => {
+      const { concept: parentConcept } = parentNode ?? {};
+      if (
+        !['test', 'patterns', 'returnExpression'].includes(parentKey) &&
+        ![
+          'Argument',
+          'Assignment',
+          'ForEachStatement',
+          'SelectMembers',
+          'UnaryExpression',
+          'BinaryExpression',
+          'StringInterpolation',
+          'MemberExpression',
+          'NewMap',
+          'NewList',
+        ].includes(parentConcept)
+      ) {
+        code += ';';
+      }
+    };
+
+    if (useIife) {
+      code += '(() =>{\nconst wrapper = ';
+
+      if (handleError) {
+        code += 'nasl.util.autoHandleError(';
+      }
+
+      if (useValidation) {
+        code += 'nasl.util.callLogicWithChecker(';
+      }
+
+      code += calleeKey;
+      setTypeArguments();
+
+      if (useValidation) {
+        code += ', (';
+        code += completeArguments.map((item) => item.keyword).join(', ');
+        code += ') => {\n';
+        code += validation.toNaturalTS(shiftState(state, code));
+        code += ';\n})';
+      }
+
+      if (handleError) {
+        code += ')';
+      }
+
+      code += ';\nreturn wrapper';
+      setInputParam();
+      code += ';\n})()';
+    } else {
+      code += calleeKey;
+      setTypeArguments();
+      setInputParam();
+    }
+
+    setEndSemi();
+
+    return code;
+  }
+
   // 通过内容过去节点
   getCallNode(): Logic | undefined {
     let node;
@@ -1342,48 +1412,26 @@ export class CallLogic extends LogicItem {
             state: Array.isArray(t) ? 2 : 1,
           };
         });
-      } catch (err) {}
+      } catch (err) { }
     }
     return node;
   }
 
   /** 获取类型标注时的偏移量 */
   getQuickInfoOffset(detail: TSPPosition, code: string): TSPPosition {
-    if (this.useIife) {
-      const characterOffset = 8;
-      const lines = code.split('\n');
-      const lineOffset = lines.findIndex((item) => item.startsWith('return wrapper'));
-
-      return {
-        line: detail.line + lineOffset,
-        offset: characterOffset,
-      };
-    } else {
-      let indexOf = 0;
+    if (!this.useIife) {
+      return { ...detail };
+    }
 
-      // 没有括号，表示括号在外面
-      if (!code.includes('(')) {
-        indexOf = -1;
-      }
-      // 有括号就找到括号的位置
-      else {
-        // 如果有泛型先往前找一找
-        if (code.includes('<')) {
-          indexOf = Math.min(code.indexOf('<'), code.indexOf('('));
-        } else {
-          indexOf = code.indexOf('(');
-        }
-        const subStr = code.substring(0, indexOf);
-        const codeArr = subStr?.split('.');
-        const lastLen = codeArr?.[codeArr.length - 1]?.length;
-        indexOf = subStr.length - lastLen;
-      }
+    const characterOffset = 8;
+    const lines = code.split('\n');
+    // @ts-ignore
+    const lineOffset = lines.findLastIndex((item) => item.startsWith('return wrapper'));
 
-      return {
-        ...detail,
-        offset: detail.offset + indexOf,
-      };
-    }
+    return {
+      line: detail.line + lineOffset,
+      offset: characterOffset,
+    };
   }
 
   /**
@@ -1425,22 +1473,21 @@ export class CallLogic extends LogicItem {
 
   static isSystem(node: CallLogic) {
     // 如此处需要更新，则需同时更新 menu-poppup 组件下的 includesSystemLogic 校验逻辑
-    return ['nasl.process', 'nasl.browser', 'nasl.auth', 'nasl.configuration', 'nasl.io'].includes(
-      node?.calleeNamespace,
-    );
+    return ['nasl.process', 'nasl.browser', 'nasl.auth', 'nasl.configuration', 'nasl.io'].includes(node?.calleeNamespace);
   }
 
   public static getDefaultOptions() {
     return {};
   }
-  //================================================================================
+  //= ===============================================================================
   // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
   // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-  //================================================================================
+  //= ===============================================================================
 }
 
+classMap.CallLogic = CallLogic;
 export default CallLogic;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================