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

package/src/concepts/CallInterface__.ts

@@ -1,41 +1,37 @@
-import { TranslatorState, shiftState, withSourceMap } from '../translator';
+import { v4 as uuidv4 } from 'uuid';
+import { TranslatorState, shiftState, withSourceMap, createCompilerState } from '../translator';
 import Interface from './Interface__';
+import NullLiteral from './NullLiteral__';
 import { ElementToVueOptions } from './ViewElement__';
-import { getNodeByNodeCallee } from '../automate/engine/utils';
+import { getConceptConstructor, concept, excludedInJSON, property } from '../decorators';
+import { getNodeByNodeCallee, getNaslNodeByNodeCallee } from '../automate/engine/utils';
 import InterfaceParam from './InterfaceParam__';
 import Structure from './Structure__';
 import Module from './Module__';
 import AuthInterface from './AuthInterface__';
-import type { MemberExpression } from './MemberExpression__';
 import Frontend from './Frontend__';
-//================================================================================
+//= ===============================================================================
 // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
 // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-//================================================================================
+//= ===============================================================================
 import { EventPayload, Params } from '../common/EventEmitter';
-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 Argument from './Argument__';
 import LogicItem from './LogicItem__';
 
 /**
  * 调用接口
  */
-@className
 @concept('调用接口')
-// @ts-ignore
 export class CallInterface extends LogicItem {
-  /** 类名 */
-  static readonly ConceptName = 'CallInterface';
-
   /**
    * 产品概念
    */
-  // @ts-ignore
-  readonly concept!: 'CallInterface' | 'CallAuthInterface';
+  @property()
+  concept: 'CallInterface' | 'CallAuthInterface' = 'CallInterface';
 
   /**
    * 调用接口标题
@@ -47,22 +43,19 @@ export class CallInterface 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;
 
   /**
    * 实际参数
@@ -74,11 +67,16 @@ export class CallInterface extends LogicItem {
    * @param source 需要合并的部分参数
    */
   constructor(source?: Partial<CallInterface>) {
-    source = Object.assign({}, CallInterface.getDefaultOptions(), source);
+    // @ts-ignore
+    source = { ...CallInterface.getDefaultOptions(), ...source };
     super(source);
     super.subConstructor(source);
   }
 
+  getClassName() {
+    return 'CallInterface';
+  }
+
   static from(source: any, parentNode?: any, parentKey?: string): CallInterface {
     return super.from(source, parentNode, parentKey) as CallInterface;
   }
@@ -112,7 +110,7 @@ export class CallInterface extends LogicItem {
           ...options,
         },
         this,
-        'arguments',
+        'arguments'
       );
     }
     this.arguments.splice(index, 0, argument);
@@ -190,12 +188,12 @@ export class CallInterface 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,
@@ -217,10 +215,10 @@ export class CallInterface extends LogicItem {
     return params;
   }
 
-  //================================================================================
+  //= ===============================================================================
   // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
   // 自动生成的代码已结束。下面可以手动编写。
-  //================================================================================
+  //= ===============================================================================
 
   /**
    * 标识当前删除是否为鉴权方式更新导致的删除。
@@ -240,8 +238,7 @@ export class CallInterface extends LogicItem {
           let refCount = 0;
           const refs = (await oldItf.findUsage()) as Map<any, any>;
 
-          const isOnlyUsedByFronted =
-            refs.size === 1 && Array.from(refs.keys()).filter((key) => key instanceof Frontend).length > 0;
+          const isOnlyUsedByFronted = refs.size === 1 && Array.from(refs.keys()).filter((key) => key instanceof Frontend).length > 0;
           const isAuthInterface = oldItf instanceof AuthInterface;
           // 当仅仅只有前端使用，且该接口是鉴权接口时需要将接口还原为普通接口
           if (isOnlyUsedByFronted && isAuthInterface) {
@@ -261,8 +258,7 @@ export class CallInterface extends LogicItem {
           if (!refCount) {
             //  删除旧的接口、接口的数据结构
             this.app.emit('collect:start', { actionMsg: '删除接口' });
-            const parentModule = oldItf.module || oldItf.connector;
-            const oldStructuresSet = oldItf.getRefStructures(parentModule.structures);
+            const oldStructuresSet = oldItf.getRefStructures(oldItf.module.structures);
             oldStructuresSet.forEach((s: Structure) => s.delete());
             oldItf.delete();
             oldItf.afterDelete();
@@ -406,7 +402,7 @@ export class CallInterface extends LogicItem {
   }
 
   @withSourceMap()
-  toEmbeddedTS(state?: TranslatorState): string {
+  toEmbeddedTS(state = createCompilerState()): string {
     const { completeArguments, parentNode, parentKey, tsCalleeKey } = this;
 
     let code = `${tsCalleeKey}(`;
@@ -440,6 +436,41 @@ export class CallInterface extends LogicItem {
     return code;
   }
 
+  @withSourceMap()
+  toNaturalTS(state = createCompilerState()): string {
+    const { completeArguments, parentNode, parentKey, tsCalleeKey } = this;
+
+    let code = `${tsCalleeKey}(`;
+
+    if (completeArguments?.length) {
+      code += '{';
+      completeArguments?.forEach((arg, index) => {
+        code += arg?.toNaturalTS(shiftState(state, code), arg.paramRequired, true);
+        if (index !== this.arguments?.length - 1) code += ', ';
+      });
+      code += '}';
+    }
+    code += ')';
+    if (
+      parentNode.concept !== 'Argument' &&
+      parentKey !== 'test' &&
+      parentKey !== 'patterns' &&
+      parentKey !== 'returnExpression' &&
+      parentNode.concept !== 'Assignment' &&
+      parentNode.concept !== 'ForEachStatement' &&
+      parentNode.concept !== 'SelectMembers' &&
+      parentNode.concept !== 'UnaryExpression' &&
+      parentNode.concept !== 'BinaryExpression' &&
+      parentNode.concept !== 'MemberExpression' &&
+      parentNode.concept !== 'StringInterpolation' &&
+      !['NewMap', 'NewList'].includes(parentNode.concept)
+    ) {
+      code += ';';
+    }
+    code += '\n';
+    return code;
+  }
+
   toBrief(): string {
     return this.arguments.map((a) => a.toBrief()).join('<br />');
   }
@@ -580,7 +611,7 @@ export class CallInterface extends LogicItem {
         folded: false,
       });
       if (parentNode.concept === 'MemberExpression') {
-        const topNode = (parentNode as MemberExpression).firstMemberExpressionNode();
+        const topNode = parentNode.firstMemberExpressionNode();
         topNode.replaceNode(this);
       }
       this.app.emit('collect:end');
@@ -772,14 +803,15 @@ export class CallInterface extends LogicItem {
     structuresSet.forEach((s) => module.addStructure(Structure.from(s)));
     this.app.emit('collect:end');
   }
-  //================================================================================
+  //= ===============================================================================
   // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
   // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-  //================================================================================
+  //= ===============================================================================
 }
 
+classMap.CallInterface = CallInterface;
 export default CallInterface;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================