Refactoring: Transfer nodes position as dictionary instead of array.

Author: CoderAllan

src/dgmlManager.ts

@@ -71,8 +71,7 @@ export class DgmlManager {
     }
   }
 
-  public addNodesAndLinks(xmlDoc: Document, nodes: Node[], nodeInfos: NetworkNode[], edges: Edge[]) {
-    const nodeInfoDictionary = Object.assign({}, ...nodeInfos.map((nodeInfo) => ({ [nodeInfo.id]: nodeInfo })));
+  public addNodesAndLinks(xmlDoc: Document, nodes: Node[], nodeInfoDictionary: { [id: string]: NetworkNode}, edges: Edge[]) {
     const nodesElement = this.addNodeToRoot(xmlDoc, "Nodes");
     const linksElement = this.addNodeToRoot(xmlDoc, "Links");
     const categoryDictionary: { [nodeType: string]: Category } = {};

templates/showHierarchy_Template.js

@@ -247,14 +247,14 @@
   }
 
   function saveAsDgml() {
-    const nodeExport = [];
+    const nodeExport = {};
     nodes.forEach(node => {
-      nodeExport.push({
+      nodeExport[node.id] = {
         id: node.id,
         label: cleanLabel(node.label),
         position: network.getPosition(node.id),
         boundingBox: network.getBoundingBox(node.id)
-      });
+      };
     });
     const direction = hierarchicalOptionsDirectionSelect.value ? hierarchicalOptionsDirectionSelect.value : 'UD';
     vscode.postMessage({