mockrtc 0.1.0

package/dist/webrtc/datachannel-stream.js

@@ -0,0 +1,86 @@
+"use strict";
+/*
+ * SPDX-FileCopyrightText: 2022 Tim Perry <tim@httptoolkit.tech>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DataChannelStream = void 0;
+const stream = require("stream");
+/**
+ * Turns a node-datachannel DataChannel into a real Node.js stream, complete with
+ * buffering, backpressure (up to a point - if the buffer fills up, messages are dropped),
+ * and support for piping data elsewhere.
+ *
+ * Read & written data may be either UTF-8 strings or Buffers - this difference exists at
+ * the protocol level, and is preserved here throughout.
+ */
+class DataChannelStream extends stream.Duplex {
+    constructor(rawChannel, streamOptions = {}) {
+        super(Object.assign(Object.assign({ allowHalfOpen: false }, streamOptions), { objectMode: true // Preserve the string/buffer distinction (WebRTC treats them differently)
+         }));
+        this.rawChannel = rawChannel;
+        this._readActive = true;
+        rawChannel.onMessage((msg) => {
+            if (!this._readActive)
+                return; // If the buffer is full, drop messages.
+            // If the push is rejected, we pause reading until the next call to _read().
+            this._readActive = this.push(msg);
+        });
+        // When the DataChannel closes, the readable & writable ends close
+        rawChannel.onClosed(() => {
+            this.push(null);
+            this.destroy();
+        });
+        rawChannel.onError((errMsg) => {
+            this.destroy(new Error(`DataChannel error: ${errMsg}`));
+        });
+        // Buffer all writes until the DataChannel opens
+        if (!rawChannel.isOpen()) {
+            this.cork();
+            rawChannel.onOpen(() => this.uncork());
+        }
+    }
+    _read() {
+        // Stop dropping messages, if the buffer filling up meant we were doing so before.
+        this._readActive = true;
+    }
+    _write(chunk, encoding, callback) {
+        let sentOk;
+        try {
+            if (Buffer.isBuffer(chunk)) {
+                sentOk = this.rawChannel.sendMessageBinary(chunk);
+            }
+            else if (typeof chunk === 'string') {
+                sentOk = this.rawChannel.sendMessage(chunk);
+            }
+            else {
+                const typeName = chunk.constructor.name || typeof chunk;
+                throw new Error(`Cannot write ${typeName} to DataChannel stream`);
+            }
+        }
+        catch (err) {
+            return callback(err);
+        }
+        if (sentOk) {
+            callback(null);
+        }
+        else {
+            callback(new Error("Failed to write to DataChannel"));
+        }
+    }
+    _final(callback) {
+        if (!this.allowHalfOpen)
+            this.destroy();
+        callback(null);
+    }
+    _destroy(maybeErr, callback) {
+        // When the stream is destroyed, we close the DataChannel.
+        this.rawChannel.close();
+        callback(maybeErr);
+    }
+    get label() {
+        return this.rawChannel.getLabel();
+    }
+}
+exports.DataChannelStream = DataChannelStream;
+//# sourceMappingURL=datachannel-stream.js.map

package/dist/webrtc/datachannel-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"datachannel-stream.js","sourceRoot":"","sources":["../../src/webrtc/datachannel-stream.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,iCAAiC;AAGjC;;;;;;;GAOG;AACH,MAAa,iBAAkB,SAAQ,MAAM,CAAC,MAAM;IAEhD,YACY,UAAuC,EAC/C,gBAKI,EAAE;QAEN,KAAK,+BACD,aAAa,EAAE,KAAK,IACjB,aAAa,KAChB,UAAU,EAAE,IAAI,CAAC,0EAA0E;YAC7F,CAAC;QAZK,eAAU,GAAV,UAAU,CAA6B;QAsC3C,gBAAW,GAAG,IAAI,CAAC;QAxBvB,UAAU,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE;YACzB,IAAI,CAAC,IAAI,CAAC,WAAW;gBAAE,OAAO,CAAC,wCAAwC;YAEvE,4EAA4E;YAC5E,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,kEAAkE;QAClE,UAAU,CAAC,QAAQ,CAAC,GAAG,EAAE;YACrB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,IAAI,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC1B,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,sBAAsB,MAAM,EAAE,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,gDAAgD;QAChD,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE;YACtB,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,UAAU,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;SAC1C;IACL,CAAC;IAGD,KAAK;QACD,kFAAkF;QAClF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC5B,CAAC;IAED,MAAM,CAAC,KAAgC,EAAE,QAAgB,EAAE,QAAuC;QAC9F,IAAI,MAAe,CAAC;QAEpB,IAAI;YACA,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;gBACxB,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;aACrD;iBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;gBAClC,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;aAC/C;iBAAM;gBACH,MAAM,QAAQ,GAAI,KAAgB,CAAC,WAAW,CAAC,IAAI,IAAI,OAAO,KAAK,CAAC;gBACpE,MAAM,IAAI,KAAK,CAAC,gBAAgB,QAAQ,wBAAwB,CAAC,CAAC;aACrE;SACJ;QAAC,OAAO,GAAQ,EAAE;YACf,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC;SACxB;QAED,IAAI,MAAM,EAAE;YACR,QAAQ,CAAC,IAAI,CAAC,CAAC;SAClB;aAAM;YACH,QAAQ,CAAC,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC,CAAC;SACzD;IACL,CAAC;IAED,MAAM,CAAC,QAAuC;QAC1C,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,IAAI,CAAC,OAAO,EAAE,CAAC;QACxC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACnB,CAAC;IAED,QAAQ,CAAC,QAAsB,EAAE,QAAuC;QACpE,0DAA0D;QAC1D,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACvB,CAAC;IAED,IAAI,KAAK;QACL,OAAO,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;IACtC,CAAC;CAEJ;AArFD,8CAqFC"}

package/dist/webrtc/mediatrack-stream.d.ts

@@ -0,0 +1,29 @@
+/// <reference types="node" />
+import * as stream from 'stream';
+import * as NodeDataChannel from 'node-datachannel';
+/**
+ * Turns a node-datachannel media track into a real Node.js stream, complete with
+ * buffering, backpressure (up to a point - if the buffer fills up, messages are dropped),
+ * and support for piping data elsewhere.
+ */
+export declare class MediaTrackStream extends stream.Duplex {
+    private rawTrack;
+    constructor(rawTrack: NodeDataChannel.Track, streamOptions?: {
+        readableHighWaterMark?: number | undefined;
+        writableHighWaterMark?: number | undefined;
+        allowHalfOpen?: boolean;
+    });
+    private close;
+    private _readActive;
+    _read(): void;
+    _write(chunk: Buffer, _encoding: string, callback: (error: Error | null) => void): void;
+    _writev(chunks: Array<{
+        chunk: any;
+        encoding: BufferEncoding;
+    }>, callback: (error?: Error | null) => void): void;
+    _final(callback: (error: Error | null) => void): void;
+    _destroy(maybeErr: Error | null, callback: (error: Error | null) => void): void;
+    get direction(): NodeDataChannel.Direction;
+    get mid(): string;
+    get type(): string;
+}

package/dist/webrtc/mediatrack-stream.js

@@ -0,0 +1,109 @@
+"use strict";
+/*
+ * SPDX-FileCopyrightText: 2022 Tim Perry <tim@httptoolkit.tech>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MediaTrackStream = void 0;
+const stream = require("stream");
+const NodeDataChannel = require("node-datachannel");
+const { Direction } = NodeDataChannel;
+/**
+ * Turns a node-datachannel media track into a real Node.js stream, complete with
+ * buffering, backpressure (up to a point - if the buffer fills up, messages are dropped),
+ * and support for piping data elsewhere.
+ */
+class MediaTrackStream extends stream.Duplex {
+    constructor(rawTrack, streamOptions = {}) {
+        super(Object.assign({ allowHalfOpen: false }, streamOptions));
+        this.rawTrack = rawTrack;
+        this._readActive = true;
+        rawTrack.onMessage((msg) => {
+            if (!this._readActive)
+                return; // If the buffer is full, drop messages.
+            // If the push is rejected, we pause reading until the next call to _read().
+            this._readActive = this.push(msg);
+        });
+        // When the DataChannel closes, the readable & writable ends close
+        rawTrack.onClosed(() => this.close());
+        rawTrack.onError((errMsg) => {
+            this.destroy(new Error(`Media track error: ${errMsg}`));
+        });
+        // Buffer all writes until the DataChannel opens
+        if (!rawTrack.isOpen()) {
+            this.cork();
+            rawTrack.onOpen(() => this.uncork());
+        }
+    }
+    close() {
+        this.push(null);
+        this.destroy();
+    }
+    _read() {
+        // Stop dropping messages, if the buffer filling up meant we were doing so before.
+        this._readActive = true;
+    }
+    _write(chunk, _encoding, callback) {
+        let sentOk;
+        if (this.rawTrack.isClosed()) {
+            // isClosed becomes true and writes start failing just before onClosed() fires, so here we
+            // drop pending writes as soon as we notice.
+            this.close();
+            return;
+        }
+        try {
+            sentOk = this.rawTrack.sendMessageBinary(chunk);
+        }
+        catch (err) {
+            return callback(err);
+        }
+        if (sentOk) {
+            callback(null);
+        }
+        else {
+            callback(new Error("Failed to write to media track"));
+        }
+    }
+    _writev(chunks, callback) {
+        let sentOk;
+        if (this.rawTrack.isClosed()) {
+            // isClosed becomes true and writes start failing just before onClosed() fires, so here we
+            // drop pending writes as soon as we notice.
+            this.close();
+            return;
+        }
+        try {
+            sentOk = this.rawTrack.sendMessageBinary(Buffer.concat(chunks.map(c => c.chunk)));
+        }
+        catch (err) {
+            return callback(err);
+        }
+        if (sentOk) {
+            callback(null);
+        }
+        else {
+            callback(new Error("Failed to write to media track"));
+        }
+    }
+    _final(callback) {
+        if (!this.allowHalfOpen)
+            this.destroy();
+        callback(null);
+    }
+    _destroy(maybeErr, callback) {
+        // When the stream is destroyed, we close the DataChannel.
+        this.rawTrack.close();
+        callback(maybeErr);
+    }
+    get direction() {
+        return this.rawTrack.direction();
+    }
+    get mid() {
+        return this.rawTrack.mid();
+    }
+    get type() {
+        return this.rawTrack.type();
+    }
+}
+exports.MediaTrackStream = MediaTrackStream;
+//# sourceMappingURL=mediatrack-stream.js.map

package/dist/webrtc/mediatrack-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mediatrack-stream.js","sourceRoot":"","sources":["../../src/webrtc/mediatrack-stream.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,iCAAiC;AACjC,oDAAoD;AAEpD,MAAM,EAAE,SAAS,EAAE,GAAG,eAAe,CAAC;AAEtC;;;;GAIG;AACH,MAAa,gBAAiB,SAAQ,MAAM,CAAC,MAAM;IAE/C,YACY,QAA+B,EACvC,gBAKI,EAAE;QAEN,KAAK,iBACD,aAAa,EAAE,KAAK,IACjB,aAAa,EAClB,CAAC;QAXK,aAAQ,GAAR,QAAQ,CAAuB;QAuCnC,gBAAW,GAAG,IAAI,CAAC;QA1BvB,QAAQ,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE;YACvB,IAAI,CAAC,IAAI,CAAC,WAAW;gBAAE,OAAO,CAAC,wCAAwC;YAEvE,4EAA4E;YAC5E,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,kEAAkE;QAClE,QAAQ,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QAEtC,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACxB,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,sBAAsB,MAAM,EAAE,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,gDAAgD;QAChD,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE;YACpB,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;SACxC;IACL,CAAC;IAEO,KAAK;QACT,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,IAAI,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;IAGD,KAAK;QACD,kFAAkF;QAClF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC5B,CAAC;IAED,MAAM,CAAC,KAAa,EAAE,SAAiB,EAAE,QAAuC;QAC5E,IAAI,MAAe,CAAC;QAEpB,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE;YAC1B,0FAA0F;YAC1F,4CAA4C;YAC5C,IAAI,CAAC,KAAK,EAAE,CAAC;YACb,OAAO;SACV;QAED,IAAI;YACA,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;SACnD;QAAC,OAAO,GAAQ,EAAE;YACf,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC;SACxB;QAED,IAAI,MAAM,EAAE;YACR,QAAQ,CAAC,IAAI,CAAC,CAAC;SAClB;aAAM;YACH,QAAQ,CAAC,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC,CAAC;SACzD;IACL,CAAC;IAED,OAAO,CAAC,MAAwD,EAAE,QAAwC;QACtG,IAAI,MAAe,CAAC;QAEpB,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE;YAC1B,0FAA0F;YAC1F,4CAA4C;YAC5C,IAAI,CAAC,KAAK,EAAE,CAAC;YACb,OAAO;SACV;QAED,IAAI;YACA,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAC1C,CAAC;SACL;QAAC,OAAO,GAAQ,EAAE;YACf,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC;SACxB;QAED,IAAI,MAAM,EAAE;YACR,QAAQ,CAAC,IAAI,CAAC,CAAC;SAClB;aAAM;YACH,QAAQ,CAAC,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC,CAAC;SACzD;IACL,CAAC;IAED,MAAM,CAAC,QAAuC;QAC1C,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,IAAI,CAAC,OAAO,EAAE,CAAC;QACxC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACnB,CAAC;IAED,QAAQ,CAAC,QAAsB,EAAE,QAAuC;QACpE,0DAA0D;QAC1D,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACvB,CAAC;IAED,IAAI,SAAS;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;IACrC,CAAC;IAED,IAAI,GAAG;QACH,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;IAC/B,CAAC;IAED,IAAI,IAAI;QACJ,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IAChC,CAAC;CAEJ;AAvHD,4CAuHC"}

package/dist/webrtc/mockrtc-connection.d.ts

@@ -0,0 +1,14 @@
+import * as NodeDataChannel from 'node-datachannel';
+import { DataChannelStream } from './datachannel-stream';
+import { RTCConnection } from './rtc-connection';
+export declare class MockRTCConnection extends RTCConnection {
+    private getExternalConnection;
+    private controlChannel;
+    private externalConnection;
+    constructor(getExternalConnection: (id: string) => RTCConnection);
+    protected trackNewChannel(channel: NodeDataChannel.DataChannel, options: {
+        isLocal: boolean;
+    }): DataChannelStream;
+    proxyTrafficToExternalConnection(): Promise<void>;
+    proxyTrafficTo(externalConnection: RTCConnection): void;
+}

package/dist/webrtc/mockrtc-connection.js

@@ -0,0 +1,147 @@
+"use strict";
+/*
+ * SPDX-FileCopyrightText: 2022 Tim Perry <tim@httptoolkit.tech>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MockRTCConnection = void 0;
+const control_channel_1 = require("./control-channel");
+const datachannel_stream_1 = require("./datachannel-stream");
+const rtc_connection_1 = require("./rtc-connection");
+class MockRTCConnection extends rtc_connection_1.RTCConnection {
+    constructor(getExternalConnection) {
+        super();
+        this.getExternalConnection = getExternalConnection;
+    }
+    trackNewChannel(channel, options) {
+        if (channel.getLabel() === control_channel_1.MOCKRTC_CONTROL_CHANNEL && !options.isLocal) {
+            // We don't track the control channel like other channels - we handle it specially.
+            if (this.controlChannel) {
+                const error = new Error('Cannot open multiple control channels simultaneously');
+                channel.sendMessage(JSON.stringify({
+                    type: 'error',
+                    error: error.message
+                }));
+                setTimeout(() => channel.close(), 100);
+                throw error;
+            }
+            this.controlChannel = new datachannel_stream_1.DataChannelStream(channel);
+            this.controlChannel.on('data', (msg) => {
+                var _a;
+                try {
+                    const controlMessage = JSON.parse(msg);
+                    if (controlMessage.type === 'attach-external') {
+                        if (this.externalConnection) {
+                            throw new Error('Cannot attach mock connection to multiple external connections');
+                        }
+                        this.externalConnection = this.getExternalConnection(controlMessage.id);
+                        this.emit('external-connection-attached');
+                        // We don't necessarily proxy traffic through to the external connection at this point,
+                        // that depends on the specific handling that's used here.
+                    }
+                    else {
+                        throw new Error(`Unrecognized control channel message: ${controlMessage.type}`);
+                    }
+                }
+                catch (e) {
+                    console.warn("Failed to handle control channel message", e);
+                    (_a = this.controlChannel) === null || _a === void 0 ? void 0 : _a.write(JSON.stringify({
+                        type: 'error',
+                        error: e.message || e
+                    }));
+                }
+            });
+            this.controlChannel.on('close', () => {
+                this.controlChannel = undefined;
+            });
+            this.controlChannel.on('error', (error) => {
+                console.error('Control channel error:', error);
+            });
+            return this.controlChannel;
+        }
+        else {
+            return super.trackNewChannel(channel, options);
+        }
+    }
+    proxyTrafficToExternalConnection() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!this.externalConnection) {
+                yield new Promise((resolve) => this.once('external-connection-attached', resolve));
+            }
+            this.proxyTrafficTo(this.externalConnection);
+        });
+    }
+    proxyTrafficTo(externalConnection) {
+        /**
+         * When proxying traffic, you effectively have four peers, each with a connection endpoint:
+         * - The incoming RTCPeerConnection that we're mocking ('internal')
+         * - This MockRTC connection, with an associated MockRTCPeer that it will actually connect to ('mock')
+         * - A MockRTC external connection that will connect to the remote peer ('external')
+         * - The original remote peer that we're connecting to ('remote')
+         *
+         * Once the proxy is set up, the the connection structure works like so:
+         * INTERNAL <--> MOCK <--> EXTERNAL <--> REMOTE
+         *
+         * Here we connect the internal & external connections together, proxying all behaviours between the
+         * two so that from this point forwards every event on one is reflected on the other.
+         *
+         * Note that this isn't necessarily the initialization of either connection: the remote peer could
+         * have been connected for a while (sending data with no response), and the internal peer could have
+         * been fully interacting with steps before this point.
+         */
+        // Mirror connection closure:
+        this.on('connection-closed', () => externalConnection.close());
+        externalConnection.on('connection-closed', () => this.close());
+        /// --- Data channels: --- ///
+        // Forward *all* existing internal channels to the external connection:
+        this.channels.forEach((channel) => {
+            const mirrorChannelStream = externalConnection.createDataChannel(channel.label);
+            channel.pipe(mirrorChannelStream).pipe(channel);
+        });
+        // Forward any existing external channels back to this peer connection. Note that we're mirroring
+        // *remote* channels only, so we skip the channels that we've just created above.
+        externalConnection.remoteChannels.forEach((channel) => {
+            const mirrorChannelStream = this.createDataChannel(channel.label);
+            channel.pipe(mirrorChannelStream).pipe(channel);
+        });
+        // If any new channels open in future, mirror them to the other peer:
+        [[this, externalConnection], [externalConnection, this]].forEach(([connA, connB]) => {
+            connA.on('remote-channel-open', (incomingChannel) => {
+                const mirrorChannelStream = connB.createDataChannel(incomingChannel.label);
+                incomingChannel.pipe(mirrorChannelStream).pipe(incomingChannel);
+            });
+        });
+        /// --- Media tracks: --- ///
+        // Note that while data channels will *not* have been negotiated before this point, so
+        // we can always assume that mock data channels need mirroring, media tracks are negotiated
+        // in the SDP, not in-band, and so any media track could already exist on the other side.
+        // For each track on the internal connection, proxy it to the corresponding external track:
+        this.mediaTracks.forEach((track) => {
+            const externalStream = externalConnection.mediaTracks.find(({ mid }) => mid === track.mid);
+            if (externalStream) {
+                if (externalStream.type === track.type) {
+                    track.pipe(externalStream).pipe(track);
+                }
+                else {
+                    throw new Error(`Mock & external streams with mid ${track.mid} have mismatched types (${track.type}/${externalStream.type})`);
+                }
+            }
+            else {
+                // A mismatch in media streams means the external & mock peer negotiation isn't in sync!
+                // For now we just reject this case - later we should try to prompt a renegotiation.
+                throw new Error(`Mock has ${track.type} ${track.mid} but external does not`);
+            }
+        });
+    }
+}
+exports.MockRTCConnection = MockRTCConnection;
+//# sourceMappingURL=mockrtc-connection.js.map

package/dist/webrtc/mockrtc-connection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mockrtc-connection.js","sourceRoot":"","sources":["../../src/webrtc/mockrtc-connection.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;;;;;;;AAIH,uDAAmF;AAEnF,6DAAyD;AAEzD,qDAAiD;AAEjD,MAAa,iBAAkB,SAAQ,8BAAa;IAOhD,YACY,qBAAoD;QAE5D,KAAK,EAAE,CAAC;QAFA,0BAAqB,GAArB,qBAAqB,CAA+B;IAGhE,CAAC;IAES,eAAe,CAAC,OAAoC,EAAE,OAA6B;QACzF,IAAI,OAAO,CAAC,QAAQ,EAAE,KAAK,yCAAuB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;YACpE,mFAAmF;YACnF,IAAI,IAAI,CAAC,cAAc,EAAE;gBACrB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;gBAChF,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC;oBAC/B,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,KAAK,CAAC,OAAO;iBACvB,CAAC,CAAC,CAAC;gBACJ,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,GAAG,CAAC,CAAC;gBACvC,MAAM,KAAK,CAAC;aACf;YAED,IAAI,CAAC,cAAc,GAAG,IAAI,sCAAiB,CAAC,OAAO,CAAC,CAAC;YAErD,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE;;gBACnC,IAAI;oBACA,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA0B,CAAC;oBAEhE,IAAI,cAAc,CAAC,IAAI,KAAK,iBAAiB,EAAE;wBAC3C,IAAI,IAAI,CAAC,kBAAkB,EAAE;4BACzB,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;yBACrF;wBACD,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;wBAExE,IAAI,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;wBAE1C,uFAAuF;wBACvF,0DAA0D;qBAC7D;yBAAM;wBACH,MAAM,IAAI,KAAK,CAAC,yCAAyC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;qBACnF;iBACJ;gBAAC,OAAO,CAAM,EAAE;oBACb,OAAO,CAAC,IAAI,CAAC,0CAA0C,EAAE,CAAC,CAAC,CAAC;oBAC5D,MAAA,IAAI,CAAC,cAAc,0CAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;wBACtC,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,CAAC,OAAO,IAAI,CAAC;qBACxB,CAAC,CAAC,CAAC;iBACP;YACL,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACjC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YACpC,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACtC,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YACnD,CAAC,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC,cAAe,CAAC;SAC/B;aAAM;YACH,OAAO,KAAK,CAAC,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;SAClD;IACL,CAAC;IAEK,gCAAgC;;YAClC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE;gBAC1B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,8BAA8B,EAAE,OAAO,CAAC,CAAC,CAAC;aACtF;YAED,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,kBAAmB,CAAC,CAAC;QAClD,CAAC;KAAA;IAED,cAAc,CAAC,kBAAiC;QAC5C;;;;;;;;;;;;;;;;WAgBG;QAGH,6BAA6B;QAC7B,IAAI,CAAC,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC,CAAC;QAC/D,kBAAkB,CAAC,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QAE/D,8BAA8B;QAE9B,uEAAuE;QACvE,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAA0B,EAAE,EAAE;YACjD,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAChF,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,iGAAiG;QACjG,iFAAiF;QACjF,kBAAkB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,OAA0B,EAAE,EAAE;YACrE,MAAM,mBAAmB,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAClE,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,qEAAqE;QACrE,CAAC,CAAC,IAAI,EAAE,kBAAkB,CAAC,EAAE,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE;YAChF,KAAK,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,eAAkC,EAAE,EAAE;gBACnE,MAAM,mBAAmB,GAAG,KAAK,CAAC,iBAAiB,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC3E,eAAe,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpE,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,6BAA6B;QAE7B,sFAAsF;QACtF,2FAA2F;QAC3F,yFAAyF;QAEzF,2FAA2F;QAC3F,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,KAAuB,EAAE,EAAE;YACjD,MAAM,cAAc,GAAG,kBAAkB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,KAAK,KAAK,CAAC,GAAG,CAAC,CAAC;YAC3F,IAAI,cAAc,EAAE;gBAChB,IAAI,cAAc,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,EAAE;oBACpC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iBAC1C;qBAAM;oBACH,MAAM,IAAI,KAAK,CAAC,oCAAoC,KAAK,CAAC,GAAG,2BACzD,KAAK,CAAC,IACV,IACI,cAAc,CAAC,IACnB,GAAG,CAAC,CAAC;iBACR;aACJ;iBAAM;gBACH,wFAAwF;gBACxF,oFAAoF;gBACpF,MAAM,IAAI,KAAK,CAAC,YAAY,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,GAAG,wBAAwB,CAAC,CAAC;aAChF;QACL,CAAC,CAAC,CAAC;IACP,CAAC;CAEJ;AAtJD,8CAsJC"}

package/dist/webrtc/peer-connection.d.ts

@@ -0,0 +1,16 @@
+/// <reference types="node" />
+import { EventEmitter } from 'events';
+import { DataChannelStream } from './datachannel-stream';
+export declare class MockRTCPeerConnection extends EventEmitter {
+    private rawConn;
+    readonly channels: Array<DataChannelStream>;
+    constructor();
+    setRemoteDescription(description: RTCSessionDescriptionInit): void;
+    /**
+     * Gets the local description for this connection, waiting until gathering is complete to provide a
+     * full result. Because this waits for gathering, it will not resolve if no DataChannel, other
+     * tracks or remote description have been provided beforehand.
+     */
+    getLocalDescription(): Promise<RTCSessionDescriptionInit>;
+    close(): Promise<void>;
+}

package/dist/webrtc/peer-connection.js

@@ -0,0 +1,81 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MockRTCPeerConnection = void 0;
+const events_1 = require("events");
+const NodeDataChannel = require("node-datachannel");
+const datachannel_stream_1 = require("./datachannel-stream");
+class MockRTCPeerConnection extends events_1.EventEmitter {
+    constructor() {
+        super();
+        this.rawConn = new NodeDataChannel.PeerConnection("MockRTCPeer", { iceServers: [] });
+        this.channels = [];
+        this.rawConn.onDataChannel((channel) => {
+            const channelStream = new datachannel_stream_1.DataChannelStream(channel);
+            this.channels.push(channelStream);
+            channel.onClosed(() => {
+                const channelIndex = this.channels.findIndex(c => c === channelStream);
+                if (channelIndex !== -1) {
+                    this.channels.splice(channelIndex, 1);
+                }
+            });
+            channelStream.on('error', (error) => {
+                console.error('Channel error:', error);
+            });
+            this.emit('channel-open', channelStream);
+        });
+    }
+    setRemoteDescription(description) {
+        const { type: offerType, sdp: offerSdp } = description;
+        if (!offerSdp)
+            throw new Error("Cannot set MockRTC peer description without providing an SDP");
+        this.rawConn.setRemoteDescription(offerSdp, offerType[0].toUpperCase() + offerType.slice(1));
+    }
+    /**
+     * Gets the local description for this connection, waiting until gathering is complete to provide a
+     * full result. Because this waits for gathering, it will not resolve if no DataChannel, other
+     * tracks or remote description have been provided beforehand.
+     */
+    getLocalDescription() {
+        return __awaiter(this, void 0, void 0, function* () {
+            // TODO: Detect offer vs answer case here.
+            const setupChannel = this.rawConn.createDataChannel('mockttp.setup.channel');
+            yield new Promise((resolve) => {
+                this.rawConn.onGatheringStateChange((state) => {
+                    if (state === 'complete')
+                        resolve();
+                });
+                // Handle race conditions where gathering has already completed
+                if (this.rawConn.gatheringState() === 'complete')
+                    resolve();
+            });
+            const sessionDescription = this.rawConn.localDescription();
+            setupChannel.close(); // TODO: Does this work? Goal is to get a channel without actually sending open msg.
+            return sessionDescription;
+        });
+    }
+    close() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const closedPromise = new Promise((resolve) => {
+                if (this.rawConn.state() === 'closed')
+                    resolve();
+                this.rawConn.onStateChange((state) => {
+                    if (state === 'closed')
+                        resolve();
+                });
+            });
+            this.rawConn.close();
+            yield closedPromise;
+        });
+    }
+}
+exports.MockRTCPeerConnection = MockRTCPeerConnection;
+//# sourceMappingURL=peer-connection.js.map

package/dist/webrtc/peer-connection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"peer-connection.js","sourceRoot":"","sources":["../../src/webrtc/peer-connection.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,mCAAsC;AACtC,oDAAoD;AAEpD,6DAAyD;AAEzD,MAAa,qBAAsB,SAAQ,qBAAY;IAMnD;QACI,KAAK,EAAE,CAAC;QALJ,YAAO,GAAG,IAAI,eAAe,CAAC,cAAc,CAAC,aAAa,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QAExE,aAAQ,GAA6B,EAAE,CAAC;QAKpD,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,EAAE;YACnC,MAAM,aAAa,GAAG,IAAI,sCAAiB,CAAC,OAAO,CAAC,CAAC;YACrD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAElC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;gBAClB,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,aAAa,CAAC,CAAC;gBACvE,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE;oBACrB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;iBACzC;YACL,CAAC,CAAC,CAAC;YAEH,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAChC,OAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACP,CAAC;IAED,oBAAoB,CAAC,WAAsC;QACvD,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,WAAW,CAAC;QACvD,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;QAC/F,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAQ,CAAC,CAAC;IACxG,CAAC;IAED;;;;OAIG;IACG,mBAAmB;;YACrB,0CAA0C;YAC1C,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,uBAAuB,CAAC,CAAC;YAE7E,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;gBAChC,IAAI,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,KAAK,EAAE,EAAE;oBAC1C,IAAI,KAAK,KAAK,UAAU;wBAAE,OAAO,EAAE,CAAC;gBACxC,CAAC,CAAC,CAAC;gBAEH,+DAA+D;gBAC/D,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,UAAU;oBAAE,OAAO,EAAE,CAAC;YAChE,CAAC,CAAC,CAAC;YAEH,MAAM,kBAAkB,GAAG,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAA+B,CAAC;YACxF,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,oFAAoF;YAC1G,OAAO,kBAAkB,CAAC;QAC9B,CAAC;KAAA;IAEK,KAAK;;YACP,MAAM,aAAa,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;gBAChD,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,QAAQ;oBAAE,OAAO,EAAE,CAAC;gBAEjD,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,KAAK,EAAE,EAAE;oBACjC,IAAI,KAAK,KAAK,QAAQ;wBAAE,OAAO,EAAE,CAAC;gBACtC,CAAC,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACrB,MAAM,aAAa,CAAC;QACxB,CAAC;KAAA;CAEJ;AAtED,sDAsEC"}

package/dist/webrtc/rtc-connection.d.ts

@@ -0,0 +1,47 @@
+/// <reference types="node" />
+import { EventEmitter } from 'events';
+import * as NodeDataChannel from 'node-datachannel';
+import { MockRTCSession } from '../mockrtc-peer';
+import { DataChannelStream } from './datachannel-stream';
+import { MediaTrackStream } from './mediatrack-stream';
+/**
+ * An RTC connection is a single connection. This base class defines the raw connection management and
+ * tracking logic for a generic connection. The MockRTCConnection subclass extends this and adds
+ * logic to support control channels, proxying and other MockRTC-specific additions.
+ */
+export declare class RTCConnection extends EventEmitter {
+    readonly id: string;
+    private rawConn;
+    private remoteDescription;
+    private readonly trackedChannels;
+    get channels(): ReadonlyArray<DataChannelStream>;
+    get localChannels(): ReadonlyArray<DataChannelStream>;
+    get remoteChannels(): ReadonlyArray<DataChannelStream>;
+    private readonly trackedMediaTracks;
+    get mediaTracks(): ReadonlyArray<MediaTrackStream>;
+    get localMediaTracks(): ReadonlyArray<MediaTrackStream>;
+    get remoteMediaTracks(): ReadonlyArray<MediaTrackStream>;
+    constructor();
+    createDataChannel(label: string): DataChannelStream;
+    protected trackNewChannel(channel: NodeDataChannel.DataChannel, options: {
+        isLocal: boolean;
+    }): DataChannelStream;
+    protected trackNewMediaTrack(track: NodeDataChannel.Track, options: {
+        isLocal: boolean;
+    }): MediaTrackStream;
+    setRemoteDescription(description: RTCSessionDescriptionInit): void;
+    /**
+     * Gets the local description for this connection, waiting until gathering is complete to provide a
+     * full result. Because this waits for gathering, it will not resolve if no DataChannel, other
+     * tracks or remote description have been provided beforehand.
+     */
+    getLocalDescription(): Promise<RTCSessionDescriptionInit>;
+    getRemoteDescription(): RTCSessionDescriptionInit | undefined;
+    getMirroredLocalOffer(sdpToMirror: string, options?: {
+        addDataStream?: boolean;
+    }): Promise<RTCSessionDescriptionInit>;
+    getMirroredLocalAnswer(sdpToMirror: string): Promise<RTCSessionDescriptionInit>;
+    waitUntilConnected(): Promise<void>;
+    readonly sessionApi: MockRTCSession;
+    close(): Promise<void>;
+}