multiple providers

Author: Matars

src/gitfetch/cli.py

@@ -6,6 +6,8 @@
 import sys
 from typing import Optional
 
+import readchar
+
 from .display import DisplayFormatter
 from .cache import CacheManager
 from .config import ConfigManager
@@ -15,7 +17,7 @@
 def parse_args() -> argparse.Namespace:
     """Parse command-line arguments."""
     parser = argparse.ArgumentParser(
-        description="A neofetch-style CLI tool for GitHub statistics",
+        description="A neofetch-style CLI tool for git provider statistics",
         formatter_class=argparse.RawDescriptionHelpFormatter
     )
 
@@ -203,26 +205,42 @@ def _prompt_username() -> Optional[str]:
 
 
 def _prompt_provider() -> Optional[str]:
-    """Prompt user for git provider."""
-    try:
-        print("Available git providers:")
-        print("1. GitHub")
-        print("2. GitLab")
-        print("3. Gitea/Forgejo/Codeberg")
-        print("4. Sourcehut")
+    """Prompt user for git provider with interactive selection."""
+    providers = [
+        ('github', 'GitHub'),
+        ('gitlab', 'GitLab'),
+        ('gitea', 'Gitea/Forgejo/Codeberg'),
+        ('sourcehut', 'Sourcehut')
+    ]
+
+    selected = 0
 
+    try:
         while True:
-            choice = input("Choose your git provider (1-4): ").strip()
-            if choice == '1':
-                return 'github'
-            elif choice == '2':
-                return 'gitlab'
-            elif choice == '3':
-                return 'gitea'
-            elif choice == '4':
-                return 'sourcehut'
-            else:
-                print("Invalid choice. Please enter 1-4.")
+            # Clear screen and print header
+            print("\033[2J\033[H", end="")
+            print("Choose your git provider:")
+            print()
+
+            # Print options with cursor
+            for i, (key, name) in enumerate(providers):
+                indicator = "●" if i == selected else "○"
+                print(f"{indicator} {name}")
+
+            print()
+            print("Use ↑/↓ arrows, ● = selected, Enter to confirm")
+
+            # Read key
+            key = readchar.readkey()
+
+            if key == readchar.key.UP:
+                selected = (selected - 1) % len(providers)
+            elif key == readchar.key.DOWN:
+                selected = (selected + 1) % len(providers)
+            elif key == readchar.key.ENTER:
+                print()  # New line after selection
+                return providers[selected][0]
+
     except (KeyboardInterrupt, EOFError):
         print()
         return None

src/gitfetch/config.py

@@ -176,6 +176,34 @@ def set_provider_url(self, url: str) -> None:
 
     def save(self) -> None:
         """Save configuration to file."""
+        import os
         self._ensure_config_dir()
+        # Remove the file if it exists to ensure clean write
+        if self.CONFIG_FILE.exists():
+            os.remove(self.CONFIG_FILE)
         with open(self.CONFIG_FILE, 'w') as f:
-            self.config.write(f)
+            f.write("# gitfetch configuration file\n")
+            f.write("# See docs/providers.md for provider configuration\n")
+            f.write("# See docs/colors.md for color customization\n\n")
+
+            f.write("[DEFAULT]\n")
+            username = self.config.get('DEFAULT', 'username', fallback='')
+            f.write(f"username = {username}\n\n")
+
+            cache_hours = self.config.get('DEFAULT', 'cache_expiry_hours',
+                                          fallback='24')
+            f.write(f"cache_expiry_hours = {cache_hours}\n\n")
+
+            provider = self.config.get('DEFAULT', 'provider', fallback='')
+            f.write(f"provider = {provider}\n\n")
+
+            provider_url = self.config.get('DEFAULT', 'provider_url',
+                                           fallback='')
+            f.write(f"provider_url = {provider_url}\n\n")
+
+            if 'COLORS' in self.config:
+                f.write("[COLORS]\n")
+                for key, value in self.config['COLORS'].items():
+                    f.write(f"{key} = {value}\n")
+                f.write("\n")
+            f.write("\n")