import logging
import tomllib

import aiohttp
import requests
from wcferry import WxMsg, Wcf

from robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager


class BotMusic:
    def __init__(self, wcf: Wcf, gbm: GroupBotManager):
        self.LOG = logging.getLogger(__name__)
        self.wcf = wcf  # 假设 wcf 对象在此类中初始化
        self.gbm = gbm  # 权限功能
        with open("music/config.toml", "rb") as f:
            plugin_config = tomllib.load(f)

        config = plugin_config["Music"]

        self.enable = config["enable"]
        self.command = config["command"]
        self.command_format = config["command-format"]
        self.LOG.info(f"[点歌台] 组件初始化完成，指令： {self.command}")

    import requests

    def get_music(self, message: WxMsg):
        if not self.enable:
            return

        content = str(message.content).strip()
        command = content.split(" ")

        if command[0] not in self.command:
            return

        if len(command) == 1:
            self.wcf.send_text(f"-----Bot-----\n❌命令格式错误！{self.command_format}",
                               (message.roomid if message.from_group() else message.sender), message.sender)
            return

        # 如果触发了指令，但是没有权限，则返回权限不足
        if self.gbm.get_group_permission(message.roomid, Feature.MUSIC) == PermissionStatus.DISABLED:
            return

        song_name = content[len(command[0]):].strip()

        try:
            response = requests.get(
                f"https://www.hhlqilongzhu.cn/api/dg_wyymusic.php?gm={song_name}&n=1&br=2&type=json",
                timeout=5  # 设置超时，防止阻塞
            )
            data = response.json()
        except requests.RequestException as e:
            self.wcf.send_text(f"-----Bot-----\n❌请求出错：{e}",
                               (message.roomid if message.from_group() else message.sender), message.sender)
            return

        if data.get("code") != 200:
            self.wcf.send_text(f"-----Bot-----\n❌点歌失败！\n{data}",
                               (message.roomid if message.from_group() else message.sender), message.sender)
            return

        title = data["title"]
        singer = data["singer"]
        url = data["link"]
        music_url = data["music_url"].split("?")[0]
        cover_url = data["cover"]
        lyric = data["lrc"]

        xml = f"""<appmsg appid="wx79f2c4418704b4f8" sdkver="0">
        <title>{title}</title>
        <des>{singer}</des>
        <action>view</action>
        <type>3</type>
        <showtype>0</showtype>
        <content/>
        <url>{url}</url>
        <dataurl>{music_url}</dataurl>
        <lowurl>{url}</lowurl>
        <lowdataurl>{music_url}</lowdataurl>
        <recorditem/>
        <thumburl>{cover_url}</thumburl>
        <messageaction/>
        <laninfo/>
        <extinfo/>
        <sourceusername/>
        <sourcedisplayname/>
        <songlyric>{lyric}</songlyric>
        <commenturl/>
        <appattach><totallen>0</totallen><attachid/><emoticonmd5/><fileext/><aeskey/></appattach>
        <webviewshared><publisherId/><publisherReqId>0</publisherReqId></webviewshared>
        <weappinfo><pagepath/><username/><appid/><appservicetype>0</appservicetype></weappinfo>
        <websearch/>
        <songalbumurl>{cover_url}</songalbumurl>
        </appmsg>
        <fromusername>{message.sender}</fromusername>
        <scene>0</scene>
        <appinfo><version>1</version><appname/></appinfo>
        <commenturl/>"""

        self.wcf.forward_msg(message.sender, xml, 3)