from machine import Pin
from umqtt.simple import MQTTClient
import ujson
import network
import utime as time
import dht

# Device Setup
DEVICE_ID = "nadira"

# WiFi Setup
WIFI_SSID = "Wokwi-GUEST"
WIFI_PASSWORD = ""

# MQTT Setup
MQTT_BROKER = "broker.emqx.io"
MQTT_CLIENT_ID = DEVICE_ID
MQTT_DATA_SENSOR_TOPIC = "SEMOGA_SEMANGATNYA_GA_DIAWAL_DOANG_TAPI_SETERUSNYA/nadira_najmi_rahmany/data_sensor"
MQTT_AKTUASI_LED_TOPIC = "SEMOGA_SEMANGATNYA_GA_DIAWAL_DOANG_TAPI_SETERUSNYA/nadira_najmi_rahmany/aktuasi_led"

# DHT Sensor Setup
DHT_PIN = Pin(15)

# LED Setup
RED_LED = Pin(12, Pin.OUT)
BLUE_LED = Pin(13, Pin.OUT)
FLASH_LED = Pin(2, Pin.OUT)

# Initialize LEDs to OFF state
RED_LED.off()
BLUE_LED.off()

# Global Variables
data_sensor_previous = ""

def mqtt_message_callback(topic, message):
    print(f"\nMessage received!\nTopic: {topic}\nMessage: {message}")

    if topic == MQTT_AKTUASI_LED_TOPIC.encode():
        if message == f"{DEVICE_ID}/red/on".encode():
            RED_LED.on()
        elif message == f"{DEVICE_ID}/red/off".encode():
            RED_LED.off()
        elif message == f"{DEVICE_ID}/blue/on".encode():
            BLUE_LED.on()
        elif message == f"{DEVICE_ID}/blue/off".encode():
            BLUE_LED.off()
        elif message == f"{DEVICE_ID}/lamp/on".encode():
            RED_LED.on()
            BLUE_LED.on()
        elif message == f"{DEVICE_ID}/lamp/off".encode():
            RED_LED.off()
            BLUE_LED.off()
        elif message == f"{DEVICE_ID}/status".encode():
            send_led_status()


def mqtt_connect():
    print("Connecting to MQTT broker...", end=" ")
    client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER)
    client.set_callback(mqtt_message_callback)
    client.connect()
    print("Connected.")
    client.subscribe(MQTT_AKTUASI_LED_TOPIC)
    return client


def create_sensor_data_json(temperature, humidity):
    return ujson.dumps({
        "device_id": DEVICE_ID,
        "temperature": temperature,
        "humidity": humidity,
        "type": "sensor"
    })


def send_led_status():
    status_data = ujson.dumps({
        "device_id": DEVICE_ID,
        "red_led": "ON" if RED_LED.value() else "OFF",
        "green_led": "ON" if BLUE_LED.value() else "OFF",
        "type": "lamp"
    })
    mqtt_client.publish(MQTT_DATA_SENSOR_TOPIC, status_data)
    print("LED status sent.")


# Connect to WiFi
wifi = network.WLAN(network.STA_IF)
wifi.active(True)
print("Connecting to WiFi...")
wifi.connect(WIFI_SSID, WIFI_PASSWORD)

while not wifi.isconnected():
    print("Waiting for connection...")
    time.sleep(0.5)
print("WiFi connected!", wifi.ifconfig())

# Connect to MQTT Broker
mqtt_client = mqtt_connect()

# Initialize Sensor
sensor = dht.DHT22(DHT_PIN)

while True:
    try:
        mqtt_client.check_msg()  # Check for incoming messages
        
        # Flash LED for activity indication
        FLASH_LED.on()
        time.sleep(0.2)
        FLASH_LED.off()

        # Read sensor data
        sensor.measure()
        temperature = sensor.temperature()
        humidity = sensor.humidity()

        # Prepare sensor data and publish if changed
        sensor_data = create_sensor_data_json(temperature, humidity)
        if sensor_data != data_sensor_previous:
            mqtt_client.publish(MQTT_DATA_SENSOR_TOPIC, sensor_data)
            print(f"Published sensor data: {sensor_data}")
            data_sensor_previous = sensor_data

    except Exception as e:
        print(f"Error: {e}")

    time.sleep(1)