# Copyright (C) 2008-2022 The Paparazzi Team
# released under GNU GPLv2 or later. See COPYING file.
import os.path

from generated.ui_program import Ui_Program
from PyQt5.QtWidgets import *
from PyQt5 import QtCore
from PyQt5.QtCore import QProcess
from PyQt5.QtGui import QIcon
import utils
from typing import List
from enum import Enum


class TabProgramsState(Enum):
    IDLE = 0
    RUNNING = 1
    ERROR = 2


class ProgramWidget(QWidget, Ui_Program):

    ready_read_stdout = QtCore.pyqtSignal()
    ready_read_stderr = QtCore.pyqtSignal()
    started = QtCore.pyqtSignal()
    finished = QtCore.pyqtSignal(int, QProcess.ExitStatus)
    remove = QtCore.pyqtSignal()

    def __init__(self, shortname: str, cmd: List[str], icon=None, parent=None):
        QWidget.__init__(self, parent=parent)
        self.setupUi(self)
        self.cmd = cmd
        self.shortname = shortname
        self.process = QProcess(self)
        self.program_lineedit.setText(" ".join(cmd))
        self.program_lineedit.returnPressed.connect(self.handle_cmd_return)
        self.run_button.clicked.connect(self.handle_run)
        self.remove_button.clicked.connect(self.handle_remove)
        self.process.readyReadStandardOutput.connect(self.ready_read_stdout)
        self.process.readyReadStandardError.connect(self.ready_read_stderr)
        self.process.finished.connect(self.handle_finished)
        self.process.started.connect(self.handle_started)
        self.process.errorOccurred.connect(self.handle_error)
        i = QIcon(os.path.join(utils.PAPARAZZI_HOME, "data", "pictures", "tools_icons", icon))
        self.icon_label.setPixmap(i.pixmap(20, 20))
        self.icon_label.setToolTip(shortname)

    def start_program(self):
        self.program_lineedit.setStyleSheet("")
        if self.process.state() == QProcess.NotRunning:
            self.process.start(self.cmd[0], self.cmd[1:])
            self.started.emit()

    def handle_cmd_return(self):
        if self.process.state() == QProcess.NotRunning:
            self.cmd = self.program_lineedit.text().split(" ")
            self.start_program()
        elif self.process.state() == QProcess.Running:
            self.terminate()

    def handle_run(self):
        if self.process.state() == QProcess.NotRunning:
            self.cmd = self.program_lineedit.text().split(" ")
            self.start_program()
        elif self.process.state() == QProcess.Running:
            self.process.terminate()

    def handle_remove(self):
        if self.process.state() == QProcess.NotRunning:
            self.remove.emit()
        elif self.process.state() == QProcess.Running:
            self.process.finished.connect(self.remove)
            self.process.terminate()

    def handle_started(self):
        self.run_button.setIcon(QIcon(":/icons/icons/stop.png"))
        self.program_lineedit.setReadOnly(True)

    def handle_finished(self, exit_code: int, exit_status: QProcess.ExitStatus):
        if exit_code not in (0, 15):
            self.program_lineedit.setStyleSheet("background: #f56464")
        self.run_button.setIcon(QIcon(":/icons/icons/play.png"))
        self.program_lineedit.setReadOnly(False)
        self.finished.emit(exit_code, exit_status)

    def handle_error(self, error: QProcess.ProcessError):
        if error == QProcess.FailedToStart:
            self.handle_finished(-1, QProcess.CrashExit)
        # FailedToStart
        # Crashed
        # Timedout
        # ReadError
        # WriteError
        # UnknownError

    def terminate(self):
        if self.process.state() != QProcess.NotRunning:
            self.process.terminate()

    def state(self):
        return self.process.state()