# -*- coding: utf-8 -*-
import re
from datetime import datetime
from typing import Any, Dict, List

import requests

from core.managers.command_manager import matcher
from core.managers.image_manager import image_manager
from core.utils.logger import logger
from models import MessageEvent, MessageSegment

# 插件元数据
__plugin_meta__ = {
    "name": "weather",
    "description": "查询天气信息，支持中国天气网数据。",
    "usage": "/天气 [城市代码] - 查询指定城市的天气信息\n例如：/天气 101190207 (南京)",
}

# 城市代码映射（可以扩展）
CITY_CODES = {
    "北京": "101010100",
    "上海": "101020100",
    "广州": "101280101",
    "深圳": "101280601",
    "南京": "101190101",
    "苏州": "101190401",
    "杭州": "101210101",
    "武汉": "101200101",
    "成都": "101270101",
    "重庆": "101040100",
    "西安": "101110101",
    "天津": "101030100",
    "沈阳": "101070101",
    "大连": "101070201",
    "青岛": "101120201",
    "济南": "101120101",
    "郑州": "101180101",
    "长沙": "101250101",
    "南昌": "101240101",
    "合肥": "101220101",
    "福州": "101230101",
    "厦门": "101230201",
    "南宁": "101300101",
    "海口": "101310101",
    "昆明": "101290101",
    "贵阳": "101260101",
    "拉萨": "101140101",
    "兰州": "101160101",
    "西宁": "101150101",
    "银川": "101170101",
    "乌鲁木齐": "101130101",
    "哈尔滨": "101050101",
    "长春": "101060101",
    "呼和浩特": "101080101",
    "太原": "101100101",
    "石家庄": "101090101",
}

HEADERS = {
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
}


def get_weather_data(city_code: str) -> Dict[str, Any]:
    """
    获取天气数据

    Args:
        city_code (str): 城市代码

    Returns:
        Dict[str, Any]: 包含城市信息和天气数据的字典
    """
    try:
        url = f"https://www.weather.com.cn/weather/{city_code}.shtml"
        response = requests.get(url, headers=HEADERS, timeout=10)
        response.encoding = "utf-8"
        html_content = response.text

        # 提取城市信息
        city_info = (
            html_content.split('<a href="http://www.weather.com.cn/forecast/" ')[-1]
            .split("</div>")[0]
            .strip()
        )

        city_parts = []
        city_parts.append(city_info.split("<a>")[-1].split("</a>")[0])

        if city_info.count("_blank") == 1:
            city_parts.append(
                city_info.split("<span>></span>")[-1]
                .replace("</span>", "")
                .replace("<span>", "")
                .strip()
            )
        else:
            additional_parts = (
                city_info.split('target="_blank">')[-1]
                .replace("<span>></span>  <span>", "")
                .replace("</span>", "")
                .split("</a>")
            )
            city_parts.extend(additional_parts)

        city_name = " ".join([part for part in city_parts if part.strip()])

        # 提取天气信息
        weather_data = []
        for i in range(7):
            try:
                weather_info = (
                    html_content.split('<ul class="t clearfix">')[-1]
                    .split('on">')[1]
                    .split("</li>")[i]
                )

                day = weather_info.split("<h1>")[-1].split("</h1>")[0].strip()
                weather = (
                    weather_info.split('<p title="')[-1]
                    .split('" class="wea">')[0]
                    .strip()
                )

                tem = (
                    weather_info.split("<span>")[-1]
                    .split("</i>")[0]
                    .replace("</span>/<i>", " / ")
                    .strip()
                )
                if len(tem) > 10:
                    tem = weather_info.split("<i>")[1].split("</i>")[0].strip()

                wind = weather_info.split('<span title="')
                wind_direction = []
                if len(wind) > 2:
                    for j in range(2):
                        wind_direction.append(
                            wind[j + 1]
                            .split('"></span>')[0]
                            .replace('" class="', " / ")
                        )
                else:
                    wind_direction.append(
                        wind[1].split('"></span>')[0].replace('" class="', " / ")
                    )
                wind_power = weather_info.split("<i>")[-1].split("</i>")[0].strip()

                wind_direction_str = (
                    " / ".join(wind_direction) if wind_direction else "未知"
                )

                weather_data.append(
                    {
                        "day": day,
                        "weather": weather,
                        "temperature": tem,
                        "wind_power": wind_power,
                        "wind_direction": wind_direction_str,
                    }
                )

            except (IndexError, ValueError) as e:
                logger.warning(f"解析第{i + 1}天天气数据失败: {e}")
                continue

        return {
            "city_name": city_name,
            "weather_data": weather_data,
            "query_time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
            "timestamp": datetime.now().strftime("%Y年%m月%d日 %H:%M"),
        }

    except Exception as e:
        logger.error(f"获取天气数据失败: {e}")
        return None


@matcher.command("天气")
async def handle_weather(bot, event: MessageEvent, args: List[str]):
    """
    处理天气查询指令

    Args:
        bot: Bot实例
        event: 消息事件
        args: 指令参数
    """
    if not args:
        # 显示支持的城市列表
        city_list = "\n".join(
            [f"{name}: {code}" for name, code in list(CITY_CODES.items())[:10]]
        )
        reply_msg = f"请指定城市名称或城市代码，例如：\n/天气 北京\n/天气 101010100\n\n支持的城市：\n{city_list}\n..."
        await event.reply(reply_msg)
        return

    city_input = args[0].strip()

    # 尝试匹配城市名称或直接使用城市代码
    city_code = None
    if city_input in CITY_CODES:
        city_code = CITY_CODES[city_input]
    elif re.match(r"^\d{9}$", city_input):
        city_code = city_input
    else:
        # 尝试模糊匹配城市名称
        for name, code in CITY_CODES.items():
            if city_input in name:
                city_code = code
                break

    if not city_code:
        await event.reply(f"未找到城市 '{city_input}'，请检查城市名称或使用城市代码。")
        return

    # 获取天气数据
    await event.reply("正在查询天气信息，请稍候...")
    weather_info = get_weather_data(city_code)

    if not weather_info or not weather_info.get("weather_data"):
        await event.reply("获取天气信息失败，请稍后重试。")
        return

    try:
        # 渲染HTML模板为图片
        base64_image = await image_manager.render_template_to_base64(
            "weather.html", weather_info, output_name="weather.png"
        )

        if base64_image:
            # 发送图片消息
            await event.reply(MessageSegment.image(base64_image))
        else:
            await event.reply("生成天气图片失败，请稍后重试。")

    except Exception as e:
        logger.error(f"渲染天气图片失败: {e}")
        await event.reply("生成天气图片时发生错误，请稍后重试。")