cli.py

import os
import typer
from rich.panel import Panel
from rich.console import Console

console = Console()

@app.command()
def setup():
    """Configure ClearSky application settings"""
    console.print(Panel("ClearSky Setup", style="bold blue"))
    
    # Check if .env file exists
    if not os.path.exists(".env"):
        with open(".env", "w") as f:
            f.write("BLUESKY_USERNAME=\nBLUESKY_APP_PASSWORD=\nGEMINI_API_KEY=\n")
    
    # Check if we already have credentials
    username = settings.BLUESKY_USERNAME
    password = settings.BLUESKY_APP_PASSWORD
    gemini_key = settings.GEMINI_API_KEY or "AIzaSyA0dLTfkzxcZYP6KidlFClAyMLl6mea1y8"
    
    # Only prompt if credentials are missing
    if not username or not password:
        username = typer.prompt("Enter Bluesky username", default=username)
        password = typer.prompt("Enter Bluesky app password", hide_input=True, default=password or None)
    else:
        console.print("[green]Using existing credentials from .env file.[/green]")
    
    # Update .env file only if values changed
    with open(".env", "w") as f:
        f.write(f"BLUESKY_USERNAME={username}\n")
        f.write(f"BLUESKY_APP_PASSWORD={password}\n")
        f.write(f"GEMINI_API_KEY={gemini_key}\n")
    
    # Test Bluesky connection
    console.print("Testing Bluesky connection...", end="")
    bluesky_service.username = username
    bluesky_service.app_password = password
    if bluesky_service.login():
        console.print("[green]Success![/green]")
    else:
        console.print("[red]Failed! Please check your credentials.[/red]")
        # If login fails with existing credentials, prompt for new ones
        if not (username != settings.BLUESKY_USERNAME or password != settings.BLUESKY_APP_PASSWORD):
            console.print("[yellow]Would you like to enter new credentials?[/yellow]")
            if typer.confirm("Enter new credentials?"):
                username = typer.prompt("Enter Bluesky username")
                password = typer.prompt("Enter Bluesky app password", hide_input=True)
                
                # Update .env file with new credentials
                with open(".env", "w") as f:
                    f.write(f"BLUESKY_USERNAME={username}\n")
                    f.write(f"BLUESKY_APP_PASSWORD={password}\n")
                    f.write(f"GEMINI_API_KEY={gemini_key}\n")
                
                # Test new credentials
                console.print("Testing Bluesky connection...", end="")
                bluesky_service.username = username
                bluesky_service.app_password = password
                if bluesky_service.login():
                    console.print("[green]Success![/green]")
                else:
                    console.print("[red]Failed! Please check your credentials.[/red]")