PlexKodiConnect/resources/lib/DownloadUtils.py

import xbmc
import xbmcgui
import xbmcaddon

import requests
import json
import logging

import Utils as utils
from ClientInformation import ClientInformation
from requests.packages.urllib3.exceptions import InsecureRequestWarning

# Disable requests logging
# requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
# logging.getLogger("requests").setLevel(logging.WARNING)

class DownloadUtils():
    
    # Borg - multiple instances, shared state
    _shared_state = {}
    clientInfo = ClientInformation()

    addonName = clientInfo.getAddonName()
    addonId = clientInfo.getAddonId()
    addon = xbmcaddon.Addon(id=addonId)
    WINDOW = xbmcgui.Window(10000)

    # Requests session
    s = None
    timeout = 30

    def __init__(self):

        self.__dict__ = self._shared_state

    def logMsg(self, msg, lvl=1):

        self.className = self.__class__.__name__
        utils.logMsg("%s %s" % (self.addonName, self.className), msg, int(lvl))

    def setUsername(self, username):
        # Reserved for UserClient only
        self.username = username
        self.logMsg("Set username: %s" % username, 2)

    def setUserId(self, userId):
        # Reserved for UserClient only
        self.userId = userId
        self.logMsg("Set userId: %s" % userId, 2)

    def setServer(self, server):
        # Reserved for UserClient only
        self.server = server
        self.logMsg("Set server: %s" % server, 2)

    def setToken(self, token):
        # Reserved for UserClient only
        self.token = token
        self.logMsg("Set token: %s" % token, 2)

    def setSSL(self, ssl, sslclient):
        # Reserved for UserClient only
        self.sslverify = ssl
        self.sslclient = sslclient
        self.logMsg("Verify SSL host certificate: %s" % ssl, 2)
        self.logMsg("SSL client side certificate: %s" % sslclient, 2)

    def postCapabilities(self, deviceId):

        # Get sessionId
        url = "{server}/mediabrowser/Sessions?DeviceId=%s&format=json" % deviceId
        result = self.downloadUrl(url)
        # sessionId result
        self.logMsg("Session result: %s" % result, 2)
        self.sessionId = result[0][u'Id']
        self.WINDOW.setProperty('sessionId%s' % self.username, self.sessionId)

        # Settings for capabilities
        playableMediaTypes = "Audio,Video"
        supportedCommands = "Play,Playstate,SendString,DisplayMessage,PlayNext"

        # Post settings to sessionId
        url = "{server}/mediabrowser/Sessions/Capabilities?Id=%s&PlayableMediaTypes=%s&SupportedCommands=%s&SupportsMediaControl=True" % (self.sessionId, playableMediaTypes, supportedCommands)
        data = {}
        self.logMsg("Capabilities URL: %s" % url, 2)
        self.logMsg("PostData: %s" % data, 2)

        self.downloadUrl(url, postBody=data, type="POST")
        self.logMsg("Posted capabilities to sessionId: %s" % self.sessionId, 1)

    def startSession(self):

        self.deviceId = self.clientInfo.getMachineId()

        # User is identified from this point
        # Attach authenticated header to the session
        verify = None
        cert = None
        header = self.getHeader()

        # If user enabled host certificate verification
        if self.sslverify:
            verify = True
            cert = self.sslclient
        
        # Start session
        self.s = requests.Session()
        self.s.headers = header
        self.s.verify = verify
        self.s.cert = cert
        # Retry connections to the server
        self.s.mount("http://", requests.adapters.HTTPAdapter(max_retries=1))
        self.s.mount("https://", requests.adapters.HTTPAdapter(max_retries=1))

        self.logMsg("Requests session started on: %s" % self.server)
        self.postCapabilities(self.deviceId)

    def imageUrl(self, id, type, index, width, height):
        # To move to API.py
        return "%s/mediabrowser/Items/%s/Images/%s?MaxWidth=%s&MaxHeight=%s&Index=%s" % (self.server, id, type, width, height, index)

    def getHeader(self, authenticate=True):

        clientInfo = self.clientInfo

        deviceName = clientInfo.getDeviceName()
        deviceId = clientInfo.getMachineId()
        version = clientInfo.getVersion()

        if not authenticate:
            # If user is not authenticated
            auth = 'MediaBrowser Client="Kodi", Device="%s", DeviceId="%s", Version="%s"' % (deviceName, deviceId, version)
            header = {'Content-type': 'application/json', 'Accept-encoding': 'gzip', 'Accept-Charset': 'UTF-8,*', 'Authorization': auth}      
            
            self.logMsg("Header: %s" % header, 2)
            return header
        
        else:
            userId = self.userId
            token = self.token
            # Attached to the requests session
            auth = 'MediaBrowser UserId="%s", Client="Kodi", Device="%s", DeviceId="%s", Version="%s"' % (userId, deviceName, deviceId, version)
            header = {'Content-type': 'application/json', 'Accept-encoding': 'gzip', 'Accept-Charset': 'UTF-8,*', 'Authorization': auth, 'X-MediaBrowser-Token': token}        
                    
            self.logMsg("Header: %s" % header, 2)
            return header

    def downloadUrl(self, url, postBody=None, type="GET", authenticate=True):
        
        self.logMsg("=== ENTER downloadUrl ===", 2)

        WINDOW = xbmcgui.Window( 10000 )
        timeout = self.timeout
        default_link = ""
        
        username = WINDOW.getProperty('currUser')
        userId = WINDOW.getProperty('userId%s' % username)
        server = WINDOW.getProperty('server%s' % username)
        url = url.replace("{server}", server, 1)
        url = url.replace("{UserId}", userId, 1)
        #url = "%s&api_key=%s" % (url, self.token)

        # If user is authenticated
        if (authenticate):
            # Get requests session
            s = self.s
            # Replace for the real values and append api_key

            self.logMsg("URL: %s" % url, 2)
            # Prepare request
            if type == "GET":
                r = s.get(url, json=postBody, timeout=timeout)
            elif type == "POST":
                r = s.post(url, json=postBody, timeout=timeout)
            elif type == "DELETE":
                r = s.delete(url, json=postBody, timeout=timeout)

        # If user is not authenticated
        elif not authenticate:
            
            self.logMsg("URL: %s" % url, 1)
            header = self.getHeader(authenticate=False)
            verifyssl = False

            # If user enables ssl verification
            try:
                verifyssl = self.sslverify
            except AttributeError:
                pass
            
            # Prepare request
            if type == "GET":
                r = requests.get(url, json=postBody, headers=header, timeout=timeout, verify=verifyssl)
            elif type == "POST":
                r = requests.post(url, json=postBody, headers=header, timeout=timeout, verify=verifyssl)
        
        # Process the response
        try:
            r.raise_for_status()

            if r.status_code == 204:
                # No response in body
                self.logMsg("====== 204 Success ======", 2)
                return default_link
            # Response code 200
            elif r.status_code == requests.codes.ok:
                try: 
                    # UTF-8 - JSON object
                    r = r.json()
                    self.logMsg("====== 200 Success ======", 2)
                    return r
                except:
                    self.logMsg("Unable to convert the response for: %s" % url, 1)

            return default_link
        
        # TO REVIEW EXCEPTIONS
        except requests.exceptions.ConnectionError as e:
            self.logMsg("Server unreachable at: %s" % url, 0)
            self.logMsg(e, 1)

        except requests.exceptions.ConnectTimeout as e:
            self.logMsg("Server timeout at: %s" % url, 0)
            self.logMsg(e, 1)

        except requests.exceptions.HTTPError as e:

            if r.status_code == 401:
                # Unauthorized
                status = WINDOW.getProperty("Server_status")
                if (status == "401") or (status == "Auth"):
                    pass
                else:
                    # Tell UserClient token has been revoked.
                    WINDOW.setProperty("Server_status", "401")
                    self.logMsg("HTTP Error: %s" % e, 0)

            elif (r.status_code == 301) or (r.status_code == 302):
                # Redirects
                pass
            elif r.status_code == 400:
                # Bad requests
                pass

        except requests.exceptions.SSLError as e:
            self.logMsg("Invalid SSL certificate for: %s" % url, 0)
            self.logMsg(e, 1)

        except requests.exceptions.RequestException as e:
            self.logMsg("Unknown error connecting to: %s" % url, 0)
            self.logMsg(e, 1)

        return default_link