mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2024-12-27 03:06:10 +00:00
138 lines
No EOL
4.4 KiB
Python
138 lines
No EOL
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
#################################################################################################
|
|
# connection manager class
|
|
#################################################################################################
|
|
|
|
import json
|
|
import socket
|
|
|
|
import xbmc
|
|
import xbmcgui
|
|
import xbmcaddon
|
|
|
|
import Utils as utils
|
|
from ClientInformation import ClientInformation
|
|
from DownloadUtils import DownloadUtils
|
|
from UserClient import UserClient
|
|
|
|
|
|
class ConnectionManager():
|
|
|
|
clientInfo = ClientInformation()
|
|
user = UserClient()
|
|
doUtils = DownloadUtils()
|
|
|
|
addonName = clientInfo.getAddonName()
|
|
addonId = clientInfo.getAddonId()
|
|
addon = xbmcaddon.Addon()
|
|
WINDOW = xbmcgui.Window(10000)
|
|
|
|
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))
|
|
|
|
def checkServer(self):
|
|
|
|
self.WINDOW.setProperty("Server_Checked", "True")
|
|
self.logMsg("Connection Manager Called", 2)
|
|
|
|
addon = self.addon
|
|
server = self.user.getServer()
|
|
|
|
if server != "":
|
|
self.logMsg("Server already set", 2)
|
|
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
|
|
self.logMsg("getServerDetails failed", 1)
|
|
xbmc.executebuiltin('Addon.OpenSettings(%s)' % self.addonId)
|
|
return
|
|
|
|
if setServer == 1:
|
|
self.logMsg("Server selected. Saving information.", 1)
|
|
addon.setSetting("ipaddress", ip.replace("/", ""))
|
|
addon.setSetting("port", port)
|
|
# If https, enable the setting
|
|
if (prefix == 'https'):
|
|
addon.setSetting('https', "true")
|
|
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
|
|
|
|
result = self.doUtils.downloadUrl(url, authenticate=False)
|
|
if result == "":
|
|
self.logMsg("Unable to connect to %s." % server, 1)
|
|
return
|
|
|
|
self.logMsg("Result: %s" % result, 2)
|
|
|
|
# Process the list returned
|
|
names = []
|
|
userList = []
|
|
for user in result:
|
|
name = user['Name']
|
|
userList.append(name)
|
|
|
|
if user['HasPassword']:
|
|
name = "%s (Secure)" % name
|
|
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)
|
|
self.addon.setSetting("username", selected_user)
|
|
else:
|
|
self.logMsg("No user selected.", 1)
|
|
xbmc.executebuiltin('Addon.OpenSettings(%s)' % self.addonId)
|
|
return
|
|
|
|
def getServerDetails(self):
|
|
|
|
self.logMsg("Getting Server Details from Network")
|
|
|
|
MULTI_GROUP = ("<broadcast>", 7359)
|
|
MESSAGE = "who is EmbyServer?"
|
|
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sock.settimeout(6.0)
|
|
|
|
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);
|
|
sock.sendto(MESSAGE, MULTI_GROUP)
|
|
|
|
try:
|
|
data, addr = sock.recvfrom(1024) # buffer size is 1024 bytes
|
|
self.logMsg("Received Response: %s" % data)
|
|
# Get the address
|
|
data = json.loads(data)
|
|
return data['Address']
|
|
except:
|
|
self.logMsg("No UDP Response")
|
|
pass
|
|
|
|
return None |