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

# MQTT Server Parameters
MQTT_CLIENT_ID = "esp32"
MQTT_BROKER    = "rasp-mirko.cloud.shiftr.io"
MQTT_USER      = "rasp-mirko"
MQTT_PASSWORD  = "EksEF8TsFRiENfgc"

TOPIC_DATA   = b"sensor/data"   # temp + humidity
TOPIC_STATUS = b"sensor/status" # hot/cold message

# Setup DHT22 sensor on GPIO4
sensor = dht.DHT22(Pin(4))

# WiFi Connection
print("Connecting to WiFi", end="")
sta_if = network.WLAN(network.STA_IF)
sta_if.active(True)
sta_if.connect("Wokwi-GUEST", "")
while not sta_if.isconnected():
    print(".", end="")
    time.sleep(0.2)
print(" Connected!")

# MQTT Server connection
print("Connecting to MQTT server... ", end="")
client = mq.MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, port=1883, user=MQTT_USER, password=MQTT_PASSWORD)
client.connect()
print("connected")

# --- Publish only 2 times in 60 seconds ---
while(True):
    sensor.measure()
    temp = sensor.temperature()
    hum = sensor.humidity()

    # first message: sensor data
    message_data = ujson.dumps({"temperature": temp, "humidity": hum})
    client.publish(TOPIC_DATA, message_data)
    print("Published to", TOPIC_DATA, ":", message_data)

    # second message: hot or cold
    status = "Hot" if temp >= 30 else "Cold"
    client.publish(TOPIC_STATUS, status)
    print("Published to", TOPIC_STATUS, ":", status)
    time.sleep(2)

print("✅ Done sending 2 messages, stopping.")