import pygame
import enums
from random import choice

BRICKS = [
    " X " + "\n" + "XXX" + "\n" + " X ",
    "  X" + "\n" + "XXX",
    "X  " + "\n" + "XXX",
    "X",
    "XXXX",
    "XX" + "\n" + "XX",
    " XX" + "\n" + "XX ",
    "XX " + "\n" + " XX",
    "X X" + "\n" + "XXX",
    "XXX" + "\n" + " X "
]

TILE_SIZE = 48

class Brick:
    def __init__(self, brick, state = enums.BrickState.Next):
        self.layout = []
        self.color = choice(list(enums.BrickColor))
        self.set_state(state)
        self.angle = 0
        self.direction = None
        self.load_brick(brick)
        self.get_image()
        self.block_image = pygame.transform.scale(self.img, (48, 48)) if self.img else None
        self.draw_brick()
        self.x, self.y = (3, 0)
        self.location = list((self.x, self.y))
    
    def get_image(self):
        match self.color:
            case enums.BrickColor.Magenta:
                self.img = pygame.image.load("assets/magenta_block.png").convert_alpha()
            case enums.BrickColor.Cyan:
                self.img = pygame.image.load("assets/cyan_block.png").convert_alpha()
            case enums.BrickColor.Yellow:
                self.img = pygame.image.load("assets/yellow_block.png").convert_alpha()
            case _:
                self.img = pygame.image.load("assets/magenta_block.png").convert_alpha()
                
    def load_brick(self, brick):
        self.layout = [l for l in BRICKS[brick].splitlines()]
        
        self.rows = len(self.layout)
        self.cols = len(self.layout[0])
        self.width = self.cols * TILE_SIZE
        self.height = self.rows * TILE_SIZE
        
    def draw_brick(self):
        self.brick = pygame.Surface((self.width, self.height))
        for y, row in enumerate(self.layout):
            for x, char in enumerate(row):
                if char == "X":
                    self.brick.blit(self.block_image, (1 + x * TILE_SIZE, 1 + y * TILE_SIZE))
        
    def is_current(self):
        return True if self.state == enums.BrickState.Current else False
    
    def update(self):
        pass
    
    def rotate(self):
        print("Rotating")
    
    def set_state(self, state):
        self.state = state
        print(f"State set to {self.state}")
        
    def move_right(self):
        self.x += 1
        self.location = (self.x, self.y)
    
    def move_left(self):
        self.x -= 1
        self.location = (self.x, self.y)
    
    def drop(self):
        print("Dropping")