Attempt to reestablished websocket communication if the connection fails

jellyfin_kodi/jellyfin/ws_client.py

@@ -5,6 +5,7 @@ from __future__ import division, absolute_import, print_function, unicode_litera
 
 import json
 import threading
+import time
 
 from kodi_six import xbmc
 
@@ -54,17 +55,22 @@ class WSClient(threading.Thread):
         LOG.info("Websocket url: %s", wsc_url)
 
         self.wsc = websocket.WebSocketApp(wsc_url,
+                                          on_open=lambda ws: self.on_open(ws),
                                           on_message=lambda ws, message: self.on_message(ws, message),
                                           on_error=lambda ws, error: self.on_error(ws, error))
-        self.wsc.on_open = self.on_open
 
+        retry_count = 0
         while not self.stop:
 
+            time.sleep(retry_count * 5)
             self.wsc.run_forever(ping_interval=10)
 
             if not self.stop and monitor.waitForAbort(5):
                 break
 
+            if retry_count < 12:
+                retry_count += 1
+
         LOG.info("---<[ websocket ]")
 
     def on_error(self, ws, error):
@@ -72,6 +78,21 @@ class WSClient(threading.Thread):
 
     def on_open(self, ws):
         LOG.info("--->[ websocket ]")
+        self.client.jellyfin.post_capabilities({
+            'PlayableMediaTypes': "Audio,Video",
+            'SupportsMediaControl': True,
+            'SupportedCommands': (
+                "MoveUp,MoveDown,MoveLeft,MoveRight,Select,"
+                "Back,ToggleContextMenu,ToggleFullscreen,ToggleOsdMenu,"
+                "GoHome,PageUp,NextLetter,GoToSearch,"
+                "GoToSettings,PageDown,PreviousLetter,TakeScreenshot,"
+                "VolumeUp,VolumeDown,ToggleMute,SendString,DisplayMessage,"
+                "SetAudioStreamIndex,SetSubtitleStreamIndex,"
+                "SetRepeatMode,"
+                "Mute,Unmute,SetVolume,"
+                "Play,Playstate,PlayNext,PlayMediaSource"
+            ),
+        })
 
     def on_message(self, ws, message):