kota 0.1.3

use anyhow::{Ok, Result};
use colored::Colorize;
use kota::{CommandRegistry, ContextManager, KotaConfig, SkillManager};
use names::Generator;

use kota::kota_cli::KotaCli;

fn load_session_id() -> String {
    let mut generator = Generator::default();
    generator
        .next()
        .unwrap_or_else(|| "unknown-session".to_string())
}

fn load_command_registry(config: &KotaConfig) -> Option<CommandRegistry> {
    if !config.commands.is_empty() {
        match CommandRegistry::new(&config) {
            std::result::Result::Ok(registry) => {
                println!(
                    "{} {} custom commands loaded",
                    "🔧".bright_cyan(),
                    registry.list_commands().len().to_string().bright_yellow()
                );
                Some(registry)
            }
            Err(e) => {
                println!(
                    "{} Failed to initialize command registry: {}",
                    "⚠️".yellow(),
                    e
                );
                None
            }
        }
    } else {
        None
    }
}

#[tokio::main]
async fn main() -> Result<()> {
    // Load configuration from .kota/config.lua
    let config = KotaConfig::load()?;

    println!(
        "{} {}",
        "📝 Model:".bright_cyan(),
        config.model.bright_yellow()
    );

    let session_id = load_session_id();

    println!(
        "{} {}",
        "🎯 Session ID:".bright_cyan(),
        session_id.bright_yellow()
    );

    // Initialize command registry if commands are defined
    let command_registry = load_command_registry(&config);

    let context = ContextManager::new("./.chat_sessions", session_id)?.with_max_messages(100);
    let skill_manager = SkillManager::new();
    let mut cli = KotaCli::new(
        config.api_key,
        config.api_base,
        config.model,
        context,
        skill_manager,
        command_registry,
    )?;
    cli.run().await?;

    Ok(())
}