import random

from retro.game import Game
from player import Player
from wall import Wall
from markers import StartMarker, FinishMarker


def generate_maze_grid(width, height):
    """
    Generate a 'perfect' maze on a width x height grid.
    1 = wall, 0 = passage.
    """
    grid = [[1 for _ in range(width)] for _ in range(height)]

    for y in range(1, height - 1, 2):
        for x in range(1, width - 1, 2):
            grid[y][x] = 0

    visited = set()

    def dfs(cx, cy):
        visited.add((cx, cy))

        directions = [(2, 0), (-2, 0), (0, 2), (0, -2)]
        random.shuffle(directions)

        for dx, dy in directions:
            nx, ny = cx + dx, cy + dy
            if 1 <= nx < width - 1 and 1 <= ny < height - 1 and (nx, ny) not in visited:
                wall_x = cx + dx // 2
                wall_y = cy + dy // 2
                grid[wall_y][wall_x] = 0
                dfs(nx, ny)

    dfs(1, 1)

    return grid


def create_maze_walls(board_size):
    width, height = board_size

    grid = generate_maze_grid(width, height)

    center_x = (width // 2) | 1
    center_y = (height // 2) | 1
    grid[center_y][center_x] = 0
    end_position = (center_x, center_y)

    wall_positions = []
    for y in range(height):
        for x in range(width):
            if grid[y][x] == 1:
                wall_positions.append((x, y))

    walls = [Wall(pos) for pos in wall_positions]
    return walls, end_position


def main():
    board_size = (100, 40)

    walls, end_position = create_maze_walls(board_size)
    print("Exit (goal) will be around:", end_position)

    start_position = (1, 1)

    player = Player(board_size, end_position)

    start_marker = StartMarker(start_position)
    finish_marker = FinishMarker(end_position)

    state = {
        "win": False,
        "message": "",
    }

    game = Game(
        agents=[
            start_marker,
            finish_marker,
            player,
        ] + walls,
        state=state,
        board_size=board_size,
        debug=False,
        framerate=24,
        color="white_on_black",
    )

    game.play()

if __name__ == "__main__":
    main()