@lcap/nasl 3.3.0-alpha.3 → 3.3.0-beta.0

package/src/concepts/CallConnector__.ts

@@ -1,44 +1,50 @@
-import { v4 as uuidv4 } from 'uuid';
-import Connector from './Connector__';
+import type Connector from './Connector__';
 import { getNodeByNodeCallee } from '../automate/engine/utils';
 import Logic from './Logic__';
 import Argument from './Argument__';
 import Identifier from './Identifier__';
 import TypeAnnotation from './TypeAnnotation__';
 import Connection from './Connection__';
-import { TranslatorState, shiftState, withSourceMap, createCompilerState } from '../translator';
+import { TranslatorState, shiftState, withSourceMap } from '../translator';
+import Namespace from './Namespace__';
 
-//= ===============================================================================
+//================================================================================
 // 从这里开始到结尾注释之间的代码由 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 CallLogic from './CallLogic__';
 
 /**
- * 连接器
+ * 调用连接器
  */
-@concept('连接器')
+@className
+@concept('调用连接器')
+// @ts-ignore
 export class CallConnector extends CallLogic {
+  /** 类名 */
+  static readonly ConceptName = 'CallConnector';
+
   /**
    * 产品概念
    */
-  @property()
-  concept: 'CallConnector' = 'CallConnector';
+  // @ts-ignore
+  readonly concept!: 'CallConnector';
 
   /**
    * calleeConnectionName
    */
   @property()
-  calleeConnectionName: string = undefined;
+  // @ts-ignore
+  calleeConnectionName!: string;
 
   /**
-   * 连接器标题
+   * 调用连接器标题
    */
   @property()
   label: string = '连接器';
@@ -47,16 +53,11 @@ export class CallConnector extends CallLogic {
    * @param source 需要合并的部分参数
    */
   constructor(source?: Partial<CallConnector>) {
-    // @ts-ignore
-    source = { ...CallConnector.getDefaultOptions(), ...source };
+    source = Object.assign({}, CallConnector.getDefaultOptions(), source);
     super(source);
     super.subConstructor(source);
   }
 
-  getClassName() {
-    return 'CallConnector';
-  }
-
   static from(source: any, parentNode?: any, parentKey?: string): CallConnector {
     return super.from(source, parentNode, parentKey) as CallConnector;
   }
@@ -73,10 +74,10 @@ export class CallConnector extends CallLogic {
     });
   }
 
-  //= ===============================================================================
+  //================================================================================
   // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
   // 自动生成的代码已结束。下面可以手动编写。
-  //= ===============================================================================
+  //================================================================================
   setCallCtor(connection: Connection) {
     const calleeConnectionName = connection.name;
     if (this.app) {
@@ -143,20 +144,30 @@ export class CallConnector extends CallLogic {
   getCallNode() {
     let node;
     try {
-      node = getNodeByNodeCallee(this.app, this.calleewholeKey, (key: any, tempPoint: any) => {
-        // 处理 NASL 中未定义的假的节点
-        if (key === tempPoint.name) {
+      // 自定义连接器走额外的检索逻辑
+      if (this.connection?.connector?.connectorKind === 'app') {
+        const {
+          connection: { connector },
+        } = this;
+        const namespaceName = this.calleeNamespace.split('.').pop();
+        const ns = (connector as Connector).namespaces.find((nsItem) => nsItem.name === namespaceName);
+        node = ns?.logics?.find?.((logic) => logic.name === this.calleeName);
+      } else {
+        node = getNodeByNodeCallee(this.app, this.calleewholeKey, (key: any, tempPoint: any) => {
+          // 处理 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,
-        };
-      });
+        });
+      }
     } catch (err) {}
     return node;
   }
@@ -169,7 +180,7 @@ export class CallConnector extends CallLogic {
           keyword: param.name,
           expression: param.isFunction ? new Identifier() : param.defaultExpression,
         },
-        this
+        this,
       );
       Object.assign(argument, relationOptions);
       return argument;
@@ -190,7 +201,7 @@ export class CallConnector extends CallLogic {
   }
 
   @withSourceMap()
-  toEmbeddedTS(state = createCompilerState()): string {
+  toEmbeddedTS(state?: TranslatorState): string {
     let code = '';
     if (this.handleError) {
       code += 'nasl.util.autoHandleError(';
@@ -228,7 +239,6 @@ export class CallConnector extends CallLogic {
       this.parentNode.concept !== 'Assignment' &&
       this.parentNode.concept !== 'ForEachStatement' &&
       this.parentNode.concept !== 'SelectMembers' &&
-      this.parentNode.concept !== 'UnaryExpression' &&
       this.parentNode.concept !== 'BinaryExpression' &&
       this.parentNode.concept !== 'StringInterpolation' &&
       this.parentNode.concept !== 'MemberExpression' &&
@@ -238,15 +248,14 @@ export class CallConnector extends CallLogic {
     }
     return code;
   }
-  //= ===============================================================================
+  //================================================================================
   // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
   // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-  //= ===============================================================================
+  //================================================================================
 }
 
-classMap.CallConnector = CallConnector;
 export default CallConnector;
-//= ===============================================================================
+//================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//= ===============================================================================
+//================================================================================

package/src/concepts/CallFunction__.ts

@@ -1,22 +1,22 @@
-import { v4 as uuidv4 } from 'uuid';
-import { TranslatorState, shiftState, withSourceMap, indent, createCompilerState } from '../translator';
+import { TranslatorState, shiftState, withSourceMap } from '../translator';
 import Function from './Function__';
 import { ElementToVueOptions } from './ViewElement__';
 import Identifier from './Identifier__';
 import { getBuildInFunction } from '../server/getFunctions';
 import AnonymousFunction from './AnonymousFunction__';
+import MemberExpression from './MemberExpression__';
 import asyncFuncMap from '../common/asyncFuncMap';
 
-//= ===============================================================================
+//================================================================================
 // 从这里开始到结尾注释之间的代码由 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 LogicItem from './LogicItem__';
@@ -24,13 +24,18 @@ import LogicItem from './LogicItem__';
 /**
  * 调用函数
  */
+@className
 @concept('调用函数')
+// @ts-ignore
 export class CallFunction extends LogicItem {
+  /** 类名 */
+  static readonly ConceptName = 'CallFunction';
+
   /**
    * 产品概念
    */
-  @property()
-  concept: 'CallFunction' = 'CallFunction';
+  // @ts-ignore
+  readonly concept!: 'CallFunction';
 
   /**
    * 调用函数标题
@@ -42,19 +47,22 @@ export class CallFunction 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;
 
   /**
    * 类型参数
@@ -72,16 +80,11 @@ export class CallFunction extends LogicItem {
    * @param source 需要合并的部分参数
    */
   constructor(source?: Partial<CallFunction>) {
-    // @ts-ignore
-    source = { ...CallFunction.getDefaultOptions(), ...source };
+    source = Object.assign({}, CallFunction.getDefaultOptions(), source);
     super(source);
     super.subConstructor(source);
   }
 
-  getClassName() {
-    return 'CallFunction';
-  }
-
   static from(source: any, parentNode?: any, parentKey?: string): CallFunction {
     return super.from(source, parentNode, parentKey) as CallFunction;
   }
@@ -115,7 +118,7 @@ export class CallFunction extends LogicItem {
           ...options,
         },
         this,
-        'typeArguments'
+        'typeArguments',
       );
     }
     this.typeArguments.splice(index, 0, typeAnnotation);
@@ -215,7 +218,7 @@ export class CallFunction extends LogicItem {
           ...options,
         },
         this,
-        'arguments'
+        'arguments',
       );
     }
     this.arguments.splice(index, 0, argument);
@@ -293,12 +296,12 @@ export class CallFunction 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,
@@ -327,12 +330,12 @@ export class CallFunction 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,
@@ -354,10 +357,10 @@ export class CallFunction extends LogicItem {
     return params;
   }
 
-  //= ===============================================================================
+  //================================================================================
   // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
   // 自动生成的代码已结束。下面可以手动编写。
-  //= ===============================================================================
+  //================================================================================
   toVue(options?: ElementToVueOptions): string {
     if (options?.finalCode === false) {
       return this.toUI();
@@ -455,7 +458,8 @@ export class CallFunction extends LogicItem {
     code += ')';
     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})`;
     }
@@ -467,7 +471,9 @@ export class CallFunction extends LogicItem {
   }
 
   get calleeKey(): string {
-    return this.calleeName ? `${this.calleeNamespace ? `${this.calleeNamespace}.` : ''}${this.calleeName}` : '__IDENTIFIER__';
+    return this.calleeName
+      ? `${this.calleeNamespace ? `${this.calleeNamespace}.` : ''}${this.calleeName}`
+      : '__IDENTIFIER__';
   }
 
   _setCallee(func: Function) {
@@ -488,7 +494,7 @@ export class CallFunction extends LogicItem {
         {
           keyword: param.name,
         },
-        this
+        this,
       );
       if (param.defaultExpression instanceof Identifier) {
         argument.expression = BaseNode.from(param.defaultExpression, argument, 'expression') as Identifier;
@@ -522,7 +528,7 @@ export class CallFunction extends LogicItem {
         folded: false,
       });
       if (parentNode.concept === 'MemberExpression') {
-        const topNode = parentNode.firstMemberExpressionNode();
+        const topNode = (parentNode as MemberExpression).firstMemberExpressionNode();
         topNode.replaceNode(this);
       }
       this.detectArgs();
@@ -541,7 +547,8 @@ export class CallFunction extends LogicItem {
 
   detectArgs(obj?: any) {
     const func = getBuildInFunction(this.calleeName);
-    const shouldJudge = !!func && ['ListDistinctBy', 'MapDistinctBy', 'ListGroupBy', 'ListToMap', 'MapFilter'].includes(this.calleeName);
+    const shouldJudge =
+      !!func && ['ListDistinctBy', 'MapDistinctBy', 'ListGroupBy', 'ListToMap', 'MapFilter'].includes(this.calleeName);
     if (shouldJudge && Array.isArray(this.arguments)) {
       const depParamIdx = func.params.findIndex((param) => ['list', 'map'].includes(param.name));
       if (depParamIdx !== -1) {
@@ -558,7 +565,7 @@ export class CallFunction extends LogicItem {
   }
 
   @withSourceMap()
-  toEmbeddedTS(state = createCompilerState()): string {
+  toEmbeddedTS(state?: TranslatorState): string {
     let code = '';
     const { calleeKey } = this;
     if (['nasl.util.Convert', 'nasl.util.FromString'].includes(calleeKey)) {
@@ -595,62 +602,23 @@ export class CallFunction extends LogicItem {
     return code;
   }
 
-  @withSourceMap()
-  toNaturalTS(state = createCompilerState()): string {
-    let code = '';
-    code = this.calleeKey;
-    if (Array.isArray(this.typeArguments) && this.typeArguments.length) {
-      // 如果有内容
-      const typeAnnotation = this.typeArguments[0];
-      code += `<`;
-      code += `${typeAnnotation.toNaturalTS(shiftState(state, code, { tabSize: 0 }))}>`;
-    }
-    if (this.arguments.length <= 3) {
-      code += '(';
-      this.arguments.forEach((arg, index) => {
-        code += arg.toNaturalTS(shiftState(state, code, { tabSize: 0 }));
-        if (index !== this.arguments.length - 1)
-          code += ', ';
-      });
-      if (this?.typeArguments?.[0]?.typeKind === 'reference') {
-        if (this.arguments?.length > 0)
-          code += ', ';
-        code += `{ typeNamespace: '${this.typeArguments[0].typeNamespace}' }`;
-      }
-      code += `)`;
-    } else {
-      code += '(\n';
-      this.arguments.forEach((arg, index) => {
-        code += arg.toNaturalTS(shiftState(state, code, { tabSize: state.tabSize + 1 }));
-        if (index !== this.arguments.length - 1) {
-          code += ',\n';
-        } else if (this?.typeArguments?.[0]?.typeKind === 'reference') {
-          code += `,\n{ typeNamespace: '${this.typeArguments[0].typeNamespace}' }`;
-        } else {
-          code += '\n';
-        }
-      });
-      code += `${indent(state.tabSize)})`;
-    }
-    return code;
-  }
-
   checkParamNameOfAnonymousFunction() {
-    const fns = this.arguments.map((arg) => arg.expression).filter((exp) => exp instanceof AnonymousFunction) as unknown as AnonymousFunction[];
+    const fns = this.arguments
+      .map((arg) => arg.expression)
+      .filter((exp) => exp instanceof AnonymousFunction) as unknown as AnonymousFunction[];
 
     fns.forEach((fn) => {
       fn.checkParamName();
     });
   }
-  //= ===============================================================================
+  //================================================================================
   // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
   // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-  //= ===============================================================================
+  //================================================================================
 }
 
-classMap.CallFunction = CallFunction;
 export default CallFunction;
-//= ===============================================================================
+//================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//= ===============================================================================
+//================================================================================