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

MQTT_CLIENT_ID = "micropython-weather-demo"
MQTT_BROKER = "test.mosquitto.org"
MQTT_USER = ""
MQTT_PASSWORD = ""
MQTT_TOPIC = "Vitaliy"

# Глобальная переменная для хранения текущей частоты
current_frequency = 0

buzzer_pin = Pin(27)
buzzer_pwm = PWM(buzzer_pin)
buzzer_pwm.duty(0)

def set_buzzer_frequency(frequency):
    global current_frequency
    if frequency > 0:
        buzzer_pwm.freq(frequency)
        buzzer_pwm.duty(512)
        current_frequency = frequency  # Сохраняем частоту
    else:
        buzzer_pwm.duty(0)
        current_frequency = 0  # Сбрасываем частоту при выключении

def message_callback(topic, msg):
    print(f"Received message: {msg} from topic: {topic}")
    try:
        value = int(msg) * 100
        print(f"Converted to integer and multiplied: {value}")
        set_buzzer_frequency(value)
    except ValueError:
        try:
            value = float(msg) * 100
            print(f"Converted to float and multiplied: {value}")
            set_buzzer_frequency(int(value))
        except ValueError:
            print("Received non-numeric message. Turning off buzzer.")
            set_buzzer_frequency(0)

# Подключение к Wi-Fi
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!")

# Подключение к MQTT брокеру
print("Connecting to MQTT server... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.set_callback(message_callback)
client.connect()
client.subscribe(MQTT_TOPIC)
print("Connected!")

# Инициализация датчика DHT22 и OLED дисплея
d = dht.DHT22(Pin(15))
led_1 = Pin(12, Pin.OUT)
led_2 = Pin(14, Pin.OUT)

i2c = I2C(0, scl=Pin(22), sda=Pin(21))
oled_width = 128
oled_height = 64
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)

prev_weather = ""
while True:
    # Очистка экрана в начале итерации
    oled.fill(0)
    
    # Измерение температуры и влажности
    d.measure()
    temp = d.temperature()
    hum = d.humidity() 
    message = ujson.dumps({
        "Temp: ": temp,
        "Hum: ": hum,
    })

    # Отображение данных на экране
    oled.text(f'Temperature:{temp}C', 0, 0)
    oled.text(f'Humidity:{hum}%', 0, 20)
    oled.text(f'Frequency:{current_frequency}Hz', 0, 40)  # Добавлена частота
    
    # Управление светодиодами
    led_1.value(temp >= 50.0)
    led_2.value(hum >= 50.0)
    
    # Обновление дисплея
    oled.show()

    # Публикация в MQTT при изменении данных
    if message != prev_weather:
        print(f"Reporting to MQTT topic {MQTT_TOPIC}: {message}")
        client.publish(MQTT_TOPIC, message)
        prev_weather = message
    else:
        print("No change")

    # Проверка входящих сообщений
    client.check_msg()
    
    # Задержка между измерениями
    time.sleep(2)