*add createDefaultMessage option to disable the default socket pipe

This commit is contained in:
Derrick Hammer 2023-03-15 08:14:07 -04:00
parent 27261fedd2
commit c1d495a54b
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
2 changed files with 22 additions and 10 deletions

View File

@ -38,6 +38,7 @@ export interface DataSocketOptions {
onclose?: OnClose;
onchannel?: OnChannel;
emulateWebsocket?: boolean;
createDefaultMessage?: boolean;
}
export interface PeerOptions {
@ -59,6 +60,7 @@ export default class Peer {
private _onclose: OnCloseBound;
private _onchannel: OnChannelBound;
private _emulateWebsocket: boolean;
private _createDefaultMessage: boolean;
constructor({
proxy,
@ -70,6 +72,7 @@ export default class Peer {
onclose,
onchannel,
emulateWebsocket = false,
createDefaultMessage = true,
}: PeerOptionsWithProxy & DataSocketOptions) {
this._proxy = proxy;
this._peer = peer;
@ -80,6 +83,7 @@ export default class Peer {
this._onclose = onclose?.bind(undefined, this);
this._onchannel = onchannel?.bind(undefined, this);
this._emulateWebsocket = emulateWebsocket;
this._createDefaultMessage = createDefaultMessage;
}
private _socket?: Socket;
@ -96,12 +100,15 @@ export default class Peer {
async init() {
const self = this;
let pipe;
this._socket = new Socket({
remoteAddress: self._peer.rawStream.remoteHost,
remotePort: self._peer.rawStream.remotePort,
remotePublicKey: self._peer.remotePublicKey,
async write(data: any, cb: Function) {
pipe.send(data);
if (pipe) {
pipe.send(data);
}
await self._onsend?.(data);
cb();
},
@ -133,15 +140,18 @@ export default class Peer {
await self._onclose?.(self._socket);
},
});
const pipe = this._channel.addMessage({
async onmessage(m: any) {
if (m instanceof Uint8Array) {
m = Buffer.from(m);
}
self._socket.emit("data", m);
await self._onreceive?.(m);
},
});
if (this._createDefaultMessage) {
pipe = this._channel.addMessage({
async onmessage(m: any) {
if (m instanceof Uint8Array) {
m = Buffer.from(m);
}
self._socket.emit("data", m);
await self._onreceive?.(m);
},
});
}
await this._onchannel?.(this._channel);
await this._channel.open();

View File

@ -24,6 +24,7 @@ export default class Proxy {
listen = false,
autostart = false,
emulateWebsocket = false,
createDefaultMessage = true,
}: ProxyOptions) {
this._swarm = swarm;
this._protocol = protocol;
@ -36,6 +37,7 @@ export default class Proxy {
onclose,
onchannel,
emulateWebsocket,
createDefaultMessage,
};
this.init();
}