348 lines
12 KiB
Python
348 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
|
###############################################################################
|
|
from logging import getLogger
|
|
from threading import Thread
|
|
|
|
from xbmc import sleep, executebuiltin, translatePath
|
|
import xbmcaddon
|
|
from xbmcvfs import exists
|
|
|
|
from utils import window, settings, language as lang, thread_methods, dialog
|
|
from downloadutils import DownloadUtils as DU
|
|
import plex_tv
|
|
import PlexFunctions as PF
|
|
import state
|
|
|
|
###############################################################################
|
|
|
|
LOG = getLogger("PLEX." + __name__)
|
|
|
|
###############################################################################
|
|
|
|
|
|
@thread_methods(add_suspends=['SUSPEND_USER_CLIENT'])
|
|
class UserClient(Thread):
|
|
"""
|
|
Manage Plex users
|
|
"""
|
|
# Borg - multiple instances, shared state
|
|
__shared_state = {}
|
|
|
|
def __init__(self):
|
|
self.__dict__ = self.__shared_state
|
|
|
|
self.auth = True
|
|
self.retry = 0
|
|
|
|
self.user = None
|
|
self.has_access = True
|
|
|
|
self.server = None
|
|
self.server_name = None
|
|
self.machine_identifier = None
|
|
self.token = None
|
|
self.ssl = None
|
|
self.sslcert = None
|
|
|
|
self.addon = xbmcaddon.Addon()
|
|
self.do_utils = None
|
|
|
|
Thread.__init__(self)
|
|
|
|
def get_server(self):
|
|
"""
|
|
Get the current PMS' URL
|
|
"""
|
|
# Original host
|
|
self.server_name = settings('plex_servername')
|
|
https = settings('https') == "true"
|
|
host = settings('ipaddress')
|
|
port = settings('port')
|
|
self.machine_identifier = settings('plex_machineIdentifier')
|
|
if not host:
|
|
LOG.debug("No server information saved.")
|
|
return False
|
|
server = host + ":" + port
|
|
# If https is true
|
|
if https:
|
|
server = "https://%s" % server
|
|
# If https is false
|
|
else:
|
|
server = "http://%s" % server
|
|
# User entered IP; we need to get the machineIdentifier
|
|
if not self.machine_identifier:
|
|
self.machine_identifier = PF.GetMachineIdentifier(server)
|
|
if not self.machine_identifier:
|
|
self.machine_identifier = ''
|
|
settings('plex_machineIdentifier', value=self.machine_identifier)
|
|
LOG.debug('Returning active server: %s', server)
|
|
return server
|
|
|
|
@staticmethod
|
|
def get_ssl_verify():
|
|
"""
|
|
Do we need to verify the SSL certificate? Return None if that is the
|
|
case, else False
|
|
"""
|
|
return None if settings('sslverify') == 'true' else False
|
|
|
|
@staticmethod
|
|
def get_ssl_certificate():
|
|
"""
|
|
Client side certificate
|
|
"""
|
|
return None if settings('sslcert') == 'None' \
|
|
else settings('sslcert')
|
|
|
|
def set_user_prefs(self):
|
|
"""
|
|
Load a user's profile picture
|
|
"""
|
|
LOG.debug('Setting user preferences')
|
|
# Only try to get user avatar if there is a token
|
|
if self.token:
|
|
url = PF.GetUserArtworkURL(self.user)
|
|
if url:
|
|
window('PlexUserImage', value=url)
|
|
|
|
@staticmethod
|
|
def check_access():
|
|
# Plex: always return True for now
|
|
return True
|
|
|
|
def load_user(self, username, user_id, usertoken, authenticated=False):
|
|
"""
|
|
Load the current user's details for PKC
|
|
"""
|
|
LOG.debug('Loading current user')
|
|
self.token = usertoken
|
|
self.server = self.get_server()
|
|
self.ssl = self.get_ssl_verify()
|
|
self.sslcert = self.get_ssl_certificate()
|
|
|
|
if authenticated is False:
|
|
if self.server is None:
|
|
return False
|
|
LOG.debug('Testing validity of current token')
|
|
res = PF.check_connection(self.server,
|
|
token=self.token,
|
|
verifySSL=self.ssl)
|
|
if res is False:
|
|
# PMS probably offline
|
|
return False
|
|
elif res == 401:
|
|
LOG.error('Token is no longer valid')
|
|
return 401
|
|
elif res >= 400:
|
|
LOG.error('Answer from PMS is not as expected. Retrying')
|
|
return False
|
|
|
|
# Set to windows property
|
|
state.PLEX_USER_ID = user_id or None
|
|
state.PLEX_USERNAME = username
|
|
# This is the token for the current PMS (might also be '')
|
|
window('pms_token', value=usertoken)
|
|
state.PMS_TOKEN = usertoken
|
|
# This is the token for plex.tv for the current user
|
|
# Is only '' if user is not signed in to plex.tv
|
|
window('plex_token', value=settings('plexToken'))
|
|
state.PLEX_TOKEN = settings('plexToken') or None
|
|
window('plex_restricteduser', value=settings('plex_restricteduser'))
|
|
state.RESTRICTED_USER = True \
|
|
if settings('plex_restricteduser') == 'true' else False
|
|
window('pms_server', value=self.server)
|
|
window('plex_machineIdentifier', value=self.machine_identifier)
|
|
window('plex_servername', value=self.server_name)
|
|
window('plex_authenticated', value='true')
|
|
state.AUTHENTICATED = True
|
|
|
|
window('useDirectPaths', value='true'
|
|
if settings('useDirectPaths') == "1" else 'false')
|
|
state.DIRECT_PATHS = True if settings('useDirectPaths') == "1" \
|
|
else False
|
|
state.INDICATE_MEDIA_VERSIONS = True \
|
|
if settings('indicate_media_versions') == "true" else False
|
|
window('plex_force_transcode_pix', value='true'
|
|
if settings('force_transcode_pix') == "1" else 'false')
|
|
|
|
# Start DownloadUtils session
|
|
self.do_utils = DU()
|
|
self.do_utils.startSession(reset=True)
|
|
# Set user preferences in settings
|
|
self.user = username
|
|
self.set_user_prefs()
|
|
|
|
# Writing values to settings file
|
|
settings('username', value=username)
|
|
settings('userid', value=user_id)
|
|
settings('accessToken', value=usertoken)
|
|
return True
|
|
|
|
def authenticate(self):
|
|
"""
|
|
Authenticate the current user
|
|
"""
|
|
LOG.debug('Authenticating user')
|
|
|
|
# Give attempts at entering password / selecting user
|
|
if self.retry >= 2:
|
|
LOG.error("Too many retries to login.")
|
|
state.PMS_STATUS = 'Stop'
|
|
dialog('ok', lang(33001), lang(39023))
|
|
executebuiltin(
|
|
'Addon.OpenSettings(plugin.video.plexkodiconnect)')
|
|
return False
|
|
|
|
# Get /profile/addon_data
|
|
addondir = translatePath(self.addon.getAddonInfo('profile'))
|
|
|
|
# If there's no settings.xml
|
|
if not exists("%ssettings.xml" % addondir):
|
|
LOG.error("Error, no settings.xml found.")
|
|
self.auth = False
|
|
return False
|
|
server = self.get_server()
|
|
# If there is no server we can connect to
|
|
if not server:
|
|
LOG.info("Missing server information.")
|
|
self.auth = False
|
|
return False
|
|
|
|
# If there is a username in the settings, try authenticating
|
|
username = settings('username')
|
|
userId = settings('userid')
|
|
usertoken = settings('accessToken')
|
|
enforceLogin = settings('enforceUserLogin')
|
|
# Found a user in the settings, try to authenticate
|
|
if username and enforceLogin == 'false':
|
|
LOG.debug('Trying to authenticate with old settings')
|
|
answ = self.load_user(username,
|
|
userId,
|
|
usertoken,
|
|
authenticated=False)
|
|
if answ is True:
|
|
# SUCCESS: loaded a user from the settings
|
|
return True
|
|
elif answ == 401:
|
|
LOG.error("User token no longer valid. Sign user out")
|
|
settings('username', value='')
|
|
settings('userid', value='')
|
|
settings('accessToken', value='')
|
|
else:
|
|
LOG.debug("Could not yet authenticate user")
|
|
return False
|
|
|
|
# Could not use settings - try to get Plex user list from plex.tv
|
|
plextoken = settings('plexToken')
|
|
if plextoken:
|
|
LOG.info("Trying to connect to plex.tv to get a user list")
|
|
userInfo = plex_tv.choose_home_user(plextoken)
|
|
if userInfo is False:
|
|
# FAILURE: Something went wrong, try again
|
|
self.auth = True
|
|
self.retry += 1
|
|
return False
|
|
username = userInfo['username']
|
|
userId = userInfo['userid']
|
|
usertoken = userInfo['token']
|
|
else:
|
|
LOG.info("Trying to authenticate without a token")
|
|
username = ''
|
|
userId = ''
|
|
usertoken = ''
|
|
|
|
if self.load_user(username, userId, usertoken, authenticated=False):
|
|
# SUCCESS: loaded a user from the settings
|
|
return True
|
|
# Something went wrong, try again
|
|
self.auth = True
|
|
self.retry += 1
|
|
return False
|
|
|
|
def reset_client(self):
|
|
"""
|
|
Reset all user settings
|
|
"""
|
|
LOG.debug("Reset UserClient authentication.")
|
|
self.do_utils.stopSession()
|
|
|
|
window('plex_authenticated', clear=True)
|
|
state.AUTHENTICATED = False
|
|
window('pms_token', clear=True)
|
|
state.PLEX_TOKEN = None
|
|
state.PLEX_TRANSIENT_TOKEN = None
|
|
state.PMS_TOKEN = None
|
|
window('plex_token', clear=True)
|
|
window('pms_server', clear=True)
|
|
window('plex_machineIdentifier', clear=True)
|
|
window('plex_servername', clear=True)
|
|
state.PLEX_USER_ID = None
|
|
state.PLEX_USERNAME = None
|
|
window('plex_restricteduser', clear=True)
|
|
state.RESTRICTED_USER = False
|
|
|
|
settings('username', value='')
|
|
settings('userid', value='')
|
|
settings('accessToken', value='')
|
|
|
|
self.token = None
|
|
self.auth = True
|
|
self.user = None
|
|
|
|
self.retry = 0
|
|
|
|
def run(self):
|
|
"""
|
|
Do the work
|
|
"""
|
|
LOG.info("----===## Starting UserClient ##===----")
|
|
stopped = self.stopped
|
|
suspended = self.suspended
|
|
while not stopped():
|
|
while suspended():
|
|
if stopped():
|
|
break
|
|
sleep(1000)
|
|
|
|
if state.PMS_STATUS == "Stop":
|
|
sleep(500)
|
|
continue
|
|
|
|
elif state.PMS_STATUS == "401":
|
|
# Unauthorized access, revoke token
|
|
state.PMS_STATUS = 'Auth'
|
|
window('plex_serverStatus', value='Auth')
|
|
self.reset_client()
|
|
sleep(3000)
|
|
|
|
if self.auth and (self.user is None):
|
|
# Try to authenticate user
|
|
if not state.PMS_STATUS or state.PMS_STATUS == "Auth":
|
|
# Set auth flag because we no longer need
|
|
# to authenticate the user
|
|
self.auth = False
|
|
if self.authenticate():
|
|
# Successfully authenticated and loaded a user
|
|
LOG.info("Successfully authenticated!")
|
|
LOG.info("Current user: %s", self.user)
|
|
LOG.info("Current userId: %s", state.PLEX_USER_ID)
|
|
self.retry = 0
|
|
state.SUSPEND_LIBRARY_THREAD = False
|
|
window('plex_serverStatus', clear=True)
|
|
state.PMS_STATUS = False
|
|
|
|
if not self.auth and (self.user is None):
|
|
# Loop if no server found
|
|
server = self.get_server()
|
|
|
|
# The status Stop is for when user cancelled password dialog.
|
|
# Or retried too many times
|
|
if server and state.PMS_STATUS != "Stop":
|
|
# Only if there's information found to login
|
|
LOG.debug("Server found: %s", server)
|
|
self.auth = True
|
|
|
|
# Minimize CPU load
|
|
sleep(100)
|
|
|
|
LOG.info("##===---- UserClient Stopped ----===##")
|