feat: native-mobile-api

src/Socket/web-socket.ts

@@ -0,0 +1,31 @@
+import { WebSocket as WS } from 'ws'
+import { DEFAULT_ORIGIN } from '../Defaults'
+import { SocketConfig } from '../Types'
+
+export class WebSocket extends WS {
+	constructor(public config: SocketConfig) {
+		super(config.waWebSocketUrl, undefined, {
+			origin: DEFAULT_ORIGIN,
+			headers: config.options.headers as Record<string, string>,
+			handshakeTimeout: config.connectTimeoutMs,
+			timeout: config.connectTimeoutMs,
+			agent: config.agent,
+		})
+	}
+
+	get isOpen() {
+		return this.readyState === WS.OPEN
+	}
+
+	get isClosed() {
+		return this.readyState === WS.CLOSED
+	}
+
+	get isClosing() {
+		return this.readyState === WS.CLOSING
+	}
+
+	get isConnecting() {
+		return this.readyState === WS.CONNECTING
+	}
+}