import paho.mqtt.client as paho
import paho.mqtt.enums as paho_enums

class vqeMqttBroker:
    def __init__(self, broker_address, broker_port, username="", password="", use_tls=True, max_connection_attempts=30):
        self.server = broker_address
        self.port = broker_port
        self.client = paho.Client(callback_api_version=paho.CallbackAPIVersion.VERSION2, client_id="my_python_client", reconnect_on_failure=True)
        self.client.username_pw_set(username, password)
        if use_tls:
            self.client.tls_set()
        self.client.on_connect = self.on_connect
        self.client.on_message = self.on_message
        self.client.on_connect_fail = self.on_connect_fail
        self.client.on_disconnect = self.on_disconnect
        self.max_connection_attempts=max_connection_attempts
        self.connection_attempt_number=0
    
    def connect_to_server(self):
        self.connection_attempt_number=0
        self.client.connect_async(self.server, self.port, 5)
        self.client.loop_start()

    def get_status(self):
        
        return self.client._state #self.client.is_connected()

    def on_connect(self, client, userdata, flags, reason_code, properties):
        print(f"Connected with result code {reason_code}")
        # Subscribe to topics here if needed
        client.subscribe("my/topic")

    def on_message(self, client, userdata, msg):
        print(f"Received message: {msg.payload.decode()} on topic {msg.topic}")

    def on_connect_fail(self, client, userdata):
        self.connection_attempt_number+=1
        print("fail")
    
    def on_disconnect(self, client, userdata, disconnect_flags, reason_code, properties):
        self.connection_attempt_number+=1
        print("disconnected") 
        if (self.connection_attempt_number > self.max_connection_attempts):
            self.client.disconnect()
            self.client.loop_stop()
            self.connection_attempt_number=0