Version 1.1.63

alpha ready for beta testing

resources/lib/initialsetup.py

@@ -0,0 +1,192 @@
+# -*- coding: utf-8 -*-
+
+#################################################################################################
+
+import json
+import socket
+
+import xbmc
+import xbmcgui
+import xbmcaddon
+
+import utils
+import clientinfo
+import downloadutils
+import userclient
+
+#################################################################################################
+
+
+class InitialSetup():
+
+
+    def __init__(self):
+
+        self.addon = xbmcaddon.Addon()
+        self.__language__ = self.addon.getLocalizedString
+
+        self.clientInfo = clientinfo.ClientInfo()
+        self.addonName = self.clientInfo.getAddonName()
+        self.addonId = self.clientInfo.getAddonId()
+        self.doUtils = downloadutils.DownloadUtils()
+        self.userClient = userclient.UserClient()
+    
+    def logMsg(self, msg, lvl=1):
+
+        className = self.__class__.__name__
+        utils.logMsg("%s %s" % (self.addonName, className), msg, lvl)
+
+
+    def setup(self):
+        # Check server, user, direct paths, music, direct stream if not direct path.
+        string = self.__language__
+        addonId = self.addonId
+
+        ##### SERVER INFO #####
+        
+        self.logMsg("Initial setup called.", 2)
+        server = self.userClient.getServer()
+
+        if server:
+            self.logMsg("Server is already set.", 2)
+            return
+        
+        self.logMsg("Looking for server...", 2)
+        server = self.getServerDetails()
+        self.logMsg("Found: %s" % server, 2)
+        try:
+            prefix, ip, port = server.replace("/", "").split(":")
+        except: # Failed to retrieve server information
+            self.logMsg("getServerDetails failed.", 1)
+            xbmc.executebuiltin('Addon.OpenSettings(%s)' % addonId)
+            return
+        else:
+            server_confirm = xbmcgui.Dialog().yesno(
+                                            heading="Emby for Kodi",
+                                            line1="Proceed with the following server?",
+                                            line2="%s %s" % (string(30169), server))
+            if server_confirm:
+                # Correct server found
+                self.logMsg("Server is selected. Saving the information.", 1)
+                utils.settings('ipaddress', value=ip)
+                utils.settings('port', value=port)
+
+                if prefix == "https":
+                    utils.settings('https', value="true")
+            else:
+                # User selected no or cancelled the dialog
+                self.logMsg("No server selected.", 1)
+                xbmc.executebuiltin('Addon.OpenSettings(%s)' % addonId)
+                return
+
+        ##### USER INFO #####
+        
+        self.logMsg("Getting user list.", 1)
+        
+        url = "%s/emby/Users/Public?format=json" % server
+        result = self.doUtils.downloadUrl(url, authenticate=False)
+        if result == "":
+            self.logMsg("Unable to connect to %s" % server, 1)
+            return
+
+        self.logMsg("Response: %s" % result, 2)
+        # Process the list of users
+        usernames = []
+        users_hasPassword = []
+
+        for user in result:
+            # Username
+            name = user['Name']
+            usernames.append(name)
+            # Password
+            if user['HasPassword']:
+                name = "%s (secure)" % name
+            users_hasPassword.append(name)
+
+        self.logMsg("Presenting user list: %s" % users_hasPassword, 1)
+        user_select = xbmcgui.Dialog().select(string(30200), users_hasPassword)
+        if user_select > -1:
+            selected_user = usernames[user_select]
+            self.logMsg("Selected user: %s" % selected_user, 1)
+            utils.settings('username', value=selected_user)
+        else:
+            self.logMsg("No user selected.", 1)
+            xbmc.executebuiltin('Addon.OpenSettings(%s)' % addonId)
+
+        ##### ADDITIONAL PROMPTS #####
+        dialog = xbmcgui.Dialog()
+
+        directPaths = dialog.yesno(
+                            heading="Playback Mode",
+                            line1=(
+                                "Caution! If you choose Native mode, you "
+                                "will lose access to certain Emby features such as: "
+                                "Emby cinema mode, direct stream/transcode options, "
+                                "parental access schedule."),
+                            nolabel="Addon (Default)",
+                            yeslabel="Native (Direct Paths)")
+        if directPaths:
+            self.logMsg("User opted to use direct paths.", 1)
+            utils.settings('useDirectPaths', value="1")
+
+            # ask for credentials
+            credentials = dialog.yesno(
+                                heading="Network credentials",
+                                line1= (
+                                    "Add network credentials to allow Kodi access to your "
+                                    "content? Note: Skipping this step may generate a message "
+                                    "during the initial scan of your content if Kodi can't "
+                                    "locate your content."))
+            if credentials:
+                self.logMsg("Presenting network credentials dialog.", 1)
+                utils.passwordsXML()
+        
+        musicDisabled = dialog.yesno(
+                            heading="Music Library",
+                            line1="Disable Emby music library?")
+        if musicDisabled:
+            self.logMsg("User opted to disable Emby music library.", 1)
+            utils.settings('enableMusic', value="false")
+        else:
+            # Only prompt if the user didn't select direct paths for videos
+            if not directPaths:
+                musicAccess = dialog.yesno(
+                                    heading="Music Library",
+                                    line1=(
+                                        "Direct stream the music library? Select "
+                                        "this option only if you plan on listening "
+                                        "to music outside of your network."))
+                if musicAccess:
+                    self.logMsg("User opted to direct stream music.", 1)
+                    utils.settings('streamMusic', value="true")
+                
+    def getServerDetails(self):
+
+        self.logMsg("Getting Server Details from Network", 1)
+        
+        MULTI_GROUP = ("<broadcast>", 7359)
+        MESSAGE = "who is EmbyServer?"
+        
+        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        sock.settimeout(6.0)
+
+        sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 20)
+        
+        sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
+        sock.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_LOOP, 1)
+        sock.setsockopt(socket.IPPROTO_IP, socket.SO_REUSEADDR, 1)
+        
+        self.logMsg("MultiGroup      : %s" % str(MULTI_GROUP), 2);
+        self.logMsg("Sending UDP Data: %s" % MESSAGE, 2);
+        sock.sendto(MESSAGE, MULTI_GROUP)
+    
+        try:
+            data, addr = sock.recvfrom(1024) # buffer size is 1024 bytes
+            self.logMsg("Received Response: %s" % data)
+        except:
+            self.logMsg("No UDP Response")
+            return None
+        else:
+            # Get the address
+            data = json.loads(data)
+            return data['Address']