1337 Move getWebSocket to utils.ts (#1877)

Author: Georgi2704

.changeset/small-vans-glow.md

@@ -0,0 +1,5 @@
+---
+'@orchestrator-ui/orchestrator-ui-components': patch
+---
+
+1337 Move getWebSocket to utils.ts

packages/orchestrator-ui-components/src/rtk/endpoints/streamMessages.ts

@@ -1,26 +1,9 @@
 import { debounce } from 'lodash';
-import { getSession } from 'next-auth/react';
 
-import type { WfoSession } from '@/hooks';
+import { getWebSocket, orchestratorApi } from '@/rtk';
 import type { RootState } from '@/rtk/store';
 import { CacheTag, CacheTagType } from '@/types';
 
-import { orchestratorApi } from '../api';
-
-const getWebSocket = async (url: string) => {
-    const session = (await getSession()) as WfoSession;
-
-    if (session?.accessToken) {
-        // Implemented authentication taking this into account: https://stackoverflow.com/questions/4361173/http-headers-in-websockets-client-api/77060459#77060459
-        return new WebSocket(url, [
-            'base64.bearer.token',
-            session?.accessToken,
-        ]);
-    } else {
-        return new WebSocket(url);
-    }
-};
-
 const PING_INTERVAL_MS = 30000;
 const NO_PONG_RECEIVED_TIMEOUT_MS = 35000;
 const INITIAL_CONNECTION_CHECK_INTERVAL_MS = 2000;

packages/orchestrator-ui-components/src/rtk/utils.ts

@@ -1,9 +1,11 @@
 import { GraphQLError } from 'graphql/error/GraphQLError';
 import { isPlainObject } from 'lodash';
+import { getSession } from 'next-auth/react';
 
 import { SerializedError } from '@reduxjs/toolkit';
 import { FetchBaseQueryError } from '@reduxjs/toolkit/query';
 
+import type { WfoSession } from '@/hooks';
 import { WfoGraphqlError } from '@/rtk/api';
 
 export function stripUndefined(obj: object): Record<string, unknown> {
@@ -77,3 +79,17 @@ export const mapRtkErrorToWfoError = (
     }
     return error;
 };
+
+export const getWebSocket = async (url: string) => {
+    const session = (await getSession()) as WfoSession;
+
+    if (session?.accessToken) {
+        // Implemented authentication taking this into account: https://stackoverflow.com/questions/4361173/http-headers-in-websockets-client-api/77060459#77060459
+        return new WebSocket(url, [
+            'base64.bearer.token',
+            session?.accessToken,
+        ]);
+    } else {
+        return new WebSocket(url);
+    }
+};