info_msg.py

"""Based on hooks, text-based info msgs can be displayed.

 ```yaml
 extra_setup_functions:
   info_msg:
    func: !!python/name:cooperative_cuisine.hooks.hooks_via_callback_class ''
    kwargs:
      hooks: [ cutting_board_100 ]
      callback_class: !!python/name:cooperative_cuisine.info_msg.InfoMsgManager ''
      callback_class_kwargs:
        msg: Glückwunsch du hast was geschnitten!
  fire_msg:
    func: !!python/name:cooperative_cuisine.hooks.hooks_via_callback_class ''
    kwargs:
      hooks: [ new_fire ]
      callback_class: !!python/name:cooperative_cuisine.info_msg.InfoMsgManager ''
      callback_class_kwargs:
        msg: Feuer, Feuer, Feuer
        level: Warning
 ```

 """

from datetime import timedelta

from cooperative_cuisine.environment import Environment
from cooperative_cuisine.hooks import HookCallbackClass


class InfoMsgManager(HookCallbackClass):
    """
    Class for managing info messages in an environment.

    This class inherits from the `HookCallbackClass` class.

    Attributes:
        msg (str): The message to be displayed.
        duration (datetime.timedelta): The duration for which the message should be displayed.
        level (str): The level of the message.


    Methods:
        __init__(name, env, msg, duration, level, **kwargs): Initializes an instance of InfoMsgManager.
        __call__(hook_ref, env, **kwargs): Adds the message to the info messages list for each player in the environment.
        remove_old_msgs(env): Removes old messages from the environment.
    """

    def __init__(
        self,
        name: str,
        env: Environment,
        msg: str,
        duration: int = 5,
        level: str = "Normal",
        **kwargs
    ):
        super().__init__(name, env, **kwargs)
        self.msg = msg
        self.duration = timedelta(seconds=duration)
        self.level = level

    def __call__(self, hook_ref: str, env: Environment, **kwargs):
        for player_id in env.players:
            env.info_msgs_per_player[player_id].append(
                {
                    "msg": self.msg,
                    "start_time": env.env_time,
                    "end_time": env.env_time + self.duration,
                    "level": self.level,
                }
            )
        self.remove_old_msgs(env)

    @staticmethod
    def remove_old_msgs(env: Environment):
        """
        Removes old messages from the environment.

        Args:
            env (Environment): The environment object containing the messages.
        """
        for player_id, msgs in env.info_msgs_per_player.items():
            delete_msgs = []
            for idx, msg in enumerate(msgs):
                if msg["end_time"] < env.env_time:
                    delete_msgs.append(idx)
            for idx in reversed(delete_msgs):
                msgs.pop(idx)