from machine import Pin
from umqtt.simple import MQTTClient
import ujson
import network # NOTE: biblioteca para conexão wifi
import utime as time

# Device setup
DEVICE_ID = "clientId-nmUDVIhgcG"

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

# MQTT setup
MQTT_CLIENT = DEVICE_ID
MQTT_BROKER = "mqtt-dashboard.com"
MQTT_TELEMETRY_TOPIC = "iot/telemetry"
MQTT_CONTROL_TOPIC = "iot/control"

# NOTE: definindo pinos
# DHT sensor setup
DHT_PIN = Pin(15)
# LED/LAMP setup
RED_LED = Pin(12, Pin.OUT)
BLUE_LED = Pin(13, Pin.OUT)

RED_LED.on() # NOTE: liga o LED
RED_LED.off() # NOTE: desliga o LED

# Methods
def did_recieve_callback(topic, message):
  print("\n\nData Recieved! \ntopic = {0}, message = {1}".format(topic, message))

# NOTE: Conexão com o servidor Broker MQTT
def mqtt_connect():
  print("Connecting to MQTT broker...", end="")
  mqtt_client = MQTTClient(MQTT_CLIENT, MQTT_BROKER, user="", password="")
  mqtt_client.set_callback(did_recieve_callback)
  mqtt_client.connect()
  print("Connected.", end="")
  mqtt_client.subscribe(MQTT_CONTROL_TOPIC)
  return mqtt_client

def create_json_data(temperature, humidity):
  data = ujson.dumps({
    "temp": temperature,
    "humidity": humidity,
  })
  return data

def mqtt_client_publish(topic, data):
  print("\nUpdating MQTT Broker...")
  mqtt_client.publish(topic, data)

# Application Logic

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

# Wait until WiFi is Connected
while not wifi_client.isconnected();
  print("Connecting")
  time.sleep(0.1)

print("WiFi Connected!")
print(wifi_client.ifconfig())

# Connect to MQTT
mqtt_client = mqtt_connect()
RED_LED.off()
BLUE_LED.off()
dht_sensor = dht.DHT22(DHT_PIN)
telemetry_data_old = ""

while True:
  mqtt_client.check_msg()
  print(". ", end="")
  dht_sensor.measure()
  telemetry_data_new = create_json_data(dht_sensor.temperature(), dht_sensor.humidity())

  if telemetry_data_new != telemetry_data_old:
    mqtt_client_publish(MQTT_TELEMETRY_TOPIC, telemetry_data_new)
    telemetry_data_old = telemetry_data_new

  time.sleep(0.1)