import network
import time
from machine import Pin
import dht
import ujson
from umqtt.simple import MQTTClient
from machine import Timer
import machine
from machine import I2C, Pin
import ssd1306

# MQTT Server Parameters
MQTT_CLIENT_ID = "shskdhjahkjhkjahk111111"
MQTT_BROKER    = "broker.emqx.io"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "licheng2206055821"
TOPIC1 = b"ledctl2206055821"  #灯控主题，为了避免主题和其他同学一样，末尾需要添加自己的学号
TOPIC2 = b"pwmled2206055821"
prev_weather = ""
sensor = dht.DHT22(Pin(15))

# 选择gpio2引脚
led_blue = Pin(2, Pin.OUT)  # 设置 GPIO2 为输出
led_blue.on()  # 初始状态熄灭LED

# OLED显示相关
i2c = I2C(0, scl=Pin(22), sda=Pin(21))
oled_width = 128
oled_height = 64
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)

# 连接WiFi配置
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!")

# 定义回调函数
def sub_cb(topic, msg):   #定义收到mqtt主题信息之后的回调函数
    if topic == TOPIC1:
        print((topic, msg))
        if msg == b"off":
            led_blue.on()  # 收到off消息时，点亮LED
        elif msg == b"on":
            led_blue.off()  # 收到on消息时，熄灭LED
    if topic == TOPIC2:
        print((topic,msg))     #亮度控制
        LEDval = int(msg)
        print(LEDval)
        led_blue.duty(LEDval)
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.subscribe(TOPIC1)  # 订阅灯控主题
print("Connected!")

# 定义采集温湿度的函数，方便后面定时器定时调用
def sample_weather(timer):
    global prev_weather
    print("Measuring weather conditions... ", end="")
    sensor.measure() 
    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.text("licheng2206055821",0,10)
        oled.text("temp:%.1f dC"%(msg_data["temp"]),0,20)
        oled.text("humidity:%.1f RH"%(msg_data["humidity"]),0,30)
        oled.show()
        oled.fill(0)
        client.publish(MQTT_TOPIC, message)
        prev_weather = message
    else:
        print("No change")

tim = Timer(-1)  # 新建一个虚拟定时器
# 此处执行时程序会等待5秒读取一次温湿度，period=5000 以毫秒为单位，mode=Timer.ONE_SHOT 表示只执行一次，Timer.PERIODIC表示循环执行，callback绑定回调函数
tim.init(period=5000, mode=Timer.PERIODIC, callback=sample_weather)

while True:
    client.wait_msg() 
    time.sleep(1)