import tomllib
import traceback

import aiohttp
from loguru import logger

from WechatAPI import WechatAPIClient
from utils.decorators import *
from utils.plugin_base import PluginBase


class RandomPicture(PluginBase):
 description = "随机图片"
 author = "HenryXiaoYang"
 version = "1.0.0"

 def __init__(self):
 super().__init__()

 with open("plugins/RandomPicture/config.toml", "rb") as f:
 plugin_config = tomllib.load(f)

 config = plugin_config["RandomPicture"]

 self.enable = config["enable"]
 self.command = config["command"]

 @on_text_message
 async def handle_text(self, bot: WechatAPIClient, message: dict):
 if not self.enable:
 return

 content = str(message["Content"]).strip()
 command = content.split(" ")

 if not len(command) or command[0] not in self.command:
 return

 api_url = "https://api.52vmy.cn/api/img/tu/man?type=text"

 try:
 conn_ssl = aiohttp.TCPConnector(ssl=False)
 async with aiohttp.request("GET", url=api_url, connector=conn_ssl) as req:
 pic_url = await req.text()

 async with aiohttp.request("GET", url=pic_url, connector=conn_ssl) as req:
 content = await req.read()

 await conn_ssl.close()

 await bot.send_image_message(message["FromWxid"], image=content)

 except Exception as error:
 out_message = f"-----XYBot-----\n出现错误❌!\n{error}"
 logger.error(traceback.format_exc())

 await bot.send_text_message(message["FromWxid"], out_message)