import pygame

pygame.init()

SCREEN_WIDTH = 800
SCREEN_HEIGHT = 600

screen_surface = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
pygame.display.set_caption('Moja Gra')
clock = pygame.time.Clock()

def load_image(img_path: str, position):
    image = pygame.image.load(img_path)
    surface = image.convert()
 
    transparent_color = (0, 0, 0)
    surface.set_colorkey(transparent_color)
 
    rect = surface.get_rect(center=position)
 
    return [image, surface, rect]
 
 
def print_image(img_list) -> None:

    image, surface, rect = img_list
    screen_surface.blit(surface, rect)
    pass

def set_position(img_list, position):
    image, surface, rect = img_list
    rect = surface.get_rect(center = position)
    return [image, surface, rect]

def calculate_player_movement(keys):
    speed = 10
    delta_x = 0
    delta_y = 0

    if keys[pygame.K_LSHIFT]:
        speed *= 2

    if keys[pygame.K_w]:
        delta_y -= speed
    if keys[pygame.K_s]:
        delta_y += speed
    if keys[pygame.K_d]:
        delta_x += speed
    if keys[pygame.K_a]:
        delta_x -= speed

    return [ delta_x, delta_y]

def limit_position(position):
    x,y = position

    x = max(0, min(x, SCREEN_WIDTH))
    y = max(0, min(y, SCREEN_HEIGHT))
    return [x,y]


player_pos = [SCREEN_WIDTH // 2, SCREEN_HEIGHT // 2]
player = load_image('player.png', player_pos)
background_color = [149, 219, 147]


game_status = True

while game_status:

    events = pygame.event.get()

    for event in events:

        
        
        if event.type == pygame.QUIT:
            game_status = False
        pass

    pressed_keys = pygame.key.get_pressed()
    delta_x, delta_y = calculate_player_movement(pressed_keys)

    player_pos[0] += delta_x
    player_pos[1] += delta_y

    player_pos = limit_position(player_pos)

    player = set_position(player, player_pos)

    screen_surface.fill(background_color)
    print_image(player)
    pygame.display.update()
    clock.tick(60)

pygame.quit()
quit()