import machine
import network
from umqtt.simple import MQTTClient
import time
import dht
import lcd_api
import i2c_lcd

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

# Servo setup
servo = machine.PWM(machine.Pin(23), freq=50)  # Servo is using Pin 23

# DHT22 setup
dht_sensor = dht.DHT22(machine.Pin(14))  # DHT22 is using Pin 14

# I2C LCD setup
I2C_ADDR = 0x27
I2C_SDA_PIN = 21
I2C_SCL_PIN = 22
i2c = machine.I2C(scl=machine.Pin(I2C_SCL_PIN), sda=machine.Pin(I2C_SDA_PIN), freq=400000)
lcd = i2c_lcd.I2cLcd(i2c, I2C_ADDR, 2, 16)

# LED setup
led_temp = machine.Pin(5, machine.Pin.OUT)  # LED for high temperature (Pin 15)
led_hum = machine.Pin(12, machine.Pin.OUT)  # LED for low humidity (Pin 16)

# Thresholds
TEMP_THRESHOLD = 40.0  # threshold for high temperature in Celsius
HUM_THRESHOLD = 30.0  # threshold for low humidity in percentage

# MQTT Broker setup
CLIENT_ID = ""
BROKER = "mqtt-dashboard.com"
TOPIC_SERVO = "AfiqServo"
TOPIC_DHT = "dht22/data"
client = MQTTClient(CLIENT_ID, BROKER)

# Connect to Wi-Fi
def connect_to_wifi():
    wlan = network.WLAN(network.STA_IF)
    wlan.active(True)
    if not wlan.isconnected():
        print('Connecting to WiFi...')
        wlan.connect(WIFI_SSID, WIFI_PASSWORD)
        while not wlan.isconnected():
            pass
    print('WiFi Connected:', wlan.ifconfig())

# Connect to MQTT Broker
def connect_to_broker():
    try:
        client.connect()
        print("Connected to MQTT broker")
        client.set_callback(on_message)
        client.subscribe(TOPIC_SERVO)
    except Exception as e:
        print("Error connecting to MQTT broker:", e)

# MQTT message callback
def on_message(topic, msg):
    topic_str = topic.decode('utf-8')  # Convert bytes to string
    if topic_str == TOPIC_SERVO:
        process_servo_command(msg)

# Process servo commands
def process_servo_command(msg):
    try:
        # Decode the message to integer
        angle = int(msg.decode('utf-8'))
        print("Received angle command:", angle)

        # Convert the angle to duty cycle values
        duty_cycle = int(((angle / 180) * 1023))  # Mapping 0-180 to duty cycle
        print("Calculated duty cycle:", duty_cycle)
        servo.duty(duty_cycle)
    except ValueError as ve:
        print("Invalid integer format for servo command:", ve)
    except Exception as e:
        print("Error processing servo command:", e)

    # Convert the angle to duty cycle
    duty = (int(((angle)/180 *2 + 0.5) / 20 * 1023))
    # Set duty cycle
    servo.duty(duty)

# Publish DHT22 sensor data
def publish_sensor_data():
    try:
        dht_sensor.measure()
        temperature = dht_sensor.temperature()
        humidity = dht_sensor.humidity()
        payload = '{"temperature": ' + str(temperature) + ', "humidity": ' + str(humidity) + '}'
        client.publish(TOPIC_DHT, payload)
        print("Published sensor data:", payload)
        
        # Display data on LCD
        lcd.clear()
        lcd.putstr("Temp: {:.1f}C".format(temperature))
        lcd.move_to(0, 1)
        lcd.putstr("Hum: {:.1f}%".format(humidity))

        # Update LEDs based on thresholds
        if temperature > TEMP_THRESHOLD:
            led_temp.on()
        else:
            led_temp.off()

        if humidity < HUM_THRESHOLD:
            led_hum.on()
        else:
            led_hum.off()


    except Exception as e:
        print("Error reading/publishing sensor data:", e)

# Main loop
def main():
    connect_to_wifi()
    connect_to_broker()
    last_publish = 0

    while True:
        client.check_msg()
        current_time = time.time()
        if current_time - last_publish > 10:
            publish_sensor_data()
            last_publish = current_time
        time.sleep(0.1)

if __name__ == "__main__":
    main()