import json

from functools import partial

from pkg_resources import resource_filename

import jsonschema
from flask import Blueprint, jsonify, request, abort, current_app

from .auth import httpauth
from .utils import first, fuzzy_list_find, pass_active_status

sensors_views = Blueprint('sensors', __name__)

ALLOWED_SENSORS_KEYS = json.load(
    open(resource_filename('spaceapi', 'schema/sensors.json'),
         encoding='utf-8')
)

RADIATON_SUBKEYS = frozenset(('alpha', 'beta', 'gamma', 'beta_gamma'))

get_identification_key = partial(first, keys=frozenset(('name', 'location')))


def set_value(data, key):

    active = ActiveStatus()

    try:
        subkey = get_identification_key(data)
    except ValueError:
        return abort(400)

    try:
        index = fuzzy_list_find(active['sensors'][key],
                                subkey, data[subkey])
        active['sensors'][key][index].update(data)
    except ValueError:
        active['sensors'][key].append(data)

    return jsonify(active)


def set_radiation_value(data):

    active = ActiveStatus()

    radiation_keys = [k for k in RADIATON_SUBKEYS if k in data]

    if not radiation_keys:
        return abort(400)

    for first_subkey in radiation_keys:

        try:
            second_subkey = get_identification_key(data[first_subkey])
        except ValueError:
            return abort(400)

        try:
            index = fuzzy_list_find(
                active['sensors']['radiation'][first_subkey],
                second_subkey,
                data[first_subkey][second_subkey])
            active['sensors']['radiation'][first_subkey][index].update(
                data)
        except ValueError:
            active['sensors']['radiaton'][first_subkey].append(data)

    return jsonify(active)


@sensors_views.route('/set/<key>', methods=['POST'])
@httpauth.login_required
@pass_active_status
def set_sensors(active, key):

    if key in ALLOWED_SENSORS_KEYS and key in active['sensors']:
        data = request.data.decode('utf-8')
        try:
            data = json.loads(data)

            try:
                jsonschema.validate(data, ALLOWED_SENSORS_KEYS[key])
            except jsonschema.ValidationError:
                current_app.logger.info('Validation Error')
                return abort(400)

            if key != 'radiation':
                return set_value(data, key)
            else:
                return set_radiation_value(data)

        except ValueError:
            return abort(400)
    else:
        return abort(400)