@@ -15,6 +15,20 @@ import java.nio.file.Path
1515import java.nio.file.Paths
1616
1717const val CODER_SSH_CONFIG_OPTIONS = " CODER_SSH_CONFIG_OPTIONS"
18+ const val CODER_URL = " CODER_URL"
19+
20+ /* *
21+ * Describes where a setting came from.
22+ */
23+ enum class Source {
24+ CONFIG , // Pulled from the global Coder CLI config.
25+ DEPLOYMENT_CONFIG , // Pulled from the config for a deployment.
26+ ENVIRONMENT , // Pulled from environment variables.
27+ LAST_USED , // Last used token.
28+ QUERY , // From the Gateway link as a query parameter.
29+ SETTINGS , // Pulled from settings.
30+ USER , // Input by the user.
31+ }
1832
1933open class CoderSettingsState (
2034 // Used to download the Coder CLI which is necessary to proxy SSH
@@ -66,6 +80,8 @@ open class CoderSettingsState(
6680 open var setupCommand : String = " "
6781 // Whether to ignore setup command failures.
6882 open var ignoreSetupFailure : Boolean = false ,
83+ // Default URL to show in the connection window.
84+ open var defaultURL : String = " "
6985)
7086
7187/* *
@@ -140,6 +156,53 @@ open class CoderSettings(
140156 val ignoreSetupFailure: Boolean
141157 get() = state.ignoreSetupFailure
142158
159+ /* *
160+ * The default URL to show in the connection window.
161+ */
162+ fun defaultURL (): Pair <String , Source >? {
163+ val defaultURL = state.defaultURL
164+ val envURL = env.get(CODER_URL )
165+ if (! defaultURL.isBlank()) {
166+ return defaultURL to Source .SETTINGS
167+ } else if (! envURL.isBlank()) {
168+ return envURL to Source .ENVIRONMENT
169+ } else {
170+ val (configUrl, _) = readConfig(coderConfigDir)
171+ if (! configUrl.isNullOrBlank()) {
172+ return configUrl to Source .CONFIG
173+ }
174+ }
175+ return null
176+ }
177+
178+ /* *
179+ * Given a deployment URL, try to find a token for it if required.
180+ */
181+ fun token (url : String ): Pair <String , Source >? {
182+ // No need to bother if we do not need token auth anyway.
183+ if (! requireTokenAuth) {
184+ return null
185+ }
186+ // Try the deployment's config directory. This could exist if someone
187+ // has entered a URL that they are not currently connected to, but have
188+ // connected to in the past.
189+ try {
190+ val (_, deploymentToken) = readConfig(dataDir(url.toURL()).resolve(" config"
191+ if (! deploymentToken.isNullOrBlank()) {
192+ return deploymentToken to Source .DEPLOYMENT_CONFIG
193+ }
194+ } catch (ex: Exception ) {
195+ // URL is invalid.
196+ }
197+ // Try the global config directory, in case they previously set up the
198+ // CLI with this URL.
199+ val (configUrl, configToken) = readConfig(coderConfigDir)
200+ if (configUrl == url && ! configToken.isNullOrBlank()) {
201+ return configToken to Source .CONFIG
202+ }
203+ return null
204+ }
205+
143206 /* *
144207 * Where the specified deployment should put its data.
145208 */
@@ -183,18 +246,20 @@ open class CoderSettings(
183246 }
184247
185248 /* *
186- * Return the URL and token from the config, if it exists. Both the url and
187- * session files must exist if using token auth, otherwise only url must
188- * exist.
249+ * Return the URL and token from the config, if they exist.
189250 */
190251 fun readConfig (dir : Path ): Pair <String ?, String ?> {
191252 logger.info(" Reading config from $dir "
192253 return try {
193- val token = if (requireTokenAuth) Files .readString(dir.resolve(" session" else null
194- Files .readString(dir.resolve(" url"
254+ Files .readString(dir.resolve(" url"
255+ } catch (e: Exception ) {
256+ // SSH has not been configured yet, or using some other authorization mechanism.
257+ null
258+ } to try {
259+ Files .readString(dir.resolve(" session"
195260 } catch (e: Exception ) {
196261 // SSH has not been configured yet, or using some other authorization mechanism.
197- null to null
262+ null
198263 }
199264 }
200265
0 commit comments