from PySide6.QtGui import QIcon, QShowEvent
from qfluentwidgets import FluentIcon, MSFluentWindow, NavigationItemPosition, MessageBox

from ui.pyui.about_ui import AboutWidget
from ui.pyui.achievement_ui import AchievementWidget
from ui.pyui.defense_ui import DefenseWidget
from ui.pyui.test_ui import TestWidget
from utils.function import is_frozen


class MainWindow(MSFluentWindow):
    def __init__(self):
        super().__init__()

        self.achievementInterface = AchievementWidget('Achievement Interface', self)
        self.defenseInterface = DefenseWidget('Defense Interface', self)
        self.aboutInterface = AboutWidget('About Interface', self)
        if not is_frozen():
            self.testInterface = TestWidget('Test Interface', self)

        self.achievementInterface.error.connect(self.showError)
        self.defenseInterface.errorSignal.connect(self.showError)

        self.initNavigation()
        self.initWindow()

    def initNavigation(self):
        self.addSubInterface(self.achievementInterface, FluentIcon.SPEED_HIGH, '达成度')
        self.addSubInterface(self.defenseInterface, FluentIcon.FEEDBACK, '答辩')
        if not is_frozen():
            self.addSubInterface(self.testInterface, FluentIcon.VIEW, '测试')

        self.addSubInterface(self.aboutInterface, FluentIcon.INFO, '关于', position=NavigationItemPosition.BOTTOM)

    def initWindow(self):
        self.resize(900, 700)
        self.setWindowTitle('建工工具箱')
        self.setWindowIcon(QIcon(':/images/logo.png'))

    def showError(self, title: str, message: str):
        MessageBox(title, message, self).exec()

    def showEvent(self, event: QShowEvent):
        super().showEvent(event)
        if is_frozen():
            import pyi_splash
            pyi_splash.update_text('正在加载...')
            pyi_splash.close()