Added GraphState class

Author: CoderAllan

src/commands/showHierarchyBase.ts

@@ -1,6 +1,6 @@
 import { CommandBase } from '@commands';
 import { Config, DgmlManager, FileSystemUtils, GraphVizManager } from '@src';
-import { Edge, Node } from '@model';
+import { Edge, GraphState, Node } from '@model';
 import { Base64 } from 'js-base64';
 import * as fs from 'fs';
 import * as path from 'path';
@@ -14,6 +14,7 @@ export class ShowHierarchyBase extends CommandBase {
   protected fsUtils: FileSystemUtils = new FileSystemUtils();
   protected config = new Config();
   protected extensionContext: vscode.ExtensionContext;
+  protected graphState: GraphState;
   protected nodes: Node[] = [];
   protected edges: Edge[] = [];
   protected templateJsFilename: string = 'showHierarchy_Template.js';
@@ -23,9 +24,10 @@ export class ShowHierarchyBase extends CommandBase {
   protected showHierarchyCssFilename: string = 'showHierarchy.css';
   protected workspaceDirectory = this.fsUtils.getWorkspaceFolder();
 
-  constructor(context: vscode.ExtensionContext) {
+  constructor(context: vscode.ExtensionContext, graphState: GraphState) {
     super();
     this.extensionContext = context;
+    this.graphState = graphState;
   }
   protected appendNodes = (nodeList: Node[]) => {
     nodeList.forEach(newNode => {

src/extension.ts

@@ -12,6 +12,7 @@ import {
   ShowModuleHierarchy,
   GenerateDependencyInjectionGraph
 } from '@commands';
+import { GraphState } from '@model';
 
 export function activate(context: vscode.ExtensionContext) {
 
@@ -48,6 +49,7 @@ export function activate(context: vscode.ExtensionContext) {
   context.subscriptions.push(componentHierarchyMarkdownDisposable);
 
   let componentHierarchyPanel: vscode.WebviewPanel | undefined = undefined;
+  const componentHierarchyGraphState: GraphState = new GraphState();
   const showComponentHierarchyDisposable = vscode.commands.registerCommand(`${cmdPrefix}.${ShowComponentHierarchy.commandName}`, () => {
     if (componentHierarchyPanel !== undefined) {
       componentHierarchyPanel.reveal(vscode.ViewColumn.One);
@@ -64,12 +66,13 @@ export function activate(context: vscode.ExtensionContext) {
       }, null, context.subscriptions);
     }
     componentHierarchyPanel.onDidDispose(() => componentHierarchyPanel = undefined, undefined, context.subscriptions);
-    const command = new ShowComponentHierarchy(context);
+    const command = new ShowComponentHierarchy(context, componentHierarchyGraphState);
     command.execute(componentHierarchyPanel.webview);
   });
   context.subscriptions.push(showComponentHierarchyDisposable);
 
   let moduleHierarchyPanel: vscode.WebviewPanel | undefined = undefined;
+  const moduleHierarchyGraphState: GraphState = new GraphState();
   const showModuleHierarchyDisposable = vscode.commands.registerCommand(`${cmdPrefix}.${ShowModuleHierarchy.commandName}`, () => {
     if (moduleHierarchyPanel !== undefined) {
       moduleHierarchyPanel.reveal(vscode.ViewColumn.One);
@@ -86,12 +89,13 @@ export function activate(context: vscode.ExtensionContext) {
       }, null, context.subscriptions);
     }
     moduleHierarchyPanel.onDidDispose(() => moduleHierarchyPanel = undefined, undefined, context.subscriptions);
-    const command = new ShowModuleHierarchy(context);
+    const command = new ShowModuleHierarchy(context, moduleHierarchyGraphState);
     command.execute(moduleHierarchyPanel.webview);
   });
   context.subscriptions.push(showModuleHierarchyDisposable);
 
   let dependencyInjectionGraphPanel: vscode.WebviewPanel | undefined = undefined;
+  const dependencyInjectionGraphState: GraphState = new GraphState();
   const generateDependencyInjectionGraphDisposable = vscode.commands.registerCommand(`${cmdPrefix}.${GenerateDependencyInjectionGraph.commandName}`, () => {
     if (dependencyInjectionGraphPanel !== undefined) {
       dependencyInjectionGraphPanel.reveal(vscode.ViewColumn.One);
@@ -108,7 +112,7 @@ export function activate(context: vscode.ExtensionContext) {
       }, null, context.subscriptions);
     }
     dependencyInjectionGraphPanel.onDidDispose(() => dependencyInjectionGraphPanel = undefined, undefined, context.subscriptions);
-    const command = new GenerateDependencyInjectionGraph(context);
+    const command = new GenerateDependencyInjectionGraph(context, dependencyInjectionGraphState);
     command.execute(dependencyInjectionGraphPanel.webview);
   });
   context.subscriptions.push(generateDependencyInjectionGraphDisposable);

src/model/GraphState.ts

@@ -0,0 +1,7 @@
+import { Position } from "./Position";
+
+export class GraphState {
+  public graphLayout: string | undefined;
+  public networkSeed: number | undefined;
+  public nodePositions: Position | undefined;
+}

src/model/index.ts

@@ -4,6 +4,7 @@ export * from './BoundingBox';
 export * from './Category';
 export * from './Component';
 export * from './Edge';
+export * from './GraphState';
 export * from './NamedEntity';
 export * from './NetworkNode';
 export * from './Node';