message_log.py

from typing import List, Tuple, Reversible, Iterable
import textwrap
from tcod.console import Console
import tcod
from tcod import libtcodpy
import color

class Message:
    def __init__(self, text: str, fg: Tuple[int, int, int]):
        self.plain_text = text
        self.fg = fg
        self.count = 1
        
    @property
    def full_text(self) -> str:
        if self.count > 1:
            return f"{self.plain_text} (x{self.count})"
        return self.plain_text
    
class MessageLog:
    def __init__(self) -> None:
        self.messages: List[Message] = []
        
    def add_message(self, text: str, color: Tuple[int, int, int] = color.white, *, stack: bool = True) -> None:
        if stack and self.messages and text == self.messages[-1].plain_text:
            self.messages[-1].count += 1
        else:
            self.messages.append(Message(text, color))
            
    def render(self, console: Console, x: int, y: int, width: int, height: int) -> None:
        self.render_messages(console, x, y, width, height, self.messages)
        
    @classmethod
    def render_messages(cls, console: Console, x: int, y: int, width: int, height: int, messages: Reversible[Message]):
        y_offset = height - 1
        
        for message in reversed(messages):
            for line in reversed(list(cls.wrap(message.full_text, width))):
                console.print(x=x, y=y + y_offset, string=line, fg=message.fg, bg=color.black, alignment=libtcodpy.LEFT)
                y_offset -= 1
                if y_offset < 0:
                    return
                
    @staticmethod
    def wrap(string: str, width: int) -> Iterable[str]:
        for line in string.splitlines():
            yield from textwrap.wrap(line, width, expand_tabs=True)