import time
time.sleep(0.1) # Wait for USB to become ready

print("Hello, Pi Pico!")

from Button import *
from Displays import *
from LightStrip import *
from SpinningWheel import *
from PrizeChest import *
from Buzzer import *

class SpinToWin:

    def __init__(self):
        self.button = Button(pin=15, name='b', handler=self)
        self.display = LCDDisplay(sda=0, scl=1)
        self.lightstrip = LightStrip(2, numleds=8,brightness=1)
        self.wheel = SpinningWheel(self.lightstrip)
        self.buzzer = PassiveBuzzer(3)
        self.prizes = PrizeChest(8)
        self.display.showText('Push the button')

    def buttonPressed(self, name):
        self.display.showText("Let's go!!!    ")
        p = self.wheel.spin()
        self.display.clear()
        t = self.prizes.get_prize_type(p)
        if t == 'no prize':
            self.display.showText('Sorry you lose')
            self.buzzer.beep(200, 1000)
        elif t == 'retry':
            self.display.showText('Try again!')
            self.buzzer.beep(500, 1000)
        else:
            v = self.prizes.get_prize_value(p)
            self.display.showText(f'You win ${v}!')
            self.buzzer.beep(1000, 2000)
        

    def buttonReleased(self, name):
        pass

    def run(self):
        while True:
            time.sleep(2)


if __name__ == '__main__':
    game = SpinToWin()

    game.run()