@lcap/nasl 1.0.2 → 2.12.0-beta

package/src/concepts/CallQueryComponent__.ts

@@ -84,13 +84,11 @@ export class CallQueryComponent extends LogicItem {
     limit: QueryLimitExpression = undefined;
 
     /**
-     * 查询组件的局部版本
+     * 局部版本
      */
     @property()
     ideVersion: string = undefined;
 
-    
-
     /**
      * @param source 需要合并的部分参数
      */
@@ -104,7 +102,6 @@ export class CallQueryComponent extends LogicItem {
         return super.from(source, parentNode, parentKey) as CallQueryComponent;
     }
 
-    
     /**
      * 设置select
      */
@@ -204,7 +201,7 @@ export class CallQueryComponent extends LogicItem {
         });
     }
 
-        getQueryGroupByExpressionExistingNames(excludedList: Array<QueryGroupByExpression> = []) {
+    getQueryGroupByExpressionExistingNames(excludedList: Array<QueryGroupByExpression> = []) {
         const excludedSet = new Set(excludedList);
         return (this.groupBy || []).filter((item) => !excludedSet.has(item)).map((item) => item.name);
     }
@@ -345,8 +342,6 @@ export class CallQueryComponent extends LogicItem {
         return node;
     }
 
-    
-
     /**
     * 插入查询排序子句
     * @internal
@@ -379,8 +374,6 @@ export class CallQueryComponent extends LogicItem {
         return queryOrderByExpression;
     }
 
-    
-
     /**
      * 插入查询排序子句
      * @param queryOrderByExpressionOptions 查询排序子句参数
@@ -403,8 +396,6 @@ export class CallQueryComponent extends LogicItem {
         return node;
     }
 
-    
-
     /**
      * 添加查询排序子句
      * @internal
@@ -424,8 +415,6 @@ export class CallQueryComponent extends LogicItem {
         return this._insertQueryOrderByExpressionInOrderByAt(options as any, index);
     }
 
-    
-
     /**
      * 添加查询排序子句
      * @param queryOrderByExpressionOptions 查询排序子句参数
@@ -449,9 +438,6 @@ export class CallQueryComponent extends LogicItem {
         return node;
     }
 
-
-    
-
     /**
     * 删除类型标注
     * @param name 类型标注名称
@@ -500,8 +486,6 @@ export class CallQueryComponent extends LogicItem {
         return params;
     }
 
-
-
     /**
     * 删除查询选择子句
     * @param name 查询选择子句名称
@@ -527,8 +511,6 @@ export class CallQueryComponent extends LogicItem {
         return select.delete();
     }
 
-    
-
     /**
     * 删除查询 From 子句
     * @param name 查询 From 子句名称
@@ -554,8 +536,6 @@ export class CallQueryComponent extends LogicItem {
         return from.delete();
     }
 
-    
-
     /**
     * 删除逻辑项
     * @param name 逻辑项名称
@@ -581,8 +561,6 @@ export class CallQueryComponent extends LogicItem {
         return where.delete();
     }
 
-    
-
     /**
     * 删除查询分组
     * @param name 查询分组名称
@@ -608,8 +586,6 @@ export class CallQueryComponent extends LogicItem {
         return queryGroupByExpression.delete();
     }
 
-    
-
     /**
     * 删除逻辑项
     * @param name 逻辑项名称
@@ -635,8 +611,6 @@ export class CallQueryComponent extends LogicItem {
         return having.delete();
     }
 
-    
-
     /**
      * 删除查询排序子句
      * @param queryOrderByExpression 已有的查询排序子句实例
@@ -644,12 +618,10 @@ export class CallQueryComponent extends LogicItem {
     removeQueryOrderByExpressionInOrderBy(queryOrderByExpression: QueryOrderByExpression): void;
 
     removeQueryOrderByExpressionInOrderBy(options: QueryOrderByExpression) {
-        let queryOrderByExpression: QueryOrderByExpression = options;
+        const queryOrderByExpression: QueryOrderByExpression = options;
         return queryOrderByExpression.delete();
     }
 
-    
-
     /**
     * 删除查询分页子句
     * @param name 查询分页子句名称
@@ -751,16 +723,16 @@ export class CallQueryComponent extends LogicItem {
         if (list.length === 0)
             return code;
 
-        code += indent(state.tabSize + 1) + `.${key}(() => `;
+        code += indent((state?.tabSize || 0) + 1) + `.${key}(() => `;
         list.forEach((item, index) => {
             if (index > 0)
-                code += `,\n${indent(state.tabSize + 2)}() => `;
+                code += `,\n${indent((state?.tabSize || 0) + 2)}() => `;
             code += item?.toEmbeddedTS?.(shiftState(state, code, { tabSize: 0 }));
         });
         if (list.length > 1) {
             if (multipleErrorType)
-                code += `,\n${indent(state.tabSize + 2)}() => ${multipleErrorType}`;
-            code += `\n${indent(state.tabSize + 1)}`;
+                code += `,\n${indent((state?.tabSize || 0) + 2)}() => ${multipleErrorType}`;
+            code += `\n${indent((state?.tabSize || 0) + 1)}`;
         }
         code += ')\n';
 
@@ -819,7 +791,7 @@ export class CallQueryComponent extends LogicItem {
         }
         // 聚合属性别名
         this.select.selectElements.forEach((item, index, array) => {
-            code += indent(state.tabSize + 1);
+            code += indent((state?.tabSize || 0) + 1);
             const { asName, aggregateName, aggregateParam } = item;
             if (asName)
                 code += `const __${asName} = `;
@@ -835,9 +807,9 @@ export class CallQueryComponent extends LogicItem {
         });
 
         if (this.from)
-            code += this.from.toEmbeddedTS(shiftState(state, code, { tabSize: state.tabSize + 1 }));
+            code += this.from.toEmbeddedTS(shiftState(state, code, { tabSize: (state?.tabSize || 0) + 1 }));
         else
-            code += `${indent(state.tabSize + 1)}nasl.langUtil.FROM()\n`;
+            code += `${indent((state?.tabSize || 0) + 1)}nasl.langUtil.FROM()\n`;
 
         // where 子句
         code = this.toEmbeddedTSOfSubPart('WHERE', this.where ? [this.where] : [], {
@@ -878,8 +850,8 @@ export class CallQueryComponent extends LogicItem {
         });
 
         code += '\n';
-        code += indent(state.tabSize + 1) + 'return;\n';
-        code += indent(state.tabSize) + '})();\n';
+        code += indent((state?.tabSize || 0) + 1) + 'return;\n';
+        code += indent((state?.tabSize || 0)) + '})();\n';
         return code;
     }
 
@@ -888,6 +860,7 @@ export class CallQueryComponent extends LogicItem {
             return [];
 
         const entityName = node.entityName;
+        const entityNamespace = node.entityNamespace;
         entities.push({
             concept: 'StructureProperty',
             name: utils.firstLowerCase(entityName),
@@ -895,7 +868,7 @@ export class CallQueryComponent extends LogicItem {
                 concept: 'TypeAnnotation',
                 typeKind: 'reference',
                 typeName: entityName,
-                typeNamespace: `${this.app.getNamespace()}.entities`,
+                typeNamespace: entityNamespace,
             },
         });
 
@@ -906,7 +879,7 @@ export class CallQueryComponent extends LogicItem {
         return entities;
     }
 
-    genPropertyListOfAggregate(node: CallQueryComponent) {
+    genPropertyListOfAggregate(node: CallQueryComponent, entities: Entity[]) {
         return node.select.selectElements
             .filter((item) => item.aggregateName && item.asName)
             .map((item) => {
@@ -917,9 +890,9 @@ export class CallQueryComponent extends LogicItem {
                     typeName = 'Double';
                 } else {
                     const { entityAsName, propertyName } = item.aggregateParam;
-                    const entity = this.app.entities.find((item: Entity) => item.name === entityAsName);
+                    const entity = entities.find((item: Entity) => item.name === entityAsName);
                     const property = entity?.properties.find((item) => item.name === propertyName);
-                    typeName = property?.typeAnnotation.typeName;
+                    typeName = property?.typeAnnotation?.typeName;
                 }
                 return {
                     concept: 'StructureProperty',
@@ -934,22 +907,23 @@ export class CallQueryComponent extends LogicItem {
             });
     }
 
-    genPropertyListOfGroupBy(node: CallQueryComponent) {
+    genPropertyListOfGroupBy(node: CallQueryComponent, entities: Entity[]) {
         return node.groupBy
             .filter((item) => item.groupElement && (item.groupElement as QueryFieldExpression).propertyName && (item.groupElement as QueryFieldExpression).asName)
             .map((item) => {
                 const { entityAsName, propertyName, asName } = item.groupElement as QueryFieldExpression;
-                const entity = this.app.entities.find((item: Entity) => item.name === entityAsName);
+                const entity = entities.find((item: Entity) => item.name === entityAsName);
                 const property = entity?.properties.find((item) => item.name === propertyName);
-                const typeName = property?.typeAnnotation.typeName;
+                const propertyTypeAnnotation = property.typeAnnotation;
+
                 return {
                     concept: 'StructureProperty',
                     name: asName,
                     typeAnnotation: {
-                        concept: 'TypeAnnotation',
-                        typeKind: 'primitive',
-                        typeName,
-                        typeNamespace: 'nasl.core',
+                        concept: propertyTypeAnnotation.concept,
+                        typeKind: propertyTypeAnnotation.typeKind,
+                        typeName: propertyTypeAnnotation.typeName,
+                        typeNamespace: propertyTypeAnnotation.typeNamespace,
                     },
                 };
             });
@@ -966,13 +940,14 @@ export class CallQueryComponent extends LogicItem {
 
     saveStructure() {
         // 没有数据源时不初始化数据结构
-        if (!this.entityNames.length) {
+        if (!this.entityInfos.length) {
             return;
         }
-
+        const entityNamespace = this.from?.entityNamespace;
+        const entities = this.app.findNodeByCompleteName(entityNamespace);
         if (!this.isAutoInfer()) {
             // 存量应用走之前的逻辑
-            return this.saveStructureNoInfer();
+            return this.saveStructureNoInfer(entities);
         }
 
         const groupBy = this.groupBy;
@@ -999,35 +974,35 @@ export class CallQueryComponent extends LogicItem {
         const groupByLength = groupBy.filter((item) => (item.groupElement as QueryFieldExpression)?.propertyName && (item.groupElement as QueryFieldExpression).asName).length;
         const aggregateLength = selectElements.filter((item) => item.aggregateName && item.asName).length;
 
+
+
         // 1. 有分组
         if (groupByLength) {
             if (!aggregateLength && groupByLength === 1) {
                 // 1.1 单分组无聚合：`PageOf<列的类型>`
-                const { entityAsName, propertyName } = groupBy[0].groupElement as any;
-                const entity = this.app.entities.find((item: Entity) => item.name === entityAsName);
-                const property = entity?.properties.find((item) => item.name === propertyName);
-                const typeName = property?.typeAnnotation.typeName;
+                const { entityAsName, propertyName } = groupBy[0].groupElement as QueryFieldExpression;
+                const entity = entities.find((item: Entity) => item.name === entityAsName);
+                const property = entity?.properties.find((item: any) => item.name === propertyName);
+                const propertyTypeAnnotation = property.typeAnnotation;
 
                 typeAnnotation = TypeAnnotation.from({
-                    text: `PageOf<${typeName}>`,
+                    text: `PageOf<${propertyTypeAnnotation.typeName}>`,
                     concept: 'TypeAnnotation',
                     typeKind: 'generic',
                     typeName: 'PageOf',
                     typeNamespace: `nasl.collection`,
-                    typeArguments: [
-                        TypeAnnotation.from({
-                            concept: 'TypeAnnotation',
-                            typeKind: 'primitive',
-                            typeName,
-                            typeNamespace: 'nasl.core',
-                        }, this.typeAnnotation, 'typeAnnotation'),
-                    ],
+                    typeArguments: [TypeAnnotation.from({
+                        concept: propertyTypeAnnotation.concept,
+                        typeKind: propertyTypeAnnotation.typeKind,
+                        typeName: propertyTypeAnnotation.typeName,
+                        typeNamespace: propertyTypeAnnotation.typeNamespace,
+                    }, this.typeAnnotation, 'typeAnnotation')],
                 }, this, 'typeAnnotation');
             } else {
                 // 1.2 多分组：`PageOf<Structure>`
                 const structure = this.genStructure(structureName, [
-                    ...this.genPropertyListOfGroupBy(this),
-                    ...this.genPropertyListOfAggregate(this),
+                    ...this.genPropertyListOfGroupBy(this, entities),
+                    ...this.genPropertyListOfAggregate(this, entities),
                 ]);
                 this.app.addStructure(structure);
 
@@ -1061,9 +1036,9 @@ export class CallQueryComponent extends LogicItem {
                     } else {
                         // 2.1.3. 其他返回列类型，包含：`Integer`、`Long`、`Double`
                         const { entityAsName, propertyName } = selectElements[0].aggregateParam;
-                        const entity = this.app.entities.find((item: Entity) => item.name === entityAsName);
-                        const property = entity?.properties.find((item) => item.name === propertyName);
-                        typeName = property?.typeAnnotation.typeName;
+                        const entity = entities.find((item: Entity) => item.name === entityAsName);
+                        const property = entity?.properties.find((item: any) => item.name === propertyName);
+                        typeName = property?.typeAnnotation?.typeName;
                     }
                     typeAnnotation = TypeAnnotation.from({
                         concept: 'TypeAnnotation',
@@ -1074,7 +1049,7 @@ export class CallQueryComponent extends LogicItem {
                 } else {
                     // 2.2. 多聚合：`Structure`（自动创建 `Structure`）
                     const structure = this.genStructure(structureName, [
-                        ...this.genPropertyListOfAggregate(this),
+                        ...this.genPropertyListOfAggregate(this, entities),
                     ]);
                     this.app.addStructure(structure);
 
@@ -1089,7 +1064,7 @@ export class CallQueryComponent extends LogicItem {
                 // 2.3. 无聚合(默认情况)：`PageOf<Structure>`（自动创建 `Structure`）
                 const structure = this.genStructure(structureName, [
                     ...this.genPropertyListOfFrom(this.from),
-                    ...this.genPropertyListOfAggregate(this),
+                    ...this.genPropertyListOfAggregate(this, entities),
                 ]);
                 this.app.addStructure(structure);
 
@@ -1114,10 +1089,10 @@ export class CallQueryComponent extends LogicItem {
         this.update({ typeAnnotation });
     }
 
-    saveStructureNoInfer() {
+    saveStructureNoInfer(entities: any[] = []) {
         const source = this.genStructure(null, [
             ...this.genPropertyListOfFrom(this.from),
-            ...this.genPropertyListOfAggregate(this),
+            ...this.genPropertyListOfAggregate(this, entities),
         ]);
         if (this.typeAnnotation) {
             const structure = this.app.structures.find((item) => item.name === this.typeAnnotation.typeName);
@@ -1162,53 +1137,61 @@ export class CallQueryComponent extends LogicItem {
         } catch {}
     }
 
-    getEntityNames(node: QueryFromExpression | QueryJoinExpression, set = new Set()) {
+    getEntityInfos(node: QueryFromExpression | QueryJoinExpression, set = new Set()) {
         if (node?.entityName)
-            set.add(node.entityName);
+            set.add({
+                namespace: node.entityNamespace,
+                name: node.entityName,
+            });
 
         node?.joinParts?.forEach((child) => {
-            this.getEntityNames(child, set);
+            this.getEntityInfos(child, set);
         });
         return Array.from(set);
     }
 
-    get entityNames() {
-        return this.getEntityNames(this.from);
+    get entityInfos() {
+        return this.getEntityInfos(this.from);
     }
 
     getCompletionPropertyOfEntities() {
-        const entities = this.app.entities;
-        const children = this.entityNames
-            .map((name) => entities.find((item) => item.name === name))
-            .filter((item) => item)
-            .map((entity) => ({
-                concept: 'Entity',
-                name: entity.name,
-                children: entity.properties.map((property) => {
-                    const completionProperty = new CompletionProperty({
-                        concept: 'CompletionProperty',
-                        name: property.name,
-                        value: `${entity.name}.${property.name}`,
-                        expression: LogicItem.from({
-                            concept: 'QueryFieldExpression',
-                            name: property.name,
-                            value: `${entity.name}.${property.name}`,
-                            entityAsName: entity.name,
-                            propertyName: property.name,
+        const firstEntityInfo: any = this.entityInfos[0];
+        if (firstEntityInfo) {
+            const namespace = firstEntityInfo.namespace;
+            const entities: Entity[] = this.app.findNodeByCompleteName(namespace);
+            if (Array.isArray(entities)) {
+                const children = this.entityInfos
+                    .map(({ name }) => entities.find((item) => item.name === name))
+                    .filter((item) => item)
+                    .map((entity) => ({
+                        concept: 'Entity',
+                        name: entity.name,
+                        children: entity.properties.map((property) => {
+                            const completionProperty = new CompletionProperty({
+                                concept: 'CompletionProperty',
+                                name: property.name,
+                                value: `${entity.name}.${property.name}`,
+                                expression: LogicItem.from({
+                                    concept: 'QueryFieldExpression',
+                                    name: property.name,
+                                    value: `${entity.name}.${property.name}`,
+                                    entityAsName: entity.name,
+                                    propertyName: property.name,
+                                }),
+                            });
+                            completionProperty.icon = 'property';
+                            return completionProperty;
                         }),
-                    });
-                    completionProperty.icon = 'property';
-                    return completionProperty;
-                }),
-            }));
-        if (children.length === 0)
-            return;
-
-        return {
-            name: '实体',
-            icon: 'category',
-            children,
-        };
+                    }));
+                if (children.length) {
+                    return {
+                        name: '实体',
+                        icon: 'category',
+                        children,
+                    };
+                }
+            }
+        }
     }
 
     getCompletionPropertyOfAggregateAsNames() {
@@ -1263,7 +1246,7 @@ export class CallQueryComponent extends LogicItem {
     //================================================================================
 }
 
-classMap['CallQueryComponent'] = CallQueryComponent;
+classMap.CallQueryComponent = CallQueryComponent;
 export default CallQueryComponent;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/Comment__.ts

@@ -79,7 +79,7 @@ export class Comment extends LogicItem {
     @withSourceMap()
     toEmbeddedTS(state?: TranslatorState): string {
         let code = '';
-        code += `\n${indent(state.tabSize)}
+        code += `\n${indent((state?.tabSize || 0))}
         /**
          * 注释不用翻译
          */
@@ -92,7 +92,7 @@ export class Comment extends LogicItem {
     //================================================================================
 }
 
-classMap['Comment'] = Comment;
+classMap.Comment = Comment;
 export default Comment;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/CompletionProperty__.ts

@@ -246,7 +246,7 @@ export class CompletionProperty extends BaseNode {
     //================================================================================
 }
 
-classMap['CompletionProperty'] = CompletionProperty;
+classMap.CompletionProperty = CompletionProperty;
 export default CompletionProperty;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/ConfigGroup__.ts

@@ -275,14 +275,16 @@ export class ConfigGroup extends BaseNode {
     // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
     // 自动生成的代码已结束。下面可以手动编写。
     //================================================================================
-
+    getProperty(name = '') {
+        return this.properties.find((item) => item.name === name);
+    }
     //================================================================================
     // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
     // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
     //================================================================================
 }
 
-classMap['ConfigGroup'] = ConfigGroup;
+classMap.ConfigGroup = ConfigGroup;
 export default ConfigGroup;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/ConfigPropertyValue__.ts

@@ -104,7 +104,7 @@ export class ConfigPropertyValue extends BaseNode {
     //================================================================================
 }
 
-classMap['ConfigPropertyValue'] = ConfigPropertyValue;
+classMap.ConfigPropertyValue = ConfigPropertyValue;
 export default ConfigPropertyValue;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/ConfigProperty__.ts

@@ -292,13 +292,17 @@ export class ConfigProperty extends BaseNode {
             ...object,
         });
     }
+
+    getPropertyEnv(env = '') {
+        return this.values.find((item) => item.env === env);
+    }
     //================================================================================
     // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
     // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
     //================================================================================
 }
 
-classMap['ConfigProperty'] = ConfigProperty;
+classMap.ConfigProperty = ConfigProperty;
 export default ConfigProperty;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/Configuration__.ts

@@ -271,6 +271,9 @@ export class Configuration extends BaseNode {
             ],
         };
     }
+    getGroup(name = '') {
+        return this.groups.find((item) => item.name === name);
+    }
 
     //================================================================================
     // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
@@ -278,7 +281,7 @@ export class Configuration extends BaseNode {
     //================================================================================
 }
 
-classMap['Configuration'] = Configuration;
+classMap.Configuration = Configuration;
 export default Configuration;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/Constant__.ts

@@ -253,7 +253,7 @@ export class Constant extends BaseNode {
     //================================================================================
 }
 
-classMap['Constant'] = Constant;
+classMap.Constant = Constant;
 export default Constant;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑

package/src/concepts/DataElement__.ts

@@ -127,7 +127,7 @@ export class DataElement extends BaseNode {
     //================================================================================
 }
 
-classMap['DataElement'] = DataElement;
+classMap.DataElement = DataElement;
 export default DataElement;
 //================================================================================
 // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑