# -*- coding: utf-8-*-
import json
import requests
from robot import config, logging
from robot.sdk.AbstractPlugin import AbstractPlugin

logger = logging.getLogger(__name__)

class Plugin(AbstractPlugin):

    SLUG = "direction"

    def request(self, url, params):
        r = requests.get(url, params=params)
        content = r.text
        return json.loads(content)

    def handle(self, text, parsed):

        def onAsk(input):
            if input is None:
                self.say(u'已取消')
                return

            profile = config.get()
            if self.SLUG not in profile or \
               'app_key' not in profile[self.SLUG] or \
               'origin' not in profile[self.SLUG]:
                self.say(u"插件配置有误，插件使用失败")
                return

            app_key = profile[self.SLUG]['app_key']
            city = config.get('location', '深圳')

            url_place = "http://api.map.baidu.com/place/v2/suggestion"
            params_place = {
                "query" : input,
                "region" : city,
                "city_limit" : "true",
                "output" : "json",
                "ak" : app_key,
            }

            res = self.request(url_place, params_place)

            if res:
                status = res["status"]
                if status == 0:
                    if len(res['result']) > 0:
                        place_name = res['result'][0]["name"]
                        destination = "%f,%f" % (res['result'][0]["location"]['lat'], res['result'][0]["location"]['lng'])
                    else:
                        self.say(u"错误的位置")
                        return
                else:
                    logger.error(u"位置接口:" + res['message'])
                    return
            else:
                logger.error(u"位置接口调用失败")
                return

            origin = profile[self.SLUG]['origin']

            url_direction = "http://api.map.baidu.com/direction/v2/transit"
            params_direction = {
                "origin" : origin,
                "destination" : destination,
                "page_size" : 1,
                "ak" : app_key,
            }

            res = self.request(url_direction, params_direction)
            if res:
                status = res["status"]
                if status == 0:
                    if len(res['result']['routes']) > 0:
                        direction = ""
                        for step in res['result']['routes'][0]['steps']:
                            direction = direction + step[0]["instructions"] + "."
                            result = place_name + u"参考路线:" + direction
                        self.say(result)
                    else:
                        self.say(u"导航错误")
                        return
                else:
                    logger.error(u"导航接口:" + res['message'])
                    return
            else:
                logger.error(u"导航接口调用失败")
                return

        self.say("去哪里", cache=True, onCompleted=lambda: onAsk(self.activeListen()))


    def isValid(self, text, parsed):
        return any(word in text for word in [u"怎么去", u"线路", u"路线"])