mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2024-11-10 12:16:12 +00:00
c76f19e42d
With .55 the resume setting is set per library. Instead query server to see if the item is played to offer delete
330 lines
10 KiB
Python
330 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
##################################################################################################
|
|
|
|
import json
|
|
import logging
|
|
import os
|
|
|
|
import xbmc
|
|
import xbmcaddon
|
|
import xbmcvfs
|
|
|
|
import client
|
|
from database import get_credentials, save_credentials
|
|
from dialogs import ServerConnect, UsersConnect, LoginConnect, LoginManual, ServerManual
|
|
from helper import _, settings, addon_id, event, api, dialog, window
|
|
from emby import Emby
|
|
from emby.core.connection_manager import get_server_address, CONNECTION_STATE
|
|
from emby.core.exceptions import HTTPException
|
|
|
|
##################################################################################################
|
|
|
|
LOG = logging.getLogger("EMBY."+__name__)
|
|
XML_PATH = (xbmcaddon.Addon(addon_id()).getAddonInfo('path'), "default", "1080i")
|
|
|
|
##################################################################################################
|
|
|
|
|
|
class Connect(object):
|
|
|
|
def __init__(self):
|
|
self.info = client.get_info()
|
|
|
|
def register(self, server_id=None, options={}):
|
|
|
|
''' Login into server. If server is None, then it will show the proper prompts to login, etc.
|
|
If a server id is specified then only a login dialog will be shown for that server.
|
|
'''
|
|
LOG.info("--[ server/%s ]", server_id or 'default')
|
|
credentials = dict(get_credentials())
|
|
servers = credentials['Servers']
|
|
|
|
if server_id is None and credentials['Servers']:
|
|
credentials['Servers'] = [credentials['Servers'][0]]
|
|
|
|
elif credentials['Servers']:
|
|
|
|
for server in credentials['Servers']:
|
|
|
|
if server['Id'] == server_id:
|
|
credentials['Servers'] = [server]
|
|
|
|
server_select = True if server_id is None and not settings('SyncInstallRunDone.bool') else False
|
|
new_credentials = self.register_client(credentials, options, server_id, server_select)
|
|
|
|
for server in servers:
|
|
if server['Id'] == new_credentials['Servers'][0]['Id']:
|
|
server = new_credentials['Servers'][0]
|
|
|
|
break
|
|
else:
|
|
servers = new_credentials['Servers']
|
|
|
|
credentials['Servers'] = servers
|
|
save_credentials(credentials)
|
|
|
|
try:
|
|
Emby(server_id).start(True)
|
|
except ValueError as error:
|
|
LOG.error(error)
|
|
|
|
def get_ssl(self):
|
|
|
|
''' Returns boolean value.
|
|
True: verify connection.
|
|
'''
|
|
return settings('sslverify.bool')
|
|
|
|
def get_client(self, server_id=None):
|
|
|
|
''' Get Emby client.
|
|
'''
|
|
client = Emby(server_id)
|
|
client['config/app']("Kodi", self.info['Version'], self.info['DeviceName'], self.info['DeviceId'])
|
|
client['config']['http.user_agent'] = "Emby-Kodi/%s" % self.info['Version']
|
|
client['config']['auth.ssl'] = self.get_ssl()
|
|
|
|
return client
|
|
|
|
def register_client(self, credentials=None, options=None, server_id=None, server_selection=False):
|
|
|
|
client = self.get_client(server_id)
|
|
self.client = client
|
|
self.connect_manager = client.auth
|
|
|
|
if server_id is None:
|
|
client['config']['app.default'] = True
|
|
|
|
try:
|
|
state = client.authenticate(credentials or {}, options or {})
|
|
|
|
if state['State'] == CONNECTION_STATE['SignedIn']:
|
|
client.callback_ws = event
|
|
|
|
if server_id is None: # Only assign for default server
|
|
|
|
client.callback = event
|
|
self.get_user(client)
|
|
|
|
settings('serverName', client['config/auth.server-name'])
|
|
settings('server', client['config/auth.server'])
|
|
|
|
event('ServerOnline', {'ServerId': server_id})
|
|
event('LoadServer', {'ServerId': server_id})
|
|
|
|
return state['Credentials']
|
|
|
|
elif (server_selection or state['State'] in (CONNECTION_STATE['ConnectSignIn'], CONNECTION_STATE['ServerSelection']) or
|
|
state['State'] == CONNECTION_STATE['Unavailable'] and not settings('SyncInstallRunDone.bool')):
|
|
|
|
self.select_servers(state)
|
|
|
|
elif state['State'] == CONNECTION_STATE['ServerSignIn']:
|
|
if 'ExchangeToken' not in state['Servers'][0]:
|
|
self.login()
|
|
|
|
elif state['State'] == CONNECTION_STATE['Unavailable']:
|
|
raise HTTPException('ServerUnreachable', {})
|
|
|
|
return self.register_client(state['Credentials'], options, server_id, False)
|
|
|
|
except RuntimeError as error:
|
|
|
|
LOG.exception(error)
|
|
xbmc.executebuiltin('Addon.OpenSettings(%s)' % addon_id())
|
|
|
|
raise Exception('User sign in interrupted')
|
|
|
|
except HTTPException as error:
|
|
|
|
if error.status == 'ServerUnreachable':
|
|
event('ServerUnreachable', {'ServerId': server_id})
|
|
|
|
return client.get_credentials()
|
|
|
|
|
|
def get_user(self, client):
|
|
|
|
''' Save user info.
|
|
'''
|
|
self.user = client['api'].get_user()
|
|
settings('username', self.user['Name'])
|
|
|
|
if 'PrimaryImageTag' in self.user:
|
|
window('EmbyUserImage', api.API(self.user, client['auth/server-address']).get_user_artwork(self.user['Id']))
|
|
|
|
def select_servers(self, state=None):
|
|
|
|
state = state or self.connect_manager.connect({'enableAutoLogin': False})
|
|
user = state.get('ConnectUser') or {}
|
|
|
|
dialog = ServerConnect("script-emby-connect-server.xml", *XML_PATH)
|
|
dialog.set_args(**{
|
|
'connect_manager': self.connect_manager,
|
|
'username': user.get('DisplayName', ""),
|
|
'user_image': user.get('ImageUrl'),
|
|
'servers': state.get('Servers', []),
|
|
'emby_connect': False if user else True
|
|
})
|
|
dialog.doModal()
|
|
|
|
if dialog.is_server_selected():
|
|
LOG.debug("Server selected: %s", dialog.get_server())
|
|
return
|
|
|
|
elif dialog.is_connect_login():
|
|
LOG.debug("Login with emby connect")
|
|
try:
|
|
self.login_connect()
|
|
except RuntimeError: pass
|
|
|
|
elif dialog.is_manual_server():
|
|
LOG.debug("Adding manual server")
|
|
try:
|
|
self.manual_server()
|
|
except RuntimeError: pass
|
|
else:
|
|
raise RuntimeError("No server selected")
|
|
|
|
return self.select_servers()
|
|
|
|
def setup_manual_server(self):
|
|
|
|
''' Setup manual servers
|
|
'''
|
|
client = self.get_client()
|
|
client.set_credentials(get_credentials())
|
|
manager = client.auth
|
|
|
|
try:
|
|
self.manual_server(manager)
|
|
except RuntimeError:
|
|
return
|
|
|
|
credentials = client.get_credentials()
|
|
save_credentials(credentials)
|
|
|
|
def manual_server(self, manager=None):
|
|
|
|
''' Return server or raise error.
|
|
'''
|
|
dialog = ServerManual("script-emby-connect-server-manual.xml", *XML_PATH)
|
|
dialog.set_args(**{'connect_manager': manager or self.connect_manager})
|
|
dialog.doModal()
|
|
|
|
if dialog.is_connected():
|
|
return dialog.get_server()
|
|
else:
|
|
raise RuntimeError("Server is not connected")
|
|
|
|
def setup_login_connect(self):
|
|
|
|
''' Setup emby connect by itself.
|
|
'''
|
|
client = self.get_client()
|
|
client.set_credentials(get_credentials())
|
|
manager = client.auth
|
|
|
|
try:
|
|
self.login_connect(manager)
|
|
except RuntimeError:
|
|
return
|
|
|
|
credentials = client.get_credentials()
|
|
save_credentials(credentials)
|
|
|
|
def login_connect(self, manager=None):
|
|
|
|
''' Return connect user or raise error.
|
|
'''
|
|
dialog = LoginConnect("script-emby-connect-login.xml", *XML_PATH)
|
|
dialog.set_args(**{'connect_manager': manager or self.connect_manager})
|
|
dialog.doModal()
|
|
|
|
if dialog.is_logged_in():
|
|
return dialog.get_user()
|
|
else:
|
|
raise RuntimeError("Connect user is not logged in")
|
|
|
|
def login(self):
|
|
|
|
users = self.connect_manager['public-users']
|
|
server = self.connect_manager['server-address']
|
|
|
|
if not users:
|
|
try:
|
|
return self.login_manual()
|
|
except RuntimeError:
|
|
raise RuntimeError("No user selected")
|
|
|
|
dialog = UsersConnect("script-emby-connect-users.xml", *XML_PATH)
|
|
dialog.set_args(**{'server': server, '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(username)
|
|
except RuntimeError: pass
|
|
else:
|
|
return self.connect_manager['login'](server, username)
|
|
|
|
elif dialog.is_manual_login():
|
|
try:
|
|
return self.login_manual()
|
|
except RuntimeError: pass
|
|
else:
|
|
raise RuntimeError("No user selected")
|
|
|
|
return self.login()
|
|
|
|
def setup_login_manual(self):
|
|
|
|
''' Setup manual login by itself for default server.
|
|
'''
|
|
client = self.get_client()
|
|
client.set_credentials(get_credentials())
|
|
manager = client.auth
|
|
|
|
try:
|
|
self.login_manual(manager=manager)
|
|
except RuntimeError:
|
|
return
|
|
|
|
credentials = client.get_credentials()
|
|
save_credentials(credentials)
|
|
|
|
def login_manual(self, user=None, manager=None):
|
|
|
|
''' Return manual login user authenticated or raise error.
|
|
'''
|
|
dialog = LoginManual("script-emby-connect-login-manual.xml", *XML_PATH)
|
|
dialog.set_args(**{'connect_manager': manager or self.connect_manager, 'username': user or {}})
|
|
dialog.doModal()
|
|
|
|
if dialog.is_logged_in():
|
|
return dialog.get_user()
|
|
else:
|
|
raise RuntimeError("User is not authenticated")
|
|
|
|
def remove_server(self, server_id):
|
|
|
|
''' Stop client and remove server.
|
|
'''
|
|
Emby(server_id).close()
|
|
credentials = get_credentials()
|
|
|
|
for server in credentials['Servers']:
|
|
if server['Id'] == server_id:
|
|
credentials['Servers'].remove(server)
|
|
|
|
break
|
|
|
|
save_credentials(credentials)
|
|
LOG.info("[ remove server ] %s", server_id)
|
|
|