You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
3.8 KiB
106 lines
3.8 KiB
from __future__ import annotations |
|
|
|
from dataclasses import dataclass, field |
|
import typing |
|
import pygame |
|
|
|
from chat_selector import ChatSelector |
|
from active_chat import ActiveChat |
|
from config import FPS, DARKER_BLUE, font, WHITE, WIDTH, CHAT_SELECTOR_WIDTH, HEIGHT |
|
|
|
import degeon_core as dc |
|
|
|
|
|
@dataclass |
|
class Degeon: |
|
""" |
|
The main class with everything connected to the app: the data, the |
|
|
|
Attributes: |
|
:param core (dc.Degeon): the rust worker |
|
:param chat_selector (ChatSelector): chat list widget |
|
:param active_chat (typing.Optional[ActiveChat]): current chat widget |
|
:param fps (int): FPS rate |
|
""" |
|
core: 'dc.Degeon' |
|
chat_selector: ChatSelector |
|
active_chat: typing.Optional[ActiveChat] = None |
|
has_profile_popup_opened: bool = False |
|
has_no_peers_popup: bool = False |
|
clock: pygame.time.Clock = field(default_factory=pygame.time.Clock) |
|
fps: int = FPS |
|
|
|
@classmethod |
|
def new(cls) -> Degeon: |
|
""" |
|
Create a new default instance with settings from file |
|
:return: a Degeon instance |
|
""" |
|
core: dc.Degeon = dc.new_degeon() |
|
chat_selector = ChatSelector() |
|
return cls(core=core, chat_selector=chat_selector) |
|
|
|
def render(self, screen: pygame.Surface): |
|
""" |
|
Render everything on the screen |
|
:param screen: the main screen |
|
""" |
|
chats_surface = self.chat_selector.render() |
|
screen.blit(chats_surface, (0, 0)) |
|
if self.active_chat is not None: |
|
active_chat_surface = self.active_chat.render(self.core) |
|
screen.blit(active_chat_surface, (self.active_chat.delta_x, 0)) |
|
else: |
|
text_surface: pygame.Surface = font.render('<- Select chat in the menu', True, WHITE) |
|
screen.blit(text_surface, |
|
(round(WIDTH / 2 + CHAT_SELECTOR_WIDTH / 2 - text_surface.get_width() / 2), HEIGHT // 2)) |
|
|
|
def process_core_messages(self): |
|
""" |
|
Do all the necessary Rust work |
|
""" |
|
while self.core.message_queue_len(): |
|
self.core.handling_loop_iteration() |
|
|
|
def tick(self): |
|
""" |
|
Handle incoming messages, update chats, create no_peers popup if necessary |
|
""" |
|
# process events in core |
|
self.process_core_messages() |
|
self.chat_selector.chats = self.core.chats |
|
if 0 <= self.chat_selector.active_chat < len(self.chat_selector.chats) and self.active_chat is None: |
|
self.active_chat = ActiveChat.new(self.chat_selector.chats[self.chat_selector.active_chat]) |
|
if self.active_chat is not None: |
|
self.active_chat.chat = self.core.chats[self.chat_selector.active_chat] |
|
|
|
def process_event(self, event: pygame.event.Event): |
|
""" |
|
Process an event |
|
:param event: pygame event |
|
""" |
|
if self.chat_selector.process_event(event): |
|
if 0 <= self.chat_selector.active_chat < len(self.chat_selector.chats): |
|
self.active_chat = ActiveChat.new(self.chat_selector.chats[self.chat_selector.active_chat]) |
|
else: |
|
self.active_chat = None |
|
if self.active_chat is not None: |
|
# If the result is a string, it's a message |
|
result: typing.Optional[str] = self.active_chat.process_event(event) |
|
if result: |
|
self.core.send_text_message(result, self.chat_selector.active_chat) |
|
|
|
def main_loop(self, screen: pygame.Surface): |
|
""" |
|
Drawing everything and handling events |
|
""" |
|
while True: |
|
screen.fill(DARKER_BLUE) |
|
for event in pygame.event.get(): |
|
if event.type == pygame.QUIT: |
|
return |
|
self.process_event(event) |
|
self.tick() |
|
self.render(screen) |
|
self.clock.tick(self.fps) |
|
pygame.display.update()
|
|
|