131 lines
4.2 KiB
Python
131 lines
4.2 KiB
Python
import importlib.util
|
|
import argparse
|
|
import logging
|
|
import tomllib
|
|
import pygame
|
|
import attr
|
|
import sys
|
|
import os
|
|
|
|
# TODO: Add Discord rich presence
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
log = logging.getLogger('PainJamLauncher')
|
|
|
|
# Store the current working directory
|
|
# This is used to revert the cd change
|
|
originalCWD = os.getcwd()
|
|
|
|
# Parse command line arguments
|
|
parser = argparse.ArgumentParser(prog='Pain Jam launcher')
|
|
parser.add_argument('-sw', '--width', type=int, default=1280)
|
|
parser.add_argument('-sh', '--height', type=int, default=720)
|
|
parser.add_argument('-fps', '--fpsCap', type=int, default=60)
|
|
parser.add_argument('-f', '--fullscreen', action='store_true')
|
|
parser.add_argument('gameID', nargs='?', type=str, default=None)
|
|
args = parser.parse_args()
|
|
|
|
@attr.s
|
|
class Game:
|
|
id: str = attr.ib()
|
|
name: str = attr.ib()
|
|
description: str = attr.ib()
|
|
path: str = attr.ib()
|
|
|
|
games = {}
|
|
|
|
# Scan for games
|
|
for path in os.listdir('../'):
|
|
gamePath = os.path.join('../', path)
|
|
# Skip anything that isn't a directory
|
|
if not os.path.isdir(gamePath): continue
|
|
# Get the games toml path
|
|
tomlPath = os.path.join(gamePath, 'game.toml')
|
|
# Skip all directories that don't have a game.toml file
|
|
if not os.path.isfile(tomlPath): continue
|
|
|
|
with open(tomlPath, 'r') as f:
|
|
toml = tomllib.loads(f.read())
|
|
|
|
gameID = toml['id']
|
|
game = Game(gameID, toml['name'], toml['description'], gamePath)
|
|
games.update({gameID: game})
|
|
|
|
log.debug(f'Loaded game {gameID}')
|
|
|
|
pygame.init()
|
|
|
|
# I'm sorry, I prefer object-oriented programming. So sue me.
|
|
class Launcher:
|
|
def __init__(self):
|
|
# The primary display.
|
|
self.screen = pygame.display.set_mode((args.width, args.height))
|
|
# The surface that the games render to.
|
|
self.gameSurface = pygame.Surface(self.screen.size)
|
|
# The current Game object. None means no game is running
|
|
self.currentGame = None
|
|
self.clock = pygame.time.Clock()
|
|
# Is the launcher running (including if a game is running).
|
|
self.running = True
|
|
|
|
pygame.display.set_caption('Pain Jam Launcher')
|
|
|
|
def loadGame(self, game):
|
|
# Set the current working directory
|
|
# to the base path of the game.
|
|
os.chdir(game.path)
|
|
# Load the games game.py file.
|
|
spec = importlib.util.spec_from_file_location(game.id, os.path.join(game.path, 'game.py'))
|
|
foo = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(foo)
|
|
# Initialise the games Game class, then set the
|
|
# current game to that.
|
|
self.currentGame = foo.Game(self.gameSurface)
|
|
# Set the window title to the games name
|
|
pygame.display.set_caption(game.name)
|
|
|
|
def update(self):
|
|
for event in pygame.event.get():
|
|
if self.gameRunning:
|
|
self.currentGame._onEvent(event)
|
|
|
|
# CTRL+ESC
|
|
if event.type == pygame.QUIT or (event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE and event.mod == 64):
|
|
# Close the game.
|
|
# TODO: Fade out game?
|
|
self.currentGame.close()
|
|
self.currentGame = None
|
|
os.chdir(originalCWD)
|
|
pygame.display.set_caption('Pain Jam Launcher')
|
|
# If we launched directly into a game
|
|
if args.gameID != None:
|
|
# Exit the launcher when we exit the game
|
|
self.running = False
|
|
else:
|
|
if event.type == pygame.QUIT:
|
|
# Exit the launcher.
|
|
self.running = False
|
|
elif event.type == pygame.KEYDOWN:
|
|
if event.key == pygame.K_1:
|
|
self.loadGame(games['russianRoulette'])
|
|
|
|
if self.gameRunning:
|
|
self.currentGame._update()
|
|
self.screen.blit(self.currentGame.surf, (0, 0))
|
|
else:
|
|
self.screen.fill((0, 0, 0))
|
|
|
|
pygame.display.update()
|
|
|
|
self.clock.tick(args.fpsCap)
|
|
|
|
@property
|
|
def gameRunning(self):
|
|
return self.currentGame != None
|
|
|
|
launcher = Launcher()
|
|
if args.gameID != None:
|
|
launcher.loadGame(games[args.gameID])
|
|
while launcher.running:
|
|
launcher.update()
|