cleaning around websocket client

src/util/websocket-util.ts

@@ -1,38 +1,84 @@
 import { WsKey } from '../types';
 
-export const wsKeyInverse = 'inverse';
-export const wsKeyLinearPrivate = 'linearPrivate';
-export const wsKeyLinearPublic = 'linearPublic';
-export const wsKeySpotPrivate = 'spotPrivate';
-export const wsKeySpotPublic = 'spotPublic';
+interface NetworkMapV3 {
+  livenet: string;
+  livenet2?: string;
+  testnet: string;
+  testnet2?: string;
+}
 
-export const WS_KEY_MAP = {
-  inverse: wsKeyInverse,
-  linearPrivate: wsKeyLinearPrivate,
-  linearPublic: wsKeyLinearPublic,
-  spotPrivate: wsKeySpotPrivate,
-  spotPublic: wsKeySpotPublic,
+type PublicPrivateNetwork = 'public' | 'private';
+
+export const WS_BASE_URL_MAP: Record<
+  string,
+  Record<PublicPrivateNetwork, NetworkMapV3>
+> = {
+  inverse: {
+    private: {
+      livenet: 'wss://stream.bybit.com/realtime',
+      testnet: 'wss://stream-testnet.bybit.com/realtime',
+    },
+    public: {
+      livenet: 'wss://stream.bybit.com/realtime',
+      testnet: 'wss://stream-testnet.bybit.com/realtime',
+    },
+  },
+  linear: {
+    private: {
+      livenet: 'wss://stream.bybit.com/realtime_private',
+      livenet2: 'wss://stream.bytick.com/realtime_private',
+      testnet: 'wss://stream-testnet.bybit.com/realtime_private',
+    },
+    public: {
+      livenet: 'wss://stream.bybit.com/realtime_public',
+      livenet2: 'wss://stream.bytick.com/realtime_public',
+      testnet: 'wss://stream-testnet.bybit.com/realtime_public',
+    },
+  },
+  spot: {
+    private: {
+      livenet: 'wss://stream.bybit.com/spot/ws',
+      testnet: 'wss://stream-testnet.bybit.com/spot/ws',
+    },
+    public: {
+      livenet: 'wss://stream.bybit.com/spot/quote/ws/v1',
+      livenet2: 'wss://stream.bybit.com/spot/quote/ws/v2',
+      testnet: 'wss://stream-testnet.bybit.com/spot/quote/ws/v1',
+      testnet2: 'wss://stream-testnet.bybit.com/spot/quote/ws/v2',
+    },
+  },
 };
 
-export const PUBLIC_WS_KEYS = [WS_KEY_MAP.linearPublic, WS_KEY_MAP.spotPublic];
+export const WS_KEY_MAP = {
+  inverse: 'inverse',
+  linearPrivate: 'linearPrivate',
+  linearPublic: 'linearPublic',
+  spotPrivate: 'spotPrivate',
+  spotPublic: 'spotPublic',
+} as const;
+
+export const PUBLIC_WS_KEYS = [
+  WS_KEY_MAP.linearPublic,
+  WS_KEY_MAP.spotPublic,
+] as string[];
 
 export function getLinearWsKeyForTopic(topic: string): WsKey {
-  const privateLinearTopics = [
+  const privateTopics = [
     'position',
     'execution',
     'order',
     'stop_order',
     'wallet',
   ];
-  if (privateLinearTopics.includes(topic)) {
-    return wsKeyLinearPrivate;
+  if (privateTopics.includes(topic)) {
+    return WS_KEY_MAP.linearPrivate;
   }
 
-  return wsKeyLinearPublic;
+  return WS_KEY_MAP.linearPublic;
 }
 
 export function getSpotWsKeyForTopic(topic: string): WsKey {
-  const privateLinearTopics = [
+  const privateTopics = [
     'position',
     'execution',
     'order',
@@ -42,9 +88,8 @@ export function getSpotWsKeyForTopic(topic: string): WsKey {
     'ticketInfo',
   ];
 
-  if (privateLinearTopics.includes(topic)) {
-    return wsKeySpotPrivate;
+  if (privateTopics.includes(topic)) {
+    return WS_KEY_MAP.spotPrivate;
   }
-
-  return wsKeySpotPublic;
+  return WS_KEY_MAP.spotPublic;
 }