mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2024-11-14 06:06:12 +00:00
241 lines
8 KiB
Python
241 lines
8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
##################################################################################################
|
|
|
|
import logging
|
|
|
|
import xbmc
|
|
import xbmcaddon
|
|
import xbmcvfs
|
|
|
|
import clientinfo
|
|
import read_embyserver as embyserver
|
|
import connect.connectionmanager as connectionmanager
|
|
from dialogs import ServerConnect, UsersConnect, LoginConnect, LoginManual, ServerManual
|
|
from ga_client import GoogleAnalytics
|
|
from utils import window
|
|
|
|
##################################################################################################
|
|
|
|
log = logging.getLogger("EMBY."+__name__)
|
|
addon = xbmcaddon.Addon(id='plugin.video.emby')
|
|
|
|
STATE = connectionmanager.ConnectionState
|
|
XML_PATH = (addon.getAddonInfo('path'), "default", "1080i")
|
|
|
|
##################################################################################################
|
|
|
|
class ConnectManager(object):
|
|
|
|
_shared_state = {} # Borg
|
|
state = {}
|
|
|
|
|
|
def __init__(self):
|
|
|
|
self.__dict__ = self._shared_state
|
|
|
|
client_info = clientinfo.ClientInfo()
|
|
self.emby = embyserver.Read_EmbyServer()
|
|
|
|
version = client_info.get_version()
|
|
device_name = client_info.get_device_name()
|
|
device_id = client_info.get_device_id()
|
|
self._connect = connectionmanager.ConnectionManager(appName="Kodi",
|
|
appVersion=version,
|
|
deviceName=device_name,
|
|
deviceId=device_id)
|
|
path = xbmc.translatePath(
|
|
"special://profile/addon_data/plugin.video.emby/").decode('utf-8')
|
|
|
|
if not xbmcvfs.exists(path):
|
|
xbmcvfs.mkdirs(path)
|
|
|
|
self._connect.setFilePath(path)
|
|
|
|
if window('emby_state.json'):
|
|
self.state = window('emby_state.json')
|
|
|
|
elif not self.state:
|
|
self.state = self._connect.connect()
|
|
log.info("Started with: %s", self.state)
|
|
window('emby_state.json', value=self.state)
|
|
|
|
|
|
def update_state(self):
|
|
self.state = self._connect.connect({'updateDateLastAccessed': False})
|
|
return self.get_state()
|
|
|
|
def get_state(self):
|
|
window('emby_state.json', value=self.state)
|
|
return self.state
|
|
|
|
def get_server(self, server, options={}):
|
|
self.state = self._connect.connectToAddress(server, options)
|
|
return self.get_state()
|
|
|
|
@classmethod
|
|
def get_address(cls, server):
|
|
return connectionmanager.getServerAddress(server, server['LastConnectionMode'])
|
|
|
|
def clear_data(self):
|
|
self._connect.clearData()
|
|
|
|
def select_servers(self):
|
|
# Will return selected server or raise error
|
|
state = self._connect.connect({'enableAutoLogin': False})
|
|
user = state.get('ConnectUser') or {}
|
|
|
|
dialog = ServerConnect("script-emby-connect-server.xml", *XML_PATH)
|
|
kwargs = {
|
|
'connect_manager': self._connect,
|
|
'username': user.get('DisplayName', ""),
|
|
'user_image': user.get('ImageUrl'),
|
|
'servers': state.get('Servers') or [],
|
|
'emby_connect': False if user else True
|
|
}
|
|
dialog.set_args(**kwargs)
|
|
dialog.doModal()
|
|
|
|
if dialog.is_server_selected():
|
|
log.debug("Server selected")
|
|
return dialog.get_server()
|
|
|
|
elif dialog.is_connect_login():
|
|
log.debug("Login with Emby Connect")
|
|
try: # Login to emby connect
|
|
self.login_connect()
|
|
except RuntimeError:
|
|
pass
|
|
return self.select_servers()
|
|
|
|
elif dialog.is_manual_server():
|
|
log.debug("Add manual server")
|
|
try: # Add manual server address
|
|
return self.manual_server()
|
|
except RuntimeError:
|
|
return self.select_servers()
|
|
else:
|
|
raise RuntimeError("No server selected")
|
|
|
|
def manual_server(self):
|
|
# Return server or raise error
|
|
dialog = ServerManual("script-emby-connect-server-manual.xml", *XML_PATH)
|
|
dialog.set_connect_manager(self._connect)
|
|
dialog.doModal()
|
|
|
|
if dialog.is_connected():
|
|
return dialog.get_server()
|
|
else:
|
|
raise RuntimeError("Server is not connected")
|
|
|
|
def login_connect(self):
|
|
# Return connect user or raise error
|
|
dialog = LoginConnect("script-emby-connect-login.xml", *XML_PATH)
|
|
dialog.set_connect_manager(self._connect)
|
|
dialog.doModal()
|
|
|
|
self.update_state()
|
|
|
|
if dialog.is_logged_in():
|
|
return dialog.get_user()
|
|
else:
|
|
raise RuntimeError("Connect user is not logged in")
|
|
|
|
def login(self, server=None):
|
|
|
|
ga = GoogleAnalytics()
|
|
ga.sendEventData("Connect", "UserLogin")
|
|
|
|
# Return user or raise error
|
|
server = server or self.state['Servers'][0]
|
|
server_address = connectionmanager.getServerAddress(server, server['LastConnectionMode'])
|
|
|
|
users = "";
|
|
try:
|
|
users = self.emby.getUsers(server_address)
|
|
except Exception as error:
|
|
log.info("Error getting users from server: " + str(error))
|
|
|
|
if not users:
|
|
try:
|
|
return self.login_manual(server_address)
|
|
except RuntimeError:
|
|
raise RuntimeError("No user selected")
|
|
|
|
dialog = UsersConnect("script-emby-connect-users.xml", *XML_PATH)
|
|
dialog.set_server(server_address)
|
|
dialog.set_users(users)
|
|
dialog.doModal()
|
|
|
|
if dialog.is_user_selected():
|
|
|
|
user = dialog.get_user()
|
|
username = user['Name']
|
|
|
|
if user['HasPassword']:
|
|
log.debug("User has password, present manual login")
|
|
try:
|
|
return self.login_manual(server_address, username)
|
|
except RuntimeError:
|
|
return self.login(server)
|
|
else:
|
|
try:
|
|
user = self.emby.loginUser(server_address, username)
|
|
except Exception as error:
|
|
log.info("Error logging in user: " + str(error))
|
|
raise
|
|
|
|
self._connect.onAuthenticated(user)
|
|
return user
|
|
|
|
elif dialog.is_manual_login():
|
|
try:
|
|
return self.login_manual(server_address)
|
|
except RuntimeError:
|
|
return self.login(server)
|
|
else:
|
|
raise RuntimeError("No user selected")
|
|
|
|
def login_manual(self, server, user=None):
|
|
# Return manual login user authenticated or raise error
|
|
dialog = LoginManual("script-emby-connect-login-manual.xml", *XML_PATH)
|
|
dialog.set_server(server)
|
|
dialog.set_user(user)
|
|
dialog.doModal()
|
|
|
|
if dialog.is_logged_in():
|
|
user = dialog.get_user()
|
|
self._connect.onAuthenticated(user)
|
|
return user
|
|
else:
|
|
raise RuntimeError("User is not authenticated")
|
|
|
|
def update_token(self, server):
|
|
|
|
credentials = self._connect.credentialProvider.getCredentials()
|
|
self._connect.credentialProvider.addOrUpdateServer(credentials['Servers'], server)
|
|
|
|
for server in self.get_state()['Servers']:
|
|
for cred_server in credentials['Servers']:
|
|
if server['Id'] == cred_server['Id']:
|
|
# Update token saved in current state
|
|
server.update(cred_server)
|
|
# Update the token in data.txt
|
|
self._connect.credentialProvider.getCredentials(credentials)
|
|
|
|
def get_connect_servers(self):
|
|
|
|
connect_servers = []
|
|
servers = self._connect.getAvailableServers()
|
|
for server in servers:
|
|
if 'ExchangeToken' in server:
|
|
result = self.connect_server(server)
|
|
if result['State'] == STATE['SignedIn']:
|
|
connect_servers.append(server)
|
|
|
|
log.info(connect_servers)
|
|
return connect_servers
|
|
|
|
def connect_server(self, server):
|
|
return self._connect.connectToServer(server, {'updateDateLastAccessed': False})
|