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

led_pin = Pin(2, Pin.OUT)
sensor_dht = dht.DHT22(Pin(15))

MQTT_CLIENT_ID = "wokwi_esp32_client"
MQTT_BROKER    = "oljrhome.tplinkdns.com" 
MQTT_USER      = "batavo"
MQTT_PASSWORD  = "thor1428"
TOPIC_PUBLISH  = b"/pucpr/iot/aula"  #ENVIAR
TOPIC_SUBSCRIBE= b"/actuator" # RECEBER

print("Conectando ao Wi-Fi...", 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.1)
print(" Conectado!")

def sub_cb(topic, msg):
  print(f"Mensagem recebida no tópico {topic.decode()}: {msg.decode()}")
  if msg == b"1":
    print("Ligando o LED")
    led_pin.value(1)
  elif msg == b"0":
    print("Desligando o LED")
    led_pin.value(0)

print("Conectando ao servidor MQTT... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.set_callback(sub_cb) 
client.connect()
client.subscribe(TOPIC_SUBSCRIBE)
print(f"Conectado! Inscrito no tópico: {TOPIC_SUBSCRIBE.decode()}")

last_publish = time.time()
publish_interval = 5 

while True:
  client.check_msg()

  if (time.time() - last_publish) >= publish_interval:
    print("Medindo condições de temperatura e umidade...")
    try:
      sensor_dht.measure()
      temp = sensor_dht.temperature()
      hum = sensor_dht.humidity()

      message = ujson.dumps({
        "temperatura": temp,
        "umidade": hum,
      })

      print(f"Publicando no tópico {TOPIC_PUBLISH.decode()}: {message}")
      client.publish(TOPIC_PUBLISH, message)
      last_publish = time.time()
    except OSError as e:
      print(f"Falha ao ler o sensor: {e}")

  time.sleep(1)