49 lines
1.1 KiB
TypeScript
49 lines
1.1 KiB
TypeScript
import { Duplex, Callback } from "streamx";
|
|
interface SocketOptions {
|
|
allowHalfOpen?: boolean;
|
|
remoteAddress?: string;
|
|
remotePort?: number;
|
|
cb?: Function;
|
|
}
|
|
|
|
export default class Socket extends Duplex {
|
|
private _ws: WebSocket;
|
|
|
|
constructor({ remoteAddress, remotePort, cb }: SocketOptions = {}) {
|
|
super();
|
|
this._ws = new WebSocket(`wss://${remoteAddress}/webirc`);
|
|
|
|
["message", "end", "close", "error"].forEach((event) => {
|
|
this._ws.addEventListener(event, (...args) => {
|
|
let sendEvent = event;
|
|
let sendData = args;
|
|
|
|
if (event === "message") {
|
|
sendEvent = "data";
|
|
sendData[0] = (args[0] as MessageEvent).data + "\n" as any;
|
|
}
|
|
// @ts-ignore
|
|
return this.emit(sendEvent, ...sendData);
|
|
});
|
|
this._ws.addEventListener("open", cb as any);
|
|
});
|
|
}
|
|
_write(data: any, cb: Callback) {
|
|
this._ws.send(data);
|
|
cb();
|
|
}
|
|
setEncoding() {}
|
|
}
|
|
export function connect(
|
|
port: number,
|
|
host: string,
|
|
opts = {},
|
|
cb: Function
|
|
): Socket {
|
|
return new Socket({
|
|
remotePort: port,
|
|
remoteAddress: host,
|
|
cb,
|
|
});
|
|
}
|