From 6fcc3a2397dc23b595f46c34a9d3482f8809046f Mon Sep 17 00:00:00 2001
From: Fabio Manganiello <blacklight86@gmail.com>
Date: Tue, 13 Nov 2018 20:20:55 +0100
Subject: [PATCH] Support for YouTube media in Chromecast plugin

---
 platypush/plugins/media/chromecast.py | 60 +++++++++++++++++++++++----
 1 file changed, 52 insertions(+), 8 deletions(-)

diff --git a/platypush/plugins/media/chromecast.py b/platypush/plugins/media/chromecast.py
index 32ba7f048..ae8c80d20 100644
--- a/platypush/plugins/media/chromecast.py
+++ b/platypush/plugins/media/chromecast.py
@@ -1,5 +1,8 @@
+import re
 import pychromecast
 
+from pychromecast.controllers.youtube import YouTubeController
+
 from platypush.plugins import Plugin, action
 
 
@@ -7,6 +10,11 @@ class MediaChromecastPlugin(Plugin):
     """
     Plugin to interact with Chromecast devices
 
+    Supported formats:
+
+        * YouTube URLs
+        * Plex (through ``media.plex`` plugin, experimental)
+
     Requires:
 
         * **pychromecast** (``pip install pychromecast``)
@@ -67,16 +75,19 @@ class MediaChromecastPlugin(Plugin):
 
 
         if chromecast not in self.chromecasts:
-            self.chromecasts = pychromecast.get_chromecasts()
-            cast = next(cc for cc in self.chromecasts
-                        if cc.device.friendly_name == chromecast)
-        else:
-            cast = self.chromecasts[chromecast]
+            self.chromecasts = {
+                cast.device.friendly_name: cast
+                for cast in pychromecast.get_chromecasts()
+            }
+
+            if chromecast not in self.chromecasts:
+                raise RuntimeError('Device {} not found'.format(chromecast))
+
+        return self.chromecasts[chromecast]
 
-        return cast
 
     @action
-    def cast_media(self, media, content_type, chromecast=None, title=None,
+    def cast_media(self, media, content_type=None, chromecast=None, title=None,
                    image_url=None, autoplay=True, current_time=0,
                    stream_type=STREAM_TYPE_BUFFERED, subtitles=None,
                    subtitles_lang='en-US', subtitles_mime='text/vtt',
@@ -121,10 +132,29 @@ class MediaChromecastPlugin(Plugin):
         :type subtitle_id: int
         """
 
+        if not chromecast:
+            chromecast = self.chromecast
+
         cast = self.get_chromecast(chromecast)
         cast.wait()
+
         mc = cast.media_controller
-        mc.namespace = 'urn:x-cast:com.google.cast.sse'
+        yt = self._get_youtube_url(media)
+
+        if yt:
+            self.logger.info('Playing YouTube video {} on {}'.format(
+                yt, chromecast))
+
+            hndl = YouTubeController()
+            cast.register_handler(hndl)
+            hndl.update_screen_id()
+            return hndl.play_video(yt)
+
+        if not content_type:
+            raise RuntimeError('content_type required to process media {}'.
+                               format(media))
+
+        self.logger.info('Playing {} on {}'.format(media, chromecast))
 
         mc.play_media(media, content_type, title=title, thumb=image_url,
                       current_time=current_time, autoplay=autoplay,
@@ -134,6 +164,20 @@ class MediaChromecastPlugin(Plugin):
 
         mc.block_until_active()
 
+
+    @classmethod
+    def _get_youtube_url(cls, url):
+        m = re.match('https?://www.youtube.com/watch\?v=([^&]+).*', url)
+        if m:
+            return m.group(1)
+
+        m = re.match('youtube:video:(.*)', url)
+        if m:
+            return m.group(1)
+
+        return None
+
+
     @action
     def disconnect(self, chromecast=None, timeout=None, blocking=True):
         """