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

package/src/concepts/Return__.ts

@@ -1,3 +1,4 @@
+import { v4 as uuidv4 } from 'uuid';
 import { dataTypesMap } from './basics/types';
 import { TranslatorState, shiftState, withSourceMap } from '../translator';
 import EntityProperty from './EntityProperty__';
@@ -5,25 +6,14 @@ import StructureProperty from './StructureProperty__';
 import Structure from './Structure__';
 import { getNodeByNodeCallee, getNaslNodeByNodeCallee } from '../automate/engine/utils';
 
-function tryJSONParse(str: string) {
-    let result;
-
-    try {
-        result = JSON.parse(str);
-    } catch (e) { }
-
-    return result;
-}
-
-//================================================================================
+//= ===============================================================================
 // 从这里开始到结尾注释之间的代码由 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 TypeAnnotation from './TypeAnnotation__';
@@ -36,360 +26,391 @@ import Frontend from './Frontend__';
 import ProcessElement from './ProcessElement__';
 import Process from './Process__';
 
+function tryJSONParse(str: string) {
+  let result;
+
+  try {
+    result = JSON.parse(str);
+  } catch (e) {}
+
+  return result;
+}
+
 /**
  * 输出参数
  */
 @concept('输出参数')
 export class Return extends BaseNode {
-    /**
-     * 产品概念
-     */
-    @property()
-    concept: 'Return' = 'Return';
-
-    /**
-     * 输出参数名称
-     */
-    @property()
-    name: string = undefined;
-
-    /**
-     * 输出参数描述
-     */
-    @property()
-    description: string = undefined;
-
-    /**
-     * 类型
-     */
-    @property('TypeAnnotation')
-    typeAnnotation: TypeAnnotation = undefined;
-
-    /**
-     * 默认值
-     */
-    @property()
-    defaultValue: string = undefined;
-
-    /**
-     * 祖先 Logic
-     */
-    get logic() {
-        return this.getAncestor('Logic') as Logic;
+  /**
+   * 产品概念
+   */
+  @property()
+  concept: 'Return' = 'Return';
+
+  /**
+   * 输出参数名称
+   */
+  @property()
+  name: string = undefined;
+
+  /**
+   * 输出参数描述
+   */
+  @property()
+  description: string = undefined;
+
+  /**
+   * 类型
+   */
+  @property('TypeAnnotation')
+  typeAnnotation: TypeAnnotation = undefined;
+
+  /**
+   * 默认值
+   */
+  @property()
+  defaultValue: string = undefined;
+
+  /**
+   * 祖先 Logic
+   */
+  get logic() {
+    return this.getAncestor('Logic') as Logic;
+  }
+
+  /**
+   * 祖先 Module
+   */
+  get module() {
+    return this.getAncestor('Module') as Module;
+  }
+
+  /**
+   * 祖先 App
+   */
+  get app() {
+    return this.getAncestor('App') as App;
+  }
+
+  /**
+   * 祖先 Connector
+   */
+  get connector() {
+    return this.getAncestor('Connector') as Connector;
+  }
+
+  /**
+   * 祖先 View
+   */
+  get view() {
+    return this.getAncestor('View') as View;
+  }
+
+  /**
+   * 祖先 Frontend
+   */
+  get frontend() {
+    return this.getAncestor('Frontend') as Frontend;
+  }
+
+  /**
+   * 祖先 ProcessElement
+   */
+  get processElement() {
+    return this.getAncestor('ProcessElement') as ProcessElement;
+  }
+
+  /**
+   * 祖先 Process
+   */
+  get process() {
+    return this.getAncestor('Process') as Process;
+  }
+
+  /**
+   * @param source 需要合并的部分参数
+   */
+  constructor(source?: Partial<Return>) {
+    // @ts-ignore
+    source = { ...Return.getDefaultOptions(), ...source };
+    super(source);
+    super.subConstructor(source);
+  }
+
+  getClassName() {
+    return 'Return';
+  }
+
+  static from(source: any, parentNode?: any, parentKey?: string): Return {
+    return super.from(source, parentNode, parentKey) as Return;
+  }
+
+  /**
+   * 从父级删除该节点
+   * @internal
+   */
+  _delete() {
+    let params: Params = null;
+    if (this.parentNode) {
+      params = (this.parentNode as any)?.__removeReturn?.(this);
     }
-    /**
-     * 祖先 Module
-     */
-    get module() {
-        return this.getAncestor('Module') as Module;
+    return params;
+  }
+
+  /**
+   * 设置输出参数名称
+   */
+  setName(name: string) {
+    const object = {
+      name,
+    };
+    this.update({
+      ...object,
+      field: 'name',
+    });
+  }
+
+  /**
+   * 设置输出参数描述
+   */
+  setDescription(description: string) {
+    const object = {
+      description,
+    };
+    this.update({
+      ...object,
+    });
+  }
+
+  /**
+   * 设置默认值
+   */
+  setDefaultValue(defaultValue: string) {
+    const object = {
+      defaultValue,
+    };
+    this.update({
+      ...object,
+    });
+  }
+
+  /**
+   * 删除类型标注
+   * @param name 类型标注名称
+   */
+  removeTypeAnnotation(name: string): void;
+
+  /**
+   * 删除类型标注
+   * @param typeAnnotation 已有的类型标注实例
+   */
+  removeTypeAnnotation(typeAnnotation: TypeAnnotation): void;
+
+  removeTypeAnnotation(options: string | TypeAnnotation) {
+    let typeAnnotation: TypeAnnotation;
+    if (typeof options === 'string') {
+      typeAnnotation = this.typeAnnotation;
+      if (!typeAnnotation) {
+        throw new Error(`找不到类型标注 ${options}`);
+      }
+    } else {
+      typeAnnotation = options;
     }
-    /**
-     * 祖先 App
-     */
-    get app() {
-        return this.getAncestor('App') as App;
+    return typeAnnotation.delete();
+  }
+
+  __removeTypeAnnotation(typeAnnotation: TypeAnnotation) {
+    const { parentKey } = typeAnnotation;
+    const params: Params = {
+      parentNode: this,
+      parentKey,
+      index: -1,
+      object: null,
+      oldObject: typeAnnotation,
+    };
+    if (parentKey) {
+      params.parentKey = parentKey;
+      if (Array.isArray((this as any)[parentKey])) {
+        const index = (this as any)[parentKey].indexOf(typeAnnotation);
+        ~index && (this as any)[parentKey].splice(index, 1);
+        params.index = index;
+      } else if ((this as any)[parentKey] === typeAnnotation) {
+        params.index = 0;
+        (this as any)[parentKey] = undefined;
+      }
     }
-    /**
-     * 祖先 Connector
-     */
-    get connector() {
-        return this.getAncestor('Connector') as Connector;
+    return params;
+  }
+
+  //= ===============================================================================
+  // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+  // 自动生成的代码已结束。下面可以手动编写。
+  //= ===============================================================================
+
+  @excludedInJSON()
+  useComponent: {
+    type: 's-member-identifier-select' | 's-enum-select' | 's-schema-select' | 's-datatype-select' | 's-options-select';
+    attrs?: any;
+  } = undefined;
+
+  /**
+   * 自动补全的属性列表
+   */
+  completionChildren: Array<EntityProperty | StructureProperty> = undefined;
+
+  toJS(state?: TranslatorState): string {
+    const context = this.view ? 'this' : 'Vue.prototype';
+    const typeAnnotation = this.typeAnnotation || this.__TypeAnnotation;
+    const needGenInitFromSchema =
+      typeAnnotation && (!['primitive', 'union'].includes(typeAnnotation.typeKind) || typeAnnotation.typeName === 'DateTime');
+    let code = `${this.name} =`;
+    if (needGenInitFromSchema) {
+      code += `${context}.$genInitFromSchema('${typeAnnotation?.sortedTypeKey}',`;
     }
-    /**
-     * 祖先 View
-     */
-    get view() {
-        return this.getAncestor('View') as View;
+    const { defaultValue } = this;
+    const defaultValueType = Object.prototype.toString.call(defaultValue);
+    const { typeKind, typeNamespace, typeName } = typeAnnotation || {};
+    // 设置成null，才能同步给后端清除该值，但是null对checkbox组件是一种特殊状态
+    let parsedValue = defaultValue ?? undefined;
+    if (
+      defaultValueType === '[object String]' &&
+      !(typeKind === 'primitive' && typeNamespace === 'nasl.core' && ['String', 'Text', 'Email'].includes(typeName)) &&
+      !this.isEnum() &&
+      !['union'].includes(typeKind)
+    ) {
+      // 一些特殊情况，特殊处理成undefined
+      // 1.defaultValue在nasl节点上错误得赋值给了空制符串
+      if ([''].includes(defaultValue)) {
+        parsedValue = undefined;
+      } else {
+        parsedValue = tryJSONParse(defaultValue) ?? defaultValue;
+      }
     }
-    /**
-     * 祖先 Frontend
-     */
-    get frontend() {
-        return this.getAncestor('Frontend') as Frontend;
+    code += JSON.stringify(parsedValue);
+    if (needGenInitFromSchema) {
+      code += `)`;
     }
-    /**
-     * 祖先 ProcessElement
-     */
-    get processElement() {
-        return this.getAncestor('ProcessElement') as ProcessElement;
-    }
-    /**
-     * 祖先 Process
-     */
-    get process() {
-        return this.getAncestor('Process') as Process;
-    }
-
-    /**
-     * @param source 需要合并的部分参数
-     */
-    constructor(source?: Partial<Return>) {
-        source = Object.assign({}, Return.getDefaultOptions(), source);
-        super(source);
-        super.subConstructor(source);
-    }
-    getClassName() {
-        return 'Return';
-    }
-
-    static from(source: any, parentNode?: any, parentKey?: string): Return {
-        return super.from(source, parentNode, parentKey) as Return;
-    }
-
-    /**
-     * 从父级删除该节点
-     * @internal
-     */
-    _delete() {
-        let params: Params = null;
-        if (this.parentNode) {
-            params = (this.parentNode as any)?.__removeReturn?.(this);
+    return code;
+  }
+
+  @withSourceMap()
+  toEmbeddedTS(state?: TranslatorState): string {
+    let code = this.name || '__return';
+    if (this.typeAnnotation) {
+      code += ': ';
+      code += this.typeAnnotation.toEmbeddedTS(shiftState(state, code));
+      // toTS暂时只需要翻译引用数据类型的，因为值都是字符串
+      // 就不在这里赋值了，这里单纯就为了快速查找
+      // 并且屏蔽了一些异常的错误
+      if (this.typeAnnotation?.typeKind !== 'primitive' && this.defaultValue) {
+        if (this.isEnum()) {
+          code += `= ${this.typeAnnotation.typeKey}['${this.defaultValue}']`;
         }
-        return params;
-    }
-    /**
-     * 设置输出参数名称
-     */
-    setName(name: string) {
-        const object = {
-            name,
-        };
-        this.update({
-            ...object,
-            field: 'name',
-        });
-    }
-
-    /**
-     * 设置输出参数描述
-     */
-    setDescription(description: string) {
-        const object = {
-            description,
-        };
-        this.update({
-            ...object,
-        });
+      }
     }
-
-    /**
-     * 设置默认值
-     */
-    setDefaultValue(defaultValue: string) {
-        const object = {
-            defaultValue,
-        };
-        this.update({
-            ...object,
-        });
-    }
-
-    
-
-    
-
-    /**
-    * 删除类型标注
-    * @param name 类型标注名称
-    */
-    removeTypeAnnotation(name: string): void;
-
-    /**
-     * 删除类型标注
-     * @param typeAnnotation 已有的类型标注实例
-     */
-    removeTypeAnnotation(typeAnnotation: TypeAnnotation): void;
-
-    removeTypeAnnotation(options: string | TypeAnnotation) {
-        let typeAnnotation: TypeAnnotation;
-        if (typeof options === 'string') {
-            typeAnnotation = this.typeAnnotation;
-            if (!typeAnnotation) {
-                throw new Error('找不到类型标注 ' + options);
-            }
-        } else {
-            typeAnnotation = options;
-        }
-        return typeAnnotation.delete();
-    }
-
-    __removeTypeAnnotation(typeAnnotation: TypeAnnotation) {
-        const parentKey = typeAnnotation.parentKey;
-        const params: Params = {
-            parentNode: this,
-            parentKey,
-            index: -1,
-            object: null,
-            oldObject: typeAnnotation,
-        };
-        if (parentKey) {
-            params.parentKey = parentKey;
-            if (Array.isArray((this as any)[parentKey])) {
-                const index = (this as any)[parentKey].indexOf(typeAnnotation);
-                ~index && (this as any)[parentKey].splice(index, 1);
-                params.index = index;
-            } else if ((this as any)[parentKey] === typeAnnotation) {
-                params.index = 0;
-                (this as any)[parentKey] = undefined;
-            }
-        }
-        return params;
-    }
-
-    //================================================================================
-    // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
-    // 自动生成的代码已结束。下面可以手动编写。
-    //================================================================================
-
-    @excludedInJSON()
-    useComponent: {
-        type: 's-member-identifier-select' | 's-enum-select' | 's-schema-select' | 's-datatype-select' | 's-options-select';
-        attrs?: any;
-    } = undefined;
-
-    /**
-     * 自动补全的属性列表
-     */
-    completionChildren: Array<EntityProperty | StructureProperty> = undefined;
-
-    toJS(state?: TranslatorState): string {
-        const context = this.view ? 'this' : 'Vue.prototype';
-        const typeAnnotation = this.typeAnnotation || this.__TypeAnnotation;
-        const needGenInitFromSchema = typeAnnotation && (
-            !['primitive', 'union'].includes(typeAnnotation.typeKind) 
-            || typeAnnotation.typeName === 'DateTime'
-        );
-        let code = `${this.name} =`;
-        if (needGenInitFromSchema) {
-            code += `${context}.$genInitFromSchema('${typeAnnotation?.sortedTypeKey}',`;
-        }
-        const defaultValue = this.defaultValue;
-        const defaultValueType = Object.prototype.toString.call(defaultValue);
-        const { typeKind, typeNamespace, typeName } = typeAnnotation || {};
-        // 设置成null，才能同步给后端清除该值，但是null对checkbox组件是一种特殊状态
-        let parsedValue = defaultValue ?? undefined;
-        if (
-            defaultValueType === '[object String]'
-            && (
-                !(typeKind === 'primitive' && typeNamespace === 'nasl.core' && ['String', 'Text', 'Email'].includes(typeName))
-                && !(this.isEnum())
-                && !['union'].includes(typeKind)
-            )
-        ) {
-            // 一些特殊情况，特殊处理成undefined
-            // 1.defaultValue在nasl节点上错误得赋值给了空制符串
-            if ([''].includes(defaultValue)) {
-                parsedValue = undefined;
-            } else {
-                parsedValue = tryJSONParse(defaultValue) ?? defaultValue;
-            }
-        }
-        code += JSON.stringify(parsedValue);
-        if (needGenInitFromSchema) {
-            code += `)`;
+    // if (this.defaultValue) {
+    //     code += ` = ${this.defaultValue}`;
+    // }
+    return code;
+  }
+
+  @withSourceMap()
+  toNaturalTS(state?: TranslatorState): string {
+    let code = this.name || '__return';
+    if (this.typeAnnotation) {
+      code += ': ';
+      code += this.typeAnnotation.toNaturalTS(shiftState(state, code));
+      // toTS暂时只需要翻译引用数据类型的，因为值都是字符串
+      // 就不在这里赋值了，这里单纯就为了快速查找
+      // 并且屏蔽了一些异常的错误
+      if (this.typeAnnotation?.typeKind !== 'primitive' && this.defaultValue) {
+        if (this.isEnum()) {
+          code += `= ${this.typeAnnotation.typeKey}['${this.defaultValue}']`;
         }
-        return code;
+      }
     }
-
-    @withSourceMap()
-    toEmbeddedTS(state?: TranslatorState): string {
-        let code = this.name || '__return';
-        if (this.typeAnnotation) {
-            code += ': ';
-            code += this.typeAnnotation.toEmbeddedTS(shiftState(state, code));
-            // toTS暂时只需要翻译引用数据类型的，因为值都是字符串
-            // 就不在这里赋值了，这里单纯就为了快速查找
-            // 并且屏蔽了一些异常的错误
-            if (this.typeAnnotation?.typeKind !== 'primitive' && this.defaultValue) {
-                if (this.isEnum()) {
-                    code += `= ${this.typeAnnotation.typeKey}['${this.defaultValue}']`;
-                }
-            }
-        }
-        // if (this.defaultValue) {
-        //     code += ` = ${this.defaultValue}`;
-        // }
-        return code;
-    }
-    isEnum() {
-        return this.typeAnnotation?.typeNamespace?.endsWith('.enums');
+    return code;
+  }
+
+  isEnum() {
+    return this.typeAnnotation?.typeNamespace?.endsWith('.enums');
+  }
+
+  @withSourceMap()
+  toEmbeddedTSInProcess(state?: TranslatorState): string {
+    let code = this.name;
+    if (this.typeAnnotation) {
+      code += ': undefined as ';
+      code += this.typeAnnotation.toEmbeddedTS(shiftState(state, code));
     }
-    @withSourceMap()
-    toEmbeddedTSInProcess(state?: TranslatorState): string {
-        let code = this.name;
-        if (this.typeAnnotation) {
-            code += ': undefined as ';
-            code += this.typeAnnotation.toEmbeddedTS(shiftState(state, code));
-        }
-        // if (this.defaultValue) {
-        //     code += `: ${this.defaultValue}`;
-        // }
-        return code;
-    }
-
-    /**
-     * 设置数据类型
-     * @param typeAnnotation 类型标注对象
-     */
-    async setDataType(typeAnnotation: TypeAnnotation): Promise<void>;
-    /**
-     * 设置数据类型
-     * @param typeAnnotation 类型唯一标识
-     */
-    async setDataType(typeKey: string): Promise<void>;
-    async setDataType(typeAnnotation: TypeAnnotation | string) {
-        if (typeAnnotation) {
-            if (typeof typeAnnotation === 'string') {
-                typeAnnotation = dataTypesMap[typeAnnotation];
-                if (!typeAnnotation)
-                    throw new Error('找不到类型：' + typeAnnotation);
-            } else {
-                typeAnnotation._delete?.();
-                const relationOptions = { parentId: this.id, parentConcept: this.concept, parentNode: this };
-                Object.assign(typeAnnotation, relationOptions);
-            }
-            // 处理下拉属性
-            try {
-                this.completionChildren = this.getSelectRef(typeAnnotation, true);
-            } catch (err) {
-                this.completionChildren = [];
-                console.log(err);
-            }
-        } else {
-            this.completionChildren = [];
-        }
-        const object: any = {
-            typeAnnotation: typeAnnotation || null,
-        };
-
-        // 切换类型的时候如果有默认值就清除掉
-        if (this.defaultValue) {
-            object.defaultValue = '';
-        }
-        return this.update({
-            ...object,
-        });
-    }
-
-    public static getDefaultOptions() {
-        return {};
+    // if (this.defaultValue) {
+    //     code += `: ${this.defaultValue}`;
+    // }
+    return code;
+  }
+
+  /**
+   * 设置数据类型
+   * @param typeAnnotation 类型标注对象
+   */
+  async setDataType(typeAnnotation: TypeAnnotation): Promise<void>;
+  /**
+   * 设置数据类型
+   * @param typeAnnotation 类型唯一标识
+   */
+  async setDataType(typeKey: string): Promise<void>;
+  async setDataType(typeAnnotation: TypeAnnotation | string) {
+    if (typeAnnotation) {
+      if (typeof typeAnnotation === 'string') {
+        typeAnnotation = dataTypesMap[typeAnnotation];
+        if (!typeAnnotation) throw new Error(`找不到类型：${typeAnnotation}`);
+      } else {
+        typeAnnotation._delete?.();
+        const relationOptions = { parentId: this.id, parentConcept: this.concept, parentNode: this };
+        Object.assign(typeAnnotation, relationOptions);
+      }
+      // 处理下拉属性
+      try {
+        this.completionChildren = this.getSelectRef(typeAnnotation, true);
+      } catch (err) {
+        this.completionChildren = [];
+        console.log(err);
+      }
+    } else {
+      this.completionChildren = [];
     }
+    const object: any = {
+      typeAnnotation: typeAnnotation || null,
+    };
 
-    /**
-     * 是否可编辑
-     * */
-    get readonly() {
-        return !!this.module;
+    // 切换类型的时候如果有默认值就清除掉
+    if (this.defaultValue) {
+      object.defaultValue = '';
     }
-    //================================================================================
-    // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
-    // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-    //================================================================================
+    return this.update({
+      ...object,
+    });
+  }
+
+  public static getDefaultOptions() {
+    return {};
+  }
+
+  /**
+   * 是否可编辑
+   * */
+  get readonly() {
+    return !!this.module;
+  }
+  //= ===============================================================================
+  // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
+  // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
+  //= ===============================================================================
 }
 
 classMap.Return = Return;
 export default Return;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================