Encapsulate process execution in CLI manager

code-asher · code-asher · commit 0e8c4e076a07 · 2023-03-31T12:52:58.000-08:00
We will need to create a new function for config-ssh anyway since we
will be doing our own thing rather than spawning the CLI in order to
support multiple deployments so this preemptively encapsulates that
logic.
diff --git a/src/main/kotlin/com/coder/gateway/sdk/CoderCLIManager.kt b/src/main/kotlin/com/coder/gateway/sdk/CoderCLIManager.kt
@@ -1,6 +1,7 @@
 package com.coder.gateway.sdk
 
 import com.intellij.openapi.diagnostic.Logger
+import org.zeroturnaround.exec.ProcessExecutor
 import java.io.InputStream
 import java.net.URL
 import java.nio.file.Files
@@ -97,6 +98,42 @@ class CoderCLIManager(deployment: URL, buildVersion: String) {
         }
     }
 
+    /**
+     * Use the provided credentials to authenticate the CLI.
+     */
+    fun login(url: String, token: String): String {
+        return exec("login", url, "--token", token)
+    }
+
+    /**
+     * Configure SSH to use this binary.
+     *
+     * TODO: Support multiple deployments; currently they will clobber each other.
+     */
+    fun configSsh(): String {
+        return exec("config-ssh", "--yes", "--use-previous-options")
+    }
+
+    /**
+     * Return the binary version.
+     */
+    fun version(): String {
+        return exec("version")
+    }
+
+    /**
+     * Execute the binary with the provided arguments.
+     *
+     * @return The command's output.
+     */
+    private fun exec(vararg args: String): String {
+        return ProcessExecutor()
+            .command(localBinaryPath.toString(), *args)
+            .readOutput(true)
+            .execute()
+            .outputUTF8()
+    }
+
     companion object {
         val logger = Logger.getInstance(CoderCLIManager::class.java.simpleName)
     }
diff --git a/src/main/kotlin/com/coder/gateway/views/steps/CoderWorkspacesStepView.kt b/src/main/kotlin/com/coder/gateway/views/steps/CoderWorkspacesStepView.kt
@@ -59,7 +59,6 @@ import kotlinx.coroutines.delay
 import kotlinx.coroutines.isActive
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.withContext
-import org.zeroturnaround.exec.ProcessExecutor
 import java.awt.Component
 import java.awt.Dimension
 import java.awt.event.MouseEvent
@@ -530,21 +529,13 @@ class CoderWorkspacesStepView(val setNextButtonEnabled: (Boolean) -> Unit) : Cod
                 text = "Configuring Coder CLI..."
                 fraction = 0.5
             }
-            val loginOutput = ProcessExecutor().command(
-                cliManager.localBinaryPath.toString(),
-                "login",
+            val loginOutput = cliManager.login(
                 localWizardModel.coderURL,
-                "--token",
                 localWizardModel.token
-            ).readOutput(true).execute().outputUTF8()
+            )
             logger.info("coder-cli login output: $loginOutput")
             this.indicator.fraction = 0.8
-            val sshConfigOutput = ProcessExecutor().command(
-                cliManager.localBinaryPath.toString(),
-                "config-ssh",
-                "--yes",
-                "--use-previous-options"
-            ).readOutput(true).execute().outputUTF8()
+            val sshConfigOutput = cliManager.configSsh()
             logger.info("Result of `${cliManager.localBinaryPath} config-ssh --yes --use-previous-options`: $sshConfigOutput")
 
             this.indicator.apply {
diff --git a/src/test/groovy/CoderCLIManagerTest.groovy b/src/test/groovy/CoderCLIManagerTest.groovy
@@ -1,6 +1,5 @@
 package com.coder.gateway.sdk
 
-import org.zeroturnaround.exec.ProcessExecutor
 import spock.lang.Unroll
 
 import java.nio.file.Files
@@ -55,7 +54,7 @@ class CoderCLIManagerTest extends spock.lang.Specification {
         downloaded
         ccm.localBinaryPath.getParent() == dir
         ccm.localBinaryPath.toFile().exists()
-        new ProcessExecutor().command(ccm.localBinaryPath.toString(), "version").readOutput(true).execute().outputUTF8().contains("Coder")
+        ccm.version().contains("Coder")
     }
 
     def "skips cli download if it already exists"() {
