[adapters] 增加小游戏适配部分源码

adapters/platforms/bytedance/wrapper/builtin/WebSocket.js

@@ -0,0 +1,87 @@
+const _socketTask = new WeakMap()
+
+export default class WebSocket {
+  static CONNECTING = 0 // The connection is not yet open.
+  static OPEN = 1 // The connection is open and ready to communicate.
+  static CLOSING = 2 // The connection is in the process of closing.
+  static CLOSED = 3 // The connection is closed or couldn't be opened.
+
+  binaryType = '' // TODO 更新 binaryType
+  bufferedAmount = 0 // TODO 更新 bufferedAmount
+  extensions = ''
+
+  onclose = null
+  onerror = null
+  onmessage = null
+  onopen = null
+
+  protocol = '' // TODO 小程序内目前获取不到，实际上需要根据服务器选择的 sub-protocol 返回
+  readyState = 3
+
+  constructor(url, protocols = []) {
+    if (typeof url !== 'string' || !(/(^ws:\/\/)|(^wss:\/\/)/).test(url)) {
+      throw new TypeError(`Failed to construct 'WebSocket': The URL '${url}' is invalid`)
+    }
+
+    this.url = url
+    this.readyState = WebSocket.CONNECTING
+
+    const socketTask = tt.connectSocket({
+      url,
+      protocols: Array.isArray(protocols) ? protocols : [protocols],
+      tcpNoDelay: true
+    })
+
+    _socketTask.set(this, socketTask)
+
+    socketTask.onClose((res) => {
+      this.readyState = WebSocket.CLOSED
+      if (typeof this.onclose === 'function') {
+        this.onclose(res)
+      }
+    })
+
+    socketTask.onMessage((res) => {
+      if (typeof this.onmessage === 'function') {
+        this.onmessage(res)
+      }
+    })
+
+    socketTask.onOpen(() => {
+      this.readyState = WebSocket.OPEN
+      if (typeof this.onopen === 'function') {
+        this.onopen()
+      }
+    })
+
+    socketTask.onError((res) => {
+      if (typeof this.onerror === 'function') {
+        this.onerror(new Error(res.errMsg))
+      }
+    })
+
+    return this
+  }
+
+  close(code, reason) {
+    this.readyState = WebSocket.CLOSING
+    const socketTask = _socketTask.get(this)
+
+    socketTask.close({
+      code,
+      reason
+    })
+  }
+
+  send(data) {
+    if (typeof data !== 'string' && !(data instanceof ArrayBuffer) && !ArrayBuffer.isView(data)) {
+      throw new TypeError(`Failed to send message: The data ${data} is invalid`)
+    }
+
+    const socketTask = _socketTask.get(this)
+
+    socketTask.send({
+      data
+    })
+  }
+}