from nonebot import Driver
from services.db_context import db
from asyncpg.exceptions import DuplicateColumnError
from models.group_info import GroupInfo
from nonebot.adapters.cqhttp import Bot
from utils.user_agent import get_user_agent
from services.log import logger
from configs.path_config import TEXT_PATH
from asyncio.exceptions import TimeoutError
from pathlib import Path
import aiohttp
import nonebot

try:
    import ujson as json
except ModuleNotFoundError:
    import json


driver: Driver = nonebot.get_driver()


@driver.on_startup
async def update_city():
    """
    部分插件需要中国省份城市
    这里直接更新，避免插件内代码重复
    :return:
    """
    china_city = Path(TEXT_PATH) / "china_city.json"
    data = {}
    try:
        async with aiohttp.ClientSession(headers=get_user_agent()) as session:
            async with session.get(
                "http://www.weather.com.cn/data/city3jdata/china.html", timeout=5
            ) as res:
                provinces_data = json.loads(await res.text(encoding="utf8"))
            for province in provinces_data.keys():
                data[provinces_data[province]] = []
                async with session.get(
                    f"http://www.weather.com.cn/data/city3jdata/provshi/{province}.html",
                    timeout=5,
                ) as res:
                    city_data = json.loads(await res.text(encoding="utf8"))
                    for city in city_data.keys():
                        data[provinces_data[province]].append(city_data[city])
        with open(china_city, "w", encoding="utf8") as f:
            json.dump(data, f, indent=4, ensure_ascii=False)
        logger.info("自动更新城市列表完成.....")
    except TimeoutError:
        logger.info("自动更新城市列表超时.....")


@driver.on_startup
async def _():
    """
    数据库表结构变换
    """
    sql_str = [
        "ALTER TABLE group_info ADD group_flag Integer NOT NULL DEFAULT 0;"     # group_info表添加一个group_flag
    ]
    for sql in sql_str:
        try:
            query = db.text(sql)
            await db.first(query)
        except DuplicateColumnError:
            pass


@driver.on_bot_connect
async def _(bot: Bot):
    """
    版本某些需要的变换
    """
    # 清空不存在的群聊信息，并将已所有已存在的群聊group_flag设置为1（认证所有已存在的群）
    if not await GroupInfo.get_group_info(114514):
        # 标识符，该功能只需执行一次
        await GroupInfo.add_group_info(
            114514,
            "114514",
            114514,
            114514,
            1
        )
        group_list = await bot.get_group_list()
        group_list = [g["group_id"] for g in group_list]
        _gl = [x.group_id for x in await GroupInfo.get_all_group()]
        if 114514 in _gl:
            _gl.remove(114514)
        for group_id in _gl:
            if group_id in group_list:
                if await GroupInfo.get_group_info(group_id):
                    await GroupInfo.set_group_flag(group_id, 1)
                else:
                    group_info = await bot.get_group_info(group_id=group_id)
                    await GroupInfo.add_group_info(
                        group_info["group_id"],
                        group_info["group_name"],
                        group_info["max_member_count"],
                        group_info["member_count"],
                        1
                    )
                logger.info(f"已将群聊 {group_id} 添加认证...")
            else:
                await GroupInfo.delete_group_info(group_id)
                logger.info(f"移除不存在的群聊信息：{group_id}")