Add helper for getting a query variable's value

Author: code-asher

src/extension.ts

@@ -11,7 +11,7 @@ import {
   handleInspectCommand,
   handleShowLogsCommand,
 } from "./logs"
-import { context, debug } from "./utils"
+import { context, debug, getQueryValue } from "./utils"
 import {
   CoderWorkspacesProvider,
   rebuildWorkspace,
@@ -43,8 +43,7 @@ export const uriHandler: vscode.UriHandler = {
     debug(`Handling URI: ${uri}`)
     switch (action) {
       case "open-workspace": {
-        const version = Array.isArray(query.version) ? query.version[0] : query.version
-        return preflight(version).then(() => openWorkspace(resource))
+        return preflight(getQueryValue(query.version)).then(() => openWorkspace(resource))
       }
       default:
         vscode.window.showErrorMessage(`Unknown action "${action}"`)

src/utils.test.ts

@@ -86,4 +86,11 @@ suite("Utils", () => {
       assert.strictEqual(content2, "archive content\n")
     }
   })
+
+  test("getQueryValue", () => {
+    assert.strictEqual(utils.getQueryValue(undefined), undefined)
+    assert.strictEqual(utils.getQueryValue("foo"), "foo")
+    assert.strictEqual(utils.getQueryValue(["bar"]), "bar")
+    assert.strictEqual(utils.getQueryValue(["bazzle", "qux"]), "bazzle")
+  })
 })

src/utils.ts

@@ -204,3 +204,10 @@ export const getAssetUrl = (version: string): string => {
     ? `https://github.com/cdr/coder-cli/releases/${version}/download/${assetFilename}`
     : `https://github.com/cdr/coder-cli/releases/download/${version}/${assetFilename}`
 }
+
+/**
+ * Get the first or only value from a query parameter.
+ */
+export const getQueryValue = (val: string[] | string | undefined): string | undefined => {
+  return Array.isArray(val) ? val[0] : val
+}