swarmit 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

testbed/swarmit/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.3.0"

testbed/swarmit/adapter.py

@@ -0,0 +1,94 @@
+"""Module containing classes for interfacing with the DotBot gateway."""
+
+import base64
+from abc import ABC, abstractmethod
+
+import paho.mqtt.client as mqtt
+from dotbot.hdlc import HDLCHandler, HDLCState, hdlc_encode
+from dotbot.protocol import PROTOCOL_VERSION
+from dotbot.serial_interface import SerialInterface
+
+
+class GatewayAdapterBase(ABC):
+    """Base class for interface adapters."""
+
+    @abstractmethod
+    def init(self, on_data_received: callable):
+        """Initialize the interface."""
+
+    @abstractmethod
+    def close(self):
+        """Close the interface."""
+
+    @abstractmethod
+    def send_data(self, data):
+        """Send data to the interface."""
+
+
+class SerialAdapter(GatewayAdapterBase):
+    """Class used to interface with the serial port."""
+
+    def __init__(self, port, baudrate):
+        self.port = port
+        self.baudrate = baudrate
+        self.hdlc_handler = HDLCHandler()
+
+    def on_byte_received(self, byte):
+        self.hdlc_handler.handle_byte(byte)
+        if self.hdlc_handler.state == HDLCState.READY:
+            self.on_data_received(self.hdlc_handler.payload)
+
+    def init(self, on_data_received: callable):
+        self.serial = SerialInterface(
+            self.port, self.baudrate, self.on_byte_received
+        )
+        self.on_data_received = on_data_received
+        # Just write a single byte to fake a DotBot gateway handshake
+        self.serial.write(int(PROTOCOL_VERSION).to_bytes(length=1))
+
+    def close(self):
+        self.serial.stop()
+
+    def send_data(self, data):
+        self.serial.write(hdlc_encode(data))
+
+
+class MQTTAdapter(GatewayAdapterBase):
+    """Class used to interface with MQTT."""
+
+    def __init__(self, host, port):
+        self.host = host
+        self.port = port
+        self.client = None
+
+    def on_message(self, client, userdata, message):
+        self.on_data_received(base64.b64decode(message.payload))
+
+    def on_log(self, client, userdata, paho_log_level, messages):
+        print(messages)
+
+    def on_connect(self, client, userdata, flags, reason_code, properties):
+        self.client.subscribe("/pydotbot/edge_to_controller")
+
+    def init(self, on_data_received: callable):
+        self.on_data_received = on_data_received
+        self.client = mqtt.Client(
+            mqtt.CallbackAPIVersion.VERSION2,
+            protocol=mqtt.MQTTProtocolVersion.MQTTv5,
+        )
+        self.client.tls_set_context(context=None)
+        # self.client.on_log = self.on_log
+        self.client.on_connect = self.on_connect
+        self.client.on_message = self.on_message
+        self.client.connect(self.host, self.port, 60)
+        self.client.loop_start()
+
+    def close(self):
+        self.client.disconnect()
+        self.client.loop_stop()
+
+    def send_data(self, data):
+        self.client.publish(
+            "/pydotbot/controller_to_edge",
+            base64.b64encode(data).decode(),
+        )