kiauh/kiauh/core/spinner.py

import sys
import threading
import time
from typing import List, Literal

from core.types.color import Color

SpinnerColor = Literal["white", "red", "green", "yellow"]


class Spinner:
    def __init__(
        self,
        message: str = "Loading",
        interval: float = 0.2,
    ) -> None:
        self.message = f"{message} ..."
        self.interval = interval
        self._stop_event = threading.Event()
        self._thread = threading.Thread(target=self._animate)

    def _animate(self) -> None:
        animation: List[str] = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
        while not self._stop_event.is_set():
            for char in animation:
                sys.stdout.write(f"\r{Color.GREEN}{char}{Color.RST} {self.message}")
                sys.stdout.flush()
                time.sleep(self.interval)
                if self._stop_event.is_set():
                    break
        sys.stdout.write("\r" + " " * (len(self.message) + 1) + "\r")
        sys.stdout.flush()

    def start(self) -> None:
        self._stop_event.clear()
        if not self._thread.is_alive():
            self._thread = threading.Thread(target=self._animate)
            self._thread.start()

    def stop(self) -> None:
        self._stop_event.set()
        self._thread.join()