154 lines
5.4 KiB
Python
154 lines
5.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
#################################################################################################
|
|
|
|
import logging
|
|
|
|
import xbmc
|
|
import xbmcgui
|
|
|
|
import clientinfo
|
|
import connectmanager
|
|
import connect.connectionmanager as connectionmanager
|
|
import userclient
|
|
from utils import settings, language as lang, passwordsXML
|
|
|
|
#################################################################################################
|
|
|
|
log = logging.getLogger("EMBY."+__name__)
|
|
STATE = connectionmanager.ConnectionState
|
|
|
|
#################################################################################################
|
|
|
|
|
|
class InitialSetup(object):
|
|
|
|
|
|
def __init__(self):
|
|
|
|
self.addon_id = clientinfo.ClientInfo().get_addon_id()
|
|
self.user_client = userclient.UserClient()
|
|
self.connectmanager = connectmanager.ConnectManager()
|
|
|
|
|
|
def setup(self):
|
|
# Check server, user, direct paths, music, direct stream if not direct path.
|
|
dialog = xbmcgui.Dialog()
|
|
|
|
log.debug("Initial setup called")
|
|
|
|
if self._server_verification() and settings('userId'):
|
|
# Setup is already completed
|
|
return
|
|
|
|
if not self._user_identification():
|
|
# User failed to identify
|
|
return
|
|
|
|
##### ADDITIONAL PROMPTS #####
|
|
|
|
direct_paths = dialog.yesno(heading=lang(30511),
|
|
line1=lang(33035),
|
|
nolabel=lang(33036),
|
|
yeslabel=lang(33037))
|
|
if direct_paths:
|
|
log.info("User opted to use direct paths")
|
|
settings('useDirectPaths', value="1")
|
|
|
|
# ask for credentials
|
|
credentials = dialog.yesno(heading=lang(30517), line1=lang(33038))
|
|
if credentials:
|
|
log.info("Presenting network credentials dialog")
|
|
passwordsXML()
|
|
|
|
music_disabled = dialog.yesno(heading=lang(29999), line1=lang(33039))
|
|
if music_disabled:
|
|
log.info("User opted to disable Emby music library")
|
|
settings('enableMusic', value="false")
|
|
else:
|
|
# Only prompt if the user didn't select direct paths for videos
|
|
if not direct_paths:
|
|
music_access = dialog.yesno(heading=lang(29999), line1=lang(33040))
|
|
if music_access:
|
|
log.info("User opted to direct stream music")
|
|
settings('streamMusic', value="true")
|
|
|
|
def _server_verification(self):
|
|
|
|
###$ Begin migration $###
|
|
if settings('server') == "":
|
|
current_server = self.user_client.get_server()
|
|
if current_server is not None:
|
|
server = self.connectmanager.get_server(current_server,
|
|
{'ssl': self.user_client.get_ssl()})
|
|
log.info("Detected: %s", server)
|
|
try:
|
|
server_id = server['Servers'][0]['Id']
|
|
settings('serverId', value=server_id)
|
|
except Exception as error:
|
|
log.error(error)
|
|
log.info("server migration completed")
|
|
|
|
self.user_client.get_userid()
|
|
self.user_client.get_token()
|
|
###$ End migration $###
|
|
|
|
if settings('server'):
|
|
current_state = self.connectmanager.get_state()
|
|
if current_state['State'] == STATE['ConnectSignIn']:
|
|
# Failed to identify server
|
|
return True
|
|
|
|
for server in current_state['Servers']:
|
|
if server['Id'] == settings('serverId'):
|
|
# Update token
|
|
server['UserId'] = settings('userId') or None
|
|
server['AccessToken'] = settings('token') or None
|
|
self.connectmanager.update_token(current_state['Servers'], server)
|
|
|
|
server_address = self.connectmanager.get_address(server)
|
|
self._set_server(server_address, server)
|
|
log.info("Found server!")
|
|
|
|
return True
|
|
|
|
return False
|
|
|
|
def _user_identification(self):
|
|
|
|
try:
|
|
server = self.connectmanager.select_servers()
|
|
log.info("Server: %s", server)
|
|
server_address = self.connectmanager.get_address(server)
|
|
self._set_server(server_address, server)
|
|
|
|
if not server.get('AccessToken') and not server.get('UserId'):
|
|
user = self.connectmanager.login(server)
|
|
log.info("User authenticated: %s", user)
|
|
settings('username', value=user['User']['Name'])
|
|
self._set_user(user['User']['Id'], user['AccessToken'])
|
|
else: # Logged with Emby Connect
|
|
user = self.connectmanager.get_state()
|
|
settings('connectUsername', value=user['ConnectUser']['Name'])
|
|
self._set_user(server['UserId'], server['AccessToken'])
|
|
|
|
return True
|
|
|
|
except RuntimeError as error:
|
|
log.exception(error)
|
|
xbmc.executebuiltin('Addon.OpenSettings(%s)' % self.addon_id)
|
|
return False
|
|
|
|
@classmethod
|
|
def _set_server(cls, server_address, server):
|
|
|
|
settings('serverName', value=server['Name'])
|
|
settings('serverId', value=server['Id'])
|
|
settings('server', value=server_address)
|
|
|
|
@classmethod
|
|
def _set_user(cls, user_id, token):
|
|
|
|
settings('userId', value=user_id)
|
|
settings('token', value=token)
|