jellyfin-kodi/resources/lib/initialsetup.py

# -*- 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().getAddonId()
        self.user_client = userclient.UserClient()
        self.connectmanager = connectmanager.ConnectManager()


    def setup(self):
        # Check server, user, direct paths, music, direct stream if not direct path.
        addon_id = self.addon_id
        dialog = xbmcgui.Dialog()

        ##### SERVER INFO #####

        log.debug("Initial setup called.")

        ###$ Begin transition phase $###
        if settings('server') == "":
            current_server = self.user_client.get_server()
            server = self.connectmanager.get_server(current_server)
            settings('ServerId', value=server['Id'])
            self.user_client.get_userid()
            self.user_client.get_token()
        ###$ End transition phase $###

        if settings('server'):
            current_state = self.connectmanager.get_state()
            for server in current_state['Servers']:
                if server['Id'] == settings('serverId'):
                    server_address = self.connectmanager.get_address(server)
                    self._set_server(server_address, server)
                    self._set_user(server['UserId'], server['AccessToken'])
                    break
            return

        try:
            server = self.connectmanager.select_servers()
            log.info("Server: %s", server)

        except RuntimeError as error:
            log.exception(error)
            xbmc.executebuiltin('Addon.OpenSettings(%s)' % addon_id)
            return

        else:
            server_address = self.connectmanager.get_address(server)
            self._set_server(server_address, server)

            if not server.get('AccessToken') and not server.get('UserId'):
                try:
                    user = self.connectmanager.login(server)
                    log.info("User authenticated: %s", user)
                except RuntimeError as error:
                    log.exception(error)
                    xbmc.executebuiltin('Addon.OpenSettings(%s)' % addon_id)
                    return
                settings('username', value=user['User']['Name'])
                self._set_user(user['User']['Id'], user['AccessToken'])
            else:
                user = self.connectmanager.get_state()
                settings('connectUsername', value=user['ConnectUser']['Name'])
                self._set_user(server['UserId'], server['AccessToken'])

        ##### 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")

    @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)