"""
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, I2C  
import dht  
import ujson  
from umqtt.simple import MQTTClient  
import ssd1306  
led_blue = Pin(2,Pin.OUT)  # 设置 GPIO2 为输出 
# MQTT Server Parameters  
MQTT_CLIENT_ID = "luochaorong/2206054010"  
MQTT_BROKER = "broker.emqx.io"  
MQTT_USER = ""  
MQTT_PASSWORD = ""  
MQTT_TOPIC = "luochaorong/2206054010"  
TOPIC1 = b"ledctl2206054010"  #灯控主题，为了避免主题和其他同学一样，末尾需要添加自己的学号
TOPIC2 = b"pwmled2206054010"
TOPIC3 = b"ledstatus2206054010"
sensor = dht.DHT22(Pin(15))
pwmval =0 
  
i2c = I2C(0, scl=Pin(22), sda=Pin(21))  
oled_width = 128  
oled_height = 64  
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)  
oled.text('Hello, Wokwi!', 10, 10)  
oled.show() 
  
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.check_msg()  
print("Connected!")  
def sub_cb(topic, msg):   #定义收到mqtt主题信息之后的回调函数
    global pwm_old, pwmval
    if topic == TOPIC1:
        print((topic, msg))
        if msg == b"on":
            led_blue.value(0)
            pwmval = 0

        elif msg == b"off":
            led_blue.value(1)
            pwmval = 1023;

    if topic == TOPIC2:  
        try:
            pwm_value = int(msg)
            if pwm_value < 0 or pwm_value > 100:
                print(f"Invalid PWM value received: {pwm_value}, must be between 0 and 100.")
            else:
                pwmval = int(pwm_value *100 / 1023)  # 将0-100映射到0-65535范围内（ESP8266 PWM通常范围是0-65535）
                led_blue.value(pwmval)
                print((topic, pwmval))        
        except ValueError:
            print(f"Unable to convert message '{msg}' to an integer for PWM control.")       
print("Connecting to MQTT server... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.set_callback(sub_cb)    #定义好client参数之后，就需要在connect之前设定回调函数。
client.connect()
print("Connected!")
client.subscribe(TOPIC1)    #connect之后就需要马上订阅主题，才能收到mqtt服务器送过来的信息
client.subscribe(TOPIC2) 
prev_weather = ""  
while True:  
    print("Measuring weather conditions... ", end="")  
    sensor.measure()
    client.check_msg() 
    msg_data = {  
        "temp": sensor.temperature(),  
        "humidity": sensor.humidity(),  
    }  
    message = ujson.dumps(msg_data)  
    if message != prev_weather:  
        print("Updated!")  
        print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC, message))  
        oled.fill(0)
        oled.text("luochaorong",10,10)  
        oled.text("%.1f C" % (msg_data["temp"]), 10, 20)   
        oled.text("%.1f %%" % (msg_data["humidity"]), 10, 30) 
        oled.show()  
        client.publish(MQTT_TOPIC, message)  
        prev_weather = message  
    else:  
        print("Updated!")  
        print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC, message))  
        oled.fill(0)
        oled.text("luochaorong",10,10)   
        oled.text("%.1f C" % (msg_data["temp"]), 10, 20)   
        oled.text("%.1f %%" % (msg_data["humidity"]), 10, 30) 
        oled.show()  
        client.publish(MQTT_TOPIC, message)  
        prev_weather = message  
    time.sleep(1)