Set frame url to either localhost or github pages

tonyherre · Jun 3, 2022 · 75cdf8a · 75cdf8a
1 parent 9842340
commit 75cdf8a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/index.html b/index.html
@@ -1,6 +1,6 @@
 <html>
   <a id="status"></a>
-  <iframe src="http://localhost:8001/frame.html" id="subframe" style="display: none;" allow="display-capture;camera"></iframe>
+  <iframe id="subframe" style="display: none;" allow="display-capture;camera"></iframe>
   <input type="button" id="sendPort" onclick="sendPort()" value="send port"/>
   <input type="button" id="startTrack" onclick="startTrack()" value="Start track"/>
   <input type="button" id="transfer" onclick="transferTrack()" value="Transfer track"/>

diff --git a/main.js b/main.js
@@ -7,6 +7,10 @@ startTrackBtn.disabled = true;
 let transferBtn = document.getElementById("transfer");
 transferBtn.disabled = true;
 
+const frameOrigin =
+  document.location.host == "localhost:8000" ? "http://localhost:8001" : "https://tonyherre.github.io/mst-transfer-sample";
+
+document.getElementById("subframe").src = frameOrigin + "/frame.html";
 
 function isTrackTransferSupported() {
   let track = document.createElement("canvas").captureStream().getTracks()[0];
@@ -32,7 +36,7 @@ onload = () => {
 function sendPort() {
   sendPortBtn.disabled = true;
   let frame = document.getElementById("subframe");
-  frame.contentWindow.postMessage({name:"port", port: channel.port2}, "http://localhost:8001", [channel.port2]);
+  frame.contentWindow.postMessage({name:"port", port: channel.port2}, frameOrigin, [channel.port2]);
   startTrackBtn.disabled = false;
 };