feat: add command to graph terraform project (#10)

rohinivsenthil · Oct 1, 2021 · 5c9a1c5 · 5c9a1c5
1 parent 0054a26
commit 5c9a1c5
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 19 deletions.
diff --git a/package.json b/package.json
@@ -33,7 +33,9 @@
   ],
   "activationEvents": [
     "onView:terrastate.terrastate",
-    "onCommand:terrastate.refresh"
+    "onCommand:terrastate.refresh",
+    "onCommand:terrastate.init",
+    "onCommand:terrastate.graph"
   ],
   "main": "./dist/extension.js",
   "contributes": {
@@ -203,10 +205,6 @@
         }
       ],
       "commandPalette": [
-        {
-          "command": "terrastate.graph",
-          "when": "false"
-        },
         {
           "command": "terrastate.validate",
           "when": "false"

diff --git a/src/extension.ts b/src/extension.ts
@@ -102,18 +102,30 @@ export async function activate(
   );
 
   context.subscriptions.push(
-    vscode.commands.registerCommand("terrastate.graph", async (arg: string) => {
-      const panel = vscode.window.createWebviewPanel(
-        "Graph",
-        "Terrastate – Graph",
-        vscode.ViewColumn.One,
-        { enableScripts: true }
-      );
-      panel.iconPath = vscode.Uri.joinPath(
-        context.extensionUri,
-        "media/terrastate.png"
-      );
-      panel.webview.html = `
+    vscode.commands.registerCommand(
+      "terrastate.graph",
+      async (arg?: string) => {
+        if (arg === undefined) {
+          arg = (await terrastateProvider.pickTopLevelModule())?.directory;
+        }
+
+        if (arg === undefined) {
+          return;
+        }
+
+        const panel = vscode.window.createWebviewPanel(
+          "Graph",
+          "Terrastate – Graph",
+          vscode.ViewColumn.One,
+          { enableScripts: true }
+        );
+
+        panel.iconPath = vscode.Uri.joinPath(
+          context.extensionUri,
+          "media/terrastate.png"
+        );
+
+        panel.webview.html = `
       <!DOCTYPE html>
       <html lang="en">
         <head>
@@ -130,6 +142,7 @@ export async function activate(
         </body>
       </html>
     `;
-    })
+      }
+    )
   );
 }
diff --git a/src/terrastateProvider.ts b/src/terrastateProvider.ts
@@ -220,7 +220,7 @@ export class TerrastateProvider implements vscode.TreeDataProvider<Item> {
     }
   }
 
-  private async pickTopLevelModule(): Promise<Item | undefined> {
+  async pickTopLevelModule(): Promise<Item | undefined> {
     const choice = await vscode.window.showQuickPick(
       [...this.topLevelModules.keys()].map((directory) => ({
         label: path.dirname(