"""
MicroPython IoT Weather Station Example for Wokwi.com

To view the data:

1. Go to http://www.hivemq.com/demos/websocket-client/
2. Click "Connect"
3. Under Subscriptions, click "Add New Topic Subscription"
4. In the Topic field, type "wokwi-weather" then click "Subscribe"

Now click on the DHT22 sensor in the simulation,
change the temperature/humidity, and you should see
the message appear on the MQTT Broker, in the "Messages" pane.

Copyright (C) 2022, Uri Shaked

https://wokwi.com/arduino/projects/322577683855704658
"""

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

# MQTT Server Parameters
MQTT_CLIENT_ID = "micropython-weather-demo"
MQTT_BROKER = "broker.mqttdashboard.com"
MQTT_USER = ""
MQTT_PASSWORD = ""

# MQTT Topics for each sensor
MQTT_TOPIC_WEATHER = "wokwi-weather"
MQTT_TOPIC_PIR = "wokwi-pir"
MQTT_TOPIC_ULTRASONIC = "wokwi-ultrasonic"

sensor_dht = dht.DHT22(Pin(15))  # DHT22 sensor on GPIO 15
sensor_pir = Pin(14, Pin.IN)  # PIR sensor on GPIO 14
sensor_ultrasonic_trig = Pin(4, Pin.OUT)  # Ultrasonic trigger on GPIO 4
sensor_ultrasonic_echo = Pin(5, Pin.IN)  # Ultrasonic echo on GPIO 5

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.1)
print(" Connected!")

print("Connecting to MQTT server... ", end="")
'''client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.connect(client_id=MQTT_CLIENT_ID, server=MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)'''
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.connect()

print("Connected!")

# Resto do código (sem alterações)


prev_weather = ""
prev_pir_state = -1  # Initialize PIR state to an invalid value
prev_ultrasonic_distance = -1.0  # Initialize Ultrasonic distance to an invalid value

while True:
    print("Measuring weather conditions... ", end="")
    sensor_dht.measure()
    weather_message = ujson.dumps({
        "temp": sensor_dht.temperature(),
        "humidity": sensor_dht.humidity(),
    })

    if weather_message != prev_weather:
        print("Updated Weather!")
        print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC_WEATHER, weather_message))
        client.publish(MQTT_TOPIC_WEATHER, weather_message)
        prev_weather = weather_message

    print("Reading PIR sensor... ", end="")
    pir_state = sensor_pir.value()

    if pir_state != prev_pir_state:
        print("Updated PIR!")
        pir_message = ujson.dumps({
            "motion": pir_state,
        })
        print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC_PIR, pir_message))
        client.publish(MQTT_TOPIC_PIR, pir_message)
        prev_pir_state = pir_state

    # Ultrasonic sensor measurement
    print("Measuring Ultrasonic distance... ", end="")
    sensor_ultrasonic_trig.on()
    time.sleep_us(10)
    sensor_ultrasonic_trig.off()

    while sensor_ultrasonic_echo.value() == 0:
        pulse_start = time.ticks_us()

    while sensor_ultrasonic_echo.value() == 1:
        pulse_end = time.ticks_us()

    pulse_duration = time.ticks_diff(pulse_end, pulse_start)
    distance = (pulse_duration * 0.0343) / 2  # Speed of sound is 343 m/s

    if distance != prev_ultrasonic_distance:
        print("Updated Ultrasonic!")
        ultrasonic_message = ujson.dumps({
            "distance_cm": distance,
        })
        print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC_ULTRASONIC, ultrasonic_message))
        client.publish(MQTT_TOPIC_ULTRASONIC, ultrasonic_message)
        prev_ultrasonic_distance = distance

    time.sleep(1)