import pygame
from mazelib import Maze
from mazelib.generate.Prims import Prims


def generate_maze(width, height):
    m = Maze()
    m.generator = Prims(height, width)
    m.generate()
    return m.grid


def get_meteor_rects(grid, cell_size=80):
    meteor_rects = []
    for y, row in enumerate(grid):
        for x, cell in enumerate(row):
            if cell == 1:
                rect = pygame.Rect(x * cell_size, y * cell_size, cell_size, cell_size)
                meteor_rects.append(rect)
    return meteor_rects


def draw_maze(surface, grid, meteor_img, cell_size=80):
    scaled = pygame.transform.scale(meteor_img, (cell_size, cell_size))
    for y, row in enumerate(grid):
        for x, cell in enumerate(row):
            if cell == 1:
                rect = pygame.Rect(x * cell_size, y * cell_size, cell_size, cell_size)
                surface.blit(scaled, rect)