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

package/src/concepts/Assignment__.ts

@@ -1,3 +1,4 @@
+import { v4 as uuidv4 } from 'uuid';
 import { TranslatorState, shiftState, withSourceMap } from '../translator';
 import Identifier from './Identifier__';
 import { ElementToVueOptions } from './ViewElement__';
@@ -5,15 +6,15 @@ import BatchAssignment from './BatchAssignment__';
 import SelectMembers from './SelectMembers__';
 import AssignmentLine from './AssignmentLine__';
 import Logic from './Logic__';
-//================================================================================
+import MemberExpression from './MemberExpression__';
+//= ===============================================================================
 // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
 // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-//================================================================================
+//= ===============================================================================
 import { EventPayload, Params } from '../common/EventEmitter';
 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 LogicItem from './LogicItem__';
@@ -23,301 +24,312 @@ import LogicItem from './LogicItem__';
  */
 @concept('赋值语句')
 export class Assignment extends LogicItem {
-    /**
-     * 产品概念
-     */
-    @property()
-    concept: 'Assignment' = 'Assignment';
+  /**
+   * 产品概念
+   */
+  @property()
+  concept: 'Assignment' = 'Assignment';
 
-    /**
-     * 赋值语句标题
-     */
-    @property()
-    label: string = '赋值';
+  /**
+   * 赋值语句标题
+   */
+  @property()
+  label: string = '赋值';
 
-    /**
-     * 左边项
-     */
-    @property('LogicItem')
-    left: LogicItem = undefined;
+  /**
+   * 左边项
+   */
+  @property('LogicItem')
+  left: LogicItem = undefined;
 
-    /**
-     * 右边项
-     */
-    @property('LogicItem')
-    right: LogicItem = undefined;
+  /**
+   * 右边项
+   */
+  @property('LogicItem')
+  right: LogicItem = undefined;
 
-    
+  /**
+   * @param source 需要合并的部分参数
+   */
+  constructor(source?: Partial<Assignment>) {
+    // @ts-ignore
+    // @ts-ignore
+    source = { ...Assignment.getDefaultOptions(), ...source };
+    super(source);
+    super.subConstructor(source);
+  }
 
-    /**
-     * @param source 需要合并的部分参数
-     */
-    constructor(source?: Partial<Assignment>) {
-        source = Object.assign({}, Assignment.getDefaultOptions(), source);
-        super(source);
-        super.subConstructor(source);
-    }
-    getClassName() {
-        return 'Assignment';
-    }
+  getClassName() {
+    return 'Assignment';
+  }
 
-    static from(source: any, parentNode?: any, parentKey?: string): Assignment {
-        return super.from(source, parentNode, parentKey) as Assignment;
-    }
-
-    
-    /**
-     * 设置左边项
-     */
-    setLeft(left: LogicItem) {
-        left._delete();
-        const relationOptions = { parentNode: this, parentKey: 'left' };
-        Object.assign(left, relationOptions);
-        const object = {
-            left,
-        };
-        this.update({
-            ...object,
-        });
-    }
-
-    /**
-     * 设置右边项
-     */
-    setRight(right: LogicItem) {
-        right._delete();
-        const relationOptions = { parentNode: this, parentKey: 'right' };
-        Object.assign(right, relationOptions);
-        const object = {
-            right,
-        };
-        this.update({
-            ...object,
-        });
-    }
+  static from(source: any, parentNode?: any, parentKey?: string): Assignment {
+    return super.from(source, parentNode, parentKey) as Assignment;
+  }
 
-    
+  /**
+   * 设置左边项
+   */
+  setLeft(left: LogicItem) {
+    left._delete();
+    const relationOptions = { parentNode: this, parentKey: 'left' };
+    Object.assign(left, relationOptions);
+    const object = {
+      left,
+    };
+    this.update({
+      ...object,
+    });
+  }
 
-    
+  /**
+   * 设置右边项
+   */
+  setRight(right: LogicItem) {
+    right._delete();
+    const relationOptions = { parentNode: this, parentKey: 'right' };
+    Object.assign(right, relationOptions);
+    const object = {
+      right,
+    };
+    this.update({
+      ...object,
+    });
+  }
 
-    /**
-    * 删除逻辑项
-    * @param name 逻辑项名称
-    */
-    removeLeft(name: string): void;
+  /**
+   * 删除逻辑项
+   * @param name 逻辑项名称
+   */
+  removeLeft(name: string): void;
 
-    /**
-     * 删除逻辑项
-     * @param left 已有的逻辑项实例
-     */
-    removeLeft(left: LogicItem): void;
+  /**
+   * 删除逻辑项
+   * @param left 已有的逻辑项实例
+   */
+  removeLeft(left: LogicItem): void;
 
-    removeLeft(options: string | LogicItem) {
-        let left: LogicItem;
-        if (typeof options === 'string') {
-            left = this.left;
-            if (!left) {
-                throw new Error('找不到逻辑项 ' + options);
-            }
-        } else {
-            left = options;
-        }
-        return left.delete();
+  removeLeft(options: string | LogicItem) {
+    let left: LogicItem;
+    if (typeof options === 'string') {
+      left = this.left;
+      if (!left) {
+        throw new Error(`找不到逻辑项 ${options}`);
+      }
+    } else {
+      left = options;
     }
+    return left.delete();
+  }
 
-    
+  /**
+   * 删除逻辑项
+   * @param name 逻辑项名称
+   */
+  removeRight(name: string): void;
 
-    /**
-    * 删除逻辑项
-    * @param name 逻辑项名称
-    */
-    removeRight(name: string): void;
+  /**
+   * 删除逻辑项
+   * @param right 已有的逻辑项实例
+   */
+  removeRight(right: LogicItem): void;
 
-    /**
-     * 删除逻辑项
-     * @param right 已有的逻辑项实例
-     */
-    removeRight(right: LogicItem): void;
-
-    removeRight(options: string | LogicItem) {
-        let right: LogicItem;
-        if (typeof options === 'string') {
-            right = this.right;
-            if (!right) {
-                throw new Error('找不到逻辑项 ' + options);
-            }
-        } else {
-            right = options;
-        }
-        return right.delete();
+  removeRight(options: string | LogicItem) {
+    let right: LogicItem;
+    if (typeof options === 'string') {
+      right = this.right;
+      if (!right) {
+        throw new Error(`找不到逻辑项 ${options}`);
+      }
+    } else {
+      right = options;
     }
+    return right.delete();
+  }
 
-    __removeLogicItem(logicItem: LogicItem) {
-        const parentKey = logicItem.parentKey;
-        const params: Params = {
-            parentNode: this,
-            parentKey,
-            index: -1,
-            object: null,
-            oldObject: logicItem,
-        };
-        if (parentKey) {
-            params.parentKey = parentKey;
-            if (Array.isArray((this as any)[parentKey])) {
-                const index = (this as any)[parentKey].indexOf(logicItem);
-                ~index && (this as any)[parentKey].splice(index, 1);
-                params.index = index;
-            } else if ((this as any)[parentKey] === logicItem) {
-                params.index = 0;
-                (this as any)[parentKey] = undefined;
-            }
-        }
-        return params;
+  __removeLogicItem(logicItem: LogicItem) {
+    const { parentKey } = logicItem;
+    const params: Params = {
+      parentNode: this,
+      parentKey,
+      index: -1,
+      object: null,
+      oldObject: logicItem,
+    };
+    if (parentKey) {
+      params.parentKey = parentKey;
+      if (Array.isArray((this as any)[parentKey])) {
+        const index = (this as any)[parentKey].indexOf(logicItem);
+        ~index && (this as any)[parentKey].splice(index, 1);
+        params.index = index;
+      } else if ((this as any)[parentKey] === logicItem) {
+        params.index = 0;
+        (this as any)[parentKey] = undefined;
+      }
     }
+    return params;
+  }
 
-    //================================================================================
-    // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
-    // 自动生成的代码已结束。下面可以手动编写。
-    //================================================================================
+  //= ===============================================================================
+  // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+  // 自动生成的代码已结束。下面可以手动编写。
+  //= ===============================================================================
 
-    public static getDefaultOptions() {
-        return {
-            left: {
-                concept: 'Identifier',
-                name: '',
-            },
-        };
-    }
+  public static getDefaultOptions() {
+    return {
+      left: {
+        concept: 'Identifier',
+        name: '',
+      },
+    };
+  }
 
-    replaceBatchAssignment() {
-        const { parentKey, parentNode, logic }: { parentKey: string, parentNode: any, logic: Logic } = this;
-        let index;
-        const leftMembers = [];
-        // 左边变量变换
-        let leftObject: any = Object.assign({}, this.left);
-        if (leftObject.concept === 'MemberExpression') {
-            while (leftObject.object) {
-                leftMembers.unshift(leftObject);
-                leftObject = Object.assign({}, leftObject.object);
-            }
-        }
-        // 右边变量变换
-        const rightMembers = [];
-        let rightObject: any = Object.assign({}, this.right);
-        if (rightObject.concept === 'MemberExpression' && rightObject.concept === 'Identifier') {
-            while (rightObject.object) {
-                rightMembers.unshift(rightObject);
-                rightObject = Object.assign({}, rightObject.object);
-            }
-        }
-        // 自动连线
-        const leftIndex = leftMembers.length ? [0, leftMembers.length - 1] : [0];
-        const rightIndex = rightMembers.length ? [0, rightMembers.length - 1] : [0];
-        // 切换成批量赋值
-        const leftSelect = new SelectMembers({
-            expression: leftObject,
-            members: leftMembers,
-        });
-        const rightSelect = new SelectMembers({
-            expression: rightObject,
-            members: rightMembers,
-        });
-        const assignmentLine = new AssignmentLine({
-            leftIndex,
-            rightIndex,
-        });
-        const batchObject = this.right ? {
-            left: leftSelect,
-            rights: [rightSelect],
-            assignmentLines: [assignmentLine],
-        } : {
-            left: leftSelect,
-            rights: [],
-            assignmentLines: [],
-        };
-        const batchAssignment = new BatchAssignment({
-            ...batchObject,
-            offsetX: this.offsetX,
-            offsetY: this.offsetY,
-        });
-        this.app.emit('collect:start', {
-            actionMsg: '普通赋值切换批量赋值',
-        });
-        if (parentNode.concept === 'Logic') {
-            if (parentKey === 'body') {
-                index = logic.body.findIndex((item: any) => item === this);
-                logic.insertItemInBodyAt(batchAssignment, index);
-            } else {
-                index = logic.playground.findIndex((item: any) => item === this);
-                logic.insertItemInPlaygroundAt(batchAssignment, index);
-            }
-        } 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`](batchAssignment, index);
-            else
-                parentNode.insertItemAt(batchAssignment, index);
-        }
-        this.delete();
-        this.app.emit('collect:end');
-        return batchAssignment;
+  replaceBatchAssignment() {
+    const { parentKey, parentNode, logic }: { parentKey: string; parentNode: any; logic: Logic } = this;
+    let index;
+    const leftMembers = [];
+    // 左边变量变换
+    let leftObject: any = { ...this.left };
+    if (leftObject.concept === 'MemberExpression') {
+      while (leftObject.object) {
+        leftMembers.unshift(leftObject);
+        leftObject = { ...leftObject.object };
+      }
     }
-
-    toVue(options?: ElementToVueOptions) {
-        let code = '';
-        code += this.left ? this.left.toVue(options) : '';
-        code += ' = ';
-        code += this.right ? this.right.toVue(options) : '';
-        code += ';';
-        return code;
+    // 右边变量变换
+    const rightMembers = [];
+    let rightObject: any = { ...this.right };
+    if (rightObject.concept === 'MemberExpression' && (this.right as MemberExpression).lastObject.concept === 'Identifier') {
+      while (rightObject.object) {
+        rightMembers.unshift(rightObject);
+        rightObject = { ...rightObject.object };
+      }
+    }
+    // 自动连线
+    const leftIndex = leftMembers.length ? [0, leftMembers.length - 1] : [0];
+    const rightIndex = rightMembers.length ? [0, rightMembers.length - 1] : [0];
+    // 切换成批量赋值
+    const leftSelect = new SelectMembers({
+      expression: leftObject,
+      members: leftMembers,
+    });
+    const rightSelect = new SelectMembers({
+      expression: rightObject,
+      members: rightMembers,
+    });
+    const assignmentLine = new AssignmentLine({
+      leftIndex,
+      rightIndex,
+    });
+    const batchObject = this.right
+      ? {
+        left: leftSelect,
+        rights: [rightSelect],
+        assignmentLines: [assignmentLine],
+      }
+      : {
+        left: leftSelect,
+        rights: [],
+        assignmentLines: [],
+      };
+    const batchAssignment = new BatchAssignment({
+      ...batchObject,
+      offsetX: this.offsetX,
+      offsetY: this.offsetY,
+    });
+    this.app.emit('collect:start', {
+      actionMsg: '普通赋值切换批量赋值',
+    });
+    if (parentNode.concept === 'Logic') {
+      if (parentKey === 'body') {
+        index = logic.body.findIndex((item: any) => item === this);
+        logic.insertItemInBodyAt(batchAssignment, index);
+      } else {
+        index = logic.playground.findIndex((item: any) => item === this);
+        logic.insertItemInPlaygroundAt(batchAssignment, index);
+      }
+    } 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`](batchAssignment, index);
+      else parentNode.insertItemAt(batchAssignment, index);
     }
+    this.delete();
+    this.app.emit('collect:end');
+    return batchAssignment;
+  }
 
-    toUI(state?: TranslatorState): string {
-        let code = '';
-        code += this.left ? this.left.toUI(shiftState(state, code, { inline: true })) : '';
-        code += ' = ';
-        code += this.right ? this.right.toUI(shiftState(state, code, { inline: true })) : '';
-        code += ';';
-        return code;
+  toVue(options?: ElementToVueOptions) {
+    let code = '';
+    code += this.left ? this.left.toVue(options) : '';
+    code += ' = ';
+    code += this.right ? this.right.toVue(options) : '';
+    code += ';';
+    return code;
+  }
+
+  toUI(state?: TranslatorState): string {
+    let code = '';
+    code += this.left ? this.left.toUI(shiftState(state, code, { inline: true })) : '';
+    code += ' = ';
+    code += this.right ? this.right.toUI(shiftState(state, code, { inline: true })) : '';
+    code += ';';
+    return code;
+  }
+
+  toJS(state?: TranslatorState): string {
+    let code = '';
+    code += this.left ? this.left.toJS(shiftState(state, code, { inline: true })) : '';
+    if (this.right) {
+      code += ' = ';
+      code += this.right.toJS(
+        shiftState(state, code, {
+          inline: true,
+        })
+      );
     }
+    return code;
+  }
 
-    toJS(state?: TranslatorState): string {
-        let code = '';
-        code += this.left ? this.left.toJS(shiftState(state, code, { inline: true })) : '';
-        if (this.right) {
-            code += ' = ';
-            code += this.right.toJS(shiftState(state, code, {
-                inline: true,
-            }));
-        }
-        return code;
+  @withSourceMap()
+  toEmbeddedTS(state?: TranslatorState): string {
+    let code = 'nasl.core.assign(';
+
+    code += this.left ? this.left.toEmbeddedTS(shiftState(state, code, { inline: true })) : '__LEFT__';
+    code += ',';
+    // 因为右侧枚举要赋值就得new 一下；
+    if (this.right && this.right instanceof Identifier && this.right.namespace && this.right.namespace.endsWith('enums')) {
+      code += 'new ';
     }
+    code += this.right && this.right.toEmbeddedTS ? this.right.toEmbeddedTS(shiftState(state, code, { inline: true })) : '__RIGHT__';
+    code += ');';
+    return code;
+  }
 
-    @withSourceMap()
-    toEmbeddedTS(state?: TranslatorState): string {
-        let code = 'nasl.core.assign(';
+  @withSourceMap()
+  toNaturalTS(state?: TranslatorState): string {
+    let code = '';
 
-        code += this.left ? this.left.toEmbeddedTS(shiftState(state, code, { inline: true })) : '__LEFT__';
-        code += ',';
-        // 因为右侧枚举要赋值就得new 一下；
-        if (this.right && this.right instanceof Identifier && this.right.namespace && this.right.namespace.endsWith('enums')) {
-            code += 'new ';
-        }
-        code += (this.right && this.right.toEmbeddedTS) ? this.right.toEmbeddedTS(shiftState(state, code, { inline: true })) : '__RIGHT__';
-        code += ');';
-        return code;
+    code += this.left ? this.left.toNaturalTS(shiftState(state, code, { inline: true })) : '__LEFT__';
+    code += ' = ';
+    // 因为右侧枚举要赋值就得new 一下；
+    if (this.right && this.right instanceof Identifier && this.right.namespace && this.right.namespace.endsWith('enums')) {
+      code += 'new ';
     }
+    code += (this.right && this.right.toNaturalTS) ? this.right.toNaturalTS(shiftState(state, code, { inline: true })) : '__RIGHT__';
+
+    return code;
+  }
 
-    //================================================================================
-    // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
-    // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-    //================================================================================
+  //= ===============================================================================
+  // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
+  // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
+  //= ===============================================================================
 }
 
 classMap.Assignment = Assignment;
 export default Assignment;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================