add hook onOffer and onAnswer

binbat · Mar 29, 2024 · fa15949 · fa15949
1 parent 13e211a
commit fa15949
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 40 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
-	"name": "whip-whep",
-	"version": "1.1.1",
+	"name": "@binbat/whip-whep",
+	"version": "1.1.1-sdp",
 	"description": "WHIP and WHEP clients",
 	"scripts": {
 		"precommit": "npm run-script lint",

diff --git a/whep.js b/whep.js
@@ -21,6 +21,9 @@ export class WHEPClient extends EventTarget
 		//Pending candidadtes
 		this.candidates = [];
 		this.endOfcandidates = false;
+
+		this.onOffer = offer => offer;
+		this.onAnswer = answer => answer;
 	}
 
 	async view(pc, url, token)
@@ -33,24 +36,6 @@ export class WHEPClient extends EventTarget
 		this.token = token;
 		this.pc = pc;
 
-		//Listen for state change events
-		pc.onconnectionstatechange = (event) =>
-		{
-			switch (pc.connectionState)
-			{
-				case "connected":
-					// The connection has become fully connected
-					break;
-				case "disconnected":
-				case "failed":
-					// One or more transports has terminated unexpectedly or in an error
-					break;
-				case "closed":
-					// The connection has been closed
-					break;
-			}
-		}
-
 		//Listen for candidates
 		pc.onicecandidate = (event) =>
 		{
@@ -73,6 +58,7 @@ export class WHEPClient extends EventTarget
 		}
 		//Create SDP offer
 		const offer = await pc.createOffer();
+		offer.sdp = this.onOffer(offer.sdp);
 
 		//Request headers
 		const headers = {
@@ -261,7 +247,7 @@ export class WHEPClient extends EventTarget
 		//}
 
 		//And set remote description
-		await pc.setRemoteDescription({ type: "answer", sdp: answer });
+		await pc.setRemoteDescription({ type: "answer", sdp: this.onAnswer(answer) });
 	}
 
 	async restart()

diff --git a/whip.js b/whip.js
@@ -12,6 +12,9 @@ export class WHIPClient
 		//Pending candidadtes
 		this.candidates = [];
 		this.endOfcandidates = false;
+
+		this.onOffer = offer => offer;
+		this.onAnswer = answer => answer;
 	}
 
 	async publish(pc, url, token)
@@ -24,24 +27,6 @@ export class WHIPClient
 		this.token = token;
 		this.pc = pc;
 
-		//Listen for state change events
-		pc.onconnectionstatechange = (event) =>
-		{
-			switch (pc.connectionState)
-			{
-				case "connected":
-					// The connection has become fully connected
-					break;
-				case "disconnected":
-				case "failed":
-					// One or more transports has terminated unexpectedly or in an error
-					break;
-				case "closed":
-					// The connection has been closed
-					break;
-			}
-		}
-
 		//Listen for candidates
 		pc.onicecandidate = (event) =>
 		{
@@ -64,6 +49,7 @@ export class WHIPClient
 		}
 		//Create SDP offer
 		const offer = await pc.createOffer();
+		offer.sdp = this.onOffer(offer.sdp);
 
 		//Request headers
 		const headers = {
@@ -210,7 +196,7 @@ export class WHIPClient
 		//}
 
 		//And set remote description
-		await pc.setRemoteDescription({ type: "answer", sdp: answer });
+		await pc.setRemoteDescription({ type: "answer", sdp: this.onAnswer(answer) });
 	}
 
 	async restart()