from time import sleep
import network
import machine
import ujson
from umqtt.simple import MQTTClient
import neopixel

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

# MQTT settings
MQTT_BROKER = "broker.mqttdashboard.com"
CLIENT_ID = "clientId-C9SvHXtTIs"
NEOPIXEL_TOPIC = b"Neopixel_led"
SERVO_TOPIC = b"Servo_motor"

# NeoPixel configuration
NEOPIXEL_PIN = 13  # D13 pin on ESP32
NUM_PIXELS = 10  # Number of NeoPixels
np = neopixel.NeoPixel(machine.Pin(NEOPIXEL_PIN), NUM_PIXELS)

# Servo configuration
SERVO_PIN = 14  # D14 pin on ESP32
servo = machine.PWM(machine.Pin(SERVO_PIN), freq=50)
SERVO_MIN_ANGLE = 0
SERVO_MAX_ANGLE = 180

def connect_wifi():
    sta_if = network.WLAN(network.STA_IF)
    if not sta_if.isconnected():
        print("Connecting to WiFi...")
        sta_if.active(True)
        sta_if.connect(WIFI_SSID)
        while not sta_if.isconnected():
            pass
        print("Connected to WiFi:", sta_if.ifconfig())

def mqtt_callback(topic, msg):
    print("Received MQTT message on topic:", topic)
    try:
        payload = ujson.loads(msg)
        if topic == NEOPIXEL_TOPIC and 'color' in payload:
            color = int(payload['color'])
            for i in range(NUM_PIXELS):
                np[i] = (color >> 16 & 255, color >> 8 & 255, color & 255)
            np.write()
            print("Set NeoPixel color:", color)
        elif topic == SERVO_TOPIC and 'angle' in payload:
            angle = int(payload['angle'])
            if angle < SERVO_MIN_ANGLE:
                angle = SERVO_MIN_ANGLE
            elif angle > SERVO_MAX_ANGLE:
                angle = SERVO_MAX_ANGLE
            duty = int((angle / 180) * 1023)
            servo.duty(duty)
            print("Set servo angle:", angle)
    except Exception as e:
        print("Error processing MQTT message:", e)

def connect_mqtt():
    client = MQTTClient(CLIENT_ID, MQTT_BROKER)
    client.set_callback(mqtt_callback)
    client.connect()
    client.subscribe(NEOPIXEL_TOPIC)
    client.subscribe(SERVO_TOPIC)
    print("Connected to MQTT broker")

    while True:
        client.check_msg()
        sleep(1)

def main():
    connect_wifi()
    connect_mqtt()

if __name__ == '__main__':
    main()