import json
import sys
from PyQt5.QtWidgets import QApplication
from gui_test import MainContentComponent
from client_socket import SocketRobotArm
import time


class MyApp:
    with open("./robots.json", "r") as file:
        robots = json.load(file)

    def __init__(self):
        self.startRobot()
        self.startGui()

    def get_status(self):
        if self.robot_app:
            return self.robot_app.get_status
        else:
            return self.def_robot_status
        
    def def_robot_status(self):
        return SocketRobotArm.Status[1]
    
    def startGui(self):
        app = QApplication(sys.argv)
        mainWindow = MainContentComponent(
            robotPanel={
                "robots": self.robots,
                "updateRobot": self.updateRobot,
                "status": self.get_status,
            },
            statusPanel={
                "status": self.get_status,
            }
        )
        mainWindow.setWindowTitle("ROBOT GUI")
        mainWindow.show()

        sys.exit(app.exec_())

    def startRobot(self):
        self.robot_app = SocketRobotArm()

    def updateRobot(self, robot):
        if robot in self.robots:
            selected_robot = robot
            selected_robot = {"name": "test", "host": "127.0.0.1", "slave_id": 11}
            
            if self.robot_app.status == "connected":
                self.robot_app.close()
                time.sleep(0.3)
            else:
                self.robot_app.connect(selected_robot["host"], selected_robot["slave_id"])


if __name__ == "__main__":
    MyApp()