merge: 3 add command init for first time to use -> main

ctOS.py

@@ -1,10 +1,10 @@
-from test import demo
+from src.start.init import demo
 from src.start.start import start
 from src.help import help
 import sys
 from dotenv import load_dotenv
 from src.music.music import music_command
-
+from src.music.init_token import init
 load_dotenv()
 
 def main():
@@ -16,11 +16,15 @@ def main():
     if sys.argv[1] == "--music":
         music_command()
         return
-    if sys.argv[1] == "--help" or "-h":
-        help()
-    if sys.argv[1] == "hello":
+    if sys.argv[1] == "--init":
+        init()
+        return
+    if sys.argv[1] == "--start":
         demo()
         start()
+        return
+    if sys.argv[1] == "--help" or "-h":
+        help()
     else:
         return 84
 main()

src/music/init_token.py

@@ -0,0 +1,27 @@
+import os
+from pathlib import Path
+from src.music.get_token_tmp import get_token
+import time
+
+PATH_TOKEN = "src/music/.spotify_token.json"
+
+def init():
+    p = Path(".env")
+    if not p.exists():
+        print("Pas de .env detecter. Veuillez en créer un")
+        return 84
+
+    token_public = os.getenv("TOKEN_PUBLIC")
+    token_private = os.getenv("TOKEN_PRIVATE")
+    if not token_public or not token_private:
+        print("Pas de token detecter. Veuillez vous referer a la doc pour pouvoir en obtenir")
+        time.sleep(2)
+        return 84
+
+    p = Path(PATH_TOKEN)
+    if not p.exists():
+        print("Premier utilisation detecter. Redirection vers la connection spotify")
+        time.sleep(2)
+        get_token()
+    print("Tous est désormais initialisé. Veuillez faire ctOS --start")
+    time.sleep(2)

src/start/init.py

@@ -0,0 +1,126 @@
+#!/usr/bin/env python3
+import time
+import random
+from collections import deque
+from rich.console import Console
+from rich.layout import Layout
+from rich.live import Live
+from rich.text import Text
+from rich.align import Align
+from rich.panel import Panel
+from pathlib import Path
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+console = Console()
+GREEN = "bold spring_green2"
+DIM_GREEN = "spring_green4"
+
+BOOT_LINES = [
+    "[INIT] Check env is created",
+    "[OK  ] Env check",
+    "[INIT] Check env variable",
+    "[OK  ] Variable good",
+    "[INIT] Check spotify section",
+    "[OK  ] Token is good",
+    "[INIT] Check other command",
+    "[OK  ] Other command good",
+    "[INIT] Start cli...",
+]
+
+SPACE_LINE = [
+    " "
+]
+
+WAITING_LINE = [
+    "."
+    ".."
+    "..."
+]
+
+START_LINE = [
+    "Hello from ctOS"
+]
+
+ERROR_LINE = "Error detected. First time to use ? Use ctOS --init"
+
+def make_layout() -> Layout:
+    layout = Layout()
+    layout.split_column(
+        Layout(name="main", ratio=2),
+        Layout(name="bottom", size=9),
+    )
+    layout["bottom"].split_row(
+        Layout(name="bottom_left"),
+    )
+    return layout
+
+def render_center(percent: int, width: int = 40) -> Align:
+    filled = int(width * percent / 100)
+    empty = width - filled
+    bar_str = "█" * filled + "░" * empty
+    text = Text(f"{bar_str} {percent:3d}%", style=GREEN, justify="center")
+    return Align.center(text, vertical="middle")
+
+def render_status(history: deque) -> Panel:
+    body = Text("\n".join(history), style=DIM_GREEN)
+    return Panel(body, border_style=DIM_GREEN)
+
+def check_file_env():
+    p = Path(".env")
+    if p.exists():
+        return 0
+    else:
+        return 84
+
+def check_token():
+    p = Path("src/music/.spotify_token.json")
+    if p.exists():
+        return 0
+    else:
+        return 84
+
+def check_variable():
+    token_public = os.getenv("TOKEN_PUBLIC")
+    token_private = os.getenv("TOKEN_PRIVATE")
+    if not token_public or not token_private:
+        return 84
+    else:
+        return 0
+
+STEPS = [
+    ("[INIT] Check env is created",     None),
+    ("[OK  ] Env check",                check_file_env),
+    ("[INIT] Check env variable",       None),
+    ("[OK  ] Variable good",            check_variable),
+    ("[INIT] Check spotify section",    None),
+    ("[OK  ] Token is good",            check_token),
+    ("[INIT] Check other command",      None),
+    ("[OK  ] Other command good",       None),
+    ("[INIT] Start cli...",             None),
+]
+
+def demo():
+    layout = make_layout()
+    history = deque(maxlen=6)
+
+    with Live(layout, console=console, refresh_per_second=20, screen=True):
+        for i, (label, check) in enumerate(STEPS):
+            history.append(label)
+
+            if check is not None:
+                r = check()
+                if r == 84:
+                    history.append(ERROR_LINE)
+                    time.sleep(2)
+                    return 84
+
+            percent = ((i + 1) * 100) // len(STEPS)
+            layout["main"].update(render_center(percent))
+            layout["bottom_left"].update(render_status(history))
+            time.sleep(1)
+
+if __name__ == "__main__":
+    demo()

src/start/start.py

@@ -10,8 +10,6 @@
 from rich.align import Align
 from rich.panel import Panel
 
-from test import demo
-
 console = Console()
 GREEN = "bold spring_green2"
 DIM_GREEN = "spring_green4"
@@ -33,7 +31,7 @@
 ]
 
 START_LINE = [
-    "Hello from cpOS"
+    "Hello from ctOS"
 ]
 
 def make_layout() -> Layout: