jellyfin-kodi/resources/lib/ConnectionManager.py

161 lines
5.7 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
2015-03-13 21:24:59 +00:00
#################################################################################################
# connection manager class
#################################################################################################
import json
import socket
2015-03-13 21:24:59 +00:00
import xbmc
import xbmcgui
import xbmcaddon
import Utils as utils
from ClientInformation import ClientInformation
2015-04-03 10:13:01 +00:00
from DownloadUtils import DownloadUtils
from UserClient import UserClient
2015-03-13 21:24:59 +00:00
class ConnectionManager():
2015-04-03 10:13:01 +00:00
clientInfo = ClientInformation()
user = UserClient()
2015-04-03 10:13:01 +00:00
doUtils = DownloadUtils()
addonName = clientInfo.getAddonName()
addonId = clientInfo.getAddonId()
addon = xbmcaddon.Addon()
WINDOW = xbmcgui.Window(10000)
2015-04-03 10:13:01 +00:00
def __init__(self):
self.__language__ = self.addon.getLocalizedString
def logMsg(self, msg, lvl=1):
className = self.__class__.__name__
utils.logMsg("%s %s" % (self.addonName, className), msg, int(lvl))
2015-03-13 21:24:59 +00:00
def checkServer(self):
2015-04-03 10:13:01 +00:00
self.WINDOW.setProperty("Server_Checked", "True")
self.logMsg("Connection Manager Called", 2)
2015-03-13 21:24:59 +00:00
server = self.user.getServer()
2015-04-03 10:13:01 +00:00
if server != "":
2015-04-03 10:13:01 +00:00
self.logMsg("Server already set", 2)
2015-03-13 21:24:59 +00:00
return
serverInfo = self.getServerDetails()
try:
prefix,ip,port = serverInfo.split(":")
setServer = xbmcgui.Dialog().yesno(self.__language__(30167), "Proceed with the following server?", self.__language__(30169) + serverInfo)
except: # serverInfo is None
2015-04-03 10:13:01 +00:00
self.logMsg("getServerDetails failed", 1)
xbmc.executebuiltin('Addon.OpenSettings(%s)' % self.addonId)
2015-03-13 21:24:59 +00:00
return
if setServer == 1:
2015-04-03 10:13:01 +00:00
self.logMsg("Server selected. Saving information.", 1)
utils.settings("ipaddress", ip.replace("/", ""))
utils.settings("port", port)
# If https, enable the setting
2015-04-03 10:13:01 +00:00
if (prefix == 'https'):
utils.settings('https', "true")
2015-04-03 10:13:01 +00:00
else:
self.logMsg("No server selected.", 1)
xbmc.executebuiltin('Addon.OpenSettings(%s)' % self.addonId)
return
# Get List of public users
self.logMsg("Getting user list", 1)
server = "%s:%s" % (ip.replace("/", ""), port)
url = "%s/mediabrowser/Users/Public?format=json" % serverInfo
2015-04-03 10:13:01 +00:00
result = self.doUtils.downloadUrl(url, authenticate=False)
if result == "":
self.logMsg("Unable to connect to %s." % server, 1)
2015-03-13 21:24:59 +00:00
return
self.logMsg("Result: %s" % result, 2)
# Process the list returned
2015-03-13 21:24:59 +00:00
names = []
userList = []
for user in result:
name = user['Name']
2015-03-13 21:24:59 +00:00
userList.append(name)
if user['HasPassword']:
name = "%s (Secure)" % name
2015-03-13 21:24:59 +00:00
names.append(name)
self.logMsg("User list: %s" % names, 1)
resp = xbmcgui.Dialog().select(self.__language__(30200), names)
if resp > -1:
selected_user = userList[resp]
self.logMsg("Selected User: %s" % selected_user, 1)
utils.settings("username", selected_user)
else:
self.logMsg("No user selected.", 1)
2015-04-03 10:13:01 +00:00
xbmc.executebuiltin('Addon.OpenSettings(%s)' % self.addonId)
return
musicDisabled = xbmcgui.Dialog().yesno("Music Setting", "Disable music library?")
if musicDisabled:
self.logMsg("User opted to disable music library.", 1)
utils.settings('enableMusicSync', "false")
else:
# Music is enabled, prompt for direct stream
musicPath = xbmcgui.Dialog().yesno("Music Setting", "Direct stream the music library?", "Select this option only if you plan on listening to music outside your network.")
if musicPath:
self.logMsg("User option to direct stream music library.", 1)
utils.settings('directstreammusic', "true")
2015-11-10 03:01:31 +00:00
directPaths = xbmcgui.Dialog().yesno(
heading="Direct paths",
line1=(
"Use direct paths? Caution! If you choose yes, you "
"will lose access to certain Emby features such as: "
"Emby cinema mode, direct stream/transcode options, "
"parental access schedule."))
if directPaths:
self.logMsg("User opted to use direct paths.", 1)
utils.settings('useDirectPaths', "true")
return
2015-03-13 21:24:59 +00:00
def getServerDetails(self):
2015-04-03 10:13:01 +00:00
self.logMsg("Getting Server Details from Network")
2015-03-13 21:24:59 +00:00
MULTI_GROUP = ("<broadcast>", 7359)
2015-04-03 10:13:01 +00:00
MESSAGE = "who is EmbyServer?"
2015-03-13 21:24:59 +00:00
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.settimeout(6.0)
2015-04-03 10:13:01 +00:00
2015-03-13 21:24:59 +00:00
sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 20)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
sock.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_LOOP, 1)
sock.setsockopt(socket.IPPROTO_IP, socket.SO_REUSEADDR, 1)
self.logMsg("MutliGroup : %s" % str(MULTI_GROUP), 2);
self.logMsg("Sending UDP Data: %s" % MESSAGE, 2);
2015-03-13 21:24:59 +00:00
sock.sendto(MESSAGE, MULTI_GROUP)
try:
data, addr = sock.recvfrom(1024) # buffer size is 1024 bytes
2015-04-03 10:13:01 +00:00
self.logMsg("Received Response: %s" % data)
# Get the address
data = json.loads(data)
return data['Address']
2015-03-13 21:24:59 +00:00
except:
2015-04-03 10:13:01 +00:00
self.logMsg("No UDP Response")
2015-03-13 21:24:59 +00:00
pass
2015-04-03 10:13:01 +00:00
return None