@lcap/nasl 3.4.0-alpha.1 → 3.4.0-alpha.2

package/src/concepts/CallLogic__.ts

@@ -1,26 +1,26 @@
 import * as JSON5 from 'json5';
-import { v4 as uuidv4 } from 'uuid';
 import { TranslatorState, shiftState, withSourceMap, createCompilerState } from '../translator';
 import Logic from './Logic__';
-import Param from './Param__';
+import type 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 { TSPPosition } from '../translator/tsp';
+import type { ViewElement, ElementToVueOptions } from './ViewElement__';
+import type { MemberExpression } from './MemberExpression__';
+import type { TSPPosition } from '../translator';
 
-//= ===============================================================================
+//================================================================================
 // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
 // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-//= ===============================================================================
+//================================================================================
 import { EventPayload, Params } from '../common/EventEmitter';
-import { concept, excludedInJSON, property } from '../decorators';
+import { concept, excludedInJSON, property, className } 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,10 +29,15 @@ import LogicItem from './LogicItem__';
 /**
  * 调用逻辑
  */
+@className
 @concept('调用逻辑')
+// @ts-ignore
 export class CallLogic extends LogicItem {
+  /** 类名 */
+  static readonly ConceptName = 'CallLogic';
+
   /**
-   * uuid
+   * 唯一标识
    */
   @property()
   uuid: string = uuidv4().replace(/-/g, '');
@@ -40,8 +45,8 @@ export class CallLogic extends LogicItem {
   /**
    * 产品概念
    */
-  @property()
-  concept: 'CallLogic' | 'CallConnector' = 'CallLogic';
+  // @ts-ignore
+  readonly concept!: 'CallLogic' | 'CallConnector';
 
   /**
    * 调用逻辑标题
@@ -53,25 +58,29 @@ export class CallLogic extends LogicItem {
    * calleeNamespace
    */
   @property()
-  calleeNamespace: string = undefined;
+  // @ts-ignore
+  calleeNamespace!: string;
 
   /**
    * calleeName
    */
   @property()
-  calleeName: string = undefined;
+  // @ts-ignore
+  calleeName!: string;
 
   /**
    * shortcut
    */
   @property()
-  shortcut: boolean = undefined;
+  // @ts-ignore
+  shortcut!: boolean;
 
   /**
    * handleError
    */
   @property()
-  handleError: boolean = undefined;
+  // @ts-ignore
+  handleError!: boolean;
 
   /**
    * 类型参数
@@ -89,28 +98,25 @@ export class CallLogic extends LogicItem {
    * 验证规则
    */
   @property('BindAttribute')
-  validation: BindAttribute = undefined;
+  // @ts-ignore
+  validation!: BindAttribute;
 
   /**
    * 是否被折叠
    */
   @property()
-  isAdvancedSettingFolded: boolean = undefined;
+  // @ts-ignore
+  isAdvancedSettingFolded!: boolean;
 
   /**
    * @param source 需要合并的部分参数
    */
   constructor(source?: Partial<CallLogic>) {
-    // @ts-ignore
-    source = { ...CallLogic.getDefaultOptions(), ...source };
+    source = Object.assign({}, 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;
   }
@@ -171,7 +177,7 @@ export class CallLogic extends LogicItem {
           ...options,
         },
         this,
-        'typeArguments'
+        'typeArguments',
       );
     }
     this.typeArguments.splice(index, 0, typeAnnotation);
@@ -271,7 +277,7 @@ export class CallLogic extends LogicItem {
           ...options,
         },
         this,
-        'arguments'
+        'arguments',
       );
     }
     this.arguments.splice(index, 0, argument);
@@ -349,12 +355,12 @@ export class CallLogic extends LogicItem {
   removeTypeAnnotation(typeAnnotation: TypeAnnotation): void;
 
   removeTypeAnnotation(options: TypeAnnotation) {
-    const typeAnnotation: TypeAnnotation = options;
+    let typeAnnotation: TypeAnnotation = options;
     return typeAnnotation.delete();
   }
 
   __removeTypeAnnotation(typeAnnotation: TypeAnnotation) {
-    const { parentKey } = typeAnnotation;
+    const parentKey = typeAnnotation.parentKey;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -383,12 +389,12 @@ export class CallLogic extends LogicItem {
   removeArgument(argument: Argument): void;
 
   removeArgument(options: Argument) {
-    const argument: Argument = options;
+    let argument: Argument = options;
     return argument.delete();
   }
 
   __removeLogicItem(logicItem: LogicItem) {
-    const { parentKey } = logicItem;
+    const parentKey = logicItem.parentKey;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -427,7 +433,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;
@@ -436,7 +442,7 @@ export class CallLogic extends LogicItem {
   }
 
   __removeBindAttribute(bindAttribute: BindAttribute) {
-    const { parentKey } = bindAttribute;
+    const parentKey = bindAttribute.parentKey;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -458,10 +464,10 @@ export class CallLogic extends LogicItem {
     return params;
   }
 
-  //= ===============================================================================
+  //================================================================================
   // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
   // 自动生成的代码已结束。下面可以手动编写。
-  //= ===============================================================================
+  //================================================================================
 
   get tsCalleeNamespace() {
     if (this.calleeNamespace) {
@@ -553,7 +559,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);
       }
     });
@@ -578,7 +584,7 @@ export class CallLogic extends LogicItem {
           keyword: param.name,
           expression: param.isFunction ? new Identifier() : param.defaultExpression,
         },
-        this
+        this,
       );
       Object.assign(argument, relationOptions);
       return argument;
@@ -612,7 +618,7 @@ export class CallLogic extends LogicItem {
         handleError,
       });
       if (parentNode.concept === 'MemberExpression') {
-        const topNode = parentNode.firstMemberExpressionNode();
+        const topNode = (parentNode as MemberExpression).firstMemberExpressionNode();
         topNode.replaceNode(this);
       }
       this.app.emit('collect:end');
@@ -634,8 +640,8 @@ export class CallLogic extends LogicItem {
         {
           ...argument,
         },
-        this
-      )
+        this,
+      ),
     );
     const relationOptions = { parentNode: this, parentKey: 'arguments' };
     /**
@@ -649,7 +655,7 @@ export class CallLogic extends LogicItem {
             {
               keyword: param.name,
             },
-            this
+            this,
           );
           Object.assign(newParam, relationOptions);
           newArgs.push(newParam);
@@ -703,11 +709,15 @@ export class CallLogic extends LogicItem {
       code += '$global.';
     }
     const completeArguments = this.completeArguments;
+
     if (this.concept === 'CallLogic' && this.getAncestor('BindAttribute')?.name === 'data-source') {
-        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}']`
+      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
+        }']`;
         code += `({
             config: {
                 download: false,
@@ -721,36 +731,41 @@ export class CallLogic extends LogicItem {
         code += `path: {},
         body: {`;
         const logic = getNodeByNodeCallee(this.app, this.calleewholeKey, (key: string, tempPoint: any) => {
-            // 处理 NASL 中未定义的假的节点
-            if (key === tempPoint?.name) {
-                return {
-                    tempPoint,
-                    state: 1,
-                };
-            }
-            const t = tempPoint?.[key];
+          // 处理 NASL 中未定义的假的节点
+          if (key === tempPoint?.name) {
             return {
-                tempPoint: t,
-                state: Array.isArray(t) ? 2 : 1,
+              tempPoint,
+              state: 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 += '(';
@@ -760,6 +775,7 @@ export class CallLogic extends LogicItem {
       });
     }
     code += ')}';
+
     return code;
   }
 
@@ -790,7 +806,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;
@@ -864,7 +880,9 @@ 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,
@@ -918,7 +936,8 @@ 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');
@@ -935,7 +954,10 @@ 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',
@@ -953,7 +975,10 @@ 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',
@@ -1122,7 +1147,8 @@ 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})`;
     }
@@ -1180,7 +1206,10 @@ 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) => {
           // 页面逻辑先都非必填
@@ -1189,8 +1218,12 @@ 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) => {
@@ -1200,7 +1233,24 @@ export class CallLogic extends LogicItem {
         });
         code += ')';
       } else {
-        code += '()';
+        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 += '()';
+        }
       }
     };
 
@@ -1419,19 +1469,41 @@ export class CallLogic extends LogicItem {
 
   /** 获取类型标注时的偏移量 */
   getQuickInfoOffset(detail: TSPPosition, code: string): TSPPosition {
-    if (!this.useIife) {
-      return { ...detail };
-    }
+    if (this.useIife) {
+      const characterOffset = 8;
+      const lines = code.split('\n');
+      const lineOffset = lines.findIndex((item) => item.startsWith('return wrapper'));
 
-    const characterOffset = 8;
-    const lines = code.split('\n');
-    // @ts-ignore
-    const lineOffset = lines.findLastIndex((item) => item.startsWith('return wrapper'));
+      return {
+        line: detail.line + lineOffset,
+        offset: characterOffset,
+      };
+    } else {
+      let indexOf = 0;
 
-    return {
-      line: detail.line + lineOffset,
-      offset: characterOffset,
-    };
+      // 没有括号，表示括号在外面
+      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;
+      }
+
+      return {
+        ...detail,
+        offset: detail.offset + indexOf,
+      };
+    }
   }
 
   /**
@@ -1473,21 +1545,22 @@ 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;
-//= ===============================================================================
+//================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//= ===============================================================================
+//================================================================================