from blinkstick import blinkstick

from random import randint, choice
from time import sleep

def random_grb():
    """
    Get a random GRB array.
    """
    # get a random initial color
    return [
        randint(1, 255),
        randint(1, 255),
        randint(1, 255)
    ]
def set_all_leds(device, grb=[0,0,0]):
    """
    Takes a GRB color array as input and sets the color on all LEDs.
    """
    return_value = device.set_led_data(0, grb*32)
    sleep(0.02) # minimal delay to not crash the blinkstick
    return return_value
def morph_channel(color, steps):
    """
    Takes an int between 0-255 and either increments or decreases it by the specified steps.
    """
    direction = choice([-1, 1])
    color_candidate = color+steps*direction

    # if we go out of bounds reverse the direction
    if color_candidate < 0:
        return color+steps*direction*-1
    elif color_candidate > 255:
        return color+steps*direction*-1
    else:
        return color_candidate
def morph_grb(grb, steps, mode='single'):
    """
    Morphs a GRB array randomly by the specified steps.

    Available modes:
        single - only morph a single color
        all - morph all colors
    """

    if mode == 'single':
        color_index = randint(0, len(grb)-1)
        grb[color_index] = morph_channel(grb[color_index], steps)
    elif mode == 'all':
        for color_index in range(len(grb)):
            grb[color_index] = morph_channel(grb[color_index], steps)
    else:
        raise Exception(f'Mode "{mode}" is not supported.')

    return grb
def rainbow(device, min_value=1, max_value=255):
    """
    Takes an int between 0-255 and either increments or decreases it by the specified steps.
    """
    direction = [
        choice([-1, 1]),
        choice([-1, 1]),
        choice([-1, 1])
    ]

    grb = [
        randint(min_value, max_value),
        randint(min_value, max_value),
        randint(min_value, max_value)
    ]

    channel = randint(0, 2)
    while True:
        #channel = (channel + 1) % 3
        channel = randint(0, 2)
        for i in range(30):
            candidate = grb[channel] + 1 * direction[channel]

            # if we go out of bounds reverse the direction
            if candidate < min_value or candidate > max_value:
                direction[channel] = direction[channel] * -1
            else:
                grb[channel] = candidate
                set_all_leds(device, grb)


device = blinkstick.find_first()

rainbow(device)