from pyrosapp.models import *
import socket

class DeviceObj():
    '''
        Generic object for the communication with all the devices (need inheritance)
    '''


    def __init__(self):
        '''
            The messages and their parameters will be defined in the objects that inherits from this class
        '''
        set_msgs = []
        get_msgs = []
        do_msgs = []

        self.ip = None
        self.port = None

    def init_socket(self):
        if self.ip is None or self.port is None:
            raise ValueError("IP and/or PORT not initialized")
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.sock.connect((self.ip, self.port))

    def set(self, cmd, *args):
        '''
            Send a SET message to the device
        '''
        msg = [msg for msg in self.set_msgs if msg[0] == cmd]
        if len(msg) == 0:
            raise ValueError("Invalid message argument %s" % (cmd,))
        msg = msg[0]
        if len(args) not in msg[1]:
            raise ValueError("Bad number of arguments")

        for arg in args:
            if type(arg) != msg[2]:
                raise TypeError("Bad type of argument: expected %s" % (msg[2],))

        self.init_socket()
        message = "SET " + cmd + " " + "".join([str(arg) for arg in args])
        self.sock.send(bytes(message, "UTF-8"))
        print("set ", cmd)


    def get(self, cmd):
        '''
            Send a GET message to the device
        '''
        if cmd not in self.get_msgs:
            raise ValueError("Invalid message argument %s" % (cmd,))

        self.init_socket()
        message = "GET " + cmd
        self.sock.send(bytes(message, "UTF-8"))
        print("get ", cmd)
        ret = self.sock.recv(1024).decode()
        print("return : ", ret)
        return ret


    def do(self, cmd, *args):
        '''
            Send a DO message to the device
        '''
        msg = [msg for msg in self.do_msgs if msg[0] == cmd]
        if len(msg) == 0:
            raise ValueError("Invalid message argument %s" % (cmd,))
        msg = msg[0]
        if len(args) not in msg[1]:
            raise ValueError("Bad number of arguments")

        for arg in args:
            if type(arg) != msg[2]:
                raise TypeError("Bad type of argument: expected %s" % (msg[2],))

        self.init_socket()
        message = "DO " + cmd + " " + "".join([str(arg) for arg in args])
        self.sock.send(bytes(message, "UTF-8"))
        print("do ", cmd)