# -*- coding: utf-8 -*- ################################################################################################## import json import logging import threading import xbmc import xbmcgui import artwork import connectmanager import downloadutils import read_embyserver as embyserver from utils import window, settings, language as lang ################################################################################################## log = logging.getLogger("EMBY."+__name__) ################################################################################################## class UserClient(threading.Thread): _shared_state = {} # Borg _stop_thread = False _user = None _server = None _auth = True _has_access = True def __init__(self): self.__dict__ = self._shared_state self.doutils = downloadutils.DownloadUtils() self.download = self.doutils.downloadUrl self.emby = embyserver.Read_EmbyServer() self.connectmanager = connectmanager.ConnectManager() threading.Thread.__init__(self) @classmethod def get_username(cls): return settings('username') or settings('connectUsername') or None def get_user(self, data=None): if data is not None: self._user = data self._set_user_server() return self._user def get_server_details(self): return self._server @classmethod def get_server(cls): ###$ Begin migration $### if settings('server') == "": http = "https" if settings('https') == "true" else "http" host = settings('ipaddress') port = settings('port') if host and port: settings('server', value="%s://%s:%s" % (http, host, port)) log.info("server address migration completed") ###$ End migration $### return settings('server') or None def verify_server(self): try: url = "%s/emby/Users/Public?format=json" % self.get_server() self.download(url, authenticate=False) return True except: # Server connection failed return False @classmethod def get_ssl(cls): """ Returns boolean value or path to certificate True: Verify ssl False: Don't verify connection """ certificate = settings('sslcert') if certificate != "None": return certificate return True if settings('sslverify') == "true" else False def get_access(self): if not self._has_access: self._set_access() return self._has_access def _set_access(self): try: self.download("{server}/emby/Users?format=json") except Warning as error: if self._has_access and "restricted" in error: self._has_access = False log.info("access is restricted") else: if not self._has_access: self._has_access = True window('emby_serverStatus', clear=True) log.info("access is granted") xbmcgui.Dialog().notification(lang(29999), lang(33007)) @classmethod def get_userid(cls): ###$ Begin migration $### if settings('userId') == "": settings('userId', value=settings('userId%s' % settings('username'))) log.info("userid migration completed") ###$ End migration $### return settings('userId') or None @classmethod def get_token(cls): ###$ Begin migration $### if settings('token') == "": settings('token', value=settings('accessToken')) log.info("token migration completed") ###$ End migration $### return settings('token') or None def _set_user_server(self): user = self.download("{server}/emby/Users/{UserId}?format=json") settings('username', value=user['Name']) self._user = user if "PrimaryImageTag" in self._user: window('EmbyUserImage', value=artwork.Artwork().get_user_artwork(self._user['Id'], 'Primary')) self._server = self.download("{server}/emby/System/Configuration?format=json") settings('markPlayed', value=str(self._server['MaxResumePct'])) def _authenticate(self): if not self.get_server() or not self.get_username(): log.info('missing server or user information') self._auth = False elif self.get_token(): try: self._load_user() except Warning: log.info("token is invalid") self._reset_client() else: log.info("current user: %s", self.get_username()) log.info("current userid: %s", self.get_userid()) log.debug("current token: %s", self.get_token()) return ##### AUTHENTICATE USER ##### server = self.get_server() username = self.get_username().decode('utf-8') try: user = self.connectmanager.login_manual(server, username) except RuntimeError: window('emby_serverStatus', value="stop") self._auth = False return else: log.info("user: %s", user) settings('username', value=user['User']['Name']) settings('token', value=user['AccessToken']) settings('userId', value=user['User']['Id']) xbmcgui.Dialog().notification(lang(29999), "%s %s!" % (lang(33000), username)) self._load_user(authenticated=True) window('emby_serverStatus', clear=True) def _load_user(self, authenticated=False): doutils = self.doutils userid = self.get_userid() server = self.get_server() token = self.get_token() # Set properties # TODO: Remove old reference once code converted window('emby_currUser', value=userid) window('emby_server%s' % userid, value=server) window('emby_accessToken%s' % userid, value=token) server_json = { 'UserId': userid, 'Server': server, 'ServerId': settings('serverId'), 'Token': token, 'SSL': self.get_ssl() } # Set downloadutils.py values doutils.set_session(**server_json) # Test the validity of the current token if not authenticated: try: self.download("{server}/emby/Users/{UserId}?format=json") except Warning as error: if "401" in error: # Token is not longer valid raise # verify user access self._set_access() # Start downloadutils.py session doutils.start_session() # Set _user and _server self._set_user_server() def load_connect_servers(self): # Set connect servers if not settings('connectUsername'): return servers = self.connectmanager.get_connect_servers() added_servers = [] for server in servers: if server['Id'] != settings('serverId'): # TODO: SSL setup self.doutils.add_server(server, False) added_servers.append(server['Id']) # Set properties log.info(added_servers) window('emby_servers.json', value=added_servers) def _reset_client(self): log.info("reset UserClient authentication") settings('accessToken', value="") window('emby_accessToken', clear=True) log.info("user token revoked.") self._user = None self.auth = None current_state = self.connectmanager.get_state() for server in current_state['Servers']: if server['Id'] == settings('serverId'): # Update token server['AccessToken'] = None self.connectmanager.update_token(server) def run(self): monitor = xbmc.Monitor() log.warn("----====# Starting UserClient #====----") while not self._stop_thread: status = window('emby_serverStatus') if status: # Verify the connection status to server if status == "restricted": # Parental control is restricting access self._has_access = False elif status == "401": # Unauthorized access, revoke token window('emby_serverStatus', value="auth") self._reset_client() if self._auth and self._user is None: # Try to authenticate user status = window('emby_serverStatus') if not status or status == "auth": # Set auth flag because we no longer need # to authenticate the user self._auth = False self._authenticate() if not self._auth and self._user is None: # If authenticate failed. server = self.get_server() username = self.get_username() status = window('emby_serverStatus') # The status Stop is for when user cancelled password dialog. if server and username and status != "stop": # Only if there's information found to login log.info("Server found: %s", server) log.info("Username found: %s", username) self._auth = True if monitor.waitForAbort(1): # Abort was requested while waiting. We should exit break self.doutils.stop_session() log.warn("#====---- UserClient Stopped ----====#") def stop_client(self): self._stop_thread = True