dune-weaver/modules/mqtt/factory.py

"""Factory for creating MQTT handlers."""
import os
from typing import Dict, Callable
from dotenv import load_dotenv
from pathlib import Path
from .base import BaseMQTTHandler
from .handler import MQTTHandler
from .mock import MockMQTTHandler
from .utils import create_mqtt_callbacks

import logging

logger = logging.getLogger(__name__)

# Load environment variables
BASE_DIR = Path(__file__).resolve().parent.parent.parent  # Go up to project root
env_path = os.path.join(BASE_DIR, '.env')
load_dotenv(env_path)

def create_mqtt_handler() -> BaseMQTTHandler:
    """Create and return an appropriate MQTT handler based on configuration.
    
    Returns:
        BaseMQTTHandler: Either a real MQTTHandler if MQTT_BROKER is configured,
                        or a MockMQTTHandler if not.
    """
    mqtt_broker = os.getenv('MQTT_BROKER')
    if mqtt_broker:
        logger.info(f"Got MQTT configuration for broker: {mqtt_broker}, instantiating MQTTHandler")
        return MQTTHandler(create_mqtt_callbacks())
    
    logger.info("MQTT Not going to be used, instantiating MockMQTTHandler")
    return MockMQTTHandler()