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

package/src/concepts/SqlQueryComponent__.ts

@@ -1,3 +1,4 @@
+import { v4 as uuidv4 } from 'uuid';
 import { TranslatorState, shiftState, withSourceMap, indent } from '../translator';
 import { getNodeByNodeCallee, getNaslNodeByNodeCallee } from '../automate/engine/utils';
 import StructureProperty from './StructureProperty__';
@@ -5,15 +6,14 @@ import Structure from './Structure__';
 import Identifier from './Identifier__';
 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 TypeAnnotation from './TypeAnnotation__';
@@ -24,264 +24,266 @@ import LogicItem from './LogicItem__';
  */
 @concept('SQL 查询')
 export class SqlQueryComponent extends LogicItem {
-    /**
-     * 产品概念
-     */
-    @property()
-    concept: 'SqlQueryComponent' = 'SqlQueryComponent';
-
-    /**
-     * SQL 查询标题
-     */
-    @property()
-    label: string = 'SQL 查询';
+  /**
+   * 产品概念
+   */
+  @property()
+  concept: 'SqlQueryComponent' = 'SqlQueryComponent';
 
-    /**
-     * SQL 查询
-     */
-    @property()
-    code: string = undefined;
+  /**
+   * SQL 查询标题
+   */
+  @property()
+  label: string = 'SQL 查询';
 
-    /**
-     * 相关类型
-     */
-    @property('TypeAnnotation')
-    typeAnnotation: TypeAnnotation = undefined;
+  /**
+   * SQL 查询
+   */
+  @property()
+  code: string = undefined;
 
-    /**
-     * 数据源
-     */
-    @property()
-    dataSource: string = undefined;
+  /**
+   * 相关类型
+   */
+  @property('TypeAnnotation')
+  typeAnnotation: TypeAnnotation = undefined;
 
-    /**
-     * 查询组件的局部版本
-     */
-    @property()
-    ideVersion: string = undefined;
+  /**
+   * 数据源
+   */
+  @property()
+  dataSource: string = undefined;
 
-    
-
-    /**
-     * @param source 需要合并的部分参数
-     */
-    constructor(source?: Partial<SqlQueryComponent>) {
-        source = Object.assign({}, SqlQueryComponent.getDefaultOptions(), source);
-        super(source);
-        super.subConstructor(source);
-    }
-    getClassName() {
-        return 'SqlQueryComponent';
-    }
+  /**
+   * 查询组件的局部版本
+   */
+  @property()
+  ideVersion: string = undefined;
 
-    static from(source: any, parentNode?: any, parentKey?: string): SqlQueryComponent {
-        return super.from(source, parentNode, parentKey) as SqlQueryComponent;
-    }
+  /**
+   * @param source 需要合并的部分参数
+   */
+  constructor(source?: Partial<SqlQueryComponent>) {
+    // @ts-ignore
+    source = { ...SqlQueryComponent.getDefaultOptions(), ...source };
+    super(source);
+    super.subConstructor(source);
+  }
 
-    
-    /**
-     * 设置SQL 查询
-     */
-    setCode(code: string) {
-        const object = {
-            code,
-        };
-        this.update({
-            ...object,
-        });
-    }
+  getClassName() {
+    return 'SqlQueryComponent';
+  }
 
-    
+  static from(source: any, parentNode?: any, parentKey?: string): SqlQueryComponent {
+    return super.from(source, parentNode, parentKey) as SqlQueryComponent;
+  }
 
-    
+  /**
+   * 设置SQL 查询
+   */
+  setCode(code: string) {
+    const object = {
+      code,
+    };
+    this.update({
+      ...object,
+    });
+  }
 
-    /**
-    * 删除类型标注
-    * @param name 类型标注名称
-    */
-    removeTypeAnnotation(name: string): void;
+  /**
+   * 删除类型标注
+   * @param name 类型标注名称
+   */
+  removeTypeAnnotation(name: string): void;
 
-    /**
-     * 删除类型标注
-     * @param typeAnnotation 已有的类型标注实例
-     */
-    removeTypeAnnotation(typeAnnotation: TypeAnnotation): 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(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;
+  __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;
+      }
     }
+    return params;
+  }
 
-    //================================================================================
-    // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
-    // 自动生成的代码已结束。下面可以手动编写。
-    //================================================================================
+  //= ===============================================================================
+  // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+  // 自动生成的代码已结束。下面可以手动编写。
+  //= ===============================================================================
 
-    /**
-     * APP数据源重命名时，同步SQL数据源；此方法仅提供给APP重命名时使用
-     */
-    renameDataSource(dataSource: string) {
-        // 此参数dataSource不带命名空间
-        const object = {
-            dataSource,
-        };
-        this.update({
-            ...object,
-        });
-    }
+  /**
+   * APP数据源重命名时，同步SQL数据源；此方法仅提供给APP重命名时使用
+   */
+  renameDataSource(dataSource: string) {
+    // 此参数dataSource不带命名空间
+    const object = {
+      dataSource,
+    };
+    this.update({
+      ...object,
+    });
+  }
 
-    getTextVariables() {
-        this._TsVariables = [];
-        const str = this.code;
-        const variables: Set<string> = new Set();
-        let temp: any;
-        const re = /\{(.+?)\}/g;
-        while (temp = re.exec(str)) {
-            let variable = temp[1];
-            variable = variable.replaceAll(' ', '');
-            variables.add(variable);
-        }
-        for (const variable of variables) { // 遍历Set
-            let newVariable: Identifier | MemberExpression;
-            if (variable.includes('.')) {
-                const arr = variable.split('.');
-                for (let i = 0; i < arr.length; i++) {
-                    if (!newVariable) {
-                        // value需要拼接生成
-                        newVariable = new Identifier({ name: arr[i] });
-                        (newVariable as any).noTsReName = true;
-                    } else {
-                        const property = new Identifier({ name: arr[i] });
-                        (property as any).noTsReName = true;
-                        newVariable = new MemberExpression({
-                            object: newVariable,
-                            property,
-                        });
-                        (newVariable as any).noTsReName = true;
-                    }
-                }
-            } else {
-                newVariable = new Identifier({ name: variable });
-                (newVariable as any).noTsReName = true;
-            }
-            newVariable.parentNode = this;
-            newVariable.parentKey = '_TsVariables';
-            this._TsVariables.push(newVariable);
+  getTextVariables() {
+    this._TsVariables = [];
+    const str = this.code;
+    const variables: Set<string> = new Set();
+    let temp: any;
+    const re = /\{(.+?)\}/g;
+    while ((temp = re.exec(str))) {
+      let variable = temp[1];
+      variable = variable.replaceAll(' ', '');
+      variables.add(variable);
+    }
+    for (const variable of variables) {
+      // 遍历Set
+      let newVariable: Identifier | MemberExpression;
+      if (variable.includes('.')) {
+        const arr = variable.split('.');
+        for (let i = 0; i < arr.length; i++) {
+          if (!newVariable) {
+            // value需要拼接生成
+            newVariable = new Identifier({ name: arr[i] });
+            (newVariable as any).noTsReName = true;
+          } else {
+            const property = new Identifier({ name: arr[i] });
+            (property as any).noTsReName = true;
+            newVariable = new MemberExpression({
+              object: newVariable,
+              property,
+            });
+            (newVariable as any).noTsReName = true;
+          }
         }
+      } else {
+        newVariable = new Identifier({ name: variable });
+        (newVariable as any).noTsReName = true;
+      }
+      newVariable.parentNode = this;
+      newVariable.parentKey = '_TsVariables';
+      this._TsVariables.push(newVariable);
     }
+  }
 
-    getTypeBanError() {
-        let sqlTypeBan = false;
-        if (this.typeAnnotation) {
-            const getSqlTypeBan = (typeAnnotation: TypeAnnotation) => {
-                const ref = this.getSelectRef(typeAnnotation);
-                if (ref?.length) {
-                    ref.forEach((item: StructureProperty) => {
-                        // 如果有类型但是类型不是基本类型，就需要打开开关，注入变量
-                        if (item?.typeAnnotation && item.typeAnnotation.typeNamespace !== 'nasl.core' && !(item.typeAnnotation.typeNamespace.endsWith('.enums'))) {
-                            sqlTypeBan = true;
-                        }
-                    });
-                }
-            };
-            if (this.typeAnnotation.typeArguments?.length) {
-                // list 就去校验一下内部的类型
-                if (this.typeAnnotation.typeName === 'List') {
-                    this.typeAnnotation.typeArguments.forEach((item) => {
-                        getSqlTypeBan(item);
-                    });
-                } else {
-                    // 别的都不放开
-                    sqlTypeBan = true;
-                }
-            } else {
-                getSqlTypeBan(this.typeAnnotation);
+  getTypeBanError() {
+    let sqlTypeBan = false;
+    if (this.typeAnnotation) {
+      const getSqlTypeBan = (typeAnnotation: TypeAnnotation) => {
+        const ref = this.getSelectRef(typeAnnotation);
+        if (ref?.length) {
+          ref.forEach((item: StructureProperty) => {
+            // 如果有类型但是类型不是基本类型，就需要打开开关，注入变量
+            if (item?.typeAnnotation && item.typeAnnotation.typeNamespace !== 'nasl.core' && !item.typeAnnotation.typeNamespace.endsWith('.enums')) {
+              sqlTypeBan = true;
             }
+          });
         }
-        return sqlTypeBan;
-    }
-
-    @withSourceMap()
-    toEmbeddedTS(state?: TranslatorState): string {
-        let code: string;
-        if (this.ideVersion === '2.11') {
-            code = `(function(): `;
-            code += `${this.typeAnnotation ? this.typeAnnotation.toEmbeddedTS(shiftState(state, code)) : '__IDENTIFIER__'} {\n`;
+      };
+      if (this.typeAnnotation.typeArguments?.length) {
+        // list 就去校验一下内部的类型
+        if (this.typeAnnotation.typeName === 'List') {
+          this.typeAnnotation.typeArguments.forEach((item) => {
+            getSqlTypeBan(item);
+          });
         } else {
-            code = `(function(): nasl.collection.List<`;
-            code += `${this.typeAnnotation ? this.typeAnnotation.toEmbeddedTS(shiftState(state, code)) : '__IDENTIFIER__'}> {\n`;
-        }
-        // app.dataSources.defaultDS;
-        if (this.dataSource) {
-            code += indent((state?.tabSize || 0) + 1);
-            code += `const dataSource = app.dataSources.${this.dataSource};\n`;
+          // 别的都不放开
+          sqlTypeBan = true;
         }
-        // 生成内部的ts内容
-        // 临时方案拼接字符串
-        this.getTextVariables();
-        // 翻译字段
-        this._TsVariables.forEach((variable) => {
-            code += variable.toEmbeddedTS(shiftState(state, code));
-            code += ';\n';
-        });
-        code += '\n';
-        code += indent((state?.tabSize || 0) + 1) + 'return;\n';
-        code += indent((state?.tabSize || 0)) + '})()';
-        return code;
+      } else {
+        getSqlTypeBan(this.typeAnnotation);
+      }
     }
+    return sqlTypeBan;
+  }
 
-    /**
-     * 获取添加时的默认选项
-     * @returns
-     */
-    public static getDefaultOptions(): any {
-        return {
-            dataSource: 'defaultDS',
-        };
+  @withSourceMap()
+  toEmbeddedTS(state?: TranslatorState): string {
+    let code: string;
+    if (this.ideVersion === '2.11') {
+      code = `(function(): `;
+      code += `${this.typeAnnotation ? this.typeAnnotation.toEmbeddedTS(shiftState(state, code)) : '__IDENTIFIER__'} {\n`;
+    } else {
+      code = `(function(): nasl.collection.List<`;
+      code += `${this.typeAnnotation ? this.typeAnnotation.toEmbeddedTS(shiftState(state, code)) : '__IDENTIFIER__'}> {\n`;
+    }
+    // app.dataSources.defaultDS;
+    if (this.dataSource) {
+      code += indent((state?.tabSize || 0) + 1);
+      code += `const dataSource = app.dataSources.${this.dataSource};\n`;
     }
+    // 生成内部的ts内容
+    // 临时方案拼接字符串
+    this.getTextVariables();
+    // 翻译字段
+    this._TsVariables.forEach((variable) => {
+      code += variable.toEmbeddedTS(shiftState(state, code));
+      code += ';\n';
+    });
+    code += '\n';
+    code += `${indent((state?.tabSize || 0) + 1)}return;\n`;
+    code += `${indent(state?.tabSize || 0)}})()`;
+    return code;
+  }
+
+  @withSourceMap()
+  toNaturalTS(state?: TranslatorState): string {
+    const code: string = `nasl.oql.query(\`${this.code}\`)`;
+    return code;
+  }
+
+  /**
+   * 获取添加时的默认选项
+   * @returns
+   */
+  public static getDefaultOptions(): any {
+    return {
+      dataSource: 'defaultDS',
+    };
+  }
 
-    _TsVariables: Array<Identifier | MemberExpression>= [];
+  _TsVariables: Array<Identifier | MemberExpression> = [];
 
-    //================================================================================
-    // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
-    // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-    //================================================================================
+  //= ===============================================================================
+  // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
+  // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
+  //= ===============================================================================
 }
 
 classMap.SqlQueryComponent = SqlQueryComponent;
 export default SqlQueryComponent;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================

package/src/concepts/Start__.ts

@@ -1,13 +1,13 @@
+import { v4 as uuidv4 } from 'uuid';
 import { TranslatorState, withSourceMap } 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,66 +17,59 @@ import LogicItem from './LogicItem__';
  */
 @concept('开始节点')
 export class Start extends LogicItem {
-    /**
-     * 产品概念
-     */
-    @property()
-    concept: 'Start' = 'Start';
-
-    /**
-     * 开始节点标题
-     */
-    @property()
-    label: string = '开始';
-
-    
-
-    /**
-     * @param source 需要合并的部分参数
-     */
-    constructor(source?: Partial<Start>) {
-        source = Object.assign({}, Start.getDefaultOptions(), source);
-        super(source);
-        super.subConstructor(source);
-    }
-    getClassName() {
-        return 'Start';
-    }
+  /**
+   * 产品概念
+   */
+  @property()
+  concept: 'Start' = 'Start';
 
-    static from(source: any, parentNode?: any, parentKey?: string): Start {
-        return super.from(source, parentNode, parentKey) as Start;
-    }
+  /**
+   * 开始节点标题
+   */
+  @property()
+  label: string = '开始';
 
-    
-    
+  /**
+   * @param source 需要合并的部分参数
+   */
+  constructor(source?: Partial<Start>) {
+    // @ts-ignore
+    source = { ...Start.getDefaultOptions(), ...source };
+    super(source);
+    super.subConstructor(source);
+  }
 
-    
+  getClassName() {
+    return 'Start';
+  }
 
-    
+  static from(source: any, parentNode?: any, parentKey?: string): Start {
+    return super.from(source, parentNode, parentKey) as Start;
+  }
 
-    //================================================================================
-    // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
-    // 自动生成的代码已结束。下面可以手动编写。
-    //================================================================================
+  //= ===============================================================================
+  // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+  // 自动生成的代码已结束。下面可以手动编写。
+  //= ===============================================================================
 
-    toJS(state?: TranslatorState): string {
-        return '';
-    }
+  toJS(state?: TranslatorState): string {
+    return '';
+  }
 
-    @withSourceMap()
-    toEmbeddedTS(state?: TranslatorState): string {
-        return '';
-    }
+  @withSourceMap()
+  toEmbeddedTS(state?: TranslatorState): string {
+    return '';
+  }
 
-    //================================================================================
-    // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
-    // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-    //================================================================================
+  //= ===============================================================================
+  // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
+  // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
+  //= ===============================================================================
 }
 
 classMap.Start = Start;
 export default Start;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================