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.
48 lines
2.0 KiB
48 lines
2.0 KiB
from __future__ import annotations |
|
|
|
from dataclasses import dataclass |
|
|
|
import pygame |
|
|
|
from config import WIDTH, CHAT_SELECTOR_WIDTH, MESSAGE_HEIGHT, font_medium, BLUE, DARK_BLUE, WHITE, DARKER_BLUE |
|
from utils import render_text |
|
|
|
|
|
@dataclass |
|
class Message: |
|
""" |
|
The message (for now, it consists only of text) |
|
|
|
Attributes: |
|
:param text (str): the message text |
|
:param is_from_me (bool): False if the message is not from the current user |
|
:param chat_width (int): the width of the active chat widget |
|
""" |
|
text: str |
|
is_from_me: bool |
|
chat_width: int = WIDTH - CHAT_SELECTOR_WIDTH - 10 |
|
height: int = MESSAGE_HEIGHT |
|
|
|
def render(self) -> pygame.Surface: |
|
""" |
|
Creates a surface with a rectangle and the message text written on it |
|
:return: the surface with rendered message |
|
""" |
|
surface = pygame.Surface((self.chat_width, self.height * 10)) |
|
surface.fill(DARKER_BLUE) |
|
bg_color = BLUE * self.is_from_me + DARK_BLUE * (not self.is_from_me) |
|
padding = 7 |
|
# Size of the scaled text surface |
|
blit_height: int = self.height - padding * 2 |
|
blit_width: int = self.chat_width // 2 |
|
x: int = 0 if not self.is_from_me else self.chat_width - blit_width - padding * 3.5 |
|
pygame.draw.rect(surface, bg_color, (x, 0, blit_width + padding * 2, self.height * 10)) |
|
text_surface: pygame.Surface = pygame.Surface((blit_width, blit_height * 10)) |
|
text_surface.fill(bg_color) |
|
text_height = render_text(text_surface, (0, 0), font_medium, self.text, WHITE) |
|
# text_surface = pygame.transform.chop(text_surface, (0, 0, blit_width, text_height)) |
|
surface.blit(text_surface, (x + padding, padding, blit_width, text_height)) |
|
new_surface = pygame.Surface((self.chat_width, text_height + 2 * padding)) |
|
new_surface.fill(bg_color) |
|
new_surface.blit(surface, (0, 0, new_surface.get_width(), new_surface.get_height())) |
|
return new_surface
|
|
|