from machine import Pin, UART
import time


uart0 = UART(0, baudrate=38400, tx=Pin(0), rx=Pin(1), timeout=100)
USER_LED_R = Pin(17, Pin.OUT)
USER_LED_G = Pin(16, Pin.OUT)
USER_LED_B = Pin(25, Pin.OUT)
BUTTON = Pin(26, Pin.IN)

color = 0
USER_LED_R.on()
USER_LED_G.on()
USER_LED_B.on()

while True:
    s = uart0.readline()
    if s != None:
        if s == b'MO1: led\r\n':
            color = (color + 1) %4
            if color == 0:
                USER_LED_R.on()
                USER_LED_G.on()
                USER_LED_B.on()
            if color == 1:
                USER_LED_R.off()
                USER_LED_G.on()
                USER_LED_B.on()
            if color == 2:
                USER_LED_R.on()
                USER_LED_G.off()
                USER_LED_B.on()
            if color == 3:
                USER_LED_R.on()
                USER_LED_G.on()
                USER_LED_B.off()
            print("LED ON done")
        else:
            print(f"Unknown command: {s}")
    if BUTTON.value() == 1:
        print("Button pressed")
        uart0.write(b'led')
        while(BUTTON.value() == 1):
              pass