# -*- coding: utf-8 -*- ################################################################################################## 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): url = "%s/emby/Users/Public?format=json" % self.get_server() result = self.download(url, authenticate=False) if result != "": # Specific verification, due to possibility of returning empty dict return True else: # 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): self._user = self.download("{server}/emby/Users/{UserId}?format=json") settings('username', value=self._user['Name']) if "PrimaryImageTag" in self._user: window('EmbyUserImage', value=artwork.Artwork().getUserArtwork(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') users = self.emby.getUsers(server) user_found = None for user in users: if username == user['Name']: user_found = user break try: user = self.connectmanager.login_manual(server, user_found) 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 window('emby_currUser', value=userid) window('emby_server%s' % userid, value=server) window('emby_accessToken%s' % userid, value=token) # 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 # Set downloadutils.py values doutils.setUserId(userid) doutils.setServer(server) doutils.setToken(token) doutils.setSSL(self.get_ssl()) # verify user access try: self._set_access() except Warning: # We don't need to raise any exceptions pass # Start downloadutils.py session doutils.startSession() # Set _user and _server self._set_user_server() 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(current_state['Servers'], 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.stopSession() log.warn("##===---- UserClient Stopped ----===##") def stop_client(self): self._stop_thread = True