Impl: action to update workspace to the latest template version

- resolves #52

Author: fioan89

CHANGELOG.md

@@ -5,7 +5,7 @@
 ## [Unreleased]
 ### Added
 - support for displaying workspace version
-- support for managing the lifecycle of a workspace, i.e. start and stop
+- support for managing the lifecycle of a workspace, i.e. start and stop and update workspace to the latest template version
 
 ### Changed
 - workspace panel is now updated every 5 seconds

src/main/kotlin/com/coder/gateway/icons/CoderIcons.kt

@@ -11,6 +11,7 @@ object CoderIcons {
 
     val RUN = IconLoader.getIcon("run.svg", javaClass)
     val STOP = IconLoader.getIcon("stop.svg", javaClass)
+    val UPDATE = IconLoader.getIcon("update.svg", javaClass)
 
     val WINDOWS = IconLoader.getIcon("windows.svg", javaClass)
     val MACOS = IconLoader.getIcon("macOS.svg", javaClass)

src/main/kotlin/com/coder/gateway/models/WorkspaceAgentModel.kt

@@ -8,9 +8,11 @@ data class WorkspaceAgentModel(
     val workspaceID: UUID,
     val workspaceName: String,
     val name: String,
+    val templateID: UUID,
     val templateName: String,
     val status: WorkspaceVersionStatus,
     val agentStatus: WorkspaceAgentStatus,
+    val lastBuildTransition: String,
     val agentOS: OS?,
     val agentArch: Arch?,
     val homeDirectory: String?

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.TemplateResponseException
 import com.coder.gateway.sdk.ex.WorkspaceResponseException
 import com.coder.gateway.sdk.getOS
 import com.coder.gateway.sdk.toURL
@@ -107,13 +108,15 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
 
     private val startWorkspaceAction = StartWorkspaceAction()
     private val stopWorkspaceAction = StopWorkspaceAction()
+    private val updateWorkspaceTemplateAction = UpdateWorkspaceTemplateAction()
 
     private val toolbar = ToolbarDecorator.createDecorator(tableOfWorkspaces)
         .disableAddAction()
         .disableRemoveAction()
         .disableUpDownActions()
         .addExtraAction(startWorkspaceAction)
         .addExtraAction(stopWorkspaceAction)
+        .addExtraAction(updateWorkspaceTemplateAction)
 
     private var poller: Job? = null
 
@@ -175,6 +178,26 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
         }
     }
 
+    private inner class UpdateWorkspaceTemplateAction : AnActionButton(CoderGatewayBundle.message("gateway.connector.view.coder.workspaces.update.text"), CoderGatewayBundle.message("gateway.connector.view.coder.workspaces.update.text"), CoderIcons.UPDATE) {
+        override fun actionPerformed(p0: AnActionEvent) {
+            if (tableOfWorkspaces.selectedObject != null) {
+                val workspace = tableOfWorkspaces.selectedObject as WorkspaceAgentModel
+                cs.launch {
+                    withContext(Dispatchers.IO) {
+                        try {
+                            coderClient.updateWorkspace(workspace.workspaceID, workspace.workspaceName, workspace.lastBuildTransition, workspace.templateID)
+                            loadWorkspaces()
+                        } catch (e: WorkspaceResponseException) {
+                            logger.warn("Could not update workspace ${workspace.name}, reason: $e")
+                        } catch (e: TemplateResponseException) {
+                            logger.warn("Could not update 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) {
             if (tableOfWorkspaces.selectedObject != null) {
@@ -203,16 +226,26 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
             RUNNING -> {
                 startWorkspaceAction.isEnabled = false
                 stopWorkspaceAction.isEnabled = true
+                when (tableOfWorkspaces.selectedObject?.status) {
+                    WorkspaceVersionStatus.OUTDATED -> updateWorkspaceTemplateAction.isEnabled = true
+                    else -> updateWorkspaceTemplateAction.isEnabled = false
+                }
+
             }
 
             STOPPED, FAILED -> {
                 startWorkspaceAction.isEnabled = true
                 stopWorkspaceAction.isEnabled = false
+                when (tableOfWorkspaces.selectedObject?.status) {
+                    WorkspaceVersionStatus.OUTDATED -> updateWorkspaceTemplateAction.isEnabled = true
+                    else -> updateWorkspaceTemplateAction.isEnabled = false
+                }
             }
 
             else -> {
                 startWorkspaceAction.isEnabled = false
                 stopWorkspaceAction.isEnabled = false
+                updateWorkspaceTemplateAction.isEnabled = false
             }
         }
         ActivityTracker.getInstance().inc()
@@ -342,9 +375,11 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
                             this.id,
                             this.name,
                             this.name,
+                            this.templateID,
                             this.templateName,
                             WorkspaceVersionStatus.from(this),
                             WorkspaceAgentStatus.from(this),
+                            this.latestBuild.workspaceTransition.name.toLowerCase(),
                             null,
                             null,
                             null
@@ -358,9 +393,11 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
                         this.id,
                         this.name,
                         workspaceWithAgentName,
+                        this.templateID,
                         this.templateName,
                         WorkspaceVersionStatus.from(this),
                         WorkspaceAgentStatus.from(this),
+                        this.latestBuild.workspaceTransition.name.toLowerCase(),
                         OS.from(agent.operatingSystem),
                         Arch.from(agent.architecture),
                         agent.directory
@@ -374,9 +411,11 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
                     this.id,
                     this.name,
                     this.name,
+                    this.templateID,
                     this.templateName,
                     WorkspaceVersionStatus.from(this),
                     WorkspaceAgentStatus.from(this),
+                    this.latestBuild.workspaceTransition.name.toLowerCase(),
                     null,
                     null,
                     null

src/main/resources/messages/CoderGatewayBundle.properties

@@ -12,6 +12,7 @@ gateway.connector.view.coder.workspaces.cli.downloader.dialog.title=Authenticate
 gateway.connector.view.coder.workspaces.next.text=Select IDE and Project
 gateway.connector.view.coder.workspaces.start.text=Start Workspace
 gateway.connector.view.coder.workspaces.stop.text=Stop Workspace
+gateway.connector.view.coder.workspaces.update.text=Update Workspace Template
 gateway.connector.view.coder.remoteproject.loading.text=Retrieving products...
 gateway.connector.view.coder.remoteproject.ide.error.text=Could not retrieve any IDE for workspace {0} because an error was encountered
 gateway.connector.view.coder.remoteproject.next.text=Start IDE and Connect