IoT-Systems/edge/main.py

import logging
import os
from app.adapters.agent_mqtt_adapter import AgentMQTTAdapter
from app.adapters.hub_http_adapter import HubHttpAdapter
from app.adapters.hub_mqtt_adapter import HubMqttAdapter
from config import (
    MQTT_BROKER_HOST,
    MQTT_BROKER_PORT,
    MQTT_TOPIC,
    HUB_URL,
    HUB_MQTT_BROKER_HOST,
    HUB_MQTT_BROKER_PORT,
    HUB_MQTT_TOPIC,
    HUB_CONNECTION_TYPE,
)

if __name__ == "__main__":
    # Configure logging settings
    logging.basicConfig(
        level=logging.INFO,
        format="[%(asctime)s] [%(levelname)s] [%(module)s] %(message)s",
        handlers=[
            logging.StreamHandler(),
            logging.FileHandler("app.log"),
        ],
    )

    # Logic to select the adapter based on configuration (SCRUM-93 & SCRUM-94)
    # This allows easy switching between HTTP and MQTT protocols
    if HUB_CONNECTION_TYPE.lower() == "http":
        logging.info("Initializing HubHttpAdapter (SCRUM-93 integration)")
        hub_adapter = HubHttpAdapter(
            api_base_url=HUB_URL,
        )
    else:
        logging.info("Initializing HubMqttAdapter (SCRUM-94 integration)")
        hub_adapter = HubMqttAdapter(
            broker=HUB_MQTT_BROKER_HOST,
            port=HUB_MQTT_BROKER_PORT,
            topic=HUB_MQTT_TOPIC,
        )

    # Create an instance of the AgentMQTTAdapter using the selected hub adapter
    # This adapter acts as a bridge between the Agent and the Hub
    agent_adapter = AgentMQTTAdapter(
        broker_host=MQTT_BROKER_HOST,
        broker_port=MQTT_BROKER_PORT,
        topic=MQTT_TOPIC,
        hub_gateway=hub_adapter,
    )

    try:
        logging.info(f"Connecting to MQTT broker at {MQTT_BROKER_HOST}:{MQTT_BROKER_PORT}")
        agent_adapter.connect()

        logging.info("Broker connection success. Waiting for data...")
        agent_adapter.loop_forever()
    except KeyboardInterrupt:
        logging.info("Interrupt signal received. Shutting down...")
        agent_adapter.disconnect()
        logging.info("Disconnected from MQTT broker.")