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

package/src/concepts/SwitchCase__.ts

@@ -1,13 +1,13 @@
+import { v4 as uuidv4 } from 'uuid';
 import { TranslatorState, shiftState, withSourceMap, indent } from '../translator';
-//================================================================================
+//= ===============================================================================
 // 从这里开始到结尾注释之间的代码由 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__';
@@ -17,308 +17,344 @@ import LogicItem from './LogicItem__';
  */
 @concept('选择分支情况')
 export class SwitchCase extends LogicItem {
-    /**
-     * 产品概念
-     */
-    @property()
-    concept: 'SwitchCase' = 'SwitchCase';
-
-    /**
-     * 条件表达式
-     */
-    @property('LogicItem')
-    test: LogicItem = undefined;
-
-    /**
-     * then
-     */
-    @property('LogicItem')
-    consequent: Array<LogicItem> = [];
-
-    
-
-    /**
-     * @param source 需要合并的部分参数
-     */
-    constructor(source?: Partial<SwitchCase>) {
-        source = Object.assign({}, SwitchCase.getDefaultOptions(), source);
-        super(source);
-        super.subConstructor(source);
+  /**
+   * 产品概念
+   */
+  @property()
+  concept: 'SwitchCase' = 'SwitchCase';
+
+  /**
+   * 条件表达式
+   */
+  @property('LogicItem')
+  test: LogicItem = undefined;
+
+  /**
+   * then
+   */
+  @property('LogicItem')
+  consequent: Array<LogicItem> = [];
+
+  /**
+   * @param source 需要合并的部分参数
+   */
+  constructor(source?: Partial<SwitchCase>) {
+    // @ts-ignore
+    source = { ...SwitchCase.getDefaultOptions(), ...source };
+    super(source);
+    super.subConstructor(source);
+  }
+
+  getClassName() {
+    return 'SwitchCase';
+  }
+
+  static from(source: any, parentNode?: any, parentKey?: string): SwitchCase {
+    return super.from(source, parentNode, parentKey) as SwitchCase;
+  }
+
+  /**
+   * 设置条件表达式
+   */
+  setTest(test: LogicItem) {
+    test._delete();
+    const relationOptions = { parentNode: this, parentKey: 'test' };
+    Object.assign(test, relationOptions);
+    const object = {
+      test,
+    };
+    this.update({
+      ...object,
+    });
+  }
+
+  /**
+   * 插入逻辑项
+   * @internal
+   * @param logicItemOptions 逻辑项参数
+   */
+  _insertItemAt(logicItemOptions: Partial<LogicItem>, index: number): LogicItem;
+
+  /**
+   * 插入逻辑项
+   * @internal
+   * @param logicItem 已有的逻辑项实例
+   */
+  _insertItemAt(logicItem: LogicItem, index: number): LogicItem;
+
+  _insertItemAt(options: Partial<LogicItem> | LogicItem, index: number) {
+    const logicItemOptions: any = {};
+    const relationOptions = { parentNode: this, parentKey: 'consequent' };
+    let logicItem: LogicItem;
+    if (options instanceof LogicItem) {
+      options.ensureDelete(); // 同一实例不支持多处存在
+      logicItem = options;
+      Object.assign(logicItem, relationOptions);
+    } else {
+      logicItem = LogicItem.from(
+        {
+          ...logicItemOptions,
+          ...options,
+        },
+        this,
+        'consequent'
+      );
     }
-    getClassName() {
-        return 'SwitchCase';
+    this.consequent.splice(index, 0, logicItem);
+    return logicItem;
+  }
+
+  /**
+   * 插入逻辑项
+   * @param logicItemOptions 逻辑项参数
+   */
+  insertItemAt(logicItemOptions: Partial<LogicItem>, index: number): LogicItem;
+
+  /**
+   * 插入逻辑项
+   * @param logicItem 已有的逻辑项实例
+   */
+  insertItemAt(logicItem: LogicItem, index: number): LogicItem;
+
+  insertItemAt(options: Partial<LogicItem> | LogicItem, index: number) {
+    const node = this._insertItemAt(options as any, index);
+    node.create({
+      index,
+      parentNode: this,
+      parentKey: 'consequent',
+    });
+    return node;
+  }
+
+  /**
+   * 添加逻辑项
+   * @internal
+   * @param logicItemOptions 逻辑项参数
+   */
+  _addItem(logicItemOptions: Partial<LogicItem>): LogicItem;
+
+  /**
+   * 添加逻辑项
+   * @internal
+   * @param logicItem 已有的逻辑项实例
+   */
+  _addItem(logicItem: LogicItem): LogicItem;
+
+  _addItem(options?: Partial<LogicItem> | LogicItem) {
+    const index = this.consequent.length;
+    return this._insertItemAt(options as any, index);
+  }
+
+  /**
+   * 添加逻辑项
+   * @param logicItemOptions 逻辑项参数
+   */
+  addItem(logicItemOptions: Partial<LogicItem>): LogicItem;
+
+  /**
+   * 添加逻辑项
+   * @param logicItem 已有的逻辑项实例
+   */
+  addItem(logicItem: LogicItem): LogicItem;
+
+  addItem(options?: Partial<LogicItem> | LogicItem) {
+    const node = this._addItem(options as any);
+    const index = this.consequent.indexOf(node);
+    node.create({
+      index,
+      parentNode: this,
+      parentKey: 'consequent',
+    });
+    return node;
+  }
+
+  /**
+   * 删除逻辑项
+   * @param name 逻辑项名称
+   */
+  removeTest(name: string): void;
+
+  /**
+   * 删除逻辑项
+   * @param test 已有的逻辑项实例
+   */
+  removeTest(test: LogicItem): void;
+
+  removeTest(options: string | LogicItem) {
+    let test: LogicItem;
+    if (typeof options === 'string') {
+      test = this.test;
+      if (!test) {
+        throw new Error(`找不到逻辑项 ${options}`);
+      }
+    } else {
+      test = options;
     }
-
-    static from(source: any, parentNode?: any, parentKey?: string): SwitchCase {
-        return super.from(source, parentNode, parentKey) as SwitchCase;
+    return test.delete();
+  }
+
+  /**
+   * 删除逻辑项
+   * @param logicItem 已有的逻辑项实例
+   */
+  removeItemInConsequent(logicItem: LogicItem): void;
+
+  removeItemInConsequent(options: LogicItem) {
+    const logicItem: LogicItem = options;
+    return logicItem.delete();
+  }
+
+  __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;
+      }
     }
-
-    
-    /**
-     * 设置条件表达式
-     */
-    setTest(test: LogicItem) {
-        test._delete();
-        const relationOptions = { parentNode: this, parentKey: 'test' };
-        Object.assign(test, relationOptions);
-        const object = {
-            test,
-        };
-        this.update({
-            ...object,
-        });
+    return params;
+  }
+
+  //= ===============================================================================
+  // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+  // 自动生成的代码已结束。下面可以手动编写。
+  //= ===============================================================================
+
+  toJS(state?: TranslatorState, index?: number, length?: number): string {
+    let code = '';
+    if (index === 0) {
+      code = 'if (';
+      code += this.test
+        ? this.test.toJS(
+          shiftState(state, code, {
+            tabSize: 0,
+          })
+        )
+        : '';
+      code += ') {\n';
+    } else if (index !== length - 1) {
+      code = 'else if (';
+      code += this.test
+        ? this.test.toJS(
+          shiftState(state, code, {
+            tabSize: 0,
+          })
+        )
+        : '';
+      code += ') {\n';
+    } else {
+      code += 'else {\n';
     }
-
-    
-
-    
-
-    /**
-    * 插入逻辑项
-    * @internal
-    * @param logicItemOptions 逻辑项参数
-    */
-    _insertItemAt(logicItemOptions: Partial<LogicItem>, index: number): LogicItem;
-
-    /**
-    * 插入逻辑项
-    * @internal
-    * @param logicItem 已有的逻辑项实例
-    */
-    _insertItemAt(logicItem: LogicItem, index: number): LogicItem;
-
-    _insertItemAt(options: Partial<LogicItem> | LogicItem, index: number) {
-        const logicItemOptions: any = {};
-        const relationOptions = { parentNode: this, parentKey: 'consequent' };
-        let logicItem: LogicItem;
-        if (options instanceof LogicItem) {
-            options.ensureDelete(); // 同一实例不支持多处存在
-            logicItem = options;
-            Object.assign(logicItem, relationOptions);
-        } else {
-            logicItem = LogicItem.from({
-                ...logicItemOptions,
-                ...options,
-            }, this, 'consequent');
-        }
-        this.consequent.splice(index, 0, logicItem);
-        return logicItem;
+    if (Array.isArray(this.consequent)) {
+      this.consequent.forEach((logicItem) => {
+        code += `${logicItem.toJS(
+          shiftState(state, code, {
+            tabSize: 0,
+          })
+        )}\n`;
+      });
     }
-
-    
-
-    /**
-     * 插入逻辑项
-     * @param logicItemOptions 逻辑项参数
-     */
-    insertItemAt(logicItemOptions: Partial<LogicItem>, index: number): LogicItem;
-
-    /**
-     * 插入逻辑项
-     * @param logicItem 已有的逻辑项实例
-     */
-    insertItemAt(logicItem: LogicItem, index: number): LogicItem;
-
-    insertItemAt(options: Partial<LogicItem> | LogicItem, index: number) {
-        const node = this._insertItemAt(options as any, index);
-        node.create({
-            index,
-            parentNode: this,
-            parentKey: 'consequent',
+    code += '}\n';
+
+    return code;
+  }
+
+  @withSourceMap()
+  toEmbeddedTS(state?: TranslatorState, isLast?: boolean): string {
+    let code = '\n';
+    if (!isLast) {
+      code += 'if (nasl.core.ensureBoolean(';
+      code += this.test
+        ? this.test.toEmbeddedTS(
+          shiftState(state, code, {
+            tabSize: (state?.tabSize || 0) + 1,
+          })
+        )
+        : '__IDENTIFIER__';
+
+      code += ')) {\n';
+
+      if (Array.isArray(this.consequent)) {
+        this.consequent.forEach((logicItem) => {
+          code += `${logicItem.toEmbeddedTS(
+            shiftState(state, code, {
+              tabSize: (state?.tabSize || 0) + 1,
+            })
+          )};\n`;
         });
-        return node;
-    }
-
-    
-
-    /**
-     * 添加逻辑项
-     * @internal
-     * @param logicItemOptions 逻辑项参数
-     */
-    _addItem(logicItemOptions: Partial<LogicItem>): LogicItem;
-
-    /**
-     * 添加逻辑项
-     * @internal
-     * @param logicItem 已有的逻辑项实例
-     */
-    _addItem(logicItem: LogicItem): LogicItem;
-
-    _addItem(options?: Partial<LogicItem> | LogicItem) {
-        const index = this.consequent.length;
-        return this._insertItemAt(options as any, index);
-    }
-
-    
-
-    /**
-     * 添加逻辑项
-     * @param logicItemOptions 逻辑项参数
-     */
-    addItem(logicItemOptions: Partial<LogicItem>): LogicItem;
-
-    /**
-     * 添加逻辑项
-     * @param logicItem 已有的逻辑项实例
-     */
-    addItem(logicItem: LogicItem): LogicItem;
-
-    addItem(options?: Partial<LogicItem> | LogicItem) {
-        const node = this._addItem(options as any);
-        const index = this.consequent.indexOf(node);
-        node.create({
-            index,
-            parentNode: this,
-            parentKey: 'consequent',
+      }
+      code += `${indent(state?.tabSize || 0)}}\n`;
+    } else {
+      code += 'else {\n';
+      if (Array.isArray(this.consequent)) {
+        this.consequent.forEach((logicItem) => {
+          code += `${logicItem.toEmbeddedTS(
+            shiftState(state, code, {
+              tabSize: (state?.tabSize || 0) + 1,
+            })
+          )};\n`;
         });
-        return node;
+      }
+      code += `${indent(state?.tabSize || 0)}}\n`;
     }
 
-
-    
-
-    /**
-    * 删除逻辑项
-    * @param name 逻辑项名称
-    */
-    removeTest(name: string): void;
-
-    /**
-     * 删除逻辑项
-     * @param test 已有的逻辑项实例
-     */
-    removeTest(test: LogicItem): void;
-
-    removeTest(options: string | LogicItem) {
-        let test: LogicItem;
-        if (typeof options === 'string') {
-            test = this.test;
-            if (!test) {
-                throw new Error('找不到逻辑项 ' + options);
-            }
-        } else {
-            test = options;
-        }
-        return test.delete();
-    }
-
-    
-
-    /**
-     * 删除逻辑项
-     * @param logicItem 已有的逻辑项实例
-     */
-    removeItemInConsequent(logicItem: LogicItem): void;
-
-    removeItemInConsequent(options: LogicItem) {
-        let logicItem: LogicItem = options;
-        return logicItem.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;
-    }
-
-    //================================================================================
-    // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
-    // 自动生成的代码已结束。下面可以手动编写。
-    //================================================================================
-
-    toJS(state?: TranslatorState, index?: number, length?: number): string {
-        let code = '';
-        if (index === 0) {
-            code = 'if (';
-            code += this.test ? this.test.toJS(shiftState(state, code, {
-                tabSize: 0,
-            })) : '';
-            code += ') {\n';
-        } else if (index !== length - 1) {
-            code = 'else if (';
-            code += this.test ? this.test.toJS(shiftState(state, code, {
-                tabSize: 0,
-            })) : '';
-            code += ') {\n';
-        } else {
-            code += 'else {\n';
-        }
-        if (Array.isArray(this.consequent)) {
-            this.consequent.forEach((logicItem) => {
-                code
-                    += logicItem.toJS(shiftState(state, code, {
-                        tabSize: 0,
-                    })) + '\n';
-            });
-        }
-        code += '}\n';
-
-        return code;
+    return code;
+  }
+
+  @withSourceMap()
+  toNaturalTS(state?: TranslatorState, isLast?: boolean, isFirst?: boolean): string {
+    let code = '\n';
+    if (!isLast) {
+      code += indent(state.tabSize + 2);
+      code += isFirst ? '' : 'else ';
+      code += 'if (';
+      code += this.test ? this.test.toNaturalTS(shiftState(state, code, { inline: true })) : '__IDENTIFIER__';
+      code += ') {\n';
+      if (Array.isArray(this.consequent)) {
+        this.consequent.forEach((logicItem) => {
+          code += indent(state.tabSize + 3);
+          code += `${logicItem.toNaturalTS(shiftState(state, code, { inline: true }))}\n`;
+        });
+      }
+      code += indent(state.tabSize + 2);
+      code += '}';
+    } else {
+      code += indent(state.tabSize + 2);
+      code += 'else {\n';
+      if (Array.isArray(this.consequent)) {
+        this.consequent.forEach((logicItem) => {
+          code += indent(state.tabSize + 2);
+          code += `${logicItem.toNaturalTS(shiftState(state, code, { inline: true }))}\n`;
+        });
+      }
+      code += indent(state.tabSize + 2);
+      code += '}';
     }
 
-    @withSourceMap()
-    toEmbeddedTS(state?: TranslatorState, isLast?: boolean): string {
-        let code = '\n';
-        if (!isLast) {
-            code += 'if (nasl.core.ensureBoolean(';
-            code += this.test ? this.test.toEmbeddedTS(shiftState(state, code, {
-                tabSize: (state?.tabSize || 0) + 1,
-            })) : '__IDENTIFIER__';
-
-            code += ')) {\n';
-
-            if (Array.isArray(this.consequent)) {
-                this.consequent.forEach((logicItem) => {
-                    code
-                        += logicItem.toEmbeddedTS(shiftState(state, code, {
-                            tabSize: (state?.tabSize || 0) + 1,
-                        })) + ';\n';
-                });
-            }
-            code += indent((state?.tabSize || 0)) + '}\n';
-        } else {
-            code += 'else {\n';
-            if (Array.isArray(this.consequent)) {
-                this.consequent.forEach((logicItem) => {
-                    code
-                        += logicItem.toEmbeddedTS(shiftState(state, code, {
-                            tabSize: (state?.tabSize || 0) + 1,
-                        })) + ';\n';
-                });
-            }
-            code += indent((state?.tabSize || 0)) + '}\n';
-        }
-
-        return code;
-    }
+    return code;
+  }
 
-    //================================================================================
-    // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
-    // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-    //================================================================================
+  //= ===============================================================================
+  // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
+  // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
+  //= ===============================================================================
 }
 
 classMap.SwitchCase = SwitchCase;
 export default SwitchCase;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================