Support youtube streams

4 files changed, 57 insertions, 26 deletions

diff --git a/src/client/Main.hx b/src/client/Main.hx
index 3400595..900f841 100644
--- a/src/client/Main.hx
+++ b/src/client/Main.hx
@@ -250,7 +250,7 @@ class Main {
 						author: personal.name,
 						duration: data.duration,
 						isTemp: isTemp,
-						isIframe: false
+						isIframe: data.isIframe == true
 					},
 					atEnd: atEnd
 			}});
@@ -263,23 +263,31 @@ class Main {
 		final iframe = iframeCode.value;
 		if (iframe.length == 0) return;
 		iframeCode.value = "";
-		final mediaName:InputElement = cast ge("#customembed-title");
-		final name = mediaName.value.length == 0 ? "Custom Media" : mediaName.value;
-		mediaName.value = "";
+		final mediaTitle:InputElement = cast ge("#customembed-title");
+		final title = mediaTitle.value;
+		mediaTitle.value = "";
 		final checkbox:InputElement = cast ge("#customembed").querySelector(".add-temp");
 		final isTemp = checkbox.checked;
-		send({
-			type: AddVideo, addVideo: {
-				item: {
-					url: iframe,
-					title: name,
-					author: personal.name,
-					duration: 99 * 60 * 60,
-					isTemp: isTemp,
-					isIframe: true
-				},
-				atEnd: atEnd
-		}});
+		player.getIframeData(iframe, (data:VideoData) -> {
+			if (data.duration == 0) {
+				serverMessage(4, Lang.get("addVideoError"));
+				return;
+			}
+			if (title.length > 0) data.title = title;
+			if (data.url == null) data.url = iframe;
+			send({
+				type: AddVideo, addVideo: {
+					item: {
+						url: data.url,
+						title: data.title,
+						author: personal.name,
+						duration: data.duration,
+						isTemp: isTemp,
+						isIframe: true
+					},
+					atEnd: atEnd
+			}});
+		});
 	}
 
 	public function toggleVideoElement():Bool {
diff --git a/src/client/Player.hx b/src/client/Player.hx
index 28b5c09..5c283f7 100644
--- a/src/client/Player.hx
+++ b/src/client/Player.hx
@@ -101,6 +101,10 @@ class Player {
 		player.getVideoData(url, callback);
 	}
 
+	public function getIframeData(iframe:String, callback:(data:VideoData)->Void):Void {
+		iframePlayer.getVideoData(iframe, callback);
+	}
+
 	public function setVideo(i:Int):Void {
 		if (!main.isSyncActive) return;
 		final item = items[i];
diff --git a/src/client/players/Iframe.hx b/src/client/players/Iframe.hx
index 3268a61..2f875fb 100644
--- a/src/client/players/Iframe.hx
+++ b/src/client/players/Iframe.hx
@@ -24,11 +24,20 @@ class Iframe implements IPlayer {
 		return true;
 	}
 
-	public function getVideoData(url:String, callback:(data:VideoData)->Void):Void {
-		callback({
-			duration: 99 * 60 * 60,
-			title: "Custom Media"
-		});
+	public function getVideoData(data:String, callback:(data:VideoData)->Void):Void {
+		final iframe = document.createDivElement();
+		iframe.innerHTML = data;
+		if (isValidIframe(iframe)) {
+			callback({duration: 99 * 60 * 60});
+		} else {
+			callback({duration: 0});
+		}
+	}
+
+	function isValidIframe(iframe:Element):Bool {
+		if (iframe.children.length != 1) return false;
+		return (iframe.firstChild.nodeName == "IFRAME"
+			|| iframe.firstChild.nodeName == "OBJECT");
 	}
 
 	public function loadVideo(item:VideoItem):Void {
@@ -36,9 +45,7 @@ class Iframe implements IPlayer {
 		video = document.createDivElement();
 		video.id = "videoplayer";
 		video.innerHTML = item.url; // actually data
-		if (video.firstChild.nodeName != "IFRAME"
-			&& video.firstChild.nodeName != "OBJECT") {
-			// TODO move to getVideoData too
+		if (!isValidIframe(video)) {
 			video = null;
 			return;
 		}
diff --git a/src/client/players/Youtube.hx b/src/client/players/Youtube.hx
index 0aa9f2b..f5f12a0 100644
--- a/src/client/players/Youtube.hx
+++ b/src/client/players/Youtube.hx
@@ -96,9 +96,21 @@ class Youtube implements IPlayer {
 			for (item in items) {
 				final title:String = item.snippet.title;
 				final duration:String = item.contentDetails.duration;
-				// TODO duration is PT0S for streams
+				final duration = convertTime(duration);
+				// duration is PT0S for streams
+				if (duration == 0) {
+					callback({
+						duration: 99 * 60 * 60,
+						title: title,
+						url: '<iframe src="https://www.youtube.com/embed/$id" frameborder="0"
+								allow="accelerometer; autoplay; encrypted-media; gyroscope; picture-in-picture"
+								allowfullscreen></iframe>',
+						isIframe: true
+					});
+					return;
+				}
 				callback({
-					duration: convertTime(duration),
+					duration: duration,
 					title: title,
 					url: url
 				});