@lcap/nasl 1.0.2 → 2.12.0

package/src/concepts/DataSource__.ts

@@ -0,0 +1,391 @@
+import App from './App__';
+import { TranslatorState, withSourceMap, createCompilerState } 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 Entity from './Entity__';
+
+/**
+ * 数据源
+ */
+@concept('数据源')
+export class DataSource extends BaseNode {
+    /**
+     * 产品概念
+     */
+    @property()
+    concept: 'DataSource' = 'DataSource';
+
+    /**
+     * 数据源名称
+     */
+    @property()
+    name: string = undefined;
+
+    /**
+     * 数据源描述
+     */
+    @property()
+    description: string = undefined;
+
+    /**
+     * 实体列表
+     */
+    @property('Entity')
+    entities: Array<Entity> = [];
+
+    
+
+    /**
+     * @param source 需要合并的部分参数
+     */
+    constructor(source?: Partial<DataSource>) {
+        source = Object.assign({}, DataSource.getDefaultOptions(), source);
+        super(source);
+        super.subConstructor(source);
+    }
+
+    static from(source: any, parentNode?: any, parentKey?: string): DataSource {
+        return super.from(source, parentNode, parentKey) as DataSource;
+    }
+
+    /**
+     * 从父级删除该节点
+     * @internal
+     */
+    _delete() {
+        let params: Params = null;
+        if (this.parentNode) {
+            params = (this.parentNode as any)?.__removeDataSource?.(this);
+        }
+        return params;
+    }
+    /**
+     * 设置数据源描述
+     */
+    setDescription(description: string) {
+        const object = {
+            description,
+        };
+        this.update({
+            ...object,
+        });
+    }
+
+        getEntityExistingNames(excludedList: Array<Entity> = []) {
+        const excludedSet = new Set(excludedList);
+        return (this.entities || []).filter((item) => !excludedSet.has(item)).map((item) => item.name);
+    }
+    getEntityUniqueName(name = 'Entity1') {
+        return utils.unique(name, this.getEntityExistingNames(), undefined, true);
+    }
+
+    /**
+    * 插入实体
+    * @internal
+    * @param name 实体名称，如果不填会自动生成一个唯一名称
+    */
+    _insertEntityAt(name: string, index: number): Entity;
+
+    /**
+    * 插入实体
+    * @internal
+    * @param entityOptions 实体参数
+    */
+    _insertEntityAt(entityOptions: Partial<Entity>, index: number): Entity;
+
+    /**
+    * 插入实体
+    * @internal
+    * @param entity 已有的实体实例
+    */
+    _insertEntityAt(entity: Entity, index: number): Entity;
+
+    _insertEntityAt(options: string | Partial<Entity> | Entity, index: number) {
+        const entityOptions: any = {};
+        const relationOptions = { parentNode: this, parentKey: 'entities' };
+        let entity: Entity;
+        if (!options) {
+            entity = Entity.from({
+                ...entityOptions,
+                name: this.getEntityUniqueName(),
+            }, this, 'entities');
+        } else if (typeof options === 'string') {
+            entity = Entity.from({
+                ...entityOptions,
+                name: options,
+            }, this, 'entities');
+        } else if (options instanceof Entity) {
+            options.ensureDelete(); // 同一实例不支持多处存在
+            entity = options;
+            Object.assign(entity, relationOptions);
+        } else {
+            entity = Entity.from({
+                ...entityOptions,
+                ...options,
+            }, this, 'entities');
+        }
+        this.entities.splice(index, 0, entity);
+        return entity;
+    }
+
+    /**
+     * 插入实体
+     * @param name 实体名称，如果不填会自动生成一个唯一名称
+     */
+    insertEntityAt(name: string, index: number): Entity;
+
+    /**
+     * 插入实体
+     * @param entityOptions 实体参数
+     */
+    insertEntityAt(entityOptions: Partial<Entity>, index: number): Entity;
+
+    /**
+     * 插入实体
+     * @param entity 已有的实体实例
+     */
+    insertEntityAt(entity: Entity, index: number): Entity;
+
+    insertEntityAt(options: string | Partial<Entity> | Entity, index: number) {
+        const node = this._insertEntityAt(options as any, index);
+        node.create({
+            index,
+            parentNode: this,
+            parentKey: 'entities',
+        });
+        return node;
+    }
+
+    /**
+     * 添加实体
+     * @internal
+     * @param name 实体名称，如果不填会自动生成一个唯一名称
+     */
+    _addEntity(name?: string): Entity;
+
+    /**
+     * 添加实体
+     * @internal
+     * @param entityOptions 实体参数
+     */
+    _addEntity(entityOptions: Partial<Entity>): Entity;
+
+    /**
+     * 添加实体
+     * @internal
+     * @param entity 已有的实体实例
+     */
+    _addEntity(entity: Entity): Entity;
+
+    _addEntity(options?: string | Partial<Entity> | Entity) {
+        const index = 0;
+        return this._insertEntityAt(options as any, index);
+    }
+
+    /**
+     * 添加实体
+     * @internal
+     * @param name 实体名称，如果不填会自动生成一个唯一名称
+     */
+    addEntity(name?: string): Entity;
+
+    /**
+     * 添加实体
+     * @param entityOptions 实体参数
+     */
+    addEntity(entityOptions: Partial<Entity>): Entity;
+
+    /**
+     * 添加实体
+     * @param entity 已有的实体实例
+     */
+    addEntity(entity: Entity): Entity;
+
+    addEntity(options?: string | Partial<Entity> | Entity) {
+        const node = this._addEntity(options as any);
+        const index = this.entities.indexOf(node);
+        node.create({
+            index,
+            parentNode: this,
+            parentKey: 'entities',
+        });
+        return node;
+    }
+
+
+    
+
+    /**
+    * 删除实体
+    * @param name 实体名称
+    */
+    removeEntity(name: string): void;
+
+    /**
+     * 删除实体
+     * @param entity 已有的实体实例
+     */
+    removeEntity(entity: Entity): void;
+
+    removeEntity(options: string | Entity) {
+        let entity: Entity;
+        if (typeof options === 'string') {
+            entity = this.entities.find((item) => item.name === options);
+            if (!entity) {
+                throw new Error('找不到实体 ' + options);
+            }
+        } else {
+            entity = options;
+        }
+        return entity.delete();
+    }
+
+    __removeEntity(entity: Entity) {
+        const parentKey = entity.parentKey;
+        const params: Params = {
+            parentNode: this,
+            parentKey,
+            index: -1,
+            object: null,
+            oldObject: entity,
+        };
+        if (parentKey) {
+            params.parentKey = parentKey;
+            if (Array.isArray((this as any)[parentKey])) {
+                const index = (this as any)[parentKey].indexOf(entity);
+                ~index && (this as any)[parentKey].splice(index, 1);
+                params.index = index;
+            } else if ((this as any)[parentKey] === entity) {
+                params.index = 0;
+                (this as any)[parentKey] = undefined;
+            }
+        }
+        return params;
+    }
+
+    //================================================================================
+    // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+    // 自动生成的代码已结束。下面可以手动编写。
+    //================================================================================
+    findEntityByName(name: string) {
+        return this.entities.find((entity) => entity.name === name);
+    }
+    /**
+     * 设置DataSource的名称
+     */
+    setName(name: string) {
+        const App = this.rootNode;
+        App.emit('collect:start', {
+            actionMsg: '修改数据源名称',
+        });
+        // 修改配置对应配置上的 datasource
+        const dataSourceGroup = (App as App).configuration?.getGroup('dataSource');
+        if (dataSourceGroup) {
+            const dataSourceConfig = dataSourceGroup.getProperty(this.name);
+            if (dataSourceConfig) {
+                dataSourceConfig.setName(name);
+            }
+        }
+
+        // 修改自己内部的名称
+        const object = {
+            name,
+        };
+        this.update({
+            ...object,
+            field: 'name',
+        });
+        // 结束修改 ，批量操作
+        App.emit('collect:end');
+    }
+    getNamespace() {
+        if (this.parentNode && (this.parentNode as App).getNamespace) {
+            const parentNamespace = (this.parentNode as App).getNamespace();
+            const parentName = this.parentNode.name;
+            const arr = [parentNamespace];
+            if (this.parentNode.concept !== 'App' && parentName) {
+                arr.push(parentName);
+            }
+            return `${arr.join('.')}.dataSources`;
+        } else
+            throw new Error('无法获取命名空间，请设置 parentNode！');
+    }
+    /**
+     * 生成宿主语言的文件路径
+     * @param name 一般不用传，用于 rename
+     */
+    getEmbeddedFilePath(name = this.name) {
+        return `/embedded/${this.rootNode?.name || this.parentNode.name}/datasources/${name}.ts`;
+    }
+    @withSourceMap()
+    toEmbeddedTS(state?: TranslatorState): string {
+        let code = `export namespace ${this.name} {\n`;
+        code += 'const name = ';
+        code += '"' + this.name + '";\n';
+        code += '}\n';
+        return code;
+    }
+    toEmbeddedTSFile() {
+        let code = `namespace ${this.getNamespace()} {\n`;
+
+        const state = createCompilerState(code, { tabSize: 1 });
+        try {
+            code += this.toEmbeddedTS(state);
+        } catch (err) {
+            code += '';
+        }
+        code += '}\n';
+
+        return {
+            code,
+            filePath: this.getEmbeddedFilePath(),
+            sourceMap: state.sourceMap,
+        };
+    }
+
+    // 获取datasource对应的数据源配置
+    get dataSourceConfig() {
+        let devProperty = null;
+        let onlineProperty = null;
+        let devValue = null;
+        let onlineValue = null;
+
+        const dataSourceGroup = (this.rootNode as App).configuration?.getGroup('dataSource');
+        if (dataSourceGroup) {
+            const property = dataSourceGroup.getProperty(this.name);
+            if (property.values) {
+                devProperty = property.values.find((item) => item.env === 'dev');
+                devValue = { ...devProperty, ...JSON.parse(devProperty.value || '{}') };
+                onlineProperty = property.values.find((item) => item.env === 'online');
+                onlineValue = { ...onlineProperty, ...JSON.parse(onlineProperty.value || '{}') };
+            }
+        }
+        return {
+            devProperty,
+            onlineProperty,
+            devValue,
+            onlineValue,
+        };
+    }
+    //================================================================================
+    // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
+    // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
+    //================================================================================
+}
+
+classMap['DataSource'] = DataSource;
+export default DataSource;
+//================================================================================
+// ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+// 自动生成的代码已结束。下面可以手动编写。
+//================================================================================

package/src/concepts/Destination__.ts

@@ -52,6 +52,8 @@ export class Destination extends LogicItem {
     @property('Argument')
     arguments: Array<Argument> = [];
 
+    
+
     /**
      * @param source 需要合并的部分参数
      */
@@ -65,6 +67,7 @@ export class Destination extends LogicItem {
         return super.from(source, parentNode, parentKey) as Destination;
     }
 
+    
     /**
      * 设置viewNamespace
      */
@@ -89,6 +92,10 @@ export class Destination extends LogicItem {
         });
     }
 
+    
+
+    
+
     /**
     * 插入实际参数
     * @internal
@@ -121,6 +128,8 @@ export class Destination extends LogicItem {
         return argument;
     }
 
+    
+
     /**
      * 插入实际参数
      * @param argumentOptions 实际参数参数
@@ -143,6 +152,8 @@ export class Destination extends LogicItem {
         return node;
     }
 
+    
+
     /**
      * 添加实际参数
      * @internal
@@ -162,6 +173,8 @@ export class Destination extends LogicItem {
         return this._insertArgumentAt(options as any, index);
     }
 
+    
+
     /**
      * 添加实际参数
      * @param argumentOptions 实际参数参数
@@ -185,6 +198,9 @@ export class Destination extends LogicItem {
         return node;
     }
 
+
+    
+
     /**
      * 删除实际参数
      * @param argument 已有的实际参数实例
@@ -192,7 +208,7 @@ export class Destination extends LogicItem {
     removeArgument(argument: Argument): void;
 
     removeArgument(options: Argument) {
-        const argument: Argument = options;
+        let argument: Argument = options;
         return argument.delete();
     }
 
@@ -388,7 +404,7 @@ export class Destination extends LogicItem {
     }
     @withSourceMap()
     toEmbeddedTSInProcess(state: TranslatorState, prependCode: string): string {
-        // const tab = indent(state.tabSize);
+        // const tab = indent((state?.tabSize || 0));
         if (this.tsCalleeNamespace && this.tsName) {
             let code = `destination: function() {\n`;
             code += prependCode;
@@ -402,7 +418,7 @@ export class Destination extends LogicItem {
                 }
                 code += `});\n`;
             }
-            code += indent(state.tabSize + 1) + `}\n`;
+            code += indent((state?.tabSize || 0) + 1) + `}\n`;
             return code;
         } else {
             return '__IDENTIFIER__';
@@ -415,7 +431,7 @@ export class Destination extends LogicItem {
 
         if (this.parentNode instanceof BindAttribute) {
             code = `${utils.firstLowerCase(this.concept)}: () => {\n`;
-            code += indent(state.tabSize + 2);
+            code += indent((state?.tabSize || 0) + 2);
             if (this.tsCalleeNamespace && this.tsName) {
                 code += `${this.tsCalleeNamespace}.${this.tsName}.$destination({`;
                 if (Array.isArray(this.arguments)) {
@@ -425,7 +441,7 @@ export class Destination extends LogicItem {
                 }
                 code += `});\n`;
             }
-            code += indent(state.tabSize + 1) + `},\n`;
+            code += indent((state?.tabSize || 0) + 1) + `},\n`;
             return code;
         } else {
             if (this.tsCalleeNamespace && this.tsName) {
@@ -445,7 +461,7 @@ export class Destination extends LogicItem {
     @withSourceMap()
     toEmbeddedTSNormalFunction(state?: TranslatorState): string {
         let code = `let ${utils.firstLowerCase(this.concept)} = function() {\n`;
-        const tab = indent(state.tabSize + 2);
+        const tab = indent((state?.tabSize || 0) + 2);
         if (this.tsCalleeNamespace && this.tsName) {
             code += `${tab}${this.tsCalleeNamespace}.${this.tsName}.$destination(`;
             if (Array.isArray(this.arguments)) {
@@ -455,7 +471,7 @@ export class Destination extends LogicItem {
             }
             code += `${tab});\n`;
         }
-        code += `${indent(state.tabSize + 1)}}\n`;
+        code += `${indent((state?.tabSize || 0) + 1)}}\n`;
         return code;
     }
     //================================================================================
@@ -464,7 +480,7 @@ export class Destination extends LogicItem {
     //================================================================================
 }
 
-classMap.Destination = Destination;
+classMap['Destination'] = Destination;
 export default Destination;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/End__.ts

@@ -70,9 +70,9 @@ export class End extends LogicItem {
         let code = '';
         const ret = this.logic.returns[0];
         if (ret) {
-            code += indent(state.tabSize) + 'return ' + ret.name + ';';
+            code += indent((state?.tabSize || 0)) + 'return ' + ret.name + ';';
         } else {
-            code += indent(state.tabSize) + 'return;';
+            code += indent((state?.tabSize || 0)) + 'return;';
         }
         return code;
     }

package/src/concepts/EntityIndex__.ts

@@ -15,6 +15,7 @@ import classMap from '../common/classMap';
 import Entity from './Entity__';
 import Module from './Module__';
 import App from './App__';
+import DataSource from './DataSource__';
 
 /**
  * 实体索引
@@ -75,6 +76,12 @@ export class EntityIndex extends BaseNode {
     get app() {
         return this.getAncestor('App') as App;
     }
+    /**
+     * 祖先 DataSource
+     */
+    get dataSource() {
+        return this.getAncestor('DataSource') as DataSource;
+    }
 
     /**
      * @param source 需要合并的部分参数

package/src/concepts/EntityProperty__.ts

@@ -17,6 +17,7 @@ import TypeAnnotation from './TypeAnnotation__';
 import Entity from './Entity__';
 import Module from './Module__';
 import App from './App__';
+import DataSource from './DataSource__';
 
 /**
  * 实体字段
@@ -154,6 +155,12 @@ export class EntityProperty extends BaseNode {
     get app() {
         return this.getAncestor('App') as App;
     }
+    /**
+     * 祖先 DataSource
+     */
+    get dataSource() {
+        return this.getAncestor('DataSource') as DataSource;
+    }
 
     /**
      * @param source 需要合并的部分参数
@@ -417,8 +424,8 @@ export class EntityProperty extends BaseNode {
         let code = '';
         if (this.relationNamespace && this.relationEntity && this.relationProperty) {
             code += `@nasl.annotation.EntityRelation(new ${this.relationNamespace}.${this.relationEntity}().${this.relationProperty})\n`;
-            code += indent(state.tabSize + 2);
-            // code += indent(state.tabSize + 1);
+            code += indent((state?.tabSize || 0) + 2);
+            // code += indent((state?.tabSize || 0) + 1);
         }
 
         code += this.name;
@@ -528,6 +535,17 @@ export class EntityProperty extends BaseNode {
         });
     }
 
+    /**
+     * 设置关联实体实体名称
+     */
+    setRelationNamespace(relationNamespace: string) {
+        const object = {
+            relationNamespace,
+        };
+        this.update({
+            ...object,
+        });
+    }
     /**
      * 设置关联实体实体名称
      */

package/src/concepts/Entity__.ts

@@ -15,6 +15,7 @@ import classMap from '../common/classMap';
 import EntityProperty from './EntityProperty__';
 import EntityIndex from './EntityIndex__';
 import Module from './Module__';
+import DataSource from './DataSource__';
 import App from './App__';
 
 /**
@@ -82,6 +83,12 @@ export class Entity extends BaseNode {
     get app() {
         return this.getAncestor('App') as App;
     }
+    /**
+     * 祖先 DataSource
+     */
+    get dataSource() {
+        return this.getAncestor('DataSource') as DataSource;
+    }
 
     /**
      * @param source 需要合并的部分参数
@@ -644,40 +651,45 @@ export class Entity extends BaseNode {
     @withSourceMap()
     toEmbeddedTS(state?: TranslatorState): string {
         let code = `export class ${this.name} {\n`;
-        code += indent(state.tabSize + 1);
+        code += indent((state?.tabSize || 0) + 1);
         code += `__name: '${this.getNamespace()}.${this.name}';\n`;
         this.properties.forEach((property) => {
-            code += indent(state.tabSize + 1);
+            code += indent((state?.tabSize || 0) + 1);
             code += property.toEmbeddedTS(shiftState(state, code, { tabSize: 0 }));
             code += ';\n';
         });
         if (this.indexes && this.indexes.length) {
-            code += indent(state.tabSize + 1);
+            code += indent((state?.tabSize || 0) + 1);
             code += `__IndexList() {\n`;
             code += `            return [`;
             this.indexes.forEach((entityIndex) => {
-                code += indent(state.tabSize + 1);
+                code += indent((state?.tabSize || 0) + 1);
                 code += '\n                ';
                 code += entityIndex.toEmbeddedTS(shiftState(state, code, { tabSize: 0 }));
             });
-            code += indent(state.tabSize) + `\n            ];\n        }\n`;
+            code += indent((state?.tabSize || 0)) + `\n            ];\n        }\n`;
         }
 
         const ns = this.ns;
+
+        /**
+         * 8个函数的总行数
+         * 如果修改这里,可能会影响到dataSource的重命名,  在naslserver中,可能要一起修改
+         */
         code += `\nstatic logics = {\n`;
         ns.logics.forEach((logic) => {
-            code += indent(state.tabSize + 1);
+            code += indent((state?.tabSize || 0) + 1);
             code += logic.toEmbeddedTS(shiftState(state, code), 'inObject');
         });
         code += '}\n';
-        code += indent(state.tabSize) + '}\n';
+        code += indent((state?.tabSize || 0)) + '}\n';
 
         return code;
     }
 
     getNamespace() {
-        if (this.parentNode && (this.parentNode as Namespace).getNamespace)
-            return `${(this.parentNode as Namespace).getNamespace()}.entities`;
+        if (this.parentNode && (this.parentNode as DataSource).getNamespace)
+            return `${(this.parentNode as DataSource).getNamespace()}.${this.parentNode.name}.entities`;
         else
             throw new Error('无法获取命名空间，请设置 parentNode！');
     }
@@ -687,12 +699,12 @@ export class Entity extends BaseNode {
      * @param name 一般不用传，用于 rename
      */
     getEmbeddedFilePath(name = this.name) {
-        return `/embedded/${this.app.name}/entities/${name}.ts`;
+        return `/embedded/${this.app.name}/datasources/${this.parentNode.name}/entities/${name}.ts`;
     }
 
     get ns() {
         const ns: Namespace = entity2LogicNamespace(this);
-        ns.path = this.getNamespace() + '.' + ns.name;
+        ns.path = this.getNamespace();
         return ns;
     }
     // @excludedInJSON()