from pylgbst.constants import COLOR_GREEN, COLOR_NONE
from . import *

robot = Vernie()
running = True


def callback(color, distance):
    robot.led.set_color(color)
    speed = (10 - distance + 1) / 10.0
    secs = (10 - distance + 1) / 10.0
    print("Distance is %.1f inches, I'm running back with %s%% speed!" % (distance, int(speed * 100)))
    if speed <= 1:
        robot.motor_AB.timed(secs / 1, -speed, is_async=True)
        robot.say("Ouch")


def on_btn(pressed):
    global running
    if pressed:
        running = False


robot.led.set_color(COLOR_GREEN)
robot.button.subscribe(on_btn)
robot.color_distance_sensor.subscribe(callback)
robot.say("Place your hand in front of sensor")

while running:
    time.sleep(1)

robot.color_distance_sensor.unsubscribe(callback)
robot.button.unsubscribe(on_btn)
robot.led.set_color(COLOR_NONE)
while robot.led.in_progress():
    time.sleep(1)