impl: support for certificate based authentication

We now skip token input screen if the user provided a public and a private key
for mTLS authentication.

Author: fioan89

CHANGELOG.md

@@ -5,6 +5,7 @@
 ### Added
 
 - support for matching workspace agent in the URI via the agent name
+- support for certificate based authentication
 
 ### Removed
 

src/main/kotlin/com/coder/toolbox/CoderRemoteProvider.kt

@@ -244,7 +244,7 @@ class CoderRemoteProvider(
         environments.value = LoadableState.Value(emptyList())
         isInitialized.update { false }
         client = null
-        CoderCliSetupWizardState.resetSteps()
+        CoderCliSetupWizardState.goToFirstStep()
     }
 
     override val svgIcon: SvgIcon =

src/main/kotlin/com/coder/toolbox/sdk/CoderRestClient.kt

@@ -94,7 +94,10 @@ open class CoderRestClient(
                 .build()
         }
 
-        if (token != null) {
+        if (context.settingsStore.requireTokenAuth) {
+            if (token.isNullOrBlank()) {
+                throw IllegalStateException("Token is required for $url deployment")
+            }
             builder = builder.addInterceptor {
                 it.proceed(
                     it.request().newBuilder().addHeader("Coder-Session-Token", token).build()

src/main/kotlin/com/coder/toolbox/views/ConnectStep.kt

@@ -67,7 +67,7 @@ class ConnectStep(
             return
         }
 
-        if (!CoderCliSetupContext.hasToken()) {
+        if (context.settingsStore.requireTokenAuth && !CoderCliSetupContext.hasToken()) {
             errorField.textState.update { context.i18n.ptrl("Token is required") }
             return
         }
@@ -77,7 +77,7 @@ class ConnectStep(
                 val client = CoderRestClient(
                     context,
                     CoderCliSetupContext.url!!,
-                    CoderCliSetupContext.token!!,
+                    if (context.settingsStore.requireTokenAuth) CoderCliSetupContext.token else null,
                     PluginManager.pluginInfo.version,
                 )
                 // allows interleaving with the back/cancel action
@@ -91,17 +91,17 @@ class ConnectStep(
                     statusField.textState.update { (context.i18n.pnotr(progress)) }
                 }
                 // We only need to log in if we are using token-based auth.
-                if (client.token != null) {
+                if (context.settingsStore.requireTokenAuth) {
                     statusField.textState.update { (context.i18n.ptrl("Configuring Coder CLI...")) }
                     // allows interleaving with the back/cancel action
                     yield()
-                    cli.login(client.token)
+                    cli.login(client.token!!)
                 }
                 statusField.textState.update { (context.i18n.ptrl("Successfully configured ${CoderCliSetupContext.url!!.host}...")) }
                 // allows interleaving with the back/cancel action
                 yield()
                 CoderCliSetupContext.reset()
-                CoderCliSetupWizardState.resetSteps()
+                CoderCliSetupWizardState.goToFirstStep()
                 onConnect(client, cli)
             } catch (ex: CancellationException) {
                 if (ex.message != USER_HIT_THE_BACK_BUTTON) {
@@ -127,10 +127,14 @@ class ConnectStep(
         } finally {
             if (shouldAutoLogin.value) {
                 CoderCliSetupContext.reset()
-                CoderCliSetupWizardState.resetSteps()
+                CoderCliSetupWizardState.goToFirstStep()
                 context.secrets.rememberMe = false
             } else {
-                CoderCliSetupWizardState.goToPreviousStep()
+                if (context.settingsStore.requireTokenAuth) {
+                    CoderCliSetupWizardState.goToPreviousStep()
+                } else {
+                    CoderCliSetupWizardState.goToFirstStep()
+                }
             }
         }
     }

src/main/kotlin/com/coder/toolbox/views/DeploymentUrlStep.kt

@@ -57,7 +57,11 @@ class DeploymentUrlStep(
             notify("URL is invalid", e)
             return false
         }
-        CoderCliSetupWizardState.goToNextStep()
+        if (context.settingsStore.requireTokenAuth) {
+            CoderCliSetupWizardState.goToNextStep()
+        } else {
+            CoderCliSetupWizardState.goToLastStep()
+        }
         return true
     }
 

src/main/kotlin/com/coder/toolbox/views/state/CoderCliSetupWizardState.kt

@@ -25,7 +25,11 @@ object CoderCliSetupWizardState {
         currentStep = WizardStep.entries.toTypedArray()[(currentStep.ordinal - 1) % WizardStep.entries.size]
     }
 
-    fun resetSteps() {
+    fun goToLastStep() {
+        currentStep = WizardStep.CONNECT
+    }
+
+    fun goToFirstStep() {
         currentStep = WizardStep.URL_REQUEST
     }
 }