@lcap/nasl 1.0.0-alpha.7 → 1.0.0

package/src/automate/engine/utils.js

@@ -2,14 +2,13 @@
 //     BaseNode,
 //     Module
 // } from '../../concepts';
-const { BaseNode } = require('../../../out/concepts');
-const { getConceptMeta } = require('../../../out/decorators/index');
-
+import { BaseNode, uiStructures, collectionStructures } from '../../concepts';
+import { getConceptMeta } from '../../decorators/index';
 /*
     根据 namespace 字符串来获取 node
     link: http://nasl.lcap.group/docs/current/semantics/Basics/Namespaces/
 */
-exports.getNodeByNodeCallee = function getNodeByNodeCallee(AppInstance, namespace, callback) {
+export function getNodeByNodeCallee(AppInstance, namespace, callback) {
     const nodePath = namespace.split('.');
     const kind = nodePath.shift();
     let tempPoint;
@@ -21,7 +20,7 @@ exports.getNodeByNodeCallee = function getNodeByNodeCallee(AppInstance, namespac
         tempPoint = AppInstance.dependencies;
         state = 2;
     }
-    if (kind === 'interfaces') {
+    if (kind === 'apis') {
         tempPoint = AppInstance.interfaceDependencies;
         state = 2;
     }
@@ -63,9 +62,19 @@ exports.getNodeByNodeCallee = function getNodeByNodeCallee(AppInstance, namespac
         }
     }
     return tempPoint;
-};
+}
 
-const TRACK_TYPE = (exports.TRACK_TYPE = {
+/**
+ *  根据 namespace 字符串来获取 node，主要是nasl的ui 和 collection
+ */
+export function getNaslNodeByNodeCallee(namespace, typeName) {
+    if (namespace === 'nasl.ui') {
+        return uiStructures.find((item) => item.name === typeName);
+    } else if (namespace === 'nasl.collection') {
+        return collectionStructures.find((item) => item.name === typeName);
+    }
+}
+export const TRACK_TYPE = {
     PLAIN: 'plain',
     PROPERTY: 'property',
     ARRAY: 'array',
@@ -73,8 +82,11 @@ const TRACK_TYPE = (exports.TRACK_TYPE = {
     CALLEE_TEMPLATE: 'callee_template',
     MODULE: 'module',
     ROLE_TEMPLATE: 'role_template',
+    PROPERTY_TEMPLATE: 'property_template',
     // INTERNAL: 'internal'
-});
+};
+
+const ENTITY_AS_NAME = 'entityAsName';
 
 const BACKTACKING_PROPERTY_NS = {
     TYPE_NAMESPACE: 'typeNamespace',
@@ -113,15 +125,16 @@ function isBACKTRACKING_IGNORE_NAMESPACE(ns) {
     return ns.startsWith('nasl') || ns.startsWith('elements');
 }
 
-const IGNORE_PROPERTY = (exports.IGNORE_PROPERTY = [
+export const IGNORE_PROPERTY = [
     'uuid', // View 属性
     'origin', // Entity 属性
     'changedTime',
     'branchName',
     'appId',
     'isIndex',
-    'tableName',
-]);
+    // 'tableName',
+    // 'columnName',
+];
 
 const ROLE_PROPERTY = 'bindRoles';
 
@@ -136,7 +149,7 @@ const ROLE_PROPERTY = 'bindRoles';
 //     'Role',
 // ];
 
-exports.tracking = function tracking(node, startNode, callback) {
+export function tracking(node, startNode, callback) {
     const { propertyMap } = getConceptMeta(node.concept);
     // console.log(propertyMap)
     for (const p of propertyMap) {
@@ -238,7 +251,7 @@ exports.tracking = function tracking(node, startNode, callback) {
                                     relatedPropertyName,
                                     AppPropertyName: 'processes',
                                 });
-                                return;
+                                break;
                             }
 
                             const entity = rootNode.entities.find((e) => e.name === nodeName);
@@ -252,7 +265,7 @@ exports.tracking = function tracking(node, startNode, callback) {
                                     relatedPropertyName,
                                     AppPropertyName: 'entities',
                                 });
-                                return;
+                                break;
                             }
                         } else {
                             const nsPath = `${ns}.${calleeN}`;
@@ -275,9 +288,26 @@ exports.tracking = function tracking(node, startNode, callback) {
                 case BACKTACKING_PROPERTY.ENTITY_NAMESPACE:
                 case BACKTACKING_PROPERTY.RELATION_NAMESPACE:
                     break;
+                case ENTITY_AS_NAME:
+                    // 现在还不存在 asName 先直接当 Entity 名称来用。
+                    callback(TRACK_TYPE.PROPERTY_TEMPLATE, propertyName, property, {
+                        getRelatedVar: () => {
+                            const queryCom = node.getAncestor('CallQueryComponent');
+                            if (queryCom) {
+                                const entities = node.app.entities;
+                                // entityAsName === entity.name
+                                return entities.find((e) => e.name === property);
+                            }
+                            return null;
+                        },
+                        relatedProperty: 'name',
+                    });
+
+                    break;
+
                 default:
-                    if (property !== undefined && property !== null) {
-                        if (objectRef) {
+                    if (property !== undefined) {
+                        if (objectRef && property !== null) {
                             callback(TRACK_TYPE.PROPERTY, propertyName, property);
                         } else {
                             callback(TRACK_TYPE.PLAIN, propertyName, property);
@@ -287,7 +317,7 @@ exports.tracking = function tracking(node, startNode, callback) {
             }
         }
     }
-};
+}
 
 function getAppPropertyName(node) {
     const { propertyMap } = getConceptMeta('App');
@@ -301,7 +331,7 @@ function getAppPropertyName(node) {
     return undefined;
 }
 
-exports.backtracking = function backtracking({ node, startNode, calleeNode, propertyName, relatedPropertyName }, callback) {
+export function backtracking({ node, startNode, calleeNode, propertyName, relatedPropertyName }, callback) {
     const m = calleeNode.getAncestor('Module');
     if (m) {
         callback(TRACK_TYPE.PLAIN, propertyName, node[propertyName]);
@@ -332,4 +362,4 @@ exports.backtracking = function backtracking({ node, startNode, calleeNode, prop
         relatedPropertyName,
         AppPropertyName,
     });
-};
+}

package/src/common/BaseNode.ts

@@ -3,7 +3,7 @@ import { v4 as uuidv4 } from 'uuid';
 
 import { config } from '../config';
 import { SourceMap } from '../translator';
-import axios from '../service/storage';
+import storageService from '../service/storage/service';
 import { getSourceFromMap } from '../service/storage/map';
 import { loadThen } from '../service/storage/storagePoint';
 
@@ -12,21 +12,7 @@ import * as utils from '../utils';
 import classMap from '../common/classMap';
 import { action, circular, excludedInJSON, property, getConceptConstructor, getConceptPropertyMap } from '../decorators';
 import { Diagnostic } from '../manager/diagnostic';
-
-/**
- * 分割数组
- * @param array 原数组
- * @param subGroupLength 长度
- * @returns
- */
-function group(array: Array<any>, subGroupLength: number) {
-    let index = 0;
-    const newArray = [];
-    while (index < array.length) {
-        newArray.push(array.slice(index, (index += subGroupLength)));
-    }
-    return newArray;
-}
+import { ViewElement } from '..';
 
 /**
  * 顶点类
@@ -168,57 +154,54 @@ export class BaseNode extends EventEmitter {
     subConstructor(source?: Partial<BaseNode>) {
         const propertyMap = getConceptPropertyMap(this.constructor.name);
 
-        const properties = Array.from(new Set([...Object.keys(this), ...Object.keys(source)])).filter((key) => {
-            const property = propertyMap.get(key);
-            return !!property;
-        });
+        const properties = Array.from(new Set([...Object.keys(this), ...Object.keys(source)])).filter((key) => propertyMap.has(key));
         properties.forEach((key) => {
             const property = propertyMap.get(key);
-            const sourceItem = (source as any)[key] || (this as any)[key];
+            const sourceItem = (source as any)[key] !== undefined ? (source as any)[key] : (this as any)[key];
             const { objectRef: propertyObjectRef } = property;
             let objectRefCtor;
             try {
                 if (propertyObjectRef) {
                     objectRefCtor = getConceptConstructor(propertyObjectRef);
                     if (objectRefCtor && sourceItem instanceof objectRefCtor) {
-                        if (!sourceItem.parentNode || !sourceItem.parentKey) {
-                            sourceItem.parentNode = this;
-                            sourceItem.parentKey = key;
-                        }
+                        sourceItem.parentNode = this;
+                        sourceItem.parentKey = key;
                     }
                 }
-                (this as any)[key] = objectRefCtor && !(sourceItem instanceof objectRefCtor) ? this.activeInside(sourceItem, objectRefCtor, this, key) : sourceItem;
+
+                (this as any)[key] = objectRefCtor && sourceItem && !(sourceItem instanceof objectRefCtor) ? this.activeInside(sourceItem, objectRefCtor, this, key) : sourceItem;
             } catch (e) {
                 // Function has non-object prototype 'undefined' in instanceof check
                 if (!e.message.includes('Function has non-object prototype'))
                     throw e;
             }
         });
-        if (config.storage.protocol === 'http' && loadThen[source.concept as 'App' | 'View']) {
-            const { childNodes } = loadThen[source.concept as 'App' | 'View'] || {};
-            const queryList = childNodes.map(({ field, concept }) => {
-                const path = [this.nodePath, field].join('.');
-                return {
-                    path,
-                    excludes: loadThen[concept as 'App' | 'View'].childNodes.map(({ field }) => field),
-                };
-            });
-            axios
-                .post('/api/storage/batchQuery', queryList, {
-                    headers: {
-                        appId: this.rootNode?.id || this.id,
-                    },
-                })
-                .then((res) =>
-                    res?.data?.result.forEach((sourceItem: any, index: number) => {
-                        const { field, concept } = childNodes[index];
-                        if (!sourceItem || (Array.isArray((this as any)[field]) && !Array.isArray(sourceItem))) {
-                            return;
-                        }
-                        const objectRefCtor = getConceptConstructor(concept);
-                        (this as any)[field] = objectRefCtor ? this.activeInside(sourceItem, objectRefCtor, this, field) : sourceItem;
-                    }));
-        }
+        // if (config.storage.protocol === 'http' && loadThen[source.concept as 'App' | 'View']) {
+        //     const { childNodes } = loadThen[source.concept as 'App' | 'View'] || {};
+        //     const queryList = childNodes.map(({ field, concept }) => {
+        //         const path = [this.nodePath, field].join('.');
+        //         return {
+        //             path,
+        //             excludes: loadThen[concept as 'App' | 'View'].childNodes.map(({ field }) => field),
+        //         };
+        //     });
+        //     storageService
+        //         .batchQuery({
+        //             body: queryList,
+        //             headers: {
+        //                 appId: this.rootNode?.id || this.id,
+        //             },
+        //         })
+        //         .then((data: any) =>
+        //             data.forEach((sourceItem: any, index: number) => {
+        //                 const { field, concept } = childNodes[index];
+        //                 if (!sourceItem || (Array.isArray((this as any)[field]) && !Array.isArray(sourceItem))) {
+        //                     return;
+        //                 }
+        //                 const objectRefCtor = getConceptConstructor(concept);
+        //                 (this as any)[field] = objectRefCtor ? this.activeInside(sourceItem, objectRefCtor, this, field) : sourceItem;
+        //             }));
+        // }
     }
 
     getIndexOfParent() {
@@ -240,35 +223,28 @@ export class BaseNode extends EventEmitter {
      * @returns
      */
     genActionListener($event: EventPayload) {
+        if (!$event.eventList) {
+            $event.eventList = [];
+        }
+        if (!$event.eventStop) {
+            $event.eventList.push($event);
+            // 页面逻辑需要冒泡到页面
+            if (['App', 'Process', 'View', 'Interface', 'Entity', 'Structure', 'Enum', 'ConfigProperty'].includes(this.concept) || (this.concept === 'Logic' && this.parentNode?.concept !== 'View')) {
+                $event.eventStop = true;
+            }
+        }
+        $event.originEvent = $event.originEvent || $event;
         this.onChange($event);
         if (this.concept === 'App') {
             this.emit('storage', $event);
         } else {
-            if (!$event.eventList) {
-                $event.eventList = [];
-            }
-            if (!$event.eventStop) {
-                $event.eventList.push($event);
-                // 页面逻辑需要冒泡到页面
-                if ([
-                    'Process',
-                    'View',
-                    'Interface',
-                    'Entity',
-                    'Structure',
-                    'Enum',
-                    'ConfigProperty',
-                ].includes(this.concept) || this.concept === 'Logic' && this.parentNode?.concept !== 'View') {
-                    $event.eventStop = true;
-                }
-            }
             const event: EventPayload = {
                 action: 'update',
                 path: this.parentNode?.getNodePath(false),
                 object: {},
                 oldObject: {},
-                originEvent: $event.originEvent || $event,
-                eventList: $event.eventList || [],
+                originEvent: $event.originEvent,
+                eventList: $event.eventList,
                 eventStop: $event.eventStop,
             };
             this.parentNode?.emit('update', event);
@@ -306,12 +282,44 @@ export class BaseNode extends EventEmitter {
         this.emit('create', {
             action: 'create',
             path: this.getNodePath(true),
-            object: this,
+            object: this.toJSON(),
             ...parentParams,
             oldObject: null,
         });
     }
 
+    /**
+     * 如果有父节点，则删除节点，在添加时调用
+     */
+    ensureDelete() {
+        if (this.parentNode && this.parentKey) {
+            const path = this.getNodePath(false);
+            const parentParams = this._delete();
+            if (parentParams && parentParams.index !== -1) {
+                if (Array.isArray((this.parentNode as any)[this.parentKey])) {
+                    this.emit('delete', {
+                        action: 'delete',
+                        path,
+                        object: null,
+                        ...parentParams,
+                    });
+                } else {
+                    const parentPath = this.parentNode.getNodePath(false);
+                    this.parentNode.emit('update', {
+                        action: 'update',
+                        path: parentPath,
+                        object: {
+                            [this.parentKey]: null,
+                        },
+                        oldObject: {
+                            [this.parentKey]: this.toJSON(),
+                        },
+                    });
+                }
+            }
+        }
+    }
+
     /**
      * 删除节点
      */
@@ -319,12 +327,26 @@ export class BaseNode extends EventEmitter {
         const path = this.getNodePath(false);
         const parentParams = this._delete();
         if (parentParams && parentParams.index !== -1) {
-            this.emit('delete', {
-                action: 'delete',
-                path,
-                object: null,
-                ...parentParams,
-            });
+            if (Array.isArray((this.parentNode as any)[this.parentKey])) {
+                this.emit('delete', {
+                    action: 'delete',
+                    path,
+                    object: null,
+                    ...parentParams,
+                });
+            } else {
+                const parentPath = this.parentNode.getNodePath(false);
+                this.parentNode.emit('update', {
+                    action: 'update',
+                    path: parentPath,
+                    object: {
+                        [this.parentKey]: null,
+                    },
+                    oldObject: {
+                        [this.parentKey]: this.toJSON(),
+                    },
+                });
+            }
         }
     }
 
@@ -336,8 +358,34 @@ export class BaseNode extends EventEmitter {
         const oldObject: any = {};
         for (const key in object) {
             if (Object.prototype.hasOwnProperty.call(object, key)) {
-                oldObject[key] = (this as any)[key] ?? null;
-                (this as any)[key] = object[key] ?? null;
+                const oldProperty = (this as any)[key] ?? null;
+                oldObject[key] = oldProperty;
+                if (Array.isArray(oldProperty)) {
+                    oldObject[key] = oldProperty.map((item) => {
+                        let propertyItem = item ?? null;
+                        if (item instanceof BaseNode) {
+                            propertyItem = item.toJSON();
+                        }
+                        return propertyItem;
+                    });
+                } else if (oldProperty instanceof BaseNode) {
+                    oldObject[key] = oldProperty.toJSON();
+                }
+                const property = object[key] ?? null;
+                (this as any)[key] = property;
+                const objectRefCtor = getConceptConstructor(property?.concept);
+                if (Array.isArray(property)) {
+                    (this as any)[key] = property.map((item) => {
+                        let propertyItem = item ?? null;
+                        const propertyObjectRefCtor = getConceptConstructor(item?.concept);
+                        if (propertyObjectRefCtor?.from) {
+                            propertyItem = (propertyObjectRefCtor as any).from(item, this, key);
+                        }
+                        return propertyItem;
+                    });
+                } else if (objectRefCtor?.from) {
+                    (this as any)[key] = (objectRefCtor as any).from(property, this, key);
+                }
             }
         }
         this.emit('update', {
@@ -345,6 +393,7 @@ export class BaseNode extends EventEmitter {
             path,
             object,
             oldObject,
+            field,
         });
     }
 
@@ -454,6 +503,21 @@ export class BaseNode extends EventEmitter {
         completeList.push(actualSource || source);
     }
 
+    /**
+     * 获取当前的TsName
+     */
+    get tsName() {
+        if (this.name) {
+            let tsName = this.name;
+            // 匹配所有特殊字符都转为_
+            tsName = tsName.replace(/\W/g, '_');
+            if (/^\d/.test(tsName)) {
+                tsName = '$' + tsName;
+            }
+            return tsName;
+        }
+        return '';
+    }
     /**
      * 激活 JSON 内部实例
      * @param source
@@ -488,7 +552,7 @@ export class BaseNode extends EventEmitter {
      */
     genSources(instance: BaseNode, filerFn?: (source: Partial<BaseNode>) => {}) {
         if (!instance) {
-            return;
+            return instance;
         }
         if (Array.isArray(instance)) {
             return instance.map((item) => (item && item._toJSON ? item._toJSON(filerFn) : item));
@@ -617,6 +681,10 @@ export class BaseNode extends EventEmitter {
         /* override */
     }
 
+    async findReadOnlyLogicUsage(): Promise<any> {
+        /* override */
+    }
+
     /**
      * 重命名
      */
@@ -640,6 +708,7 @@ export class BaseNode extends EventEmitter {
     async addParamsPrepare(): Promise<any> {
         /* override */
     }
+
     /**
      * 销毁
      * 从 Map 中删除点和子节点
@@ -661,6 +730,63 @@ export class BaseNode extends EventEmitter {
         /* override */
     }
 
+    /**
+     * 移动节点
+     * @param sourceNode
+     * @param position
+     */
+    moveNode(options: {
+        sourceNode: BaseNode,
+        position: 'append' | 'insertBefore' | 'insertAfter',
+        action?: string
+    }) {
+        const { sourceNode, position } = options;
+        if (this.concept === 'App') {
+            return;
+        }
+        let actionMsg = `移动${(this.constructor as any).nodeTitle}`;
+        if (this.name) {
+            actionMsg += `“${this.name}”`;
+        }
+        (sourceNode as any).app.emit('collect:start', {
+            actionMsg,
+        });
+        // 如果是兄弟节点之前的移动
+        if (position !== 'append' && this.concept !== sourceNode.concept) {
+            return;
+        }
+        // 先删
+        this.delete();
+        if (position === 'append') {
+            let { action } = options;
+            if (!action)
+                action = `add${this.concept}`;
+            const parentNode = sourceNode;
+            (parentNode as any)[action](this as any);
+        } else {
+            const parentNode = sourceNode?.parentNode;
+            const parentKey = sourceNode?.parentKey || '';
+            const list = (parentNode as any)[parentKey];
+            if (Array.isArray(list)) {
+                let index = list.indexOf(sourceNode);
+                if (position === 'insertAfter') {
+                    index += 1;
+                }
+                Object.assign(this, {
+                    parentNode,
+                    parentKey,
+                });
+                list.splice(index, 0, this);
+                this.create({
+                    index,
+                    parentNode,
+                    parentKey,
+                });
+            }
+        }
+        (sourceNode as any).app.emit('collect:end');
+    }
+
     /**
      * 获取添加时的默认选项
      * @returns
@@ -668,6 +794,28 @@ export class BaseNode extends EventEmitter {
     public static getDefaultOptions(): any {
         return {};
     }
+
+    public traverseChildren(cb: (ele: BaseNode) => void): void {
+        function traverse(node: BaseNode) {
+            cb(node);
+
+            const propertyMap = getConceptPropertyMap(node.concept);
+            if (!propertyMap)
+                return;
+
+            for (const property of propertyMap.keys()) {
+                const child = node[property as keyof BaseNode];
+                if (!child)
+                    continue;
+                else if (Array.isArray(child))
+                    child.forEach((node: BaseNode) => traverse(node));
+                else
+                    traverse(child);
+            }
+        }
+
+        traverse(this);
+    }
 }
 
 export default BaseNode;