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

package/src/concepts/BatchAssignment__.ts

@@ -1,18 +1,19 @@
-import { TranslatorState, shiftState, withSourceMap } from '../translator';
+import { v4 as uuidv4 } from 'uuid';
+import { TranslatorState, shiftState, withSourceMap, createCompilerState } from '../translator';
 import { ElementToVueOptions } from './ViewElement__';
 import MemberExpression from './MemberExpression__';
 import Assignment from './Assignment__';
 import Logic from './Logic__';
-//================================================================================
+//= ===============================================================================
 // 从这里开始到结尾注释之间的代码由 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 SelectMembers from './SelectMembers__';
 import AssignmentLine from './AssignmentLine__';
 import LogicItem from './LogicItem__';
@@ -20,18 +21,13 @@ import LogicItem from './LogicItem__';
 /**
  * 批量赋值语句
  */
-@className
 @concept('批量赋值语句')
-// @ts-ignore
 export class BatchAssignment extends LogicItem {
-  /** 类名 */
-  static readonly ConceptName = 'BatchAssignment';
-
   /**
    * 产品概念
    */
-  // @ts-ignore
-  readonly concept!: 'BatchAssignment';
+  @property()
+  concept: 'BatchAssignment' = 'BatchAssignment';
 
   /**
    * 批量赋值语句标题
@@ -43,8 +39,7 @@ export class BatchAssignment extends LogicItem {
    * 左边项
    */
   @property('SelectMembers')
-  // @ts-ignore
-  left!: SelectMembers;
+  left: SelectMembers = undefined;
 
   /**
    * 右边项
@@ -68,11 +63,16 @@ export class BatchAssignment extends LogicItem {
    * @param source 需要合并的部分参数
    */
   constructor(source?: Partial<BatchAssignment>) {
-    source = Object.assign({}, BatchAssignment.getDefaultOptions(), source);
+    // @ts-ignore
+    source = { ...BatchAssignment.getDefaultOptions(), ...source };
     super(source);
     super.subConstructor(source);
   }
 
+  getClassName() {
+    return 'BatchAssignment';
+  }
+
   static from(source: any, parentNode?: any, parentKey?: string): BatchAssignment {
     return super.from(source, parentNode, parentKey) as BatchAssignment;
   }
@@ -132,6 +132,7 @@ export class BatchAssignment extends LogicItem {
     const excludedSet = new Set(excludedList);
     return ((this.rights as SelectMembers[]) || []).filter((item) => !excludedSet.has(item)).map((item) => item.name);
   }
+
   getSelectMembersUniqueName(name = 'selectMembers1') {
     return utils.unique(name, this.getSelectMembersExistingNames(), undefined, false);
   }
@@ -168,7 +169,7 @@ export class BatchAssignment extends LogicItem {
           name: this.getSelectMembersUniqueName(),
         },
         this,
-        'rights',
+        'rights'
       );
     } else if (typeof options === 'string') {
       selectMembers = SelectMembers.from(
@@ -177,7 +178,7 @@ export class BatchAssignment extends LogicItem {
           name: options,
         },
         this,
-        'rights',
+        'rights'
       );
     } else if (options instanceof SelectMembers) {
       options.ensureDelete(); // 同一实例不支持多处存在
@@ -190,7 +191,7 @@ export class BatchAssignment extends LogicItem {
           ...options,
         },
         this,
-        'rights',
+        'rights'
       );
     }
     this.rights.splice(index, 0, selectMembers);
@@ -283,10 +284,9 @@ export class BatchAssignment extends LogicItem {
 
   getAssignmentLineExistingNames(excludedList: Array<AssignmentLine> = []) {
     const excludedSet = new Set(excludedList);
-    return ((this.assignmentLines as AssignmentLine[]) || [])
-      .filter((item) => !excludedSet.has(item))
-      .map((item) => item.name);
+    return ((this.assignmentLines as AssignmentLine[]) || []).filter((item) => !excludedSet.has(item)).map((item) => item.name);
   }
+
   getAssignmentLineUniqueName(name = 'assignmentLine1') {
     return utils.unique(name, this.getAssignmentLineExistingNames(), undefined, false);
   }
@@ -323,7 +323,7 @@ export class BatchAssignment extends LogicItem {
           name: this.getAssignmentLineUniqueName(),
         },
         this,
-        'assignmentLines',
+        'assignmentLines'
       );
     } else if (typeof options === 'string') {
       assignmentLine = AssignmentLine.from(
@@ -332,7 +332,7 @@ export class BatchAssignment extends LogicItem {
           name: options,
         },
         this,
-        'assignmentLines',
+        'assignmentLines'
       );
     } else if (options instanceof AssignmentLine) {
       options.ensureDelete(); // 同一实例不支持多处存在
@@ -345,7 +345,7 @@ export class BatchAssignment extends LogicItem {
           ...options,
         },
         this,
-        'assignmentLines',
+        'assignmentLines'
       );
     }
     this.assignmentLines.splice(index, 0, assignmentLine);
@@ -453,7 +453,7 @@ export class BatchAssignment extends LogicItem {
     if (typeof options === 'string') {
       left = this.left;
       if (!left) {
-        throw new Error('找不到选择成员 ' + options);
+        throw new Error(`找不到选择成员 ${options}`);
       }
     } else {
       left = options;
@@ -478,7 +478,7 @@ export class BatchAssignment extends LogicItem {
     if (typeof options === 'string') {
       selectMembers = (this.rights as SelectMembers[]).find((item) => item.name === options);
       if (!selectMembers) {
-        throw new Error('找不到选择成员 ' + options);
+        throw new Error(`找不到选择成员 ${options}`);
       }
     } else {
       selectMembers = options;
@@ -487,7 +487,7 @@ export class BatchAssignment extends LogicItem {
   }
 
   __removeSelectMembers(selectMembers: SelectMembers) {
-    const parentKey = selectMembers.parentKey;
+    const { parentKey } = selectMembers;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -526,7 +526,7 @@ export class BatchAssignment extends LogicItem {
     if (typeof options === 'string') {
       assignmentLine = (this.assignmentLines as AssignmentLine[]).find((item) => item.name === options);
       if (!assignmentLine) {
-        throw new Error('找不到赋值连线 ' + options);
+        throw new Error(`找不到赋值连线 ${options}`);
       }
     } else {
       assignmentLine = options;
@@ -535,7 +535,7 @@ export class BatchAssignment extends LogicItem {
   }
 
   __removeAssignmentLine(assignmentLine: AssignmentLine) {
-    const parentKey = assignmentLine.parentKey;
+    const { parentKey } = assignmentLine;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -557,10 +557,10 @@ export class BatchAssignment extends LogicItem {
     return params;
   }
 
-  //================================================================================
+  //= ===============================================================================
   // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
   // 自动生成的代码已结束。下面可以手动编写。
-  //================================================================================
+  //= ===============================================================================
 
   public static getDefaultOptions() {
     return {
@@ -588,11 +588,7 @@ export class BatchAssignment extends LogicItem {
     for (let i = 0; i < this.left.members.length - 1; i++) {
       const { completeName } = this.left.members[i];
       const preCompleteName = this.left.members[i + 1].completeName;
-      if (
-        preCompleteName.includes(completeName) &&
-        completeName.split('.').length !== preCompleteName.split('.').length
-      )
-        leftLineSet.add(i);
+      if (preCompleteName.includes(completeName) && completeName.split('.').length !== preCompleteName.split('.').length) leftLineSet.add(i);
     }
     // await new Promise((resolve) => setTimeout(resolve, 400));
     const autoConnection = () => {
@@ -614,11 +610,7 @@ export class BatchAssignment extends LogicItem {
           for (let i = 0; i < members.length - 1; i++) {
             const { completeName } = members[i];
             const preCompleteName = members[i + 1].completeName;
-            if (
-              preCompleteName.includes(completeName) &&
-              completeName.split('.').length !== preCompleteName.split('.').length
-            )
-              rightLineSet.add(i);
+            if (preCompleteName.includes(completeName) && completeName.split('.').length !== preCompleteName.split('.').length) rightLineSet.add(i);
           }
 
           if (leftName === expression.name && expression.__TypeAnnotation?.typeKind === leftTypeAnnotation.typeKind) {
@@ -655,8 +647,7 @@ export class BatchAssignment extends LogicItem {
     const typeAnnotationEnd = () => {
       const l = this.left.members.every((item) => item.__isCorrectTypeAnnotation);
       const r = this.rights.every(
-        (right) =>
-          right.members.every((item) => item.__isCorrectTypeAnnotation) && right.expression.__isCorrectTypeAnnotation,
+        (right) => right.members.every((item) => item.__isCorrectTypeAnnotation) && right.expression.__isCorrectTypeAnnotation
       );
       count++;
       if (l && r) {
@@ -691,14 +682,12 @@ export class BatchAssignment extends LogicItem {
       });
       sameLevelIndex = sameLevelIndex === -1 ? members.length : sameLevelIndex;
       newMembers = members.filter((member: any, index: number) => index <= secondIndex || index >= sameLevelIndex);
-      filterLines = this.assignmentLines.filter(
-        ({ leftIndex }) => leftIndex[1] <= secondIndex || leftIndex[1] >= sameLevelIndex,
-      );
+      filterLines = this.assignmentLines.filter(({ leftIndex }) => leftIndex[1] <= secondIndex || leftIndex[1] >= sameLevelIndex);
       // members成员减少，需要重新计算后续连线下标
       filterLines.forEach((line) => {
         const distance = sameLevelIndex - secondIndex - 1;
         if (line.leftIndex[1] > secondIndex) {
-          const newLine = { ...line?.toJSON(), leftIndex: [line.leftIndex[0], line.leftIndex[1] - distance] };
+          const newLine = { ...line, leftIndex: [line.leftIndex[0], line.leftIndex[1] - distance] };
           newLines.push(newLine as AssignmentLine);
           return;
         }
@@ -720,9 +709,7 @@ export class BatchAssignment extends LogicItem {
     }
     // 左边同一个属性只能有一个连线
     if (rightToLeft) {
-      newLines = newLines.filter(
-        ({ leftIndex: index }) => !(newLeftIndex[0] === index[0] && newLeftIndex[1] === index[1]),
-      );
+      newLines = newLines.filter(({ leftIndex: index }) => !(newLeftIndex[0] === index[0] && newLeftIndex[1] === index[1]));
     }
 
     this.update({
@@ -778,10 +765,10 @@ export class BatchAssignment extends LogicItem {
       filterLine.forEach((line) => {
         if (line.rightIndex[0] > index) {
           if (line.rightIndex[1] === undefined) {
-            const newLine = { ...line?.toJSON(), rightIndex: [line.rightIndex[0] - 1] };
+            const newLine = { ...line, rightIndex: [line.rightIndex[0] - 1] };
             newLines.push(newLine as AssignmentLine);
           } else {
-            const newLine = { ...line?.toJSON(), rightIndex: [line.rightIndex[0] - 1, line.rightIndex[1]] };
+            const newLine = { ...line, rightIndex: [line.rightIndex[0] - 1, line.rightIndex[1]] };
             newLines.push(newLine as AssignmentLine);
           }
         } else {
@@ -801,7 +788,7 @@ export class BatchAssignment extends LogicItem {
     });
     const selectMember = oldNode.parentNode;
     const autoConnectionIndex = this.rights.findIndex((right) => right === selectMember);
-    const newSelectMember = { ...selectMember?.toJSON?.(), expression, members };
+    const newSelectMember = { ...selectMember, expression, members };
     let newLines: any = [];
     // 修改连线
     if (selectMember.parentKey === 'left') {
@@ -809,7 +796,7 @@ export class BatchAssignment extends LogicItem {
         if (keepMembers?.has(item.completeName)) {
           const lines = this.findlines(item, selectMember);
           lines.forEach((line) => {
-            const newLine = { ...line?.toJSON(), leftIndex: [line.leftIndex[0], index] };
+            const newLine = { ...line, leftIndex: [line.leftIndex[0], index] };
             newLines.push(newLine);
           });
         }
@@ -825,18 +812,14 @@ export class BatchAssignment extends LogicItem {
         if (keepMembers?.has(item.completeName)) {
           const lines = this.findlines(item, selectMember);
           lines.forEach((line) => {
-            const newLine = { ...line?.toJSON(), rightIndex: [line.rightIndex[0], index] };
+            const newLine = { ...line, rightIndex: [line.rightIndex[0], index] };
             newLines.push(newLine);
           });
         }
       });
       this.update({
         assignmentLines: newLines,
-        rights: [
-          ...this.rights.slice(0, splitIndex),
-          newSelectMember,
-          ...this.rights.slice(splitIndex + 1, this.rights.length),
-        ],
+        rights: [...this.rights.slice(0, splitIndex), newSelectMember, ...this.rights.slice(splitIndex + 1, this.rights.length)],
       });
     }
     this.autoConnection && this.batchAutoConnection(autoConnectionIndex);
@@ -864,23 +847,17 @@ export class BatchAssignment extends LogicItem {
       if (member === this.left) return this.assignmentLines;
 
       const firstIndex = this.rights.findIndex((right) => right === member);
-      return this.assignmentLines.filter(
-        ({ rightIndex }) => rightIndex[1] === undefined && rightIndex[0] === firstIndex,
-      );
+      return this.assignmentLines.filter(({ rightIndex }) => rightIndex[1] === undefined && rightIndex[0] === firstIndex);
     }
 
     if (this.left === selectMember) {
-      const index2 = selectMember.members.findIndex(
-        (item: MemberExpression) => item.completeName === member.completeName,
-      );
+      const index2 = selectMember.members.findIndex((item: MemberExpression) => item.completeName === member.completeName);
       this.assignmentLines.forEach((line) => {
         if (line.leftIndex[1] === index2) lines.push(line);
       });
     } else {
       const index1 = this.rights.findIndex((right) => right === selectMember);
-      const index2 = selectMember.members.findIndex(
-        (item: MemberExpression) => item.completeName === member.completeName,
-      );
+      const index2 = selectMember.members.findIndex((item: MemberExpression) => item.completeName === member.completeName);
       this.assignmentLines.forEach((line) => {
         if (line.rightIndex[0] === index1 && line.rightIndex[1] === index2) lines.push(line);
       });
@@ -899,10 +876,10 @@ export class BatchAssignment extends LogicItem {
     let index;
     if (this.assignmentLines.length === 1) {
       const { leftIndex, rightIndex } = this.assignmentLines[0];
-      leftExpression = { ...this.findMember(leftIndex, 'left')?.toJSON?.() };
-      rightExpression = { ...this.findMember(rightIndex, 'right')?.toJSON?.() };
+      leftExpression = { ...this.findMember(leftIndex, 'left') };
+      rightExpression = { ...this.findMember(rightIndex, 'right') };
     } else {
-      leftExpression = { ...this.left.expression?.toJSON?.() };
+      leftExpression = { ...this.left.expression };
     }
     const assignment = new Assignment({
       left: leftExpression as any,
@@ -921,8 +898,7 @@ export class BatchAssignment extends LogicItem {
     } else {
       index = parentNode[parentKey].findIndex((item: any) => item === this);
       const firstUpperText = parentKey[0].toUpperCase() + parentKey.slice(1, parentKey.length);
-      if (parentNode[`insertItemIn${firstUpperText}At`])
-        parentNode[`insertItemIn${firstUpperText}At`](assignment, index);
+      if (parentNode[`insertItemIn${firstUpperText}At`]) parentNode[`insertItemIn${firstUpperText}At`](assignment, index);
       else parentNode.insertItemAt(assignment, index);
     }
     this.delete();
@@ -933,8 +909,7 @@ export class BatchAssignment extends LogicItem {
   toVue(options?: ElementToVueOptions) {
     let code = '';
     this.assignmentLines.forEach(({ leftIndex, rightIndex }) => {
-      const leftValue =
-        leftIndex.length === 1 ? this.left.expression.toVue(options) : this.left.members[leftIndex[1]]?.toVue(options);
+      const leftValue = leftIndex.length === 1 ? this.left.expression.toVue(options) : this.left.members[leftIndex[1]]?.toVue(options);
       code += `${leftValue}`;
       code += ' = ';
       const rightValue =
@@ -974,7 +949,7 @@ export class BatchAssignment extends LogicItem {
     this.assignmentLines.forEach(({ leftIndex, rightIndex }) => {
       const rightIndexJson = JSON.stringify(rightIndex);
       if (rightMap.has(rightIndexJson)) {
-        rightMap.set(rightIndexJson, [...rightMap.get(rightIndexJson), leftIndex]);
+        rightMap.set(rightIndexJson, [...rightMap.get(rightIndexJson), leftIndex])
       } else {
         rightMap.set(rightIndexJson, [leftIndex]);
       }
@@ -986,12 +961,11 @@ export class BatchAssignment extends LogicItem {
         return leftItem;
       });
       const [rightFirst, rightSecond] = [rightIndex[0], rightIndex[1]];
-      const rightItem =
-        rightSecond !== undefined ? this.rights[rightFirst].members[rightSecond] : this.rights[rightFirst].expression;
+      const rightItem = rightSecond !== undefined ? this.rights[rightFirst].members[rightSecond] : this.rights[rightFirst].expression;
       leftItems.forEach((leftItem: LogicItem) => {
         code += `${leftItem.toJS(shiftState(state, code, { inline: true }))}`;
         code += ` = `;
-      });
+      })
       code += `${rightItem.toJS(shiftState(state, code, { inline: true }))}`;
       code += ';\n';
     });
@@ -999,12 +973,10 @@ export class BatchAssignment extends LogicItem {
   }
 
   @withSourceMap()
-  toEmbeddedTS(state?: TranslatorState): string {
+  toEmbeddedTS(state = createCompilerState()): string {
     let code = '// 以下是批量赋值的翻译代码\n';
     // 提供sourcemap，获取类型标注和上下文可选变量
-    code += this.left.expression
-      ? `${this.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))}`
-      : '';
+    code += this.left.expression ? `${this.left.expression.toEmbeddedTS(shiftState(state, code, { inline: true }))}` : '';
     code += ';\n';
     this.left.members.forEach((member) => {
       code += member.toEmbeddedTS(shiftState(state, code, { inline: true }));
@@ -1042,14 +1014,32 @@ export class BatchAssignment extends LogicItem {
     code += '// 批量赋值的翻译代码结束---------\n';
     return code;
   }
-  //================================================================================
+
+  @withSourceMap()
+  toNaturalTS(state = createCompilerState()): string {
+    let code = '';
+    // 根据连线状态，拆分成多个单赋值
+    this.assignmentLines.forEach(({ leftIndex, rightIndex }) => {
+      const leftValue = leftIndex.length === 1 ? this.left.expression.toNaturalTS(shiftState(state, code, { inline: true })) : this.left.members[leftIndex[1]]?.toNaturalTS(shiftState(state, code, { inline: true }));
+      code += `${leftValue}`;
+      code += ' = ';
+      const rightValue = rightIndex.length === 1 ? this.rights[rightIndex[0]]?.expression?.toNaturalTS(shiftState(state, code, { inline: true })) : this.rights[rightIndex[0]]?.members[rightIndex[1]]?.toNaturalTS(shiftState(state, code, { inline: true }));
+      code += `${rightValue}`;
+      code += ';';
+      code += '\n';
+    });
+    return code;
+  }
+
+  //= ===============================================================================
   // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
   // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-  //================================================================================
+  //= ===============================================================================
 }
 
+classMap.BatchAssignment = BatchAssignment;
 export default BatchAssignment;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================