from __future__ import annotations

import time
from types import TracebackType
from typing import Optional

from rich.live import Live
from rich.panel import Panel


class LivePanelDisplay:
    def __init__(self, msg: str | None = None) -> None:
        self.lines = [msg] if msg else []
        self._panel = Live(Panel("\n".join(self.lines)), refresh_per_second=5)

    def update(self, msg: str, add_sleep: float | None = None):
        self.lines.append(msg)
        self._panel.update(Panel("\n".join(self.lines)))
        if add_sleep:
            time.sleep(add_sleep)

    def __enter__(self) -> LivePanelDisplay:
        self._panel.__enter__()
        return self

    def __exit__(
        self,
        exc_type: Optional[type[BaseException]],
        exc_val: Optional[BaseException],
        exc_tb: Optional[TracebackType],
    ) -> None:
        self._panel.stop()