import os

import requests

import core
from core import logger


def get_nzoid(input_name):
    nzoid = None
    slots = []
    logger.debug('Searching for nzoid from SAbnzbd ...')
    if 'http' in core.SABNZBDHOST:
        base_url = '{0}:{1}/api'.format(core.SABNZBDHOST, core.SABNZBDPORT)
    else:
        base_url = 'http://{0}:{1}/api'.format(core.SABNZBDHOST, core.SABNZBDPORT)
    url = base_url
    params = {
        'apikey': core.SABNZBDAPIKEY,
        'mode': 'queue',
        'output': 'json',
    }
    try:
        r = requests.get(url, params=params, verify=False, timeout=(30, 120))
    except requests.ConnectionError:
        logger.error('Unable to open URL')
        return nzoid  # failure
    try:
        result = r.json()
        clean_name = os.path.splitext(os.path.split(input_name)[1])[0]
        slots.extend([(slot['nzo_id'], slot['filename']) for slot in result['queue']['slots']])
    except Exception:
        logger.warning('Data from SABnzbd queue could not be parsed')
    params['mode'] = 'history'
    try:
        r = requests.get(url, params=params, verify=False, timeout=(30, 120))
    except requests.ConnectionError:
        logger.error('Unable to open URL')
        return nzoid  # failure
    try:
        result = r.json()
        clean_name = os.path.splitext(os.path.split(input_name)[1])[0]
        slots.extend([(slot['nzo_id'], slot['name']) for slot in result['history']['slots']])
    except Exception:
        logger.warning('Data from SABnzbd history could not be parsed')
    try:
        for nzo_id, name in slots:
            if name in [input_name, clean_name]:
                nzoid = nzo_id
                logger.debug('Found nzoid: {0}'.format(nzoid))
                break
    except Exception:
        logger.warning('Data from SABnzbd could not be parsed')
    return nzoid


def report_nzb(failure_link, client_agent):
    # Contact indexer site
    logger.info('Sending failure notification to indexer site')
    if client_agent == 'nzbget':
        headers = {'User-Agent': 'NZBGet / nzbToMedia.py'}
    elif client_agent == 'sabnzbd':
        headers = {'User-Agent': 'SABnzbd / nzbToMedia.py'}
    else:
        return
    try:
        requests.post(failure_link, headers=headers, timeout=(30, 300))
    except Exception as e:
        logger.error('Unable to open URL {0} due to {1}'.format(failure_link, e))
    return