from utils.message_builder import image
from configs.path_config import TEXT_PATH
from configs.config import NICKNAME
from typing import List
from nonebot import Driver
from utils.http_utils import AsyncHttpx
import ujson as json
import nonebot

driver: Driver = nonebot.get_driver()

china_city = TEXT_PATH / "china_city.json"

data = {}


async def get_weather_of_city(city: str) -> str:
    """
    获取城市天气数据
    :param city: 城市
    """
    code = _check_exists_city(city)
    if code == 999:
        return "不要查一个省份的天气啊，很累人的！"
    elif code == 998:
        return f"{NICKNAME}没查到!!试试查火星的天气？"
    else:
        data_json = (
            await AsyncHttpx.get(
                f"https://v0.yiketianqi.com/api?unescape=1&version=v91&appid=43656176&appsecret=I42og6Lm&ext=&cityid=&city={city[:-1]}"
            )
        ).json()
        if wh := data_json.get('data'):
            w_type = wh[0]["wea_day"]
            w_max = wh[0]["tem1"]
            w_min = wh[0]["tem2"]
            fengli = wh[0]["win_speed"]
            ganmao = wh[0]["narrative"]
            fengxiang = ','.join(wh[0].get('win', []))
            repass = f"{city}的天气是 {w_type} 天\n最高温度: {w_max}\n最低温度: {w_min}\n风力: {fengli} {fengxiang}\n{ganmao}"
            return repass
        else:
            return data_json.get("errmsg") or "好像出错了？再试试？"


def _check_exists_city(city: str) -> int:
    """
    检测城市是否存在合法
    :param city: 城市名称
    """
    global data
    city = city if city[-1] != "市" else city[:-1]
    for province in data.keys():
        for city_ in data[province]:
            if city_ == city:
                return 200
    for province in data.keys():
        if city == province:
            return 999
    return 998


def get_city_list() -> List[str]:
    """
    获取城市列表
    """
    global data
    if not data:
        try:
            with open(china_city, "r", encoding="utf8") as f:
                data = json.load(f)
        except FileNotFoundError:
            data = {}
    city_list = []
    for p in data.keys():
        for c in data[p]:
            city_list.append(c)
        city_list.append(p)
    return city_list