#   AUTO_MAA:A MAA Multi Account Management and Automation Tool
#   Copyright © 2024-2025 DLmaster361

#   This file is part of AUTO_MAA.

#   AUTO_MAA is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published
#   by the Free Software Foundation, either version 3 of the License,
#   or (at your option) any later version.

#   AUTO_MAA is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty
#   of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
#   the GNU General Public License for more details.

#   You should have received a copy of the GNU General Public License
#   along with AUTO_MAA. If not, see <https://www.gnu.org/licenses/>.

#   Contact: DLmaster_361@163.com

"""
AUTO_MAA
AUTO_MAA信息通知栏
v4.4
作者：DLmaster_361
"""

from PySide6.QtCore import Qt
from qfluentwidgets import InfoBar, InfoBarPosition

from .logger import logger
from .config import Config
from .sound_player import SoundPlayer


class _MainInfoBar:
    """信息通知栏"""

    # 模式到 InfoBar 方法的映射
    mode_mapping = {
        "success": InfoBar.success,
        "warning": InfoBar.warning,
        "error": InfoBar.error,
        "info": InfoBar.info,
    }

    def push_info_bar(
        self, mode: str, title: str, content: str, time: int, if_force: bool = False
    ) -> None:
        """
        推送消息到吐司通知栏

        :param mode: 通知栏模式，支持 "success", "warning", "error", "info"
        :param title: 通知栏标题
        :type title: str
        :param content: 通知栏内容
        :type content: str
        :param time: 显示时长，单位为毫秒
        :type time: int
        :param if_force: 是否强制推送
        :type if_force: bool
        """

        if Config.main_window is None:
            logger.error("信息通知栏未设置父窗口", module="吐司通知栏")
            return None

        # 根据 mode 获取对应的 InfoBar 方法
        info_bar_method = self.mode_mapping.get(mode)

        if not info_bar_method:
            logger.error(f"未知的通知栏模式: {mode}", module="吐司通知栏")
            return None

        if Config.main_window.isVisible():
            # 主窗口可见时直接推送通知
            info_bar_method(
                title=title,
                content=content,
                orient=Qt.Horizontal,
                isClosable=True,
                position=InfoBarPosition.TOP_RIGHT,
                duration=time,
                parent=Config.main_window,
            )

        elif if_force:
            # 如果主窗口不可见且强制推送，则录入消息队列等待窗口显示后推送
            info_bar_item = {
                "mode": mode,
                "title": title,
                "content": content,
                "time": time,
            }
            if info_bar_item not in Config.info_bar_list:
                Config.info_bar_list.append(info_bar_item)

            logger.info(
                f"主窗口不可见，已将通知栏消息录入队列: {info_bar_item}",
                module="吐司通知栏",
            )

        if mode == "warning":
            SoundPlayer.play("发生异常")
        if mode == "error":
            SoundPlayer.play("发生错误")


MainInfoBar = _MainInfoBar()