from PyQt5.QtCore import Qt
from PyQt5.QtGui import QGuiApplication
from PyQt5.QtWidgets import (
    QApplication,
    QMainWindow,
    QFrame,
    QVBoxLayout,
    QGroupBox,
    QLabel,
    QStatusBar,
    QPushButton,
    QGridLayout
)

from cricket.lang import SimpleLang

result = False

class SpeakerTestWindow(SimpleLang):
    def __init__(self, play_routine, stop_routine, languages) -> None:
        super().__init__()

        self.play_routine = play_routine
        self.stop_routine = stop_routine
        self.languages = languages

        self.app = QApplication([])

        self.window = QMainWindow()
        self.window.setWindowTitle(self._get_text('title'))
        # self.window.setWindowFlags(Qt.WindowStaysOnTopHint)

        # Main content
        content = QFrame(self.window)
        content_layout = QVBoxLayout(content)

        # Test step
        box = QGroupBox(self.get_text('test_step'), content)
        box_layout = QVBoxLayout(box)

        label = QLabel(self._get_text('test_step'), box)
        box_layout.addWidget(label)

        content_layout.addWidget(box)

        # Status
        self.status = QStatusBar(content)
        self.status.showMessage('Not running')
        content_layout.addWidget(self.status)

        # Toolbar
        toolbar = QFrame(content)
        toolbar_layout = QGridLayout(toolbar)

        self.play_button = QPushButton(self.get_text('play_button'), toolbar)
        self.play_button.clicked.connect(self.cmd_play)
        self.play_button.setFocus()
        toolbar_layout.addWidget(self.play_button, 0, 0)

        self.stop_button = QPushButton(self.get_text('stop_button'), toolbar)
        self.stop_button.setDisabled(True)
        self.stop_button.clicked.connect(self.cmd_stop)
        toolbar_layout.addWidget(self.stop_button, 0, 1)

        self.pass_button = QPushButton(self.get_text('pass_button'), toolbar)
        self.pass_button.setDisabled(True)
        self.pass_button.clicked.connect(self.cmd_pass)
        toolbar_layout.addWidget(self.pass_button, 0, 2)

        self.fail_button = QPushButton(self.get_text('fail_button'), toolbar)
        self.fail_button.setDisabled(True)
        self.fail_button.clicked.connect(self.cmd_fail)
        toolbar_layout.addWidget(self.fail_button, 0, 3)

        content_layout.addWidget(toolbar)

        self.window.setCentralWidget(content)

        # Move to center
        # screen = QGuiApplication.primaryScreen()
        # if screen:
        #     screen_geometry = screen.geometry()
        #     window_geometry = self.window.geometry()
        #     x = (screen_geometry.width() - window_geometry.width()) // 2
        #     y = (screen_geometry.height() - window_geometry.height()) // 2
        #     self.window.move(x, y)

        # self.window.show()
        self.window.showFullScreen()

        self.app.exec_()

    def _get_text(self, key):
        if self.languages:
            lang = self.languages.get(self.current_lang)
            if lang is not None:
                text = lang.get(key)
                if text is not None:
                    return text

        return 'Undefined'

    def cmd_play(self):
        self.status.showMessage('Running...')
        self.play_button.setDisabled(True)
        self.play_routine()
        self.stop_button.setDisabled(False)
        self.pass_button.setDisabled(False)
        self.fail_button.setDisabled(False)

    def cmd_stop(self):
        self.status.showMessage('Stopping...')
        self.stop_button.setDisabled(True)
        self.stop_routine()
        self.play_button.setDisabled(False)
        self.status.showMessage('Stopped')

    def cmd_pass(self):
        global result

        self.status.showMessage('Stopping...')
        self.stop_routine()
        result = True
        self.app.quit()

    def cmd_fail(self):
        global result

        self.status.showMessage('Stopping...')
        self.stop_routine()
        result = False
        self.app.quit()

def GetTestResult():
    return result