# coding=utf-8

import requests

from six import iteritems

import core
from core import logger


def autoFork(section, inputCategory):
    # auto-detect correct section
    # config settings

    cfg = dict(core.CFG[section][inputCategory])

    host = cfg.get("host")
    port = cfg.get("port")
    username = cfg.get("username")
    password = cfg.get("password")
    apikey = cfg.get("apikey")
    ssl = int(cfg.get("ssl", 0))
    web_root = cfg.get("web_root", "")
    try:
        fork = core.FORKS.items()[core.FORKS.keys().index(cfg.get("fork", "auto"))]
    except:
        fork = "auto"
    protocol = "https://" if ssl else "http://"

    detected = False
    if section == "NzbDrone":
        logger.info("Attempting to verify {category} fork".format
                    (category=inputCategory))
        url = "{protocol}{host}:{port}{root}/api/rootfolder".format(
                    protocol=protocol, host=host, port=port, root=web_root)
        headers = {"X-Api-Key": apikey}
        try:
            r = requests.get(url, headers=headers, stream=True, verify=False)
        except requests.ConnectionError:
            logger.warning("Could not connect to {0}:{1} to verify fork!".format(section, inputCategory))

        if not r.ok:
            logger.warning("Connection to {section}:{category} failed! "
                           "Check your configuration".format
                           (section=section, category=inputCategory))

        fork = ['default', {}]

    elif fork == "auto":
        params = core.ALL_FORKS
        rem_params = []
        logger.info("Attempting to auto-detect {category} fork".format(category=inputCategory))
        # define the order to test. Default must be first since the default fork doesn't reject parameters.
        # then in order of most unique parameters.
        url = "{protocol}{host}:{port}{root}/home/postprocess/".format(
                    protocol=protocol, host=host, port=port, root=web_root)
        # attempting to auto-detect fork
        try:
            if username and password:
                s = requests.Session()
                login = "{protocol}{host}:{port}{root}/login".format(
                    protocol=protocol, host=host, port=port, root=web_root)
                login_params = {'username': username, 'password': password}
                s.post(login, data=login_params, stream=True, verify=False)
                r = s.get(url, auth=(username, password), verify=False)
            else:
                r = requests.get(url, verify=False)
        except requests.ConnectionError:
            logger.info("Could not connect to {section}:{category} to perform auto-fork detection!".format
                        (section=section, category=inputCategory))
            r = []
        if r and r.ok:
            for param in params:
                if not 'name="{param}"'.format(param=param) in r.text:
                    rem_params.append(param)
            for param in rem_params:
                params.pop(param)
            for fork in sorted(iteritems(core.FORKS), reverse=False):
                if params == fork[1]:
                    detected = True
                    break
        if detected:
            logger.info("{section}:{category} fork auto-detection successful ...".format
                        (section=section, category=inputCategory))
        elif rem_params:
            logger.info("{section}:{category} fork auto-detection found custom params {params}".format
                        (section=section, category=inputCategory, params=params))
            fork = ['custom', params]
        else:
            logger.info("{section}:{category} fork auto-detection failed".format
                        (section=section, category=inputCategory))
            fork = core.FORKS.items()[core.FORKS.keys().index(core.FORK_DEFAULT)]

    logger.info("{section}:{category} fork set to {fork}".format
                (section=section, category=inputCategory, fork=fork[0]))
    return fork[0], fork[1]