import pygame
import random
from time import sleep

# game speed
SPEED = 1


# Initialize Pygame
pygame.init()

# Define colors
GREEN = (0, 255, 0)
RED = (255, 0, 0)
BLACK = (2, 2, 2)
WHITE = (255, 255, 255)

# Define screen dimensions
SCREEN_WIDTH = 640
SCREEN_HEIGHT = 480

# Create the screen
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
pygame.display.set_caption("Joystick Game")

# known button values
# A = 0
# B = 1
# X = 2
# Y = 3
# LB = 4
# RB = 5
# 2square = 6
# select = 7
# LSB = 8
# RSB = 9



# Define joystick buttons
BUTTON_A = 0
BUTTON_B = 1
BUTTON_X = 2
BUTTON_Y = 3
BUTTON_LB = 4
BUTTON_RB = 5
BUTTON_SQ = 6
BUTTON_SELECT = 7
BUTTON_LSB = 8
BUTTON_RSB = 9

BUTTON_EJECT = BUTTON_RB
BUTTON_HIGH = BUTTON_Y
BUTTON_MID = BUTTON_X
BUTTON_LOW = BUTTON_A
BUTTON_INTAKE = BUTTON_LB
BUTTON_ARM_DOWN = BUTTON_LSB
BUTTON_ARM_INTAKE = BUTTON_SELECT


# Define joystick axes
AXIS_X = 0
AXIS_Y = 1

# Initialize joystick
joystick = None
if pygame.joystick.get_count() > 0:
    joystick = pygame.joystick.Joystick(0)
    joystick.init()

def display_text(text1):
    # Create a font object
    font = pygame.font.Font(None, 36)

    # Render the text
    text = font.render(text1, True, WHITE)

    # Get the rect for the text surface
    text_rect = text.get_rect()

    # Center the text on the screen
    text_rect.centerx = screen.get_rect().centerx
    text_rect.centery = screen.get_rect().centery

    # Fill the screen with black
    screen.fill(BLACK)

    # Blit the text onto the surface
    screen.blit(text, text_rect)

    # Update the screen
    pygame.display.update()

# Define game loop
def game_loop():
    while True:
        sleep(SPEED)
        # Handle events
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                quit()

        # reset screen
        screen.fill(BLACK)
        pygame.display.update()

        # Check if joystick is connected
        if joystick is None:
            continue

        # Get random action
        action = random.randint(0, 6)

        # good or bad
        win = True

        skip = False

        # Handle action
        if action == 0:  # Press eject
            #print("press A")
            display_text("press: HIGH")
            while not joystick.get_button(BUTTON_HIGH) and not skip:
                for event in pygame.event.get():
                    print(event.type)
                    if event.type == 1540:
                        screen.fill(RED)
                        win = False
                        skip = True
                    if event.type == pygame.QUIT:
                        pygame.quit()
                        quit()
            if win:
                screen.fill(GREEN)
        elif action == 1:  # Press high
            #print("press B")
            display_text("press: MID")
            while not joystick.get_button(BUTTON_MID) and not skip:
                for event in pygame.event.get():
                    print(event.type)
                    if event.type == 1540:
                        screen.fill(RED)
                        win = False
                        skip = True
                    if event.type == pygame.QUIT:
                        pygame.quit()
                        quit()
            if win:
                screen.fill(GREEN)
        elif action == 2:  # Press mid
            #print("press X")
            display_text("press: LOW")
            while not joystick.get_button(BUTTON_LOW) and not skip:
                for event in pygame.event.get():
                    print(event.type)
                    if event.type == 1540:
                        screen.fill(RED)
                        win = False
                        skip = True
                    if event.type == pygame.QUIT:
                        pygame.quit()
                        quit()
            if win:
                screen.fill(GREEN)
        elif action == 3:  # Press low
            #print("press Y")
            display_text("press: EJECT")
            while not joystick.get_button(BUTTON_EJECT) and not skip:
                for event in pygame.event.get():
                    print(event.type)
                    if event.type == 1540:
                        screen.fill(RED)
                        win = False
                        skip = True
                    if event.type == pygame.QUIT:
                        pygame.quit()
                        quit()
            if win:
                screen.fill(GREEN)
        elif action == 4:  # Press low
            #print("press Y")
            display_text("press: INTAKE")
            while not joystick.get_button(BUTTON_INTAKE) and not skip:
                for event in pygame.event.get():
                    print(event.type)
                    if event.type == 1540:
                        screen.fill(RED)
                        win = False
                        skip = True
                    if event.type == pygame.QUIT:
                        pygame.quit()
                        quit()
            if win:
                screen.fill(GREEN)
        elif action == 5:  # Press low
            #print("press Y")
            display_text("press: ARM DOWN")
            while not joystick.get_button(BUTTON_ARM_DOWN) and not skip:
                for event in pygame.event.get():
                    print(event.type)
                    if event.type == 1540:
                        screen.fill(RED)
                        win = False
                        skip = True
                    if event.type == pygame.QUIT:
                        pygame.quit()
                        quit()
            if win:
                screen.fill(GREEN)
        elif action == 6:  # Press low
            #print("press Y")
            display_text("press: ARM INTAKE")
            while not joystick.get_button(BUTTON_ARM_INTAKE) and not skip:
                for event in pygame.event.get():
                    print(event.type)
                    if event.type == 1540:
                        screen.fill(RED)
                        win = False
                        skip = True
                    if event.type == pygame.QUIT:
                        pygame.quit()
                        quit()
            if win:
                screen.fill(GREEN)

        # Update the screen
        pygame.display.update()
        print("sleeping")
        #sleep(1)

# Start the game
if joystick is not None:
    display_text("Press A to start")
    while not joystick.get_button(BUTTON_A):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                quit()
    display_text("Starting...")
    game_loop()
else:
    print("No joystick found")