@lcap/nasl 2.16.1 → 2.17.1

package/src/common/BaseNode.ts

@@ -12,7 +12,8 @@ import * as utils from '../utils';
 import classMap from '../common/classMap';
 import { action, circular, excludedInJSON, property, getConceptConstructor, getConceptPropertyMap } from '../decorators';
 import { Diagnostic } from '../manager/diagnostic';
-import { ViewElement, App } from '..';
+import { ViewElement, App, TypeAnnotation } from '..';
+import * as jsoner from '../service/storage/jsoner';
 
 /**
  * 顶点类
@@ -143,6 +144,12 @@ export class BaseNode extends EventEmitter {
     @property()
     name: string = '';
 
+    /**
+     * 是否正在编辑
+     * 前端 UI 状态
+     */
+    @excludedInJSON()
+    __TypeAnnotation: TypeAnnotation = undefined;
     /**
      * @param source 需要合并的部分参数
      */
@@ -165,27 +172,21 @@ export class BaseNode extends EventEmitter {
         properties.forEach((key) => {
             const property = propertyMap.get(key);
             const sourceItem = (source as any)[key] !== undefined ? (source as any)[key] : (this as any)[key];
-            const { objectRef: propertyObjectRef } = property;
             let objectRefCtor: any;
             try {
-                if (propertyObjectRef) {
-                    objectRefCtor = getConceptConstructor(propertyObjectRef);
-                    if (objectRefCtor) {
-                        // if (Array.isArray(sourceItem)) {
-                        //     sourceItem.forEach((item) => {
-                        //         if (item instanceof objectRefCtor) {
-                        //             item.parentNode = this;
-                        //             item.parentKey = key;
-                        //         }
-                        //     });
-                        // } else
+                if (property.objectRef) {
+                    const arr = property.objectRef.split(' | ');
+                    for (let i = 0; i < arr.length; i++) {
+                        objectRefCtor = getConceptConstructor(arr[i]);
                         if (sourceItem instanceof objectRefCtor) {
                             sourceItem.parentNode = this;
                             sourceItem.parentKey = key;
+                            break;
                         }
                     }
                 }
-                (this as any)[key] = objectRefCtor && sourceItem && !(sourceItem instanceof objectRefCtor) ? this.activeInside(sourceItem, objectRefCtor, this, key) : sourceItem;
+                // 数组的情况也在这里做掉了
+                (this as any)[key] = objectRefCtor && sourceItem && !(sourceItem instanceof objectRefCtor) ? this.instantiateSub(sourceItem, key) : sourceItem;
             } catch (e) {
                 // Function has non-object prototype 'undefined' in instanceof check
                 if (!e.message.includes('Function has non-object prototype'))
@@ -215,11 +216,43 @@ export class BaseNode extends EventEmitter {
         //                     return;
         //                 }
         //                 const objectRefCtor = getConceptConstructor(concept);
-        //                 (this as any)[field] = objectRefCtor ? this.activeInside(sourceItem, objectRefCtor, this, field) : sourceItem;
+        //                 (this as any)[field] = objectRefCtor ? this.instantiateSub(sourceItem, field) : sourceItem;
         //             }));
         // }
     }
 
+    fixInstantiate() {
+        const propertyMap = getConceptPropertyMap(this.constructor.name);
+
+        const properties = Object.keys(this).filter((key) => propertyMap.has(key));
+        properties.forEach((key) => {
+            const property = propertyMap.get(key);
+            const sourceItem = (this as any)[key];
+            const objectRefCtor = property.objectRef && getConceptConstructor(property.objectRef);
+            if (objectRefCtor) {
+                if (Array.isArray(sourceItem)) {
+                    sourceItem.forEach((item, index) => {
+                        if (item instanceof objectRefCtor) {
+                            item.parentNode = this;
+                            item.parentKey = key;
+                            item.fixInstantiate();
+                        } else {
+                            sourceItem[index] = this.instantiateSub(item instanceof BaseNode ? item.toJSON() : item, key);
+                        }
+                    });
+                } else {
+                    if (sourceItem instanceof objectRefCtor) {
+                        sourceItem.parentNode = this;
+                        sourceItem.parentKey = key;
+                        sourceItem.fixInstantiate();
+                    } else {
+                        (this as any)[key] = this.instantiateSub(sourceItem instanceof BaseNode ? sourceItem.toJSON() : sourceItem, key);
+                    }
+                }
+            }
+        });
+    }
+
     getIndexOfParent() {
         let index = -1;
         const parentNode = this.parentNode;
@@ -253,7 +286,7 @@ export class BaseNode extends EventEmitter {
         this.onChange($event);
         if (this.concept === 'App') {
             this.emit('storage', $event);
-        } else {
+        } else if (this.parentNode && this.parentNode.emit) {
             const event: EventPayload = {
                 action: 'update',
                 path: this.parentNode?.getNodePath(false),
@@ -308,6 +341,12 @@ export class BaseNode extends EventEmitter {
      * 如果有父节点，则删除节点，在添加时调用
      */
     ensureDelete() {
+        // 批量赋值因为连线的问题需要特殊处理一下.
+        if (this.parentNode?.concept === 'SelectMembers') {
+            this.parentNode.ensureDelete();
+            return;
+        }
+
         if (this.parentNode && this.parentKey) {
             const path = this.getNodePath(false);
             const parentParams = this._delete();
@@ -340,6 +379,16 @@ export class BaseNode extends EventEmitter {
      * 删除节点
      */
     delete() {
+        // 批量赋值因为连线的问题需要特殊处理一下.
+        if (this.parentNode?.concept === 'SelectMembers') {
+            const node: any = this.parentNode;
+            node.delete();
+            return;
+        } else if (this.concept === 'SelectMembers') {
+            const node: any = this.parentNode;
+            node.changeLinesLocation(this);
+        }
+
         const useIndex = ['BindEvent', 'Identifier'].includes(this.concept);
         const path = this.getNodePath(useIndex);
         const parentParams = this._delete();
@@ -545,32 +594,6 @@ export class BaseNode extends EventEmitter {
         }
         return '';
     }
-    /**
-     * 激活 JSON 内部实例
-     * @param source
-     */
-    activeInside(source: any, Ctor: any, parentNode: any, parentKey: string) {
-        if (!source) {
-            return;
-        }
-        const completeList: any[] = [];
-        let isArray = false;
-        // 暂时通过是否有name判断数据是否全
-        if (Array.isArray(source)) {
-            isArray = true;
-            source.forEach((sourceItem) => {
-                this.handleSource(sourceItem, completeList);
-            });
-        } else {
-            this.handleSource(source, completeList);
-        }
-        const instanceList: any[] = completeList.map((item) => this.instantiate(item, this, parentKey));
-        if (isArray) {
-            return instanceList;
-        } else if (Array.isArray(instanceList) && instanceList.length) {
-            return instanceList[0];
-        }
-    }
 
     /**
      * 生成json
@@ -613,6 +636,33 @@ export class BaseNode extends EventEmitter {
         }
     }
 
+    /**
+     * 激活 JSON 内部实例
+     * @param source
+     */
+    instantiateSub(source: any, parentKey: string) {
+        if (!source) {
+            return;
+        }
+        const completeList: any[] = [];
+        let isArray = false;
+        // 暂时通过是否有name判断数据是否全
+        if (Array.isArray(source)) {
+            isArray = true;
+            source.forEach((sourceItem) => {
+                this.handleSource(sourceItem, completeList);
+            });
+        } else {
+            this.handleSource(source, completeList);
+        }
+        const instanceList: any[] = completeList.map((item) => this.instantiate(item, this, parentKey));
+        if (isArray) {
+            return instanceList;
+        } else if (Array.isArray(instanceList) && instanceList.length) {
+            return instanceList[0];
+        }
+    }
+
     plainAssign(source: any) {
         this._mutable = true;
 
@@ -736,6 +786,14 @@ export class BaseNode extends EventEmitter {
         /* override */
     }
 
+    /**
+     * 获取当前节点类型
+     */
+    @action('获取当前节点类型')
+    async getCurrentTypeAnnotation(): Promise<any> {
+        /* override */
+    }
+
     /**
      * 销毁
      * 从 Map 中删除点和子节点
@@ -803,8 +861,8 @@ export class BaseNode extends EventEmitter {
                     parentNode,
                     parentKey,
                 });
-                
-                if(this.concept === 'ViewElement')
+
+                if (this.concept === 'ViewElement')
                     this.name = (this as unknown as ViewElement).getViewElementUniqueName(this.name);
 
                 list.splice(index, 0, this);
@@ -848,6 +906,12 @@ export class BaseNode extends EventEmitter {
         traverse(this);
     }
 
+    /**
+     *  获取当前节点在nasl标注节点上的节点
+     */
+    public getCurrentAnnotatedJSON() {
+        return this.__TypeAnnotation;
+    }
     /**
      *  获取
      *  当前this上下文 的Source 和 父级文件级别的节点