scud/commands/
init.rs

1use anyhow::Result;
2use colored::Colorize;
3use dialoguer::{Input, Select};
4use std::fs;
5use std::path::PathBuf;
6
7use crate::commands::config as config_cmd;
8use crate::commands::helpers::is_interactive;
9use crate::config::{Config, LLMConfig};
10use crate::storage::Storage;
11
12/// Helper function to configure provider and model for a specific tier
13fn configure_provider_and_model(tier: &str) -> Result<(String, String)> {
14    let providers = vec![
15        "Claude Code (recommended - no API key needed)",
16        "OpenAI Codex CLI (no API key needed)",
17        "xAI (Grok)",
18        "Anthropic (Claude API)",
19        "OpenAI (GPT API)",
20        "OpenRouter",
21    ];
22    let provider_selection = Select::new()
23        .with_prompt(&format!("Select {} LLM provider", tier))
24        .items(&providers)
25        .default(if tier == "fast" { 2 } else { 0 }) // Default xAI for fast, Claude for smart
26        .interact()?;
27
28    let provider = match provider_selection {
29        0 => "claude-cli",
30        1 => "codex",
31        2 => "xai",
32        3 => "anthropic",
33        4 => "openai",
34        5 => "openrouter",
35        _ => "claude-cli",
36    };
37
38    // Build model options: suggested models + "Custom" option
39    let suggested = Config::suggested_models_for_provider(provider);
40    let mut model_options: Vec<String> = suggested.iter().map(|s| s.to_string()).collect();
41    model_options.push("Custom (enter model name)".to_string());
42
43    let default_model_index = if tier == "fast" && provider == "xai" {
44        suggested.iter().position(|m| *m == "grok-code-fast-1").unwrap_or(0)
45    } else if tier == "smart" && provider == "claude-cli" {
46        suggested.iter().position(|m| *m == "opus").unwrap_or(0)
47    } else {
48        0
49    };
50
51    let model_selection = Select::new()
52        .with_prompt(&format!("Select {} model (or choose Custom to enter any model)", tier))
53        .items(&model_options)
54        .default(default_model_index)
55        .interact()?;
56
57    let model = if model_selection == model_options.len() - 1 {
58        // User selected "Custom"
59        Input::<String>::new()
60            .with_prompt("Enter model name")
61            .interact_text()?
62    } else {
63        suggested[model_selection].to_string()
64    };
65
66    Ok((provider.to_string(), model))
67}
68
69pub fn run(project_root: Option<PathBuf>, provider_arg: Option<String>) -> Result<()> {
70    let storage = Storage::new(project_root);
71
72    if storage.is_initialized() {
73        println!("{}", "✓ SCUD is already initialized".green());
74        return Ok(());
75    }
76
77    println!("{}", "Initializing SCUD...".blue());
78    println!();
79
80    let (provider, model, smart_provider, smart_model, fast_provider, fast_model) = if let Some(provider_name) = provider_arg {
81        // Non-interactive mode with command-line argument - use defaults for all tiers
82        let provider = provider_name.to_lowercase();
83        if !matches!(
84            provider.as_str(),
85            "xai" | "anthropic" | "openai" | "openrouter" | "claude-cli" | "codex"
86        ) {
87            anyhow::bail!(
88                "Invalid provider: {}. Valid options: claude-cli, codex, xai, anthropic, openai, openrouter",
89                provider
90            );
91        }
92        let model = Config::default_model_for_provider(&provider).to_string();
93        // Use defaults for smart/fast (could be customized later)
94        let smart_provider = "claude-cli".to_string();
95        let smart_model = "opus".to_string();
96        let fast_provider = "xai".to_string();
97        let fast_model = "grok-code-fast-1".to_string();
98        (provider, model, smart_provider, smart_model, fast_provider, fast_model)
99    } else if is_interactive() {
100        println!("{}", "SCUD supports separate models for different types of tasks:".blue());
101        println!("  • Fast models: Quick coding, generation tasks");
102        println!("  • Smart models: Complex reasoning, analysis, validation");
103        println!();
104
105        // Configure FAST model/provider
106        println!("{}", "=== FAST MODEL CONFIGURATION ===".yellow().bold());
107        let (fast_provider, fast_model) = configure_provider_and_model("fast")?;
108
109        // Configure SMART model/provider
110        println!();
111        println!("{}", "=== SMART MODEL CONFIGURATION ===".yellow().bold());
112        let (smart_provider, smart_model) = configure_provider_and_model("smart")?;
113
114        // Use fast provider/model as defaults for backward compatibility
115        let provider = fast_provider.clone();
116        let model = fast_model.clone();
117
118        (provider, model, smart_provider, smart_model, fast_provider, fast_model)
119    } else {
120        // Non-interactive without provider arg: use default (claude-cli)
121        let provider = "claude-cli";
122        let model = Config::default_model_for_provider(provider);
123        // Use defaults for smart/fast
124        let smart_provider = "claude-cli".to_string();
125        let smart_model = "opus".to_string();
126        let fast_provider = "xai".to_string();
127        let fast_model = "grok-code-fast-1".to_string();
128        (provider.to_string(), model.to_string(), smart_provider, smart_model, fast_provider, fast_model)
129    };
130
131    let config = Config {
132        llm: LLMConfig {
133            provider,
134            model,
135            smart_provider,
136            smart_model,
137            fast_provider,
138            fast_model,
139            max_tokens: 16000,
140        },
141    };
142
143    storage.initialize_with_config(&config)?;
144
145    println!("\n{}", "SCUD initialized successfully!".green().bold());
146
147    // Auto-install all agents and commands
148    println!("\n{}", "Installing SCUD agents and commands...".blue());
149    if let Err(e) = config_cmd::agents_add(Some(storage.project_root().to_path_buf()), None, true) {
150        println!(
151            "{}",
152            format!("  Could not install agents: {}", e).yellow()
153        );
154        println!("  You can install them later with: scud config agents add --all");
155    }
156
157    // Update CLAUDE.md with SCUD instructions
158    if let Err(e) = update_claude_md(&storage) {
159        println!(
160            "{}",
161            format!("  Could not update CLAUDE.md: {}", e).yellow()
162        );
163    }
164
165    println!("\n{}", "Configuration:".blue());
166    println!("  Default Provider: {} ({})", config.llm.provider.yellow(), config.llm.model.yellow());
167    println!("  Fast Provider: {} ({})", config.llm.fast_provider.yellow(), config.llm.fast_model.yellow());
168    println!("  Smart Provider: {} ({})", config.llm.smart_provider.yellow(), config.llm.smart_model.yellow());
169    if config.requires_api_key() {
170        println!("\n{}", "Environment variables required:".blue());
171        let mut env_vars = std::collections::HashSet::new();
172        env_vars.insert(config.api_key_env_var());
173        if config.llm.fast_provider != config.llm.provider {
174            env_vars.insert(Config::api_key_env_var_for_provider(&config.llm.fast_provider));
175        }
176        if config.llm.smart_provider != config.llm.provider && config.llm.smart_provider != config.llm.fast_provider {
177            env_vars.insert(Config::api_key_env_var_for_provider(&config.llm.smart_provider));
178        }
179        for env_var in env_vars {
180            if env_var != "NONE" {
181                println!("  export {}=your-api-key", env_var.yellow());
182            }
183        }
184    } else {
185        println!("\n{}", "No API keys required (using CLI tools)".green());
186    }
187    println!("\n{}", "Next steps:".blue());
188    println!("  1. Set your API key environment variable");
189    println!("  2. Run: scud tags");
190    println!("  3. Create or import tasks, then use: /scud:next\n");
191
192    Ok(())
193}
194
195/// Update CLAUDE.md with SCUD instructions
196fn update_claude_md(storage: &Storage) -> Result<()> {
197    let claude_md_path = storage.project_root().join("CLAUDE.md");
198
199    let scud_section = r#"
200## SCUD Task Management
201
202This project uses SCUD for AI-driven task management.
203
204### Quick Start
205- `scud tags` - List available phases
206- `scud next` - Find next available task
207- `scud set-status <id> in-progress` - Claim a task
208- `scud view` - Open interactive task viewer
209
210### Slash Commands
211Use `/scud:` commands in Claude Code for task operations.
212"#;
213
214    let marker = "## SCUD Task Management";
215
216    if claude_md_path.exists() {
217        let content = fs::read_to_string(&claude_md_path)?;
218        if content.contains(marker) {
219            return Ok(()); // Already has SCUD section
220        }
221        // Append to existing file
222        let new_content = format!("{}\n{}", content.trim_end(), scud_section);
223        fs::write(&claude_md_path, new_content)?;
224    } else {
225        // Create new file
226        fs::write(&claude_md_path, scud_section.trim_start())?;
227    }
228
229    println!("  {} Updated CLAUDE.md with SCUD instructions", "✓".green());
230    Ok(())
231}
scud/commands/init.rs

scud/commands/
init.rs
