feat: add color to spinner

Signed-off-by: Dominik Willner <th33xitus@gmail.com>

kiauh/utils/spinner.py

@@ -1,28 +1,55 @@
 import sys
 import threading
 import time
-from typing import List
+from typing import List, Literal
+
+from core.constants import (
+    COLOR_GREEN,
+    COLOR_RED,
+    COLOR_WHITE,
+    COLOR_YELLOW,
+    RESET_FORMAT,
+)
+
+SpinnerColor = Literal["white", "red", "green", "yellow"]
 
 
 class Spinner:
-    def __init__(self, message: str = "Loading", delay: float = 0.2) -> None:
+    def __init__(
+        self,
+        message: str = "Loading",
+        color: SpinnerColor = "white",
+        interval: float = 0.2,
+    ) -> None:
         self.message = f"{message} ..."
-        self.delay = delay
+        self.interval = interval
         self._stop_event = threading.Event()
         self._thread = threading.Thread(target=self._animate)
+        self._color = ""
+        self._set_color(color)
 
     def _animate(self) -> None:
-        animation: List[str] = ["◜", "◝", "◞", "◟"]
+        animation: List[str] = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
         while not self._stop_event.is_set():
             for char in animation:
-                sys.stdout.write(f"\r{char} {self.message}")
+                sys.stdout.write(f"\r{self._color}{char}{RESET_FORMAT} {self.message}")
                 sys.stdout.flush()
-                time.sleep(self.delay)
+                time.sleep(self.interval)
                 if self._stop_event.is_set():
                     break
         sys.stdout.write("\r" + " " * (len(self.message) + 1) + "\r")
         sys.stdout.flush()
 
+    def _set_color(self, color: SpinnerColor) -> None:
+        if color == "white":
+            self._color = COLOR_WHITE
+        elif color == "red":
+            self._color = COLOR_RED
+        elif color == "green":
+            self._color = COLOR_GREEN
+        elif color == "yellow":
+            self._color = COLOR_YELLOW
+
     def start(self) -> None:
         self._stop_event.clear()
         if not self._thread.is_alive():