You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.4 KiB
41 lines
1.4 KiB
import * as ws from "ws"; |
|
import { BaseWS } from "./websocket.js"; |
|
/** |
|
* WebSocket transport based on the `WebSocket` object provided by the `ws` package. |
|
* |
|
* Usage: Node.js, Deno (compat), Bun (compat) |
|
* |
|
* @see https://developer.mozilla.org/en-US/docs/Web/API/WebSocket |
|
* @see https://caniuse.com/mdn-api_websocket |
|
*/ |
|
export class WS extends BaseWS { |
|
createSocket(uri, protocols, opts) { |
|
var _a; |
|
if ((_a = this.socket) === null || _a === void 0 ? void 0 : _a._cookieJar) { |
|
opts.headers = opts.headers || {}; |
|
opts.headers.cookie = |
|
typeof opts.headers.cookie === "string" |
|
? [opts.headers.cookie] |
|
: opts.headers.cookie || []; |
|
for (const [name, cookie] of this.socket._cookieJar.cookies) { |
|
opts.headers.cookie.push(`${name}=${cookie.value}`); |
|
} |
|
} |
|
return new ws.WebSocket(uri, protocols, opts); |
|
} |
|
doWrite(packet, data) { |
|
const opts = {}; |
|
if (packet.options) { |
|
opts.compress = packet.options.compress; |
|
} |
|
if (this.opts.perMessageDeflate) { |
|
const len = |
|
// @ts-ignore |
|
"string" === typeof data ? Buffer.byteLength(data) : data.length; |
|
if (len < this.opts.perMessageDeflate.threshold) { |
|
opts.compress = false; |
|
} |
|
} |
|
this.ws.send(data, opts); |
|
} |
|
}
|
|
|