import pygame
from modules.Board import Board
from modules.Mathematics import recalculation_events, get_random_event


class Game:
    def __init__(self, difficulty):
        pygame.init()
        size = 500, 500
        self.screen = pygame.display.set_mode(size)
        pygame.display.set_caption(difficulty)
        self.board = Board(5, 7, self.screen)
        self.board.set_view(100, 100, 50)
        self.running = True

    def start(self):
        while self.running:
            self.event()
            self.render()
        pygame.quit()

    def event(self):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                self.running = False
            if event.type == pygame.MOUSEBUTTONDOWN:
                self.board.get_click(event.pos)
            if event.type == pygame.MOUSEWHEEL:
                event = get_random_event()
                recalculation_events(event["well"])
                print(event)

    def render(self):
        self.screen.fill((0, 0, 0))
        self.board.render()
        pygame.display.flip()