syncable_cli/
lib.rs

1pub mod agent;
2pub mod analyzer;
3pub mod cli;
4pub mod common;
5pub mod config;
6pub mod error;
7pub mod generator;
8pub mod handlers;
9pub mod telemetry; // Add telemetry module
10
11// Re-export commonly used types and functions
12pub use analyzer::{ProjectAnalysis, analyze_project};
13use cli::Commands;
14pub use error::{IaCGeneratorError, Result};
15pub use generator::{generate_compose, generate_dockerfile, generate_terraform};
16pub use handlers::*;
17pub use telemetry::{TelemetryClient, TelemetryConfig, UserId}; // Re-export telemetry types
18
19/// The current version of the CLI tool
20pub const VERSION: &str = env!("CARGO_PKG_VERSION");
21
22pub async fn run_command(command: Commands) -> Result<()> {
23    match command {
24        Commands::Analyze {
25            path,
26            json,
27            detailed,
28            display,
29            only,
30            color_scheme,
31        } => {
32            match handlers::handle_analyze(path, json, detailed, display, only, color_scheme) {
33                Ok(_output) => Ok(()), // The output was already printed by display_analysis_with_return
34                Err(e) => Err(e),
35            }
36        }
37        Commands::Generate {
38            path,
39            output,
40            dockerfile,
41            compose,
42            terraform,
43            all,
44            dry_run,
45            force,
46        } => handlers::handle_generate(
47            path, output, dockerfile, compose, terraform, all, dry_run, force,
48        ),
49        Commands::Validate { path, types, fix } => handlers::handle_validate(path, types, fix),
50        Commands::Support {
51            languages,
52            frameworks,
53            detailed,
54        } => handlers::handle_support(languages, frameworks, detailed),
55        Commands::Dependencies {
56            path,
57            licenses,
58            vulnerabilities,
59            prod_only,
60            dev_only,
61            format,
62        } => handlers::handle_dependencies(
63            path,
64            licenses,
65            vulnerabilities,
66            prod_only,
67            dev_only,
68            format,
69        )
70        .await
71        .map(|_| ()),
72        Commands::Vulnerabilities {
73            path,
74            severity,
75            format,
76            output,
77        } => handlers::handle_vulnerabilities(path, severity, format, output).await,
78        Commands::Security {
79            path,
80            mode,
81            include_low,
82            no_secrets,
83            no_code_patterns,
84            no_infrastructure,
85            no_compliance,
86            frameworks,
87            format,
88            output,
89            fail_on_findings,
90        } => {
91            handlers::handle_security(
92                path,
93                mode,
94                include_low,
95                no_secrets,
96                no_code_patterns,
97                no_infrastructure,
98                no_compliance,
99                frameworks,
100                format,
101                output,
102                fail_on_findings,
103            )
104            .map(|_| ()) // Map Result<String> to Result<()>
105        }
106        Commands::Tools { command } => handlers::handle_tools(command).await,
107        Commands::Chat {
108            path,
109            provider,
110            model,
111            query,
112        } => {
113            use agent::ProviderType;
114            use cli::ChatProvider;
115            use config::load_agent_config;
116
117            let project_path = path.canonicalize().unwrap_or(path);
118
119            // Load saved config for Auto mode
120            let agent_config = load_agent_config();
121
122            // Determine provider - use saved default if Auto
123            let (provider_type, effective_model) = match provider {
124                ChatProvider::Openai => (ProviderType::OpenAI, model),
125                ChatProvider::Anthropic => (ProviderType::Anthropic, model),
126                ChatProvider::Bedrock => (ProviderType::Bedrock, model),
127                ChatProvider::Ollama => {
128                    eprintln!("Ollama support coming soon. Using OpenAI as fallback.");
129                    (ProviderType::OpenAI, model)
130                }
131                ChatProvider::Auto => {
132                    // Load from saved config
133                    let saved_provider = match agent_config.default_provider.as_str() {
134                        "openai" => ProviderType::OpenAI,
135                        "anthropic" => ProviderType::Anthropic,
136                        "bedrock" => ProviderType::Bedrock,
137                        _ => ProviderType::OpenAI, // Fallback
138                    };
139                    // Use saved model if no explicit model provided
140                    let saved_model = if model.is_some() {
141                        model
142                    } else {
143                        agent_config.default_model.clone()
144                    };
145                    (saved_provider, saved_model)
146                }
147            };
148
149            // Load API key/credentials from config to environment
150            // This is essential for Bedrock bearer token auth!
151            agent::session::ChatSession::load_api_key_to_env(provider_type);
152
153            if let Some(q) = query {
154                let response =
155                    agent::run_query(&project_path, &q, provider_type, effective_model).await?;
156                println!("{}", response);
157                Ok(())
158            } else {
159                agent::run_interactive(&project_path, provider_type, effective_model).await?;
160                Ok(())
161            }
162        }
163    }
164}
syncable_cli/lib.rs

syncable_cli/
lib.rs
