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

package/src/concepts/EntityProperty__.ts

@@ -1,19 +1,20 @@
+import { v4 as uuidv4 } from 'uuid';
 import { dataTypesMap } from './basics/types';
-import { TranslatorState, shiftState, withSourceMap, indent } from '../translator';
+import { TranslatorState, shiftState, withSourceMap, indent, createCompilerState } from '../translator';
+import { getNodeByNodeCallee, getNaslNodeByNodeCallee } from '../automate/engine/utils';
 import CallQueryComponent from './CallQueryComponent__';
-
-type changeType = 'up' | 'down';
-
-//================================================================================
+import StructureProperty from './StructureProperty__';
+import Structure from './Structure__';
+//= ===============================================================================
 // 从这里开始到结尾注释之间的代码由 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 TypeAnnotation from './TypeAnnotation__';
 import DatabaseTypeAnnotation from './DatabaseTypeAnnotation__';
 import Entity from './Entity__';
@@ -21,28 +22,24 @@ import Module from './Module__';
 import App from './App__';
 import DataSource from './DataSource__';
 
+type changeType = 'up' | 'down';
+
 /**
  * 实体属性
  */
-@className
 @concept('实体属性')
-// @ts-ignore
 export class EntityProperty extends BaseNode {
-  /** 类名 */
-  static readonly ConceptName = 'EntityProperty';
-
   /**
    * 产品概念
    */
-  // @ts-ignore
-  readonly concept!: 'EntityProperty';
+  @property()
+  concept: 'EntityProperty' = 'EntityProperty';
 
   /**
    * 实体属性名称
    */
   @property()
-  // @ts-ignore
-  name!: string;
+  name: string = undefined;
 
   /**
    * 唯一标识
@@ -54,85 +51,73 @@ export class EntityProperty extends BaseNode {
    * 数据库列名
    */
   @property()
-  // @ts-ignore
-  columnName!: string;
+  columnName: string = undefined;
 
   /**
    * 实体属性标题
    */
   @property()
-  // @ts-ignore
-  label!: string;
+  label: string = undefined;
 
   /**
    * 实体属性描述
    */
   @property()
-  // @ts-ignore
-  description!: string;
+  description: string = undefined;
 
   /**
    * 类型
    */
   @property('TypeAnnotation')
-  // @ts-ignore
-  typeAnnotation!: TypeAnnotation;
+  typeAnnotation: TypeAnnotation = undefined;
 
   /**
    * 数据库存储类型
    */
   @property('DatabaseTypeAnnotation')
-  // @ts-ignore
-  databaseTypeAnnotation!: DatabaseTypeAnnotation;
+  databaseTypeAnnotation: DatabaseTypeAnnotation = undefined;
 
   /**
    * 是否必填
    */
   @property()
-  // @ts-ignore
-  required!: boolean;
+  required: boolean = undefined;
 
   /**
    * 默认值
    */
   @property()
-  // @ts-ignore
-  defaultValue!: string;
+  defaultValue: string = undefined;
 
   /**
    * 当前属性是否为主键
    */
   @property()
-  // @ts-ignore
-  primaryKey!: boolean;
+  primaryKey: boolean = undefined;
 
   /**
    * 关联实体的命名空间
    */
   @property()
-  // @ts-ignore
-  relationNamespace!: string;
+  relationNamespace: string = undefined;
 
   /**
    * 关联实体
    */
   @property()
-  // @ts-ignore
-  relationEntity!: string;
+  relationEntity: string = undefined;
 
   /**
    * 关联属性
    */
   @property()
-  // @ts-ignore
-  relationProperty!: string;
+  relationProperty: string = undefined;
 
   /**
    * 关联属性实体记录删除规则
    */
   @property()
-  // @ts-ignore
-  deleteRule!: 'protect' | 'cascade';
+  deleteRule: 'protect' | 'cascade' = undefined;
 
   /**
    * 显示方式
@@ -161,8 +146,7 @@ export class EntityProperty extends BaseNode {
    * 表示 Oracle 主键自增时的序列名称
    */
   @property()
-  // @ts-ignore
-  sequence!: string;
+  sequence: string = undefined;
 
   /**
    * 祖先 Entity
@@ -170,18 +154,21 @@ export class EntityProperty extends BaseNode {
   get entity() {
     return this.getAncestor('Entity') as Entity;
   }
+
   /**
    * 祖先 Module
    */
   get module() {
     return this.getAncestor('Module') as Module;
   }
+
   /**
    * 祖先 App
    */
   get app() {
     return this.getAncestor('App') as App;
   }
+
   /**
    * 祖先 DataSource
    */
@@ -193,11 +180,16 @@ export class EntityProperty extends BaseNode {
    * @param source 需要合并的部分参数
    */
   constructor(source?: Partial<EntityProperty>) {
-    source = Object.assign({}, EntityProperty.getDefaultOptions(), source);
+    // @ts-ignore
+    source = { ...EntityProperty.getDefaultOptions(), ...source };
     super(source);
     super.subConstructor(source);
   }
 
+  getClassName() {
+    return 'EntityProperty';
+  }
+
   static from(source: any, parentNode?: any, parentKey?: string): EntityProperty {
     return super.from(source, parentNode, parentKey) as EntityProperty;
   }
@@ -213,6 +205,7 @@ export class EntityProperty extends BaseNode {
     }
     return params;
   }
+
   /**
    * 设置实体属性标题
    */
@@ -317,7 +310,7 @@ export class EntityProperty extends BaseNode {
     if (typeof options === 'string') {
       typeAnnotation = this.typeAnnotation;
       if (!typeAnnotation) {
-        throw new Error('找不到类型标注 ' + options);
+        throw new Error(`找不到类型标注 ${options}`);
       }
     } else {
       typeAnnotation = options;
@@ -326,7 +319,7 @@ export class EntityProperty extends BaseNode {
   }
 
   __removeTypeAnnotation(typeAnnotation: TypeAnnotation) {
-    const parentKey = typeAnnotation.parentKey;
+    const { parentKey } = typeAnnotation;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -365,7 +358,7 @@ export class EntityProperty extends BaseNode {
     if (typeof options === 'string') {
       databaseTypeAnnotation = this.databaseTypeAnnotation;
       if (!databaseTypeAnnotation) {
-        throw new Error('找不到数据库存储类型 ' + options);
+        throw new Error(`找不到数据库存储类型 ${options}`);
       }
     } else {
       databaseTypeAnnotation = options;
@@ -374,7 +367,7 @@ export class EntityProperty extends BaseNode {
   }
 
   __removeDatabaseTypeAnnotation(databaseTypeAnnotation: DatabaseTypeAnnotation) {
-    const parentKey = databaseTypeAnnotation.parentKey;
+    const { parentKey } = databaseTypeAnnotation;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -396,10 +389,10 @@ export class EntityProperty extends BaseNode {
     return params;
   }
 
-  //================================================================================
+  //= ===============================================================================
   // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
   // 自动生成的代码已结束。下面可以手动编写。
-  //================================================================================
+  //= ===============================================================================
 
   /**
    * 设置实体属性名称
@@ -412,9 +405,7 @@ export class EntityProperty extends BaseNode {
     // 列名为初始化时需赋值为属性名，同时需要与其他列名中去重
     if ((this as any).__columnNameInit === false) {
       delete (this as any).__columnNameInit;
-      const columnNames = (this.parentNode as Entity).properties
-        .filter((item) => item !== this)
-        .map((item) => item.columnName);
+      const columnNames = (this.parentNode as Entity).properties.filter((item) => item !== this).map((item) => item.columnName);
       const columnName = utils.unique(name, columnNames, undefined, false);
       object.columnName = columnName;
 
@@ -487,7 +478,7 @@ export class EntityProperty extends BaseNode {
     });
     callQueryComponents?.forEach((node) => {
       const targetField = node.select.selectFieldElements.find(
-        (item) => item.entityAsName === this.parentNode.name && item.propertyName === this.name,
+        (item) => item.entityAsName === this.parentNode.name && item.propertyName === this.name
       );
       if (targetField) {
         if (options.action === 'delete') {
@@ -551,6 +542,14 @@ export class EntityProperty extends BaseNode {
    * 此方法不会 update
    */
   __formatColumnName() {
+    const parentNode = this.parentNode as Entity;
+    if (parentNode.isIgnoreFormatTableAndColumnName()) {
+      // if (!this.columnName) {
+      //     this.columnName = this.name;
+      // }
+      return;
+    }
+
     const newColumnName = this.genFormatColumnName();
     if (newColumnName && (!this.columnName || (this.columnName && newColumnName !== this.columnName))) {
       this.columnName = newColumnName;
@@ -561,6 +560,14 @@ export class EntityProperty extends BaseNode {
    * 根据全局设置中的数据库命名规则，将 属性列名 重命名
    */
   formatColumnName() {
+    const parentNode = this.parentNode as Entity;
+    if (parentNode.isIgnoreFormatTableAndColumnName()) {
+      // if (!this.columnName) {
+      //     this.setColumnName(this.name);
+      // }
+      return;
+    }
+
     const newColumnName = this.genFormatColumnName();
     if (newColumnName && (!this.columnName || (this.columnName && newColumnName !== this.columnName))) {
       this.setColumnName(newColumnName);
@@ -568,7 +575,7 @@ export class EntityProperty extends BaseNode {
   }
 
   @withSourceMap()
-  toEmbeddedTS(state?: TranslatorState): string {
+  toEmbeddedTS(state = createCompilerState()): string {
     let code = '';
     if (this.relationNamespace && this.relationEntity && this.relationProperty) {
       code += `@nasl.annotation.EntityRelation(new ${this.relationNamespace}.${this.relationEntity}().${this.relationProperty})\n`;
@@ -587,6 +594,22 @@ export class EntityProperty extends BaseNode {
     return code;
   }
 
+  @withSourceMap()
+  toNaturalTS(state = createCompilerState()): string {
+    let code = '';
+    if (this.relationNamespace && this.relationEntity && this.relationProperty) {
+      code += `@EntityRelation(() => new ${this.relationEntity}().${this.relationProperty})\n`;
+      code += indent(state.tabSize);
+    }
+
+    code += this.name;
+    if (this.typeAnnotation) {
+      code += ': ';
+      code += this.typeAnnotation.toNaturalTS(shiftState(state, code, { inline: true }));
+    }
+    return code;
+  }
+
   /**
    * 设置数据类型
    * @param typeAnnotation 类型标注对象
@@ -671,11 +694,7 @@ export class EntityProperty extends BaseNode {
    */
 
   changeSort(arr: any[], index: number, type: changeType) {
-    arr.splice(
-      type !== 'up' ? index : index - 1,
-      1,
-      ...arr.splice(type !== 'up' ? index + 1 : index, 1, arr[type !== 'up' ? index : index - 1]),
-    );
+    arr.splice(type !== 'up' ? index : index - 1, 1, ...arr.splice(type !== 'up' ? index + 1 : index, 1, arr[type !== 'up' ? index : index - 1]));
     return arr;
   }
 
@@ -832,14 +851,15 @@ export class EntityProperty extends BaseNode {
 
   __showAdvanced: boolean = false;
 
-  //================================================================================
+  //= ===============================================================================
   // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
   // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-  //================================================================================
+  //= ===============================================================================
 }
 
+classMap.EntityProperty = EntityProperty;
 export default EntityProperty;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================

package/src/concepts/Entity__.ts

@@ -1,17 +1,18 @@
+import { v4 as uuidv4 } from 'uuid';
 import Namespace from './Namespace__';
 import { TranslatorState, shiftState, withSourceMap, indent, createCompilerState } from '../translator';
 import { entity2LogicNamespace } from '../server/entity2LogicNamespace';
 import CallQueryComponent from './CallQueryComponent__';
-//================================================================================
+//= ===============================================================================
 // 从这里开始到结尾注释之间的代码由 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 EntityProperty from './EntityProperty__';
 import EntityIndex from './EntityIndex__';
 import Module from './Module__';
@@ -21,25 +22,19 @@ import App from './App__';
 /**
  * 实体
  */
-@className
 @concept('实体')
-// @ts-ignore
 export class Entity extends BaseNode {
-  /** 类名 */
-  static readonly ConceptName = 'Entity';
-
   /**
    * 产品概念
    */
-  // @ts-ignore
-  readonly concept!: 'Entity';
+  @property()
+  concept: 'Entity' = 'Entity';
 
   /**
    * 实体名称
    */
   @property()
-  // @ts-ignore
-  name!: string;
+  name: string = undefined;
 
   /**
    * 唯一标识
@@ -51,15 +46,13 @@ export class Entity extends BaseNode {
    * 数据库表名
    */
   @property()
-  // @ts-ignore
-  tableName!: string;
+  tableName: string = undefined;
 
   /**
    * 实体描述
    */
   @property()
-  // @ts-ignore
-  description!: string;
+  description: string = undefined;
 
   /**
    * 实体来源
@@ -85,12 +78,14 @@ export class Entity extends BaseNode {
   get module() {
     return this.getAncestor('Module') as Module;
   }
+
   /**
    * 祖先 App
    */
   get app() {
     return this.getAncestor('App') as App;
   }
+
   /**
    * 祖先 DataSource
    */
@@ -102,11 +97,16 @@ export class Entity extends BaseNode {
    * @param source 需要合并的部分参数
    */
   constructor(source?: Partial<Entity>) {
-    source = Object.assign({}, Entity.getDefaultOptions(), source);
+    // @ts-ignore
+    source = { ...Entity.getDefaultOptions(), ...source };
     super(source);
     super.subConstructor(source);
   }
 
+  getClassName() {
+    return 'Entity';
+  }
+
   static from(source: any, parentNode?: any, parentKey?: string): Entity {
     return super.from(source, parentNode, parentKey) as Entity;
   }
@@ -122,6 +122,7 @@ export class Entity extends BaseNode {
     }
     return params;
   }
+
   /**
    * 设置实体描述
    */
@@ -136,10 +137,9 @@ export class Entity extends BaseNode {
 
   getPropertyExistingNames(excludedList: Array<EntityProperty> = []) {
     const excludedSet = new Set(excludedList);
-    return ((this.properties as EntityProperty[]) || [])
-      .filter((item) => !excludedSet.has(item))
-      .map((item) => item.name);
+    return ((this.properties as EntityProperty[]) || []).filter((item) => !excludedSet.has(item)).map((item) => item.name);
   }
+
   getPropertyUniqueName(name = 'property1') {
     return utils.unique(name, this.getPropertyExistingNames(), undefined, false);
   }
@@ -176,7 +176,7 @@ export class Entity extends BaseNode {
           name: this.getPropertyUniqueName(),
         },
         this,
-        'properties',
+        'properties'
       );
     } else if (typeof options === 'string') {
       property = EntityProperty.from(
@@ -185,7 +185,7 @@ export class Entity extends BaseNode {
           name: options,
         },
         this,
-        'properties',
+        'properties'
       );
     } else if (options instanceof EntityProperty) {
       options.ensureDelete(); // 同一实例不支持多处存在
@@ -198,7 +198,7 @@ export class Entity extends BaseNode {
           ...options,
         },
         this,
-        'properties',
+        'properties'
       );
     }
     this.properties.splice(index, 0, property);
@@ -293,6 +293,7 @@ export class Entity extends BaseNode {
     const excludedSet = new Set(excludedList);
     return ((this.indexes as EntityIndex[]) || []).filter((item) => !excludedSet.has(item)).map((item) => item.name);
   }
+
   getEntityIndexUniqueName(name = 'entityIndex1') {
     return utils.unique(name, this.getEntityIndexExistingNames(), undefined, false);
   }
@@ -329,7 +330,7 @@ export class Entity extends BaseNode {
           name: this.getEntityIndexUniqueName(),
         },
         this,
-        'indexes',
+        'indexes'
       );
     } else if (typeof options === 'string') {
       entityIndex = EntityIndex.from(
@@ -338,7 +339,7 @@ export class Entity extends BaseNode {
           name: options,
         },
         this,
-        'indexes',
+        'indexes'
       );
     } else if (options instanceof EntityIndex) {
       options.ensureDelete(); // 同一实例不支持多处存在
@@ -351,7 +352,7 @@ export class Entity extends BaseNode {
           ...options,
         },
         this,
-        'indexes',
+        'indexes'
       );
     }
     this.indexes.splice(index, 0, entityIndex);
@@ -459,7 +460,7 @@ export class Entity extends BaseNode {
     if (typeof options === 'string') {
       property = (this.properties as EntityProperty[]).find((item) => item.name === options);
       if (!property) {
-        throw new Error('找不到实体属性 ' + options);
+        throw new Error(`找不到实体属性 ${options}`);
       }
     } else {
       property = options;
@@ -468,7 +469,7 @@ export class Entity extends BaseNode {
   }
 
   __removeEntityProperty(property: EntityProperty) {
-    const parentKey = property.parentKey;
+    const { parentKey } = property;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -507,7 +508,7 @@ export class Entity extends BaseNode {
     if (typeof options === 'string') {
       entityIndex = (this.indexes as EntityIndex[]).find((item) => item.name === options);
       if (!entityIndex) {
-        throw new Error('找不到实体索引 ' + options);
+        throw new Error(`找不到实体索引 ${options}`);
       }
     } else {
       entityIndex = options;
@@ -516,7 +517,7 @@ export class Entity extends BaseNode {
   }
 
   __removeEntityIndex(entityIndex: EntityIndex) {
-    const parentKey = entityIndex.parentKey;
+    const { parentKey } = entityIndex;
     const params: Params = {
       parentNode: this,
       parentKey,
@@ -538,16 +539,14 @@ export class Entity extends BaseNode {
     return params;
   }
 
-  //================================================================================
+  //= ===============================================================================
   // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
   // 自动生成的代码已结束。下面可以手动编写。
-  //================================================================================
+  //= ===============================================================================
 
   getPropertyExistingColumnNames(excludedList: Array<EntityProperty> = []) {
     const excludedSet = new Set(excludedList);
-    return ((this.properties as EntityProperty[]) || [])
-      .filter((item) => !excludedSet.has(item))
-      .map((item) => item.columnName);
+    return ((this.properties as EntityProperty[]) || []).filter((item) => !excludedSet.has(item)).map((item) => item.columnName);
   }
 
   getPropertyUniqueColumnName(name = 'property1', excludedList: Array<EntityProperty> = []) {
@@ -810,7 +809,7 @@ export class Entity extends BaseNode {
   }
 
   @withSourceMap()
-  toEmbeddedTS(state?: TranslatorState): string {
+  toEmbeddedTS(state = createCompilerState()): string {
     let code = `export class ${this.name} {\n`;
     code += indent((state?.tabSize || 0) + 1);
     code += `__name: '${this.getNamespace()}.${this.name}';\n`;
@@ -848,6 +847,32 @@ export class Entity extends BaseNode {
     return code;
   }
 
+  @withSourceMap()
+  toNaturalTS(state = createCompilerState()): string {
+    let code = `class ${this.name} {\n`;
+    // code += `${indent(state.tabSize + 1)}__name: '${this.getNamespace()}.${this.name}';\n`;
+    this.properties.forEach((property) => {
+      code += `${property.toNaturalTS(shiftState(state, code, { tabSize: state.tabSize + 1 }))};\n`;
+    });
+    if (this.indexes && this.indexes.length) {
+      code += indent(state.tabSize + 1);
+      code += `__IndexList() {\n`;
+      code += `            return [`;
+      this.indexes.forEach((entityIndex) => {
+        code += indent(state.tabSize + 1);
+        code += '\n                ';
+        code += entityIndex.toEmbeddedTS(shiftState(state, code, { inline: true }));
+      });
+      code += `${indent(state.tabSize)}\n            ];\n        }\n`;
+    }
+
+    code += `${indent(state.tabSize)}}\n`;
+
+    code += `const ${this.name}Entity = createEntity<${this.name}>();\n`;
+
+    return code;
+  }
+
   getNamespace() {
     if (this.parentNode && (this.parentNode as DataSource).getNamespace)
       return `${(this.parentNode as DataSource).getNamespace()}.${this.parentNode.name}.entities`;
@@ -916,14 +941,15 @@ export class Entity extends BaseNode {
     const arr = this.getEntityIndexExistingNames(excludedList);
     return [...arr, 'primaryIndex'];
   }
-  //================================================================================
+  //= ===============================================================================
   // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
   // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
-  //================================================================================
+  //= ===============================================================================
 }
 
+classMap.Entity = Entity;
 export default Entity;
-//================================================================================
+//= ===============================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
 // 自动生成的代码已结束。下面可以手动编写。
-//================================================================================
+//= ===============================================================================