Break out getting terminal websocket url

code-asher · code-asher · commit 81acc6dfce82 · 2023-10-11T18:03:54.000-08:00
To make it easier to move away from this service.
diff --git a/site/src/utils/terminal.ts b/site/src/utils/terminal.ts
@@ -0,0 +1,37 @@
+import * as API from "api/api";
+
+export const terminalWebsocketUrl = async (
+  baseUrl: string | undefined,
+  reconnect: string,
+  agentId: string,
+  command: string | undefined,
+): Promise<string> => {
+  const query = new URLSearchParams({ reconnect });
+  if (command) {
+    query.set("command", command);
+  }
+
+  const url = new URL(baseUrl || `${location.protocol}//${location.host}`);
+  url.protocol = url.protocol === "https:" ? "wss:" : "ws:";
+  if (!url.pathname.endsWith("/")) {
+    url.pathname + "/";
+  }
+  url.pathname += `api/v2/workspaceagents/${agentId}/pty`;
+  url.search = "?" + query.toString();
+
+  // If the URL is just the primary API, we don't need a signed token to
+  // connect.
+  if (!baseUrl) {
+    return url.toString();
+  }
+
+  // Do ticket issuance and set the query parameter.
+  const tokenRes = await API.issueReconnectingPTYSignedToken({
+    url: url.toString(),
+    agentID: agentId,
+  });
+  query.set("coder_signed_app_token_23db1dde", tokenRes.signed_token);
+  url.search = "?" + query.toString();
+
+  return url.toString();
+};
diff --git a/site/src/xServices/terminal/terminalXService.ts b/site/src/xServices/terminal/terminalXService.ts
@@ -1,6 +1,7 @@
 import { assign, createMachine } from "xstate";
 import * as API from "api/api";
 import * as TypesGen from "api/typesGenerated";
+import { terminalWebsocketUrl } from "utils/terminal";
 import { getMatchingAgentOrFirst } from "utils/workspace";
 
 interface ReconnectingPTYRequest {
@@ -213,42 +214,12 @@ export const terminalMachine =
           if (!context.reconnection) {
             throw new Error("reconnection ID is not set");
           }
-
-          let baseURL = context.baseURL || "";
-          if (!baseURL) {
-            baseURL = `${location.protocol}//${location.host}`;
-          }
-
-          const query = new URLSearchParams({
-            reconnect: context.reconnection,
-          });
-          if (context.command) {
-            query.set("command", context.command);
-          }
-
-          const url = new URL(baseURL);
-          url.protocol = url.protocol === "https:" ? "wss:" : "ws:";
-          if (!url.pathname.endsWith("/")) {
-            url.pathname + "/";
-          }
-          url.pathname += `api/v2/workspaceagents/${context.workspaceAgent.id}/pty`;
-          url.search = "?" + query.toString();
-
-          // If the URL is just the primary API, we don't need a signed token to
-          // connect.
-          if (!context.baseURL) {
-            return url.toString();
-          }
-
-          // Do ticket issuance and set the query parameter.
-          const tokenRes = await API.issueReconnectingPTYSignedToken({
-            url: url.toString(),
-            agentID: context.workspaceAgent.id,
-          });
-          query.set("coder_signed_app_token_23db1dde", tokenRes.signed_token);
-          url.search = "?" + query.toString();
-
-          return url.toString();
+          return terminalWebsocketUrl(
+            context.baseURL,
+            context.reconnection,
+            context.workspaceAgent.id,
+            context.command,
+          );
         },
         connect: (context) => (send) => {
           return new Promise<WebSocket>((resolve, reject) => {
