309 lines
10 KiB
Python
309 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import division, absolute_import, print_function, unicode_literals
|
|
|
|
##################################################################################################
|
|
|
|
import xbmc
|
|
import xbmcaddon
|
|
|
|
from . import client
|
|
from .database import get_credentials, save_credentials
|
|
from .dialogs import ServerConnect, UsersConnect, LoginManual, ServerManual
|
|
from .helper import settings, addon_id, event, api, window, LazyLogger
|
|
from .jellyfin import Jellyfin
|
|
from .jellyfin.connection_manager import CONNECTION_STATE
|
|
from .helper.exceptions import HTTPException
|
|
|
|
##################################################################################################
|
|
|
|
LOG = LazyLogger(__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 = server_id is None and not settings("SyncInstallRunDone.bool")
|
|
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:
|
|
Jellyfin(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 Jellyfin client."""
|
|
client = Jellyfin(server_id)
|
|
client.config.app(
|
|
"Kodi", self.info["Version"], self.info["DeviceName"], self.info["DeviceId"]
|
|
)
|
|
client.config.data["http.user_agent"] = (
|
|
"Jellyfin-Kodi/%s" % self.info["Version"]
|
|
)
|
|
client.config.data["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.data["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.data["auth.server-name"])
|
|
settings("server", client.config.data["auth.server"])
|
|
|
|
event("ServerOnline", {"ServerId": server_id})
|
|
event("LoadServer", {"ServerId": server_id})
|
|
|
|
return state["Credentials"]
|
|
|
|
elif (
|
|
server_selection
|
|
or state["State"] == CONNECTION_STATE["ServerSelection"]
|
|
or state["State"] == CONNECTION_STATE["Unavailable"]
|
|
and not settings("SyncInstallRunDone.bool")
|
|
):
|
|
state["Credentials"]["Servers"] = [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"]
|
|
and state.get("Status_Code", 0) == 401
|
|
):
|
|
# If the saved credentials don't work, restart the addon to force the password dialog to open
|
|
window("jellyfin.restart", clear=True)
|
|
|
|
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.jellyfin.get_user()
|
|
settings("username", self.user["Name"])
|
|
|
|
if "PrimaryImageTag" in self.user:
|
|
server_address = client.auth.get_server_info(client.auth.server_id)[
|
|
"address"
|
|
]
|
|
window(
|
|
"JellyfinUserImage",
|
|
api.API(self.user, server_address).get_user_artwork(self.user["Id"]),
|
|
)
|
|
|
|
def select_servers(self, state=None):
|
|
|
|
state = state or self.connect_manager.connect({"enableAutoLogin": False})
|
|
user = {}
|
|
|
|
dialog = ServerConnect("script-jellyfin-connect-server.xml", *XML_PATH)
|
|
dialog.set_args(
|
|
connect_manager=self.connect_manager,
|
|
username=user.get("DisplayName", ""),
|
|
user_image=user.get("ImageUrl"),
|
|
servers=self.connect_manager.get_available_servers(),
|
|
)
|
|
|
|
dialog.doModal()
|
|
|
|
if dialog.is_server_selected():
|
|
LOG.debug("Server selected: %s", dialog.get_server())
|
|
return dialog.get_server()
|
|
|
|
elif dialog.is_manual_server():
|
|
LOG.debug("Adding manual server")
|
|
try:
|
|
return 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-jellyfin-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 login(self):
|
|
|
|
users = self.connect_manager.get_public_users()
|
|
server = self.connect_manager.get_server_info(self.connect_manager.server_id)[
|
|
"address"
|
|
]
|
|
|
|
if not users:
|
|
try:
|
|
return self.login_manual()
|
|
except RuntimeError:
|
|
raise RuntimeError("No user selected")
|
|
|
|
dialog = UsersConnect("script-jellyfin-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
|
|
|
|
username = settings("username")
|
|
try:
|
|
self.login_manual(user=username, 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-jellyfin-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."""
|
|
Jellyfin(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)
|