Moved saveAsDgml to base class

Author: CoderAllan

src/commands/generateDependencyInjectionGraph.ts

@@ -15,16 +15,8 @@ export class GenerateDependencyInjectionGraph extends ShowHierarchyBase {
           case 'saveAsPng':
             this.saveAsPng('DependencyInjectionGraph.png', message.text);
             return;
-        }
-      },
-      undefined,
-      this.extensionContext.subscriptions
-    );
-    webview.onDidReceiveMessage(
-      message => {
-        switch (message.command) {
           case 'saveAsDgml':
-            console.log(message.text);
+            this.saveAsDgml(this.config.dgmlGraphFilename, message.text);
             return;
         }
       },

src/commands/showComponentHierarchy.ts

@@ -13,18 +13,10 @@ export class ShowComponentHierarchy extends ShowHierarchyBase {
       message => {
         switch (message.command) {
           case 'saveAsPng':
-            this.saveAsPng(this.config.componentHierarchyFilename, message.text);
+            this.saveAsPng(this.config.moduleHierarchyFilename, message.text);
             return;
-        }
-      },
-      undefined,
-      this.extensionContext.subscriptions
-    );
-    webview.onDidReceiveMessage(
-      message => {
-        switch (message.command) {
           case 'saveAsDgml':
-            console.log(message.text);
+            this.saveAsDgml(this.config.dgmlGraphFilename, message.text);
             return;
         }
       },

src/commands/showHierarchyBase.ts

@@ -165,6 +165,9 @@ export class ShowHierarchyBase extends CommandBase {
     }
   }
 
+  protected saveAsDgml(dgmlGraphFilename: string, messageText: string) {
+    
+  }
 
   protected generateHtmlContent(webview: vscode.Webview, outputJsFilename: string): string {
     let htmlContent = fs.readFileSync(this.extensionContext?.asAbsolutePath(path.join('templates', this.templateHtmlFilename)), 'utf8');

src/commands/showModuleHierarchy.ts

@@ -15,16 +15,8 @@ export class ShowModuleHierarchy extends ShowHierarchyBase {
           case 'saveAsPng':
             this.saveAsPng(this.config.moduleHierarchyFilename, message.text);
             return;
-        }
-      },
-      undefined,
-      this.extensionContext.subscriptions
-    );
-    webview.onDidReceiveMessage(
-      message => {
-        switch (message.command) {
           case 'saveAsDgml':
-            console.log(message.text);
+            this.saveAsDgml(this.config.dgmlGraphFilename, message.text);
             return;
         }
       },
@@ -34,7 +26,7 @@ export class ShowModuleHierarchy extends ShowHierarchyBase {
     var workspaceFolder = this.fsUtils.getWorkspaceFolder();
     const errors: string[] = [];
     const project: Project = ModuleManager.scanProject(workspaceFolder, errors, this.isTypescriptFile);
-    
+
     this.nodes = [];
     this.edges = [];
     this.addNodesAndEdges(project, this.appendNodes, this.appendEdges);
@@ -44,8 +36,8 @@ export class ShowModuleHierarchy extends ShowHierarchyBase {
     const edgesJson = this.edges
       .map((edge, index, arr) => { return edge.toJsonString(); })
       .join(',\n');
-    
-      try {
+
+    try {
       const jsContent = this.generateJavascriptContent(nodesJson, edgesJson);
       const outputJsFilename = this.showModuleHierarchyJsFilename;
       let htmlContent = this.generateHtmlContent(webview, this.showModuleHierarchyJsFilename);