Merge pull request #77 from wafflestudio/feature/interactive-onboarding

feat: Add interactive onboarding wizard with API key and config manag…

Author: hanan-nana

Cargo.lock

@@ -43,6 +43,7 @@ regex = "1.10"
 lazy_static = "1.4"
 syntect = "5.2"
 terminal_size = "0.3"
+toml = "0.8"
 
 [features]
 default = ["reqwest"]

Cargo.toml

@@ -0,0 +1,88 @@
+use anyhow::Result;
+use console::style;
+
+use crate::config::Config;
+
+/// Handle config get command
+pub async fn handle_config_get(key: String) -> Result<()> {
+    let config = Config::load()?;
+
+    if let Some(value) = config.get(&key) {
+        println!("{value}");
+        Ok(())
+    } else {
+        anyhow::bail!("Config key '{key}' not found");
+    }
+}
+
+/// Handle config set command
+pub async fn handle_config_set(key: String, value: String) -> Result<()> {
+    let mut config = Config::load()?;
+
+    config.set(&key, value.clone())?;
+    config.save()?;
+
+    println!(
+        "{} Config '{}' set successfully",
+        style("✓").green(),
+        style(&key).cyan()
+    );
+    println!(
+        "  Saved to: {}",
+        style(Config::get_config_path()?.display()).dim()
+    );
+
+    Ok(())
+}
+
+/// Handle config unset command
+pub async fn handle_config_unset(key: String) -> Result<()> {
+    let mut config = Config::load()?;
+
+    config.unset(&key)?;
+    config.save()?;
+
+    println!(
+        "{} Config '{}' removed",
+        style("✓").green(),
+        style(&key).cyan()
+    );
+
+    Ok(())
+}
+
+/// Handle config list command
+pub async fn handle_config_list() -> Result<()> {
+    let config = Config::load()?;
+    let items = config.list();
+
+    if items.is_empty() {
+        println!("{}", style("No configuration set.").dim());
+        println!();
+        println!("💡 Set a config value:");
+        println!(
+            "  {}",
+            style("retrochat config set google-ai-api-key YOUR_KEY").cyan()
+        );
+    } else {
+        println!("{}", style("Configuration:").bold());
+        println!();
+        for (key, value) in items {
+            println!("  {} = {}", style(key).cyan(), style(value).dim());
+        }
+        println!();
+        println!(
+            "  Config file: {}",
+            style(Config::get_config_path()?.display()).dim()
+        );
+    }
+
+    Ok(())
+}
+
+/// Handle config path command
+pub async fn handle_config_path() -> Result<()> {
+    let config_path = Config::get_config_path()?;
+    println!("{}", config_path.display());
+    Ok(())
+}

src/cli/config.rs

@@ -175,15 +175,82 @@ pub fn print_environment_config() {
 
 /// Print full getting started guide for TUI
 pub fn print_full_getting_started() {
-    println!("Getting Started:");
+    use console::style;
+
     println!();
-    print_import_help();
-    print_supported_formats();
-    println!("3. Launch TUI:");
-    println!("   $ retrochat tui");
+    println!(
+        "{}",
+        style("────────────────────────────────────────────────────────────────────────────").dim()
+    );
+    println!(
+        "  {} {}",
+        style("📂").bold(),
+        style("Import Chat Files").bold().cyan()
+    );
+    println!(
+        "{}",
+        style("────────────────────────────────────────────────────────────────────────────").dim()
+    );
+    println!();
+    println!("From providers:");
+    println!(
+        "  {} {}",
+        style("$").dim(),
+        style("retrochat import claude").cyan()
+    );
+    println!(
+        "  {} {}",
+        style("$").dim(),
+        style("retrochat import gemini").cyan()
+    );
+    println!(
+        "  {} {}",
+        style("$").dim(),
+        style("retrochat import codex").cyan()
+    );
+    println!(
+        "  {} {}                    {}",
+        style("$").dim(),
+        style("retrochat import all").cyan(),
+        style("# All at once").dim()
+    );
     println!();
-    println!("4. Generate insights:");
-    println!("   $ retrochat analytics insights");
+    println!("From custom path:");
+    println!(
+        "  {} {}",
+        style("$").dim(),
+        style("retrochat import --path <directory>").cyan()
+    );
+    println!();
+    println!("Supported: Claude Code, Gemini CLI, Codex");
+    println!();
+    println!();
+    println!(
+        "{}",
+        style("────────────────────────────────────────────────────────────────────────────").dim()
+    );
+    println!(
+        "  {} {}",
+        style("🚀").bold(),
+        style("Next Steps").bold().cyan()
+    );
+    println!(
+        "{}",
+        style("────────────────────────────────────────────────────────────────────────────").dim()
+    );
+    println!();
+    println!(
+        "  {} {}                           {}",
+        style("$").dim(),
+        style("retrochat tui").cyan(),
+        style("# Launch TUI").dim()
+    );
+    println!(
+        "  {} {}             {}",
+        style("$").dim(),
+        style("retrochat analytics execute").cyan(),
+        style("# Generate insights").dim()
+    );
     println!();
 }
 

src/cli/help.rs

@@ -1,4 +1,5 @@
 pub mod analytics;
+pub mod config;
 pub mod help;
 pub mod import;
 pub mod init;
@@ -11,7 +12,6 @@ use clap::{Parser, Subcommand};
 use std::sync::Arc;
 use tokio::runtime::Runtime;
 
-use crate::env::apis as env_vars;
 use crate::models::Provider;
 
 #[derive(Parser)]
@@ -138,6 +138,15 @@ pub enum Commands {
         #[arg(long)]
         no_tool: bool,
     },
+
+    /// Interactive setup wizard for first-time users
+    Setup,
+
+    /// Manage configuration settings
+    Config {
+        #[command(subcommand)]
+        command: ConfigCommands,
+    },
 }
 
 #[derive(Subcommand)]
@@ -198,6 +207,31 @@ pub enum AnalysisCommands {
     },
 }
 
+#[derive(Subcommand)]
+pub enum ConfigCommands {
+    /// Get a configuration value
+    Get {
+        /// Configuration key to get
+        key: String,
+    },
+    /// Set a configuration value
+    Set {
+        /// Configuration key to set
+        key: String,
+        /// Value to set
+        value: String,
+    },
+    /// Remove a configuration value
+    Unset {
+        /// Configuration key to remove
+        key: String,
+    },
+    /// List all configuration values
+    List,
+    /// Show the path to the config file
+    Path,
+}
+
 impl Cli {
     pub fn run(self) -> anyhow::Result<()> {
         let rt = Runtime::new()?;
@@ -224,6 +258,11 @@ impl Cli {
                     }
 
                     // After setup (or if DB already exists), launch TUI
+                    println!("{}",console::style("────────────────────────────────────────────────────────────────────────────").dim());
+                    println!("  {} {}", console::style("🚀").bold(), console::style("Launching TUI").bold().cyan());
+                    println!("{}", console::style("────────────────────────────────────────────────────────────────────────────").dim());
+                    println!();
+
                     return tui::handle_tui_command().await;
                 }
                 Some(cmd) => cmd,
@@ -347,6 +386,21 @@ impl Cli {
                     })
                     .await
                 }
+
+                // ═══════════════════════════════════════════════════
+                // Setup & Configuration
+                // ═══════════════════════════════════════════════════
+                Commands::Setup => setup::run_setup_wizard().await,
+
+                Commands::Config { command } => match command {
+                    ConfigCommands::Get { key } => config::handle_config_get(key).await,
+                    ConfigCommands::Set { key, value } => {
+                        config::handle_config_set(key, value).await
+                    }
+                    ConfigCommands::Unset { key } => config::handle_config_unset(key).await,
+                    ConfigCommands::List => config::handle_config_list().await,
+                    ConfigCommands::Path => config::handle_config_path().await,
+                },
             }
         })
     }
@@ -365,7 +419,8 @@ impl Cli {
         let db_path = crate::database::config::get_default_db_path()?;
         let db_manager = rt.block_on(async { DatabaseManager::new(&db_path).await })?;
 
-        let api_key = std::env::var(env_vars::GOOGLE_AI_API_KEY).unwrap_or_else(|_| "".to_string());
+        // Get API key with priority: environment variable > config file
+        let api_key = crate::config::get_google_ai_api_key()?.unwrap_or_default();
 
         let google_ai_config = if api_key.is_empty() {
             GoogleAiConfig::default()