coder ssh strategy finally working

Author: bcpeinhardt

package.json

@@ -211,8 +211,8 @@
         "when": "coder.authenticated"
       },
       {
-        "command": "coder.openAITask",
-        "title": "Coder: Open AI Task",
+        "command": "coder.openAppStatus",
+        "title": "Coder: Open App Status",
         "icon": "$(robot)",
         "when": "coder.authenticated"
       }
@@ -244,7 +244,7 @@
           "group": "navigation"
         },
         {
-          "command": "coder.openAITask",
+          "command": "coder.openAppStatus",
           "when": "coder.authenticated && view == myWorkspaces",
           "group": "navigation"
         }
@@ -280,11 +280,6 @@
           "command": "coder.createWorkspace",
           "group": "remote_11_ssh_coder@2",
           "when": "coder.authenticated"
-        },
-        {
-          "command": "coder.openAITask",
-          "group": "remote_11_ssh_coder@3",
-          "when": "coder.authenticated"
         }
       ]
     }

src/commands.ts

@@ -412,14 +412,32 @@ export class Commands {
     status?: string
     url?: string
     agent_name?: string
+    command?: string
+    workspace_name?: string
   }): Promise<void> {
+    // Launch and run command in terminal if command is provided
+    if (app.command) {
+      const terminal = vscode.window.createTerminal(`${app.name || "Application"} Status`)
+      terminal.show(false)
+      vscode.commands.executeCommand("workbench.action.toggleMaximizedPanel")
+      // If workspace_name is provided, run coder ssh before the command
+      if (app.workspace_name) {
+        terminal.sendText(`coder ssh ${app.workspace_name}`)
+        // Sleep for 5 seconds
+        await new Promise((resolve) => setTimeout(resolve, 5000))
+        terminal.sendText(app.command)
+      } else {
+        terminal.sendText("need workspace name")
+      }
+      return
+    }
     // Check if app has a URL to open
     if (app.url) {
       await vscode.env.openExternal(vscode.Uri.parse(app.url))
       return
     }
 
-    // If no URL, show information about the app status
+    // If no URL or command, show information about the app status
     vscode.window.showInformationMessage(`${app.name || "Application"}: ${app.status || "Running"}`, {
       detail: `Agent: ${app.agent_name || "Unknown"}`,
     })

src/workspacesProvider.ts

@@ -168,6 +168,8 @@ export class WorkspaceProvider implements vscode.TreeDataProvider<vscode.TreeIte
                 url: app.url,
                 agent_id: agent.id,
                 agent_name: agent.name,
+                command: app.command,
+                workspace_name: workspace.name,
               }))
             }
           })
@@ -253,21 +255,25 @@ export class WorkspaceProvider implements vscode.TreeDataProvider<vscode.TreeIte
 
         // Add app status section with collapsible header
         if (element.agent.apps && element.agent.apps.length > 0) {
-          let needsAttention = []
+          const needsAttention = []
           for (const app of element.agent.apps) {
             if (app.statuses && app.statuses.length > 0) {
               for (const status of app.statuses) {
                 if (status.needs_user_attention) {
-                  needsAttention.push(new AppStatusTreeItem(status))
+                  needsAttention.push(
+                    new AppStatusTreeItem({
+                      name: status.message,
+                      command: app.command,
+                      status: status.state,
+                      workspace_name: element.workspaceName,
+                    }),
+                  )
                 }
               }
             }
           }
 
-          const appStatusSection = new SectionTreeItem(
-            "Applications in need of attention",
-            needsAttention,
-          )
+          const appStatusSection = new SectionTreeItem("Applications in need of attention", needsAttention)
           items.push(appStatusSection)
         }
 
@@ -372,17 +378,15 @@ class AgentMetadataTreeItem extends vscode.TreeItem {
 class AppStatusTreeItem extends vscode.TreeItem {
   constructor(
     public readonly app: {
-      name?: string
-      display_name?: string
+      name: string
       status?: string
-      icon?: string
       url?: string
-      agent_id?: string
-      agent_name?: string
+      command?: string
+      workspace_name?: string
     },
   ) {
-    super(app.icon || "$(pulse)", vscode.TreeItemCollapsibleState.None)
-    this.description = app.status || "Running"
+    super(app.name, vscode.TreeItemCollapsibleState.None)
+    this.description = app.status
     this.contextValue = "coderAppStatus"
 
     // Add command to handle clicking on the app
@@ -449,7 +453,16 @@ class AgentTreeItem extends OpenableTreeItem {
 }
 
 export class WorkspaceTreeItem extends OpenableTreeItem {
-  public appStatus: { name: string; status: string; icon?: string }[] = []
+  public appStatus: {
+    name: string
+    status: string
+    icon?: string
+    url?: string
+    agent_id?: string
+    agent_name?: string
+    command?: string
+    workspace_name?: string
+  }[] = []
 
   constructor(
     public readonly workspace: Workspace,