jellyfin-kodi/jellyfin_kodi/jellyfin/http.py

# -*- coding: utf-8 -*-
from __future__ import division, absolute_import, print_function, unicode_literals

#################################################################################################

import json
import logging
import time

import requests
from six import string_types

from .exceptions import HTTPException
from helper.utils import JsonDebugPrinter

#################################################################################################

LOG = logging.getLogger('Jellyfin.' + __name__)

#################################################################################################


class HTTP(object):

    session = None
    keep_alive = False

    def __init__(self, client):

        self.client = client
        self.config = client.config

    def start_session(self):

        self.session = requests.Session()

        max_retries = self.config.data['http.max_retries']
        self.session.mount("http://", requests.adapters.HTTPAdapter(max_retries=max_retries))
        self.session.mount("https://", requests.adapters.HTTPAdapter(max_retries=max_retries))

    def stop_session(self):

        if self.session is None:
            return

        try:
            LOG.info("--<[ session/%s ]", id(self.session))
            self.session.close()
        except Exception as error:
            LOG.warning("The requests session could not be terminated: %s", error)

    def _replace_user_info(self, config, string):

        if '{server}' in string:
            if config.get('auth.server', None):
                string = string.replace("{server}", config['auth.server'])
            else:
                LOG.debug("Server address not set")

        if '{UserId}'in string:
            if config.get('auth.user_id', None):
                string = string.replace("{UserId}", config['auth.user_id'])
            else:
                LOG.debug("UserId is not set.")

        return string


    def get_handler_url(self, handler):
        server = self.config.data.get("auth.server", "")
        return "%s/%s" % (server, handler)


    def REQUEST(self, url, type, params=None, json=None, session=None, \
                    headers={}, verify=None, timeout=None, retry=5):
        config = self.config.data

        url = self._replace_user_info(config, url)
        optional = {}
        if params is not None:
            optional['params'] = params
        if json is not None:
            optional['json'] = json

        timeout = timeout if timeout is not None else config['http.timeout']
        verify = verify if verify is not None else config.get('auth.ssl', False)

        headers = self._get_header(config, headers)
        self._process_params(optional.get('params') or {})
        self._process_params(optional.get('json') or {})

        LOG.debug("--->[ http ] %s", JsonDebugPrinter({
            'url': url,
            'headers': headers,
            'optional': optional,
        }))

        while True:

            try:
                r = self._requests(session or self.session or requests, type, 
                        url=url, headers=headers, verify=verify, timeout=timeout, **optional)
                r.content  # release the connection

                if not self.keep_alive:
                    self.stop_session()

                r.raise_for_status()

            except requests.exceptions.ConnectionError as error:
                if retry:

                    retry -= 1
                    time.sleep(1)

                    continue

                LOG.error(error)
                self.client.callback("ServerUnreachable", {'ServerId': config['auth.server-id']})

                raise HTTPException("ServerUnreachable", error)

            except requests.exceptions.ReadTimeout as error:
                if retry:

                    retry -= 1
                    time.sleep(1)

                    continue

                LOG.error(error)

                raise HTTPException("ReadTimeout", error)

            except requests.exceptions.HTTPError as error:
                LOG.error(error)

                if r.status_code == 401:

                    if 'X-Application-Error-Code' in r.headers:
                        self.client.callback("AccessRestricted", {'ServerId': config['auth.server-id']})

                        raise HTTPException("AccessRestricted", error)
                    else:
                        self.client.callback("Unauthorized", {'ServerId': config['auth.server-id']})
                        self.client.auth.revoke_token()

                        raise HTTPException("Unauthorized", error)

                elif r.status_code == 500:  # log and ignore.
                    LOG.error("--[ 500 response ] %s", error)

                    return

                elif r.status_code == 502:
                    if retry:

                        retry -= 1
                        time.sleep(1)

                        continue

                raise HTTPException(r.status_code, error)

            except requests.exceptions.MissingSchema as error:
                LOG.error("Request missing Schema. " + str(error))
                raise HTTPException("MissingSchema", {'Id': config.get('auth.server', "None")})

            except Exception as error:
                raise

            else:
                try:
                    config['server-time'] = r.headers['Date']
                    elapsed = int(r.elapsed.total_seconds() * 1000)
                    response = r.json()
                    LOG.debug("---<[ http ][%s ms]", elapsed)
                    LOG.debug(JsonDebugPrinter(response))

                    return response
                except ValueError:
                    return

    def _process_params(self, params):

        for key in params:
            value = params[key]

            if isinstance(value, dict):
                self._process_params(value)

            if isinstance(value, string_types):
                params[key] = self._replace_user_info(self.config.data, value)

    def _get_header(self, config, headers):
        if headers:
            headers = dict(headers)
        else:
            headers = {
                'Content-type': "application/json",
                'Accept-Charset': "UTF-8,*",
                'Accept-encoding': "gzip",
                'User-Agent': config['http.user_agent'] or "%s/%s" % (config.get('app.name', 'Jellyfin for Kodi'), config.get('app.version', "0.0.0"))
            }

        if 'x-emby-authorization' not in headers:
            auth_headers = self._authorization(config)
            headers.update(auth_headers)

        return headers

    def _authorization(self, config):

        auth = "MediaBrowser "
        auth += "Client=%s, " % config.get('app.name', "Jellyfin for Kodi")
        auth += "Device=%s, " % config.get('app.device_name', 'Unknown Device')
        auth += "DeviceId=%s, " % config.get('app.device_id', 'Unknown Device id')
        auth += "Version=%s" % config.get('app.version', '0.0.0')


        if config.get('auth.token') and config.get('auth.user_id'):
            auth += ', UserId=%s' % config.get('auth.user_id')
            headers = {'x-emby-authorization': auth, 'X-MediaBrowser-Token': config.get('auth.token')}
        else:
            headers = {'x-emby-authorization': auth}
        return headers

    def _requests(self, session, action, **kwargs):

        if action == "GET":
            return session.get(**kwargs)
        elif action == "POST":
            return session.post(**kwargs)
        elif action == "HEAD":
            return session.head(**kwargs)
        elif action == "DELETE":
            return session.delete(**kwargs)