#!/usr/bin/env python3

import sys
import time

import pygame


def zpiece():
    return ((blue_square, blue_square, None), (None, blue_square, blue_square))


def refresh_screen():
    screen.fill(black)

    for row_idx, row in enumerate(grid):
        for col_idx, element in enumerate(row):
            if element is not None:
                screen.blit(element, (col_idx * 50 + 1, row_idx * 50 + 1))

    # Display the current piece
    for row_idx, row in enumerate(current_piece):
        for col_idx, element in enumerate(row):
            if element is not None:
                screen.blit(element, ((col_idx + current_piece_position[1]) * 50 + 1, (row_idx + current_piece_position[0]) * 50 + 1))

pygame.init()

black = (0, 0, 0)
white = (0xff, 0xff, 0xff)
blue = (0x10, 0x20, 0xbb)

white_square = pygame.Surface((48, 48))
white_square.fill(white)

blue_square = pygame.Surface((48, 48))
blue_square.fill(blue)

screen = pygame.display.set_mode((500, 1000))

grid = [[None for _ in range(10)] for _ in range(20)]

# Let's start with a Z piece
current_piece = zpiece()
current_piece_position = [0, 4]

while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            sys.exit()
        elif event.type == pygame.KEYDOWN:
            if event.key in (pygame.K_DOWN, pygame.K_j):
                current_piece_position[0] += 1
            if event.key in (pygame.K_UP, pygame.K_k):
                current_piece_position[0] -= 1
            if event.key in (pygame.K_RIGHT, pygame.K_l):
                current_piece_position[1] += 1
            if event.key in (pygame.K_LEFT, pygame.K_h):
                current_piece_position[1] -= 1

    refresh_screen()
    pygame.display.flip()
    time.sleep(.1)