#!/usr/local/bin/python
# -*- coding: utf-8 -*-

import json
import requests

from heremaps.heremapsexceptions import BadGeocodingParams
from heremaps.heremapsexceptions import EmptyGeocoderResponse
from heremaps.heremapsexceptions import NoGeocodingParams
from heremaps.heremapsexceptions import MalformedResult


class Geocoder:
    'A Here Maps Geocoder wrapper for python'

    PRODUCTION_GEOCODE_JSON_URL = 'https://geocoder.api.here.com/6.2/geocode.json'
    STAGING_GEOCODE_JSON_URL = 'https://geocoder.cit.api.here.com/6.2/geocode.json'
    DEFAULT_MAXRESULTS = 1
    DEFAULT_GEN = 9

    ADDRESS_PARAMS = [
        'city',
        'country',
        'county',
        'district',
        'housenumber',
        'postalcode',
        'searchtext',
        'state',
        'street'
        ]

    ADMITTED_PARAMS = [
        'additionaldata',
        'app_id',
        'app_code',
        'bbox',
        'countryfocus',
        'gen',
        'jsonattributes',
        'jsoncallback',
        'language',
        'locationattributes',
        'locationid',
        'mapview',
        'maxresults',
        'pageinformation',
        'politicalview',
        'prox',
        'strictlanguagemode'
        ] + ADDRESS_PARAMS

    app_id = ''
    app_code = ''
    maxresults = ''

    def __init__(self, app_id, app_code, maxresults=DEFAULT_MAXRESULTS,
                 gen=DEFAULT_GEN, host=PRODUCTION_GEOCODE_JSON_URL):
        self.app_id = app_id
        self.app_code = app_code
        self.maxresults = maxresults
        self.gen = gen
        self.host = host

    def geocode(self, params):
        if not set(params.keys()).issubset(set(self.ADDRESS_PARAMS)):
            raise BadGeocodingParams(params)
        response = self.perform_request(params)
        try:
            results = response['Response']['View'][0]['Result']
        except IndexError:
            raise EmptyGeocoderResponse()

        return results

    def perform_request(self, params):
        request_params = {
            'app_id': self.app_id,
            'app_code': self.app_code,
            'maxresults': self.maxresults,
            'gen': self.gen
        }
        request_params.update(params)
        response = requests.get(self.host, params=request_params)
        if response.status_code == requests.codes.ok:
            return json.loads(response.text)
        else:
            response.raise_for_status()

    def geocode_address(self, **kwargs):
        params = {}
        for key, value in kwargs.iteritems():
            if value:
                params[key] = value
        if not params:
            raise NoGeocodingParams()
        return self.geocode(params)

    def extract_lng_lat_from_result(self, result):
        try:
            location = result['Location']
        except KeyError:
            raise MalformedResult()

        longitude = location['DisplayPosition']['Longitude']
        latitude = location['DisplayPosition']['Latitude']

        return [longitude, latitude]