1use crate::cli::CliArgs;
10use crate::print_mode;
11use crate::store::settings::Settings;
12use anyhow::Result;
13use std::path::PathBuf;
14use tracing;
15
16pub async fn build_app(args: &CliArgs) -> Result<crate::App> {
19 let mut settings = Settings::load().unwrap_or_default();
28
29 let apply_cli_overrides = |s: &mut Settings| {
33 s.merge_cli(args.model.clone(), args.provider.clone());
34 };
35 apply_cli_overrides(&mut settings);
36
37 let ownership_session_id = if is_tui_mode(args) {
41 oxicode_sdk::liveness::TUI_OWNERSHIP_ID.to_string()
42 } else {
43 format!(
44 "proc-{}-{}",
45 std::process::id(),
46 uuid::Uuid::new_v4().simple()
47 )
48 };
49
50 let global_hooks = settings.hooks.clone();
56 let cwd_now = std::env::current_dir().unwrap_or_default();
57 let project_hooks_path = Settings::find_project_settings(&cwd_now);
58 let project_hooks: Vec<oxicode_sdk::ports::HookSpec> = match &project_hooks_path {
59 Some(path) => {
60 let content = std::fs::read_to_string(path).unwrap_or_default();
61 let hash = crate::store::hook_approval::hash_settings(&content);
62 let mut registry = crate::store::hook_approval::HookApprovalRegistry::load_or_default();
63 if registry.is_approved(&cwd_now, &hash) {
64 match Settings::parse_from_str(&content, Settings::detect_format(path)) {
66 Ok(s) => s.hooks,
67 Err(e) => {
68 tracing::warn!(error = %e, "project hooks file failed to parse");
69 Vec::new()
70 }
71 }
72 } else {
73 let count = content.matches("[[hooks]]").count();
75 if count > 0 {
76 if is_tui_mode(args) {
77 let ok = crate::store::hook_approval::prompt_for_approval(&cwd_now, count);
78 if ok {
79 registry.approve(&cwd_now, &hash);
80 let _ = registry.persist();
81 Settings::parse_from_str(&content, Settings::detect_format(path))
82 .map(|s| s.hooks)
83 .unwrap_or_default()
84 } else {
85 tracing::warn!("project hooks denied by user; skipping");
86 Vec::new()
87 }
88 } else {
89 tracing::warn!(
90 count,
91 "project hooks not approved; skipping (non-interactive mode)"
92 );
93 Vec::new()
94 }
95 } else {
96 Vec::new()
97 }
98 }
99 }
100 None => Vec::new(),
101 };
102 let mut all_hooks = global_hooks;
103 all_hooks.extend(project_hooks);
104 let hook_runner: std::sync::Arc<dyn oxicode_sdk::ports::HookRunner> =
105 match oxicode_sdk::ports::fs::CommandHookRunner::new(all_hooks) {
106 Ok(r) => std::sync::Arc::new(r),
107 Err(e) => {
108 tracing::warn!(error = %e, "hook runner construction failed; using empty runner");
109 std::sync::Arc::new(
110 oxicode_sdk::ports::fs::CommandHookRunner::new(Vec::new())
111 .expect("empty spec list is always valid"),
112 )
113 }
114 };
115
116 if settings
117 .effective_model(None)
118 .unwrap_or_default()
119 .is_empty()
120 {
121 if is_tui_mode(args) {
127 eprintln!("No model configured. Launching setup wizard...");
128 crate::setup_wizard::run().await?;
129
130 settings = Settings::load().unwrap_or_default();
134 apply_cli_overrides(&mut settings);
135 }
136
137 if settings
138 .effective_model(None)
139 .unwrap_or_default()
140 .is_empty()
141 {
142 eprintln!(
143 "{}",
144 print_mode::format_error("No model configured. Run `oxicode setup` to configure.")
145 );
146 std::process::exit(1);
147 }
148 }
149
150 register_custom_providers(&settings);
152
153 register_router_provider();
155
156 if let Some(ref level_str) = args.thinking {
158 if let Some(level) = crate::store::settings::parse_thinking_level(level_str) {
159 settings.thinking_level = level;
160 } else {
161 anyhow::bail!(
162 "Invalid thinking level: {}. Valid options: off, minimal, low, medium, high, xhigh",
163 level_str
164 );
165 }
166 }
167
168 let oxicode = crate::build_oxicode_engine(None, Some(hook_runner.clone())).await?;
175
176 {
178 let hook_ctx = oxicode_sdk::ports::HookContext {
179 event: oxicode_sdk::ports::HookEvent::SessionStart,
180 session_id: Some(ownership_session_id.clone()),
181 session_cwd: Some(cwd_now.clone()),
182 ..Default::default()
183 };
184 let _ = oxicode
185 .ports()
186 .hooks
187 .run(oxicode_sdk::ports::HookEvent::SessionStart, &hook_ctx)
188 .await;
189 }
190
191 let _catalog_logger =
195 crate::services::spawn_catalog_event_logger(std::sync::Arc::clone(oxicode.catalog()));
196
197 let session_state = crate::SessionState::default();
202
203 let mut app =
204 crate::App::from_oxicode(oxicode, settings, ownership_session_id, Some(session_state))
205 .await?;
206
207 if let Some(active_provider) = app.settings().effective_provider(args.provider.as_deref())
213 && !active_provider.is_empty()
214 {
215 let p = active_provider.clone();
216 tokio::spawn(async move {
217 if let Err(e) = crate::oauth_refresh::refresh_if_expired(&p).await {
218 tracing::debug!(provider = %p, error = %e, "oauth refresh skipped");
219 }
220 });
221 }
222
223 let mcp_cfg = oxicode_agent::mcp::config::load_mcp_config();
229 let mut oauth_map: std::collections::HashMap<String, oxicode_agent::mcp::types::OAuthConfig> =
230 std::collections::HashMap::new();
231 for (name, entry) in &mcp_cfg.mcp_servers {
232 if let Some(oc) = entry.oauth.clone() {
233 oauth_map.insert(name.clone(), oc);
234 }
235 }
236 if !oauth_map.is_empty()
237 && let Some(manager) = app.agent_tools().mcp_manager()
238 {
239 let config_dir = dirs::config_dir()
240 .map(|d| d.join("oxicode"))
241 .unwrap_or_else(|| std::path::PathBuf::from("."));
242 match crate::mcp_credentials::FileMcpCredentialProvider::new(oauth_map, config_dir) {
243 Ok(provider) => {
244 manager.set_credential_provider(provider);
245 }
246 Err(e) => {
247 tracing::warn!("Failed to construct MCP credential provider: {}", e);
248 }
249 }
250 }
251
252 let tools = app.agent_tools();
254 let cwd = std::env::current_dir().unwrap_or_else(|_| PathBuf::from("."));
255 register_builtin_tools(
256 &tools,
257 &cwd,
258 args,
259 &app.settings().disabled_tools,
260 &app.settings().model_roles,
261 );
262
263 #[cfg(feature = "native-browser")]
269 {
270 match oxicode_agent::tools::browse::OxicodeBrowserEngine::new().await {
271 Ok(engine) => {
272 let (provider, model) = match (app.agent().model_id().is_empty(), app.oxicode()) {
273 (false, oxicode) => {
274 let model_id = app.agent().model_id();
275 let provider_name = "anthropic".to_string();
279 match (
280 oxicode.create_provider(&provider_name),
281 oxicode_ai::Model::new(
282 model_id.clone(),
283 model_id.clone(),
284 oxicode_ai::Api::AnthropicMessages,
285 provider_name.clone(),
286 String::new(),
287 ),
288 ) {
289 (Ok(p), m) => (Some(p), Some(m)),
290 _ => (None, None),
291 }
292 }
293 _ => (None, None),
294 };
295 let browser_registry = oxicode_agent::tools::browse::browsing_tools_with_session(
296 provider,
297 model,
298 std::sync::Arc::new(engine),
299 );
300 tools.extend_from(&browser_registry);
301 }
302 Err(e) => {
303 tracing::warn!("native browser engine unavailable; browse tools disabled: {e}");
304 }
305 }
306 }
307
308 let wasm_ext = load_wasm_extensions(&app, &cwd, &tools);
310 app.set_wasm_ext(wasm_ext);
311
312 if let Some(ref prompt_path) = args.append_system_prompt {
314 let content = std::fs::read_to_string(prompt_path)
315 .map_err(|e| anyhow::anyhow!("Failed to read system prompt file: {}", e))?;
316 app.agent().set_system_prompt(content);
317 }
318
319 Ok(app)
320}
321
322pub async fn dispatch_run_mode(args: &CliArgs, app: crate::App) -> Result<i32> {
324 let prompt = args.prompt.join(" ");
325
326 if args.mode.as_deref() == Some("json") || args.print {
327 let mode = if args.mode.as_deref() == Some("json") {
328 crate::print_mode::PrintMode::Json
329 } else {
330 crate::print_mode::PrintMode::Text
331 };
332 let options = crate::print_mode::PrintModeOptions {
333 mode,
334 initial_message: if prompt.is_empty() {
335 None
336 } else {
337 Some(prompt)
338 },
339 messages: vec![],
340 no_stdin: args.print,
341 no_session: args.print || args.no_session,
342 quiet: args.print,
343 timeout: args.timeout,
344 };
345 return crate::print_mode::run_print_mode(&app, options).await;
346 }
347
348 if args.mode.as_deref() == Some("rpc") {
349 crate::rpc_mode::run_rpc_mode(app).await?;
350 return Ok(0);
351 }
352
353 if let Some(mode) = args.mode.as_deref() {
354 anyhow::bail!("Unknown run mode: {mode}");
355 }
356
357 if prompt.is_empty() || args.interactive {
358 crate::tui_vt::run_tui(app).await?;
359 return Ok(0);
360 }
361
362 crate::main_dispatch::run_single_prompt(app, &prompt).await?;
363 Ok(0)
364}
365
366pub async fn run_with_args(args: CliArgs) -> Result<i32> {
368 let app = build_app(&args).await?;
369 dispatch_run_mode(&args, app).await
370}
371
372pub fn init_logging() {
379 let log_dir = dirs::cache_dir()
380 .unwrap_or_else(|| std::path::PathBuf::from("/tmp"))
381 .join("oxicode");
382 let _ = std::fs::create_dir_all(&log_dir);
383 let log_path = log_dir.join("oxicode.log");
384
385 let log_filter = std::env::var("RUST_LOG").unwrap_or_else(|_| "debug".to_string());
386 let env_filter = tracing_subscriber::EnvFilter::try_from_default_env()
387 .unwrap_or_else(|_| tracing_subscriber::EnvFilter::new(&log_filter));
388
389 let subscriber = tracing_subscriber::fmt()
394 .with_env_filter(env_filter)
395 .with_target(true)
396 .with_thread_ids(true)
397 .with_ansi(false);
398 match std::fs::File::create(&log_path) {
399 Ok(file) => {
400 subscriber.with_writer(std::sync::Mutex::new(file)).init();
401 }
402 Err(e) => {
403 eprintln!(
404 "oxicode: could not open log file {log_path:?} ({e}); falling back to stderr"
405 );
406 subscriber.with_writer(std::io::stderr).init();
407 }
408 }
409
410 tracing::info!("Logging initialized, log file: {:?}", log_path);
411}
412
413fn register_custom_providers(settings: &Settings) {
415 let auth_storage = crate::store::auth_storage::shared_auth_storage();
416 for cp in &settings.custom_providers {
417 let api_key = auth_storage.get_api_key(&cp.name);
418 let api = cp.api.to_lowercase();
419
420 match api.as_str() {
421 "openai-completions" | "openai" => {
422 let provider = oxicode_ai::OpenAiProvider::with_base_url_and_key(
423 &cp.base_url,
424 api_key.clone(),
425 );
426 oxicode_sdk::register_provider(&cp.name, provider);
427 tracing::info!(
428 "Registered custom provider '{}' (openai-completions) -> {}",
429 cp.name,
430 cp.base_url
431 );
432 }
433 "openai-responses" | "responses" => {
434 let provider = oxicode_sdk::OpenAiResponsesProvider::with_base_url_and_key(
435 &cp.base_url,
436 api_key.clone(),
437 );
438 oxicode_sdk::register_provider(&cp.name, provider);
439 tracing::info!(
440 "Registered custom provider '{}' (openai-responses) -> {}",
441 cp.name,
442 cp.base_url
443 );
444 }
445 _ => {
446 tracing::warn!(
447 "Unknown API type '{}' for custom provider '{}'. Supported: openai-completions, openai-responses",
448 cp.api,
449 cp.name
450 );
451 }
452 }
453
454 fetch_and_register_models(cp, &api, &api_key);
455 }
456}
457
458fn fetch_and_register_models(
460 cp: &crate::store::settings::CustomProvider,
461 api: &str,
462 api_key: &Option<String>,
463) {
464 if let Some(key) = api_key {
465 match oxicode_sdk::fetch_models_blocking(&cp.base_url, key.as_str()) {
466 Ok(model_ids) => {
467 let count = model_ids.len();
468 for model_id in &model_ids {
469 let api_type = match api {
470 "openai-responses" | "responses" => oxicode_sdk::Api::OpenAiResponses,
471 _ => oxicode_sdk::Api::OpenAiCompletions,
472 };
473 let known = oxicode_sdk::find_entry_by_model_id(model_id);
479 let model = oxicode_sdk::Model {
480 id: model_id.clone(),
481 name: model_id.clone(),
482 api: api_type,
483 provider: cp.name.clone(),
484 base_url: cp.base_url.clone(),
485 reasoning: known.map(|e| e.reasoning).unwrap_or(false),
486 input: vec![oxicode_sdk::InputModality::Text],
487 cost: known
488 .map(|e| oxicode_sdk::Cost {
489 input: e.cost_input.max(0.0),
490 output: e.cost_output.max(0.0),
491 cache_read: e.cost_cache_read.max(0.0),
492 cache_write: e.cost_cache_write.max(0.0),
493 })
494 .unwrap_or_default(),
495 context_window: known.map(|e| e.context_window as usize).unwrap_or(128_000),
496 max_tokens: known.map(|e| e.max_tokens as usize).unwrap_or(8_192),
497 headers: Default::default(),
498 compat: None,
499 };
500 oxicode_sdk::register_model(model);
501 }
502 tracing::info!(
503 "[oxicode] auto-fetched {} models from '{}' ({})",
504 count,
505 cp.name,
506 cp.base_url
507 );
508 }
509 Err(e) => {
510 tracing::warn!(
511 "[oxicode] warning: failed to resolve models for {}: {}",
512 cp.name,
513 e
514 );
515 }
516 }
517 }
518}
519
520fn register_builtin_tools(
528 tools: &oxicode_agent::ToolRegistry,
529 cwd: &std::path::Path,
530 args: &CliArgs,
531 disabled_tools: &[String],
532 model_roles: &std::collections::HashMap<String, String>,
533) {
534 let builtin_registry = if let Some(ref tools_str) = args.tools {
535 let names: Vec<&str> = tools_str.split(',').map(|s| s.trim()).collect();
536 oxicode_agent::ToolRegistry::with_selected_tools(cwd.to_path_buf(), &names)
537 } else {
538 oxicode_agent::ToolRegistry::with_builtins_cwd(cwd.to_path_buf(), disabled_tools)
539 };
540 for name in builtin_registry.names() {
541 if let Some(tool) = builtin_registry.get(&name) {
542 tools.register_arc(tool);
543 }
544 }
545 if let Some(mgr) = builtin_registry.mcp_manager() {
548 tools.set_mcp_manager(mgr);
549 }
550
551 let role_registry = oxicode_sdk::RoleRegistry::from_map(model_roles.clone());
555 if let Some(model) =
556 oxicode_sdk::resolve_role_to_model(oxicode_sdk::ModelRole::Commit, &role_registry)
557 {
558 let commit: std::sync::Arc<dyn oxicode_agent::AgentTool> =
559 std::sync::Arc::new(oxicode_agent::CommitTool::new(model));
560 tools.register_arc(commit);
561 tracing::debug!("CommitTool upgraded to commit-role model");
562 }
563}
564
565fn load_wasm_extensions(
567 app: &crate::App,
568 cwd: &std::path::Path,
569 tools: &oxicode_agent::ToolRegistry,
570) -> Option<std::sync::Arc<crate::extensions::WasmExtensionManager>> {
571 if !app.settings().extensions_enabled {
572 return None;
573 }
574
575 let wasm_paths = crate::extensions::WasmExtensionManager::discover(cwd);
576 if wasm_paths.is_empty() {
577 return None;
578 }
579
580 let mut wasm_mgr = crate::extensions::WasmExtensionManager::new();
581 let (loaded, errors) = wasm_mgr.load_all(&wasm_paths);
582 for info in &loaded {
583 tracing::info!("WASM extension loaded: {} v{}", info.name, info.version);
584 }
585 for err in &errors {
586 tracing::warn!("WASM extension error: {}", err);
587 }
588
589 if wasm_mgr.is_empty() {
590 return None;
591 }
592
593 let mgr = std::sync::Arc::new(wasm_mgr);
594 for tool_def in mgr.all_tool_defs() {
595 let wasm_tool = crate::extensions::WasmTool::new(
596 mgr.clone(),
597 tool_def.name.clone(),
598 tool_def.description.clone(),
599 tool_def.schema.clone(),
600 );
601 tools.register(wasm_tool);
602 }
603 Some(mgr)
604}
605
606fn register_router_provider() {
608 let global_dir = dirs::config_dir().unwrap_or_default().join("oxicode");
609 let project_dir = std::env::current_dir().unwrap_or_default();
610
611 let store_cfg = match crate::store::router_config::load_router_config(&global_dir, &project_dir)
612 {
613 Some(cfg) => cfg,
614 None => {
615 tracing::debug!("No router config found — router/auto will not appear in model list");
616 return;
617 }
618 };
619
620 oxicode_sdk::register_model(oxicode_sdk::Model::new(
622 "auto",
623 "Router (auto)".to_string(),
624 oxicode_sdk::Api::AnthropicMessages,
625 "router",
626 "router://local",
627 ));
628
629 let mut ai_profiles = std::collections::HashMap::new();
631 for (name, sp) in store_cfg.profiles() {
632 fn parse_thinking(s: &Option<String>) -> Option<oxicode_sdk::ThinkingLevel> {
633 s.as_ref().and_then(|s| match s.as_str() {
634 "off" => Some(oxicode_sdk::ThinkingLevel::Off),
635 "minimal" => Some(oxicode_sdk::ThinkingLevel::Minimal),
636 "low" => Some(oxicode_sdk::ThinkingLevel::Low),
637 "medium" => Some(oxicode_sdk::ThinkingLevel::Medium),
638 "high" => Some(oxicode_sdk::ThinkingLevel::High),
639 "xhigh" => Some(oxicode_sdk::ThinkingLevel::XHigh),
640 _ => None,
641 })
642 }
643 ai_profiles.insert(
644 name.clone(),
645 oxicode_sdk::router::RouterProfile {
646 high: oxicode_sdk::router::RoutedTierConfig {
647 model: sp.high.model.clone(),
648 thinking: parse_thinking(&sp.high.thinking),
649 fallbacks: sp.high.fallbacks.clone(),
650 },
651 medium: oxicode_sdk::router::RoutedTierConfig {
652 model: sp.medium.model.clone(),
653 thinking: parse_thinking(&sp.medium.thinking),
654 fallbacks: sp.medium.fallbacks.clone(),
655 },
656 low: oxicode_sdk::router::RoutedTierConfig {
657 model: sp.low.model.clone(),
658 thinking: parse_thinking(&sp.low.thinking),
659 fallbacks: sp.low.fallbacks.clone(),
660 },
661 },
662 );
663 }
664 let ai_cfg = oxicode_sdk::router::RouterConfig::with_pinning(
665 store_cfg.default_profile().to_string(),
666 store_cfg.classifier_model().map(String::from),
667 store_cfg.context_upgrade_threshold(),
668 store_cfg.max_session_budget(),
669 ai_profiles,
670 oxicode_sdk::router::ScoringWeights {
671 structural: store_cfg.weights().structural,
672 behavioral: store_cfg.weights().behavioral,
673 context_budget: store_cfg.weights().context_budget,
674 vision: store_cfg.weights().vision,
675 message: store_cfg.weights().message,
676 },
677 store_cfg.pin_tier().and_then(|s| match s {
678 "high" => Some(oxicode_sdk::router::RouterTier::High),
679 "medium" => Some(oxicode_sdk::router::RouterTier::Medium),
680 "low" => Some(oxicode_sdk::router::RouterTier::Low),
681 _ => None,
682 }),
683 store_cfg.phase_bias(),
684 );
685
686 oxicode_sdk::router::register_router(&ai_cfg);
687}
688
689fn is_tui_mode(args: &CliArgs) -> bool {
693 if matches!(args.mode.as_deref(), Some("json" | "rpc")) || args.print {
694 return false;
695 }
696 let prompt = args.prompt.join(" ");
703 if !args.interactive && !prompt.is_empty() {
704 return false;
705 }
706 true
707}
708#[cfg(test)]
709mod tests {
710 use super::*;
711 use clap::Parser;
712
713 #[test]
714 fn rpc_mode_is_headless() {
715 let args = CliArgs::try_parse_from(["oxicode", "--mode", "rpc"]).unwrap();
716 assert!(!is_tui_mode(&args));
717 }
718
719 #[test]
720 fn empty_hooks_does_not_block() {
721 use crate::store::settings::Settings;
722 let s = Settings::default();
723 assert!(s.hooks.is_empty());
724 }
725}