Impl: add support for starting&stopping a workspace

fioan89 · fioan89 · commit f9db23e7b367 · 2022-08-13T00:42:13.000+03:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,7 @@
 ## [Unreleased]
 ### Added
 - support for displaying workspace version
+- support for managing the lifecycle of a workspace, i.e. start and stop
 
 ### Changed
 - workspace panel is now updated every 5 seconds
diff --git a/src/main/kotlin/com/coder/gateway/models/WorkspaceAgentModel.kt b/src/main/kotlin/com/coder/gateway/models/WorkspaceAgentModel.kt
@@ -2,8 +2,11 @@ package com.coder.gateway.models
 
 import com.coder.gateway.sdk.Arch
 import com.coder.gateway.sdk.OS
+import java.util.UUID
 
 data class WorkspaceAgentModel(
+    val workspaceID: UUID,
+    val workspaceName: String,
     val name: String,
     val templateName: String,
     val status: WorkspaceVersionStatus,
diff --git a/src/main/kotlin/com/coder/gateway/views/steps/CoderWorkspacesStepView.kt b/src/main/kotlin/com/coder/gateway/views/steps/CoderWorkspacesStepView.kt
@@ -17,6 +17,7 @@ import com.coder.gateway.sdk.CoderCLIManager
 import com.coder.gateway.sdk.CoderRestClientService
 import com.coder.gateway.sdk.OS
 import com.coder.gateway.sdk.ex.AuthenticationResponseException
+import com.coder.gateway.sdk.ex.WorkspaceResponseException
 import com.coder.gateway.sdk.getOS
 import com.coder.gateway.sdk.toURL
 import com.coder.gateway.sdk.v2.models.Workspace
@@ -171,13 +172,37 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
 
     private inner class StartWorkspaceAction : AnActionButton(CoderGatewayBundle.message("gateway.connector.view.coder.workspaces.start.text"), CoderGatewayBundle.message("gateway.connector.view.coder.workspaces.start.text"), CoderIcons.RUN) {
         override fun actionPerformed(p0: AnActionEvent) {
-            TODO("Not yet implemented")
+            if (tableOfWorkspaces.selectedObject != null) {
+                val workspace = tableOfWorkspaces.selectedObject as WorkspaceAgentModel
+                cs.launch {
+                    withContext(Dispatchers.IO) {
+                        try {
+                            coderClient.startWorkspace(workspace.workspaceID, workspace.workspaceName)
+                            startWorkspaceAction.isEnabled = false
+                        } catch (e: WorkspaceResponseException) {
+                            logger.warn("Could not build workspace ${workspace.name}, reason: $e")
+                        }
+                    }
+                }
+            }
         }
     }
 
     private inner class StopWorkspaceAction : AnActionButton(CoderGatewayBundle.message("gateway.connector.view.coder.workspaces.stop.text"), CoderGatewayBundle.message("gateway.connector.view.coder.workspaces.stop.text"), CoderIcons.STOP) {
         override fun actionPerformed(p0: AnActionEvent) {
-            TODO("Not yet implemented")
+            if (tableOfWorkspaces.selectedObject != null) {
+                val workspace = tableOfWorkspaces.selectedObject as WorkspaceAgentModel
+                cs.launch {
+                    withContext(Dispatchers.IO) {
+                        try {
+                            coderClient.stopWorkspace(workspace.workspaceID, workspace.workspaceName)
+                            stopWorkspaceAction.isEnabled = false
+                        } catch (e: WorkspaceResponseException) {
+                            logger.warn("Could not stop workspace ${workspace.name}, reason: $e")
+                        }
+                    }
+                }
+            }
         }
     }
 
@@ -310,6 +335,8 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
                 0 -> {
                     listOf(
                         WorkspaceAgentModel(
+                            this.id,
+                            this.name,
                             this.name,
                             this.templateName,
                             WorkspaceVersionStatus.from(this),
@@ -324,6 +351,8 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
                 1 -> {
                     listOf(
                         WorkspaceAgentModel(
+                            this.id,
+                            this.name,
                             this.name,
                             this.templateName,
                             WorkspaceVersionStatus.from(this),
@@ -338,6 +367,8 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
                 else -> agents.map { agent ->
                     val workspaceName = "${this.name}.${agent.name}"
                     WorkspaceAgentModel(
+                        this.id,
+                        this.name,
                         workspaceName,
                         this.templateName,
                         WorkspaceVersionStatus.from(this),
@@ -353,6 +384,8 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
             logger.warn("Agent(s) for ${this.name} could not be retrieved. Reason: $e")
             listOf(
                 WorkspaceAgentModel(
+                    this.id,
+                    this.name,
                     this.name,
                     this.templateName,
                     WorkspaceVersionStatus.from(this),
