260 lines
11 KiB
Python
260 lines
11 KiB
Python
#################################################################################################
|
|
# WebSocket Client thread
|
|
#################################################################################################
|
|
|
|
import xbmc
|
|
import xbmcgui
|
|
import xbmcaddon
|
|
|
|
import json
|
|
import threading
|
|
import urllib
|
|
import socket
|
|
import websocket
|
|
from ClientInformation import ClientInformation
|
|
from DownloadUtils import DownloadUtils
|
|
|
|
class WebSocketThread(threading.Thread):
|
|
|
|
logLevel = 0
|
|
client = None
|
|
keepRunning = True
|
|
|
|
def __init__(self, *args):
|
|
addonSettings = xbmcaddon.Addon(id='plugin.video.mbcon')
|
|
level = addonSettings.getSetting('logLevel')
|
|
self.logLevel = 0
|
|
if(level != None):
|
|
self.logLevel = int(level)
|
|
|
|
xbmc.log("MBCon WebSocketThread -> Log Level:" + str(self.logLevel))
|
|
|
|
threading.Thread.__init__(self, *args)
|
|
|
|
def logMsg(self, msg, level = 1):
|
|
if(self.logLevel >= level):
|
|
xbmc.log("MBCon WebSocketThread -> " + msg)
|
|
|
|
def playbackStarted(self, itemId):
|
|
if(self.client != None):
|
|
try:
|
|
self.logMsg("Sending Playback Started")
|
|
messageData = {}
|
|
messageData["MessageType"] = "PlaybackStart"
|
|
messageData["Data"] = itemId + "|true|audio,video"
|
|
messageString = json.dumps(messageData)
|
|
self.logMsg("Message Data : " + messageString)
|
|
self.client.send(messageString)
|
|
except Exception, e:
|
|
self.logMsg("Exception : " + str(e), level=0)
|
|
else:
|
|
self.logMsg("Sending Playback Started NO Object ERROR")
|
|
|
|
def playbackStopped(self, itemId, ticks):
|
|
if(self.client != None):
|
|
try:
|
|
self.logMsg("Sending Playback Stopped : " + str(ticks))
|
|
messageData = {}
|
|
messageData["MessageType"] = "PlaybackStopped"
|
|
messageData["Data"] = itemId + "|" + str(ticks)
|
|
messageString = json.dumps(messageData)
|
|
self.client.send(messageString)
|
|
except Exception, e:
|
|
self.logMsg("Exception : " + str(e), level=0)
|
|
else:
|
|
self.logMsg("Sending Playback Stopped NO Object ERROR")
|
|
|
|
def sendProgressUpdate(self, itemId, ticks):
|
|
if(self.client != None):
|
|
try:
|
|
self.logMsg("Sending Progress Update : " + str(ticks))
|
|
messageData = {}
|
|
messageData["MessageType"] = "PlaybackProgress"
|
|
messageData["Data"] = itemId + "|" + str(ticks) + "|false|false"
|
|
messageString = json.dumps(messageData)
|
|
self.logMsg("Message Data : " + messageString)
|
|
self.client.send(messageString)
|
|
except Exception, e:
|
|
self.logMsg("Exception : " + str(e), level=0)
|
|
else:
|
|
self.logMsg("Sending Progress Update NO Object ERROR")
|
|
|
|
def stopClient(self):
|
|
# stopping the client is tricky, first set keep_running to false and then trigger one
|
|
# more message by requesting one SessionsStart message, this causes the
|
|
# client to receive the message and then exit
|
|
if(self.client != None):
|
|
self.logMsg("Stopping Client")
|
|
self.keepRunning = False
|
|
self.client.keep_running = False
|
|
self.client.close()
|
|
self.logMsg("Stopping Client : KeepRunning set to False")
|
|
'''
|
|
try:
|
|
self.keepRunning = False
|
|
self.client.keep_running = False
|
|
self.logMsg("Stopping Client")
|
|
self.logMsg("Calling Ping")
|
|
self.client.sock.ping()
|
|
|
|
self.logMsg("Calling Socket Shutdown()")
|
|
self.client.sock.sock.shutdown(socket.SHUT_RDWR)
|
|
self.logMsg("Calling Socket Close()")
|
|
self.client.sock.sock.close()
|
|
self.logMsg("Stopping Client Done")
|
|
self.logMsg("Calling Ping")
|
|
self.client.sock.ping()
|
|
|
|
except Exception, e:
|
|
self.logMsg("Exception : " + str(e), level=0)
|
|
'''
|
|
else:
|
|
self.logMsg("Stopping Client NO Object ERROR")
|
|
|
|
def on_message(self, ws, message):
|
|
self.logMsg("Message : " + str(message))
|
|
result = json.loads(message)
|
|
|
|
messageType = result.get("MessageType")
|
|
playCommand = result.get("PlayCommand")
|
|
data = result.get("Data")
|
|
|
|
if(messageType != None and messageType == "Play" and data != None):
|
|
itemIds = data.get("ItemIds")
|
|
playCommand = data.get("PlayCommand")
|
|
if(playCommand != None and playCommand == "PlayNow"):
|
|
|
|
startPositionTicks = data.get("StartPositionTicks")
|
|
self.logMsg("Playing Media With ID : " + itemIds[0])
|
|
|
|
addonSettings = xbmcaddon.Addon(id='plugin.video.mbcon')
|
|
mb3Host = addonSettings.getSetting('ipaddress')
|
|
mb3Port = addonSettings.getSetting('port')
|
|
|
|
url = mb3Host + ":" + mb3Port + ',;' + itemIds[0]
|
|
if(startPositionTicks == None):
|
|
url += ",;" + "-1"
|
|
else:
|
|
url += ",;" + str(startPositionTicks)
|
|
|
|
playUrl = "plugin://plugin.video.mbcon/?url=" + url + '&mode=PLAY'
|
|
playUrl = playUrl.replace("\\\\","smb://")
|
|
playUrl = playUrl.replace("\\","/")
|
|
|
|
xbmc.Player().play(playUrl)
|
|
|
|
elif(messageType != None and messageType == "Playstate"):
|
|
command = data.get("Command")
|
|
if(command != None and command == "Stop"):
|
|
self.logMsg("Playback Stopped")
|
|
xbmc.executebuiltin('xbmc.activatewindow(10000)')
|
|
xbmc.Player().stop()
|
|
|
|
if(command != None and command == "Seek"):
|
|
seekPositionTicks = data.get("SeekPositionTicks")
|
|
self.logMsg("Playback Seek : " + str(seekPositionTicks))
|
|
seekTime = (seekPositionTicks / 1000) / 10000
|
|
xbmc.Player().seekTime(seekTime)
|
|
|
|
def on_error(self, ws, error):
|
|
self.logMsg("Error : " + str(error))
|
|
|
|
def on_close(self, ws):
|
|
self.logMsg("Closed")
|
|
|
|
def on_open(self, ws):
|
|
try:
|
|
clientInfo = ClientInformation()
|
|
machineId = clientInfo.getMachineId()
|
|
version = clientInfo.getVersion()
|
|
messageData = {}
|
|
messageData["MessageType"] = "Identity"
|
|
|
|
addonSettings = xbmcaddon.Addon(id='plugin.video.mbcon')
|
|
deviceName = addonSettings.getSetting('deviceName')
|
|
deviceName = deviceName.replace("\"", "_")
|
|
|
|
messageData["Data"] = "XBMC|" + machineId + "|" + version + "|" + deviceName
|
|
messageString = json.dumps(messageData)
|
|
self.logMsg("Opened : " + str(messageString))
|
|
ws.send(messageString)
|
|
|
|
downloadUtils = DownloadUtils()
|
|
|
|
# get session ID
|
|
addonSettings = xbmcaddon.Addon(id='plugin.video.mbcon')
|
|
mb3Host = addonSettings.getSetting('ipaddress')
|
|
mb3Port = addonSettings.getSetting('port')
|
|
|
|
url = "http://" + mb3Host + ":" + mb3Port + "/mediabrowser/Sessions?DeviceId=" + machineId + "&format=json"
|
|
self.logMsg("Session URL : " + url);
|
|
jsonData = downloadUtils.downloadUrl(url)
|
|
self.logMsg("Session JsonData : " + jsonData)
|
|
result = json.loads(jsonData)
|
|
self.logMsg("Session JsonData : " + str(result))
|
|
sessionId = result[0].get("Id")
|
|
self.logMsg("Session Id : " + str(sessionId))
|
|
|
|
url = "http://" + mb3Host + ":" + mb3Port + "/mediabrowser/Sessions/Capabilities?Id=" + sessionId + "&PlayableMediaTypes=Video&SupportedCommands=Play"
|
|
postData = {}
|
|
postData["Id"] = sessionId;
|
|
postData["PlayableMediaTypes"] = "Video";
|
|
stringdata = json.dumps(postData)
|
|
self.logMsg("Capabilities URL : " + url);
|
|
self.logMsg("Capabilities Data : " + stringdata)
|
|
downloadUtils.downloadUrl(url, postBody=stringdata, type="POST")
|
|
|
|
except Exception, e:
|
|
self.logMsg("Exception : " + str(e), level=0)
|
|
|
|
def run(self):
|
|
|
|
while(self.keepRunning and xbmc.abortRequested == False):
|
|
|
|
addonSettings = xbmcaddon.Addon(id='plugin.video.mbcon')
|
|
mb3Host = addonSettings.getSetting('ipaddress')
|
|
mb3Port = addonSettings.getSetting('port')
|
|
|
|
if(self.logLevel >= 1):
|
|
websocket.enableTrace(True)
|
|
|
|
if(mb3Host != None and len(mb3Host) > 0):
|
|
|
|
try:
|
|
|
|
wsPort = mb3Port
|
|
self.logMsg("WebSocketPortNumber = " + str(wsPort))
|
|
|
|
downloadUtils = DownloadUtils()
|
|
authHeaders = downloadUtils.getAuthHeader()
|
|
flatHeaders = []
|
|
for header in authHeaders:
|
|
flatHeaders.append(header + ": " + authHeaders[header])
|
|
self.logMsg("Flat Header : " + str(flatHeaders))
|
|
|
|
# Make a call to /System/Info. WebSocketPortNumber is the port hosting the web socket.
|
|
webSocketUrl = "ws://" + mb3Host + ":" + str(wsPort) + "/mediabrowser"
|
|
self.logMsg("WebSocket URL : " + webSocketUrl)
|
|
self.client = websocket.WebSocketApp(webSocketUrl,
|
|
header = flatHeaders,
|
|
on_message = self.on_message,
|
|
on_error = self.on_error,
|
|
on_close = self.on_close)
|
|
|
|
self.client.on_open = self.on_open
|
|
|
|
self.logMsg("Client Starting")
|
|
if(self.keepRunning):
|
|
self.client.run_forever()
|
|
except:
|
|
self.logMsg("Error thrown in Web Socket Setup")
|
|
|
|
if(self.keepRunning and xbmc.abortRequested == False):
|
|
self.logMsg("Client Needs To Restart")
|
|
xbmc.sleep(5000)
|
|
|
|
self.logMsg("Thread Exited")
|
|
|
|
|
|
|
|
|