WIP: adding tree view to support crud workspaces

rodrimaia · rodrimaia · commit 66bd03a58f12 · 2023-03-28T15:29:50.000-03:00
diff --git a/package.json b/package.json
@@ -42,7 +42,7 @@
       "coder": [
         {
           "id": "coderRemote",
-          "name": "",
+          "name": "Workspaces",
           "visibility": "visible",
           "icon": "media/logo.svg",
           "contextualTitle": "Coder Remote"
@@ -54,11 +54,6 @@
         "view": "coderRemote",
         "contents": "Coder is a platform that provisions remote development environments. \n[Login](command:coder.login)",
         "when": "!coder.authenticated && coder.loaded"
-      },
-      {
-        "view": "coderRemote",
-        "contents": "You're logged in! \n[Open Workspace](command:coder.open)",
-        "when": "coder.authenticated && coder.loaded"
       }
     ],
     "commands": [
@@ -79,7 +74,27 @@
         "title": "Coder: Update Workspace",
         "when": "coder.workspace.updatable"
       }
-    ]
+    ],
+    "menus": {
+      "view/title": [
+        {
+          "command": "coder.logout",
+          "when": "coder.authenticated"
+        },
+        {
+          "command": "coder.createWorkspace",
+          "when": "coder.authenticated",
+          "group": "navigation"
+        }
+      ],
+      "view/item/context": [
+        {
+          "command": "coder.open",
+          "when": "coder.authenticated",
+          "group": "inline"
+        }
+      ]
+    }
   },
   "scripts": {
     "vscode:prepublish": "yarn package",
diff --git a/src/extension.ts b/src/extension.ts
@@ -6,12 +6,15 @@ import * as vscode from "vscode"
 import { Commands } from "./commands"
 import { Remote } from "./remote"
 import { Storage } from "./storage"
+import { WorkspaceProvider } from "./workspacesProvider"
 
 export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
   const output = vscode.window.createOutputChannel("Coder")
   const storage = new Storage(output, ctx.globalState, ctx.secrets, ctx.globalStorageUri, ctx.logUri)
   await storage.init()
 
+  vscode.window.registerTreeDataProvider("coderRemote", new WorkspaceProvider())
+
   getAuthenticatedUser()
     .then(() => {
       vscode.commands.executeCommand("setContext", "coder.authenticated", true)
diff --git a/src/workspacesProvider.ts b/src/workspacesProvider.ts
@@ -0,0 +1,32 @@
+import { getWorkspaces } from "coder/site/src/api/api"
+import * as path from "path"
+import * as vscode from "vscode"
+
+export class WorkspaceProvider implements vscode.TreeDataProvider<Dependency> {
+  getTreeItem(element: Dependency): vscode.TreeItem {
+    return element
+  }
+
+  getChildren(): Thenable<Dependency[]> {
+    return getWorkspaces({
+      q: "owner:me",
+    }).then((workspaces) => {
+      const exampleWorkspaces = [{ name: "example1" }, { name: "example2" }]
+      return [...workspaces.workspaces, ...exampleWorkspaces].map(
+        (workspace) => new Dependency(workspace.name, vscode.TreeItemCollapsibleState.None),
+      )
+    })
+  }
+}
+
+class Dependency extends vscode.TreeItem {
+  constructor(public readonly label: string, public readonly collapsibleState: vscode.TreeItemCollapsibleState) {
+    super(label, collapsibleState)
+    this.tooltip = `${this.label}`
+  }
+
+  iconPath = {
+    light: path.join(__filename, "..", "..", "media", "logo.svg"),
+    dark: path.join(__filename, "..", "..", "media", "logo.svg"),
+  }
+}
