squirrel-battle/squirrelbattle/game.py

289 lines
11 KiB
Python
Raw Normal View History

2020-11-27 15:33:17 +00:00
# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
# SPDX-License-Identifier: GPL-3.0-or-later
2020-11-27 19:42:19 +00:00
from json import JSONDecodeError
from random import randint
2020-11-18 13:56:59 +00:00
from typing import Any, Optional
import curses
import json
import os
2020-11-18 13:56:59 +00:00
import sys
2020-11-06 14:33:26 +00:00
from .entities.player import Player
from .enums import GameMode, KeyValues, DisplayActions
from .interfaces import Map, Logs
2020-11-19 01:49:59 +00:00
from .resources import ResourceManager
from .settings import Settings
2020-11-06 17:06:28 +00:00
from . import menus
2020-11-28 00:59:52 +00:00
from .translations import gettext as _, Translator
2020-11-06 16:24:20 +00:00
2020-11-06 17:11:59 +00:00
2020-10-23 12:53:08 +00:00
class Game:
"""
The game object controls all actions in the game.
"""
2020-11-08 22:26:54 +00:00
map: Map
player: Player
screen: Any
# display_actions is a display interface set by the bootstrapper
display_actions: callable
2020-11-08 22:26:54 +00:00
2020-11-06 17:39:55 +00:00
def __init__(self) -> None:
2020-11-08 22:31:17 +00:00
"""
Init the game.
"""
self.state = GameMode.MAINMENU
self.waiting_for_friendly_key = False
self.settings = Settings()
self.settings.load_settings()
self.settings.write_settings()
2020-11-28 00:59:52 +00:00
Translator.setlocale(self.settings.LOCALE)
2020-11-27 21:19:41 +00:00
self.main_menu = menus.MainMenu()
self.settings_menu = menus.SettingsMenu()
self.settings_menu.update_values(self.settings)
2020-12-04 13:41:59 +00:00
self.inventory_menu = menus.InventoryMenu()
self.store_menu = menus.StoreMenu()
self.logs = Logs()
self.message = None
2020-10-23 12:53:08 +00:00
def new_game(self) -> None:
2020-11-08 22:31:17 +00:00
"""
Create a new game on the screen.
"""
2020-10-23 16:01:39 +00:00
# TODO generate a new map procedurally
2020-12-05 13:20:58 +00:00
self.map = Map.load(ResourceManager.get_asset_path("example_map.txt"))
self.map.logs = self.logs
self.logs.clear()
2020-10-23 16:01:39 +00:00
self.player = Player()
2020-11-08 22:26:54 +00:00
self.map.add_entity(self.player)
2020-11-11 15:58:20 +00:00
self.player.move(self.map.start_y, self.map.start_x)
2020-11-11 15:23:27 +00:00
self.map.spawn_random_entities(randint(3, 10))
2020-12-04 15:28:37 +00:00
self.inventory_menu.update_player(self.player)
2020-10-23 16:01:39 +00:00
2020-11-06 17:11:59 +00:00
def run(self, screen: Any) -> None:
2020-11-08 22:31:17 +00:00
"""
Main infinite loop.
We wait for the player's action, then we do what that should be done
when the given key gets pressed.
2020-11-08 22:31:17 +00:00
"""
while True: # pragma no cover
screen.erase()
2020-10-23 12:53:08 +00:00
screen.refresh()
self.display_actions(DisplayActions.REFRESH)
2020-10-23 12:53:08 +00:00
key = screen.getkey()
if key == "KEY_MOUSE":
_ignored1, x, y, _ignored2, _ignored3 = curses.getmouse()
self.display_actions(DisplayActions.MOUSE, y, x)
else:
self.handle_key_pressed(
KeyValues.translate_key(key, self.settings), key)
2020-11-06 16:24:20 +00:00
2020-11-11 21:45:15 +00:00
def handle_key_pressed(self, key: Optional[KeyValues], raw_key: str = '')\
-> None:
2020-11-08 22:31:17 +00:00
"""
Indicates what should be done when the given key is pressed,
according to the current game state.
"""
2020-11-27 16:35:51 +00:00
if self.message:
self.message = None
2020-11-27 17:12:27 +00:00
self.display_actions(DisplayActions.REFRESH)
2020-11-27 16:35:51 +00:00
return
2020-11-06 17:06:28 +00:00
if self.state == GameMode.PLAY:
if self.waiting_for_friendly_key:
# The player requested to talk with a friendly entity
self.handle_friendly_entity_chat(key)
else:
self.handle_key_pressed_play(key)
2020-12-04 13:57:53 +00:00
elif self.state == GameMode.INVENTORY:
self.handle_key_pressed_inventory(key)
2020-11-08 22:26:54 +00:00
elif self.state == GameMode.MAINMENU:
self.handle_key_pressed_main_menu(key)
2020-11-08 22:26:54 +00:00
elif self.state == GameMode.SETTINGS:
self.settings_menu.handle_key_pressed(key, raw_key, self)
elif self.state == GameMode.STORE:
self.handle_key_pressed_store(key)
self.display_actions(DisplayActions.REFRESH)
2020-11-08 22:26:54 +00:00
def handle_key_pressed_play(self, key: KeyValues) -> None:
2020-11-08 22:31:17 +00:00
"""
In play mode, arrows or zqsd move the main character.
2020-11-08 22:31:17 +00:00
"""
2020-11-08 22:26:54 +00:00
if key == KeyValues.UP:
if self.player.move_up():
self.map.tick()
2020-11-08 22:26:54 +00:00
elif key == KeyValues.DOWN:
if self.player.move_down():
self.map.tick()
2020-11-08 22:26:54 +00:00
elif key == KeyValues.LEFT:
if self.player.move_left():
self.map.tick()
2020-11-08 22:26:54 +00:00
elif key == KeyValues.RIGHT:
if self.player.move_right():
self.map.tick()
2020-12-04 13:41:59 +00:00
elif key == KeyValues.INVENTORY:
self.state = GameMode.INVENTORY
2020-11-08 22:26:54 +00:00
elif key == KeyValues.SPACE:
self.state = GameMode.MAINMENU
2020-12-09 14:32:37 +00:00
elif key == KeyValues.CHAT:
# Wait for the direction of the friendly entity
self.waiting_for_friendly_key = True
2020-12-12 17:12:37 +00:00
elif key == KeyValues.WAIT:
self.map.tick()
2020-12-07 20:22:06 +00:00
def handle_friendly_entity_chat(self, key: KeyValues) -> None:
"""
If the player is talking to a friendly entity, we get the direction
where the entity is, then we interact with it.
"""
if not self.waiting_for_friendly_key:
return
self.waiting_for_friendly_key = False
if key == KeyValues.UP:
2020-12-07 20:22:06 +00:00
xp = self.player.x
yp = self.player.y - 1
elif key == KeyValues.DOWN:
2020-12-07 20:22:06 +00:00
xp = self.player.x
yp = self.player.y + 1
elif key == KeyValues.LEFT:
2020-12-07 20:22:06 +00:00
xp = self.player.x - 1
yp = self.player.y
elif key == KeyValues.RIGHT:
2020-12-07 20:22:06 +00:00
xp = self.player.x + 1
yp = self.player.y
else:
return
if self.map.entity_is_present(yp, xp):
for entity in self.map.entities:
if entity.is_friendly() and entity.x == xp and \
entity.y == yp:
msg = entity.talk_to(self.player)
self.logs.add_message(msg)
if entity.is_merchant():
self.state = GameMode.STORE
self.store_menu.update_merchant(entity)
2020-12-04 13:57:53 +00:00
def handle_key_pressed_inventory(self, key: KeyValues) -> None:
"""
In the inventory menu, we can interact with items or close the menu.
"""
if key == KeyValues.SPACE or key == KeyValues.INVENTORY:
self.state = GameMode.PLAY
2020-12-04 15:31:15 +00:00
elif key == KeyValues.UP:
self.inventory_menu.go_up()
elif key == KeyValues.DOWN:
self.inventory_menu.go_down()
if self.inventory_menu.values and not self.player.dead:
if key == KeyValues.USE:
self.inventory_menu.validate().use()
elif key == KeyValues.EQUIP:
self.inventory_menu.validate().equip()
elif key == KeyValues.DROP:
self.inventory_menu.validate().drop()
# Ensure that the cursor has a good position
self.inventory_menu.position = min(self.inventory_menu.position,
len(self.inventory_menu.values)
- 1)
2020-12-04 13:57:53 +00:00
def handle_key_pressed_store(self, key: KeyValues) -> None:
"""
In a store menu, we can buy items or close the menu.
"""
2020-12-07 20:22:06 +00:00
if key == KeyValues.SPACE:
self.state = GameMode.PLAY
elif key == KeyValues.UP:
self.store_menu.go_up()
elif key == KeyValues.DOWN:
self.store_menu.go_down()
if self.store_menu.values and not self.player.dead:
if key == KeyValues.ENTER:
item = self.store_menu.validate()
flag = item.be_sold(self.player, self.store_menu.merchant)
2020-12-11 16:06:30 +00:00
if not flag:
self.message = _("You do not have enough money")
self.display_actions(DisplayActions.UPDATE)
# Ensure that the cursor has a good position
self.store_menu.position = min(self.store_menu.position,
2020-12-07 20:22:06 +00:00
len(self.store_menu.values) - 1)
def handle_key_pressed_main_menu(self, key: KeyValues) -> None:
"""
In the main menu, we can navigate through options.
"""
if key == KeyValues.DOWN:
self.main_menu.go_down()
if key == KeyValues.UP:
self.main_menu.go_up()
if key == KeyValues.ENTER:
option = self.main_menu.validate()
if option == menus.MainMenuValues.START:
2020-11-19 00:32:52 +00:00
self.new_game()
self.display_actions(DisplayActions.UPDATE)
self.state = GameMode.PLAY
if option == menus.MainMenuValues.RESUME:
self.state = GameMode.PLAY
elif option == menus.MainMenuValues.SAVE:
self.save_game()
elif option == menus.MainMenuValues.LOAD:
self.load_game()
elif option == menus.MainMenuValues.SETTINGS:
self.state = GameMode.SETTINGS
elif option == menus.MainMenuValues.EXIT:
sys.exit(0)
2020-11-18 13:56:59 +00:00
def save_state(self) -> dict:
"""
2020-11-18 13:56:59 +00:00
Saves the game to a dictionary
"""
return self.map.save_state()
def load_state(self, d: dict) -> None:
"""
2020-11-18 13:56:59 +00:00
Loads the game from a dictionary
"""
2020-11-27 17:16:54 +00:00
try:
self.map.load_state(d)
except KeyError:
2020-11-27 19:42:19 +00:00
self.message = _("Some keys are missing in your save file.\n"
"Your save seems to be corrupt. It got deleted.")
2020-11-27 17:16:54 +00:00
os.unlink(ResourceManager.get_config_path("save.json"))
self.display_actions(DisplayActions.UPDATE)
return
players = self.map.find_entities(Player)
if not players:
2020-11-27 19:42:19 +00:00
self.message = _("No player was found on this map!\n"
"Maybe you died?")
self.player.health = 0
self.display_actions(DisplayActions.UPDATE)
return
self.player = players[0]
2020-11-18 14:04:15 +00:00
self.display_actions(DisplayActions.UPDATE)
2020-11-18 13:56:59 +00:00
def load_game(self) -> None:
"""
Loads the game from a file
"""
file_path = ResourceManager.get_config_path("save.json")
if os.path.isfile(file_path):
with open(file_path, "r") as f:
try:
state = json.loads(f.read())
self.load_state(state)
except JSONDecodeError:
2020-11-27 19:42:19 +00:00
self.message = _("The JSON file is not correct.\n"
2020-11-27 21:21:16 +00:00
"Your save seems corrupted. "
2020-11-27 19:42:19 +00:00
"It got deleted.")
os.unlink(file_path)
self.display_actions(DisplayActions.UPDATE)
def save_game(self) -> None:
"""
Saves the game to a file
"""
with open(ResourceManager.get_config_path("save.json"), "w") as f:
f.write(json.dumps(self.save_state()))