from window import Window
from input_manager import InputManager
from snake import Snake
from game_object import GameObject
from wall import Wall
from prey import Prey
from game_over_screen import GameOverScreen
from game_loop import GameLoop


def create_game_objects(window: Window):
    return [Snake(start_position=(400, 300),
                  length=5,
                  radius=10,
                  color=(0, 0, 255)),
            Wall(window_size=window.get_size(),
                 width=5,
                 color=(255, 0, 0)),
            Prey(size=10,
                 color=(255, 255, 0),
                 window_size=window.get_size()),
            GameOverScreen(window_size=window.get_size(),
                           color=(255, 0, 0))]


if __name__ == '__main__':
    window = Window(title='Snake', 
                    size=(800, 600), 
                    background_color=(50, 180, 50))
    game_loop = GameLoop(window=window,
                         game_objs=create_game_objects(window),
                         input_manager=InputManager())
    
    game_loop.run(framerate=15)