import pygame, sys, random

class Crosshair(pygame.sprite.Sprite):
    def __init__(self, pos_x, pos_y, picture_path):
        super().__init__()
        self.image = pygame.image.load(picture_path)
        self.rect = self.image.get_rect()
        self.rect.topleft = [pos_x, pos_y]

    def update(self):
        self.rect.center = pygame.mouse.get_pos()

    def shoot(self):
        pygame.sprite.spritecollide(crosshair, target_sprites, True)

class Target(pygame.sprite.Sprite):
    def __init__(self, picture_path, pos_x, pos_y):
        super().__init__()
        self.image = pygame.image.load(picture_path)
        self.rect = self.image.get_rect()
        self.rect.x = pos_x
        self.rect.y = pos_y

class GameState():
    def __init__(self):
        self.state = "intro"

    def intro(self):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()
            if event.type == pygame.MOUSEBUTTONDOWN:
                self.state = "main_game"
    
        # Draw background tiles
        for y in range(num_tiles_y):
            for x in range(num_tiles_x):
                screen.blit(background, (x * background.get_width(), y * background.get_height()))
        
        crosshair_group.draw(screen)
        crosshair_group.update()
        pygame.display.flip()


    def main_game(self):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()
            if event.type == pygame.MOUSEBUTTONDOWN:
                crosshair.shoot()
    
        # Draw background tiles
        for y in range(num_tiles_y):
            for x in range(num_tiles_x):
                screen.blit(background, (x * background.get_width(), y * background.get_height()))
        
        target_sprites.draw(screen)
        crosshair_group.draw(screen)
        crosshair_group.update()
        pygame.display.flip()

    def state_manager(self):
        if self.state == "intro":
            self.intro()
        if self.state == "main_game":
            self.main_game()



#General Setup
pygame.init()
clock = pygame.time.Clock()
game_state = GameState()

#Game Screen
screen_width = 1920
screen_height = 1080
screen = pygame.display.set_mode((screen_width,screen_height))

# Load background image
background = pygame.image.load("BG.png").convert()

pygame.mouse.set_visible(False)

#Crosshair
crosshair = Crosshair(200, 200, "crosshair.png")
crosshair_group = pygame.sprite.Group()
crosshair_group.add(crosshair)

# Target
target_sprites = pygame.sprite.Group()
for target in range(50):
    new_target = Target("target.png", random.randrange(0, screen_width), random.randrange(0, screen_height))
    target_sprites.add(new_target)

# Calculate number of tiles needed to cover the screen
num_tiles_x = (screen_width // background.get_width()) + 1
num_tiles_y = (screen_height // background.get_height()) + 1

while True:
    game_state.state_manager()
    clock.tick(60)