62 lines
1.6 KiB
TypeScript
62 lines
1.6 KiB
TypeScript
|
import BasePeer from "../../peer.js";
|
||
|
|
import { maybeGetAsyncProperty } from "../../util.js";
|
||
|
|
import Socket from "../../socket.js";
|
||
|
|
import { Buffer } from "buffer";
|
||
|
|
|
||
|
|
export default class Peer extends BasePeer {
|
||
|
|
private _pipe?: any;
|
||
|
|
protected async initSocket() {
|
||
|
|
const self = this;
|
||
|
|
|
||
|
|
const raw = await maybeGetAsyncProperty(self._peer.rawStream);
|
||
|
|
this._socket = new Socket({
|
||
|
|
remoteAddress: raw.remoteHost,
|
||
|
|
remotePort: raw.remotePort,
|
||
|
|
remotePublicKey: await maybeGetAsyncProperty(self._peer.remotePublicKey),
|
||
|
|
async write(data: any, cb: Function) {
|
||
|
|
self._pipe?.send(data);
|
||
|
|
await self._onsend?.(data);
|
||
|
|
cb();
|
||
|
|
},
|
||
|
|
emulateWebsocket: self._emulateWebsocket,
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
protected async handleChannelOnOpen(m: any) {
|
||
|
|
if (!m) {
|
||
|
|
m = Buffer.from([]);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (m instanceof Uint8Array) {
|
||
|
|
m = Buffer.from(m);
|
||
|
|
}
|
||
|
|
|
||
|
|
this._socket?.on("end", () => this._channel.close());
|
||
|
|
let ret = await this._onopen?.(this._socket, m);
|
||
|
|
if (!ret || (ret && ret.connect === false)) {
|
||
|
|
// @ts-ignore
|
||
|
|
self._socket?.emit("connect");
|
||
|
|
}
|
||
|
|
|
||
|
|
this._socket?.emit("data", m);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected async handleChannelOnClose(socket: Socket): Promise<void> {
|
||
|
|
this._socket?.destroy();
|
||
|
|
await this._onclose?.(this._socket);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected async initMessages(): Promise<void> {
|
||
|
|
const self = this;
|
||
|
|
this._pipe = await this._channel.addMessage({
|
||
|
|
async onmessage(m: any) {
|
||
|
|
if (m instanceof Uint8Array) {
|
||
|
|
m = Buffer.from(m);
|
||
|
|
}
|
||
|
|
self._socket.emit("data", m);
|
||
|
|
await self._onreceive?.(m);
|
||
|
|
},
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|