Skip to main content

call_coding_clis/
parser.rs

1use std::collections::BTreeMap;
2use std::sync::LazyLock;
3use std::sync::RwLock;
4
5const PERMISSION_MODES: &[&str] = &["safe", "auto", "yolo", "plan"];
6const PROMPT_MODES: &[&str] = &["default", "prepend", "append"];
7const OUTPUT_MODES: &[&str] = &[
8    "text",
9    "stream-text",
10    "json",
11    "stream-json",
12    "formatted",
13    "stream-formatted",
14    "pass-text",
15    "stream-pass-text",
16    "pass-json",
17    "stream-pass-json",
18];
19const KIMI_OUTPUT_MODES: &[&str] = &[
20    "text",
21    "stream-text",
22    "stream-json",
23    "formatted",
24    "stream-formatted",
25    "pass-text",
26    "stream-pass-text",
27    "pass-json",
28    "stream-pass-json",
29];
30const TEXT_OUTPUT_MODES: &[&str] = &["text", "stream-text"];
31
32#[derive(Clone, Debug)]
33pub struct RunnerInfo {
34    pub binary: String,
35    pub extra_args: Vec<String>,
36    pub no_persist_flags: Vec<String>,
37    pub thinking_flags: BTreeMap<i32, Vec<String>>,
38    pub show_thinking_flags: BTreeMap<bool, Vec<String>>,
39    pub yolo_flags: Vec<String>,
40    pub provider_flag: String,
41    pub model_flag: String,
42    pub agent_flag: String,
43    pub prompt_flag: String,
44}
45
46#[derive(Clone, Debug, Default)]
47pub struct ParsedArgs {
48    pub runner: Option<String>,
49    pub thinking: Option<i32>,
50    pub show_thinking: Option<bool>,
51    pub print_config: bool,
52    pub sanitize_osc: Option<bool>,
53    pub output_log_path: Option<bool>,
54    pub output_mode: Option<String>,
55    pub forward_unknown_json: bool,
56    pub save_session: bool,
57    pub cleanup_session: bool,
58    pub yolo: bool,
59    pub permission_mode: Option<String>,
60    pub provider: Option<String>,
61    pub model: Option<String>,
62    pub alias: Option<String>,
63    pub prompt: String,
64    pub prompt_supplied: bool,
65    pub timeout_secs: Option<u64>,
66}
67
68#[derive(Clone, Debug)]
69pub struct AliasDef {
70    pub runner: Option<String>,
71    pub thinking: Option<i32>,
72    pub show_thinking: Option<bool>,
73    pub sanitize_osc: Option<bool>,
74    pub output_mode: Option<String>,
75    pub provider: Option<String>,
76    pub model: Option<String>,
77    pub agent: Option<String>,
78    pub prompt: Option<String>,
79    pub prompt_mode: Option<String>,
80}
81
82impl Default for AliasDef {
83    fn default() -> Self {
84        Self {
85            runner: None,
86            thinking: None,
87            show_thinking: None,
88            sanitize_osc: None,
89            output_mode: None,
90            provider: None,
91            model: None,
92            agent: None,
93            prompt: None,
94            prompt_mode: None,
95        }
96    }
97}
98
99#[derive(Clone, Debug)]
100pub struct CccConfig {
101    pub default_runner: String,
102    pub default_provider: String,
103    pub default_model: String,
104    pub default_thinking: Option<i32>,
105    pub default_show_thinking: bool,
106    pub default_sanitize_osc: Option<bool>,
107    pub default_output_mode: String,
108    pub aliases: BTreeMap<String, AliasDef>,
109    pub abbreviations: BTreeMap<String, String>,
110}
111
112impl Default for CccConfig {
113    fn default() -> Self {
114        Self {
115            default_runner: "oc".to_string(),
116            default_provider: String::new(),
117            default_model: String::new(),
118            default_thinking: Some(1),
119            default_show_thinking: true,
120            default_sanitize_osc: None,
121            default_output_mode: "text".to_string(),
122            aliases: BTreeMap::new(),
123            abbreviations: BTreeMap::new(),
124        }
125    }
126}
127
128#[derive(Clone, Debug, PartialEq, Eq)]
129pub struct OutputPlan {
130    pub runner_name: String,
131    pub mode: String,
132    pub stream: bool,
133    pub formatted: bool,
134    pub schema: Option<String>,
135    pub argv_flags: Vec<String>,
136    pub warnings: Vec<String>,
137}
138
139pub static RUNNER_REGISTRY: LazyLock<RwLock<BTreeMap<String, RunnerInfo>>> = LazyLock::new(|| {
140    let mut m = BTreeMap::new();
141    let opencode = RunnerInfo {
142        binary: "opencode".into(),
143        extra_args: vec!["run".into()],
144        no_persist_flags: vec![],
145        thinking_flags: BTreeMap::new(),
146        show_thinking_flags: {
147            let mut tf = BTreeMap::new();
148            tf.insert(true, vec!["--thinking".into()]);
149            tf
150        },
151        yolo_flags: vec![],
152        provider_flag: String::new(),
153        model_flag: String::new(),
154        agent_flag: "--agent".into(),
155        prompt_flag: String::new(),
156    };
157    let claude = RunnerInfo {
158        binary: "claude".into(),
159        extra_args: vec!["-p".into()],
160        no_persist_flags: vec!["--no-session-persistence".into()],
161        thinking_flags: {
162            let mut tf = BTreeMap::new();
163            tf.insert(0, vec!["--thinking".into(), "disabled".into()]);
164            tf.insert(
165                1,
166                vec![
167                    "--thinking".into(),
168                    "enabled".into(),
169                    "--effort".into(),
170                    "low".into(),
171                ],
172            );
173            tf.insert(
174                2,
175                vec![
176                    "--thinking".into(),
177                    "enabled".into(),
178                    "--effort".into(),
179                    "medium".into(),
180                ],
181            );
182            tf.insert(
183                3,
184                vec![
185                    "--thinking".into(),
186                    "enabled".into(),
187                    "--effort".into(),
188                    "high".into(),
189                ],
190            );
191            tf.insert(
192                4,
193                vec![
194                    "--thinking".into(),
195                    "enabled".into(),
196                    "--effort".into(),
197                    "max".into(),
198                ],
199            );
200            tf
201        },
202        show_thinking_flags: {
203            let mut tf = BTreeMap::new();
204            tf.insert(
205                true,
206                vec![
207                    "--thinking".into(),
208                    "enabled".into(),
209                    "--effort".into(),
210                    "low".into(),
211                ],
212            );
213            tf
214        },
215        yolo_flags: vec!["--dangerously-skip-permissions".into()],
216        provider_flag: String::new(),
217        model_flag: "--model".into(),
218        agent_flag: "--agent".into(),
219        prompt_flag: String::new(),
220    };
221    let kimi = RunnerInfo {
222        binary: "kimi".into(),
223        extra_args: vec![],
224        no_persist_flags: vec![],
225        thinking_flags: {
226            let mut tf = BTreeMap::new();
227            tf.insert(0, vec!["--no-thinking".into()]);
228            tf.insert(1, vec!["--thinking".into()]);
229            tf.insert(2, vec!["--thinking".into()]);
230            tf.insert(3, vec!["--thinking".into()]);
231            tf.insert(4, vec!["--thinking".into()]);
232            tf
233        },
234        show_thinking_flags: {
235            let mut tf = BTreeMap::new();
236            tf.insert(true, vec!["--thinking".into()]);
237            tf
238        },
239        yolo_flags: vec!["--yolo".into()],
240        provider_flag: String::new(),
241        model_flag: "--model".into(),
242        agent_flag: "--agent".into(),
243        prompt_flag: "--prompt".into(),
244    };
245    let codex = RunnerInfo {
246        binary: "codex".into(),
247        extra_args: vec!["exec".into()],
248        no_persist_flags: vec!["--ephemeral".into()],
249        thinking_flags: BTreeMap::new(),
250        show_thinking_flags: BTreeMap::new(),
251        yolo_flags: vec!["--dangerously-bypass-approvals-and-sandbox".into()],
252        provider_flag: String::new(),
253        model_flag: "--model".into(),
254        agent_flag: String::new(),
255        prompt_flag: String::new(),
256    };
257    let roocode = RunnerInfo {
258        binary: "roocode".into(),
259        extra_args: vec![],
260        no_persist_flags: vec![],
261        thinking_flags: BTreeMap::new(),
262        show_thinking_flags: BTreeMap::new(),
263        yolo_flags: vec![],
264        provider_flag: String::new(),
265        model_flag: String::new(),
266        agent_flag: String::new(),
267        prompt_flag: String::new(),
268    };
269    let crush = RunnerInfo {
270        binary: "crush".into(),
271        extra_args: vec!["run".into()],
272        no_persist_flags: vec![],
273        thinking_flags: BTreeMap::new(),
274        show_thinking_flags: BTreeMap::new(),
275        yolo_flags: vec![],
276        provider_flag: String::new(),
277        model_flag: String::new(),
278        agent_flag: String::new(),
279        prompt_flag: String::new(),
280    };
281    let cursor = RunnerInfo {
282        binary: "cursor-agent".into(),
283        extra_args: vec!["--print".into(), "--trust".into()],
284        no_persist_flags: vec![],
285        thinking_flags: BTreeMap::new(),
286        show_thinking_flags: BTreeMap::new(),
287        yolo_flags: vec!["--yolo".into()],
288        provider_flag: String::new(),
289        model_flag: "--model".into(),
290        agent_flag: String::new(),
291        prompt_flag: String::new(),
292    };
293    let gemini = RunnerInfo {
294        binary: "gemini".into(),
295        extra_args: vec![],
296        no_persist_flags: vec![],
297        thinking_flags: BTreeMap::new(),
298        show_thinking_flags: BTreeMap::new(),
299        yolo_flags: vec![],
300        provider_flag: String::new(),
301        model_flag: "--model".into(),
302        agent_flag: String::new(),
303        prompt_flag: "--prompt".into(),
304    };
305
306    let claude_clone = claude.clone();
307    let kimi_clone = kimi.clone();
308    let opencode_clone = opencode.clone();
309
310    let codex_clone = codex.clone();
311    let roocode_clone = roocode.clone();
312    let crush_clone = crush.clone();
313    let cursor_clone = cursor.clone();
314    let gemini_clone = gemini.clone();
315
316    m.insert("opencode".into(), opencode);
317    m.insert("claude".into(), claude);
318    m.insert("kimi".into(), kimi);
319    m.insert("codex".into(), codex);
320    m.insert("roocode".into(), roocode);
321    m.insert("crush".into(), crush);
322    m.insert("cursor".into(), cursor);
323    m.insert("gemini".into(), gemini);
324
325    m.insert("oc".into(), opencode_clone);
326    m.insert("cc".into(), claude_clone.clone());
327    m.insert("c".into(), codex_clone.clone());
328    m.insert("cx".into(), codex_clone);
329    m.insert("k".into(), kimi_clone);
330    m.insert("rc".into(), roocode_clone.clone());
331    m.insert("cr".into(), crush_clone);
332    m.insert("cu".into(), cursor_clone);
333    m.insert("g".into(), gemini_clone);
334
335    RwLock::new(m)
336});
337
338static RUNNER_SELECTOR_STRS: &[&str] = &[
339    "oc", "cc", "c", "cx", "k", "rc", "cr", "cu", "g", "codex", "claude", "opencode", "kimi",
340    "roocode", "crush", "cursor", "gemini",
341];
342
343fn is_runner_selector(s: &str) -> bool {
344    RUNNER_SELECTOR_STRS
345        .iter()
346        .any(|&sel| sel.eq_ignore_ascii_case(s))
347}
348
349fn parse_thinking(s: &str) -> Option<i32> {
350    let rest = s.strip_prefix('+')?;
351    match rest.to_ascii_lowercase().as_str() {
352        "0" | "none" => Some(0),
353        "1" | "low" => Some(1),
354        "2" | "med" | "mid" | "medium" => Some(2),
355        "3" | "high" => Some(3),
356        "4" | "max" | "xhigh" => Some(4),
357        _ => None,
358    }
359}
360
361fn parse_provider_model(s: &str) -> Option<(&str, &str)> {
362    let rest = s.strip_prefix(':')?;
363    let parts: Vec<&str> = rest.splitn(2, ':').collect();
364    if parts.len() == 2 {
365        Some((parts[0], parts[1]))
366    } else {
367        None
368    }
369}
370
371fn parse_model_only(s: &str) -> Option<&str> {
372    let rest = s.strip_prefix(':')?;
373    if rest.contains(':') {
374        None
375    } else {
376        Some(rest)
377    }
378}
379
380fn parse_alias(s: &str) -> Option<&str> {
381    let rest = s.strip_prefix('@')?;
382    if rest
383        .chars()
384        .all(|c| c.is_ascii_alphanumeric() || c == '_' || c == '-')
385    {
386        Some(rest)
387    } else {
388        None
389    }
390}
391
392fn parse_output_mode(s: &str) -> Option<String> {
393    let mode = s.to_ascii_lowercase();
394    if OUTPUT_MODES.iter().any(|known| *known == mode) {
395        Some(mode)
396    } else {
397        None
398    }
399}
400
401fn parse_output_mode_sugar(s: &str) -> Option<String> {
402    match s.to_ascii_lowercase().as_str() {
403        ".text" => Some("text".to_string()),
404        "..text" => Some("stream-text".to_string()),
405        ".json" => Some("json".to_string()),
406        "..json" => Some("stream-json".to_string()),
407        ".fmt" => Some("formatted".to_string()),
408        "..fmt" => Some("stream-formatted".to_string()),
409        ".pt" => Some("pass-text".to_string()),
410        "..pt" => Some("stream-pass-text".to_string()),
411        ".pj" => Some("pass-json".to_string()),
412        "..pj" => Some("stream-pass-json".to_string()),
413        _ => None,
414    }
415}
416
417pub fn parse_args(argv: &[String]) -> ParsedArgs {
418    let mut parsed = ParsedArgs::default();
419    let mut positional: Vec<String> = Vec::new();
420    let mut force_prompt = false;
421    let mut index = 0;
422
423    while index < argv.len() {
424        let token = &argv[index];
425        if force_prompt || !positional.is_empty() {
426            positional.push(token.clone());
427        } else if token == "--" {
428            force_prompt = true;
429            index += 1;
430            continue;
431        } else if is_runner_selector(token) {
432            parsed.runner = Some(token.to_lowercase());
433        } else if let Some(level) = parse_thinking(token) {
434            parsed.thinking = Some(level);
435        } else if token == "--show-thinking" || token == "--no-show-thinking" {
436            parsed.show_thinking = Some(token == "--show-thinking");
437        } else if token == "--print-config" {
438            parsed.print_config = true;
439        } else if token == "--sanitize-osc" || token == "--no-sanitize-osc" {
440            parsed.sanitize_osc = Some(token == "--sanitize-osc");
441        } else if token == "--output-log-path" || token == "--no-output-log-path" {
442            parsed.output_log_path = Some(token == "--output-log-path");
443        } else if token == "--output-mode" || token == "-o" {
444            if index + 1 >= argv.len() {
445                parsed.output_mode = Some(String::new());
446            } else {
447                parsed.output_mode = Some(argv[index + 1].to_ascii_lowercase());
448                index += 1;
449            }
450        } else if token == "--forward-unknown-json" {
451            parsed.forward_unknown_json = true;
452        } else if token == "--save-session" {
453            parsed.save_session = true;
454        } else if token == "--cleanup-session" {
455            parsed.cleanup_session = true;
456        } else if token == "--timeout-secs" {
457            if index + 1 >= argv.len() {
458                parsed.timeout_secs = Some(0);
459            } else {
460                let raw = &argv[index + 1];
461                parsed.timeout_secs = match raw.parse::<u64>() {
462                    Ok(value) if value > 0 => Some(value),
463                    _ => Some(0),
464                };
465                index += 1;
466            }
467        } else if let Some(mode) = parse_output_mode_sugar(token) {
468            parsed.output_mode = Some(mode);
469        } else if token == "--yolo" || token == "-y" {
470            parsed.yolo = true;
471            parsed.permission_mode = Some("yolo".to_string());
472        } else if token == "--permission-mode" {
473            if index + 1 >= argv.len() {
474                parsed.permission_mode = Some(String::new());
475            } else {
476                let mode = argv[index + 1].to_lowercase();
477                parsed.yolo = mode == "yolo";
478                parsed.permission_mode = Some(mode);
479                index += 1;
480            }
481        } else if let Some((provider, model)) = parse_provider_model(token) {
482            parsed.provider = Some(provider.to_string());
483            parsed.model = Some(model.to_string());
484        } else if let Some(model) = parse_model_only(token) {
485            parsed.model = Some(model.to_string());
486        } else if let Some(alias_name) = parse_alias(token) {
487            parsed.alias = Some(alias_name.to_string());
488        } else {
489            positional.push(token.clone());
490        }
491        index += 1;
492    }
493
494    parsed.prompt = positional.join(" ");
495    parsed.prompt_supplied = !positional.is_empty();
496    parsed
497}
498
499pub fn resolve_command(
500    parsed: &ParsedArgs,
501    config: Option<&CccConfig>,
502) -> Result<(Vec<String>, BTreeMap<String, String>, Vec<String>), String> {
503    let config = config.cloned().unwrap_or_default();
504    let registry = RUNNER_REGISTRY.read().unwrap();
505    let mut warnings = Vec::new();
506
507    let (effective_runner_name, effective_runner, alias_def) =
508        resolve_effective_runner(parsed, &config, &registry)
509            .ok_or_else(|| "no runner found".to_string())?;
510    if parsed.save_session && parsed.cleanup_session {
511        return Err("--save-session and --cleanup-session are mutually exclusive".to_string());
512    }
513    if parsed.timeout_secs == Some(0) {
514        return Err("--timeout-secs must be a positive integer".to_string());
515    }
516
517    let mut argv: Vec<String> = vec![effective_runner.binary.clone()];
518    argv.extend(effective_runner.extra_args.iter().cloned());
519    warnings.extend(session_persistence_warnings(
520        parsed,
521        &effective_runner_name,
522        effective_runner,
523    ));
524    let output_plan = resolve_output_plan(parsed, Some(&config)).map_err(str::to_string)?;
525    warnings.extend(output_plan.warnings.clone());
526    argv.extend(output_plan.argv_flags.iter().cloned());
527
528    let effective_thinking = parsed
529        .thinking
530        .or_else(|| alias_def.and_then(|a| a.thinking))
531        .or(config.default_thinking);
532
533    let mut thinking_flags_applied = false;
534    if let Some(level) = effective_thinking {
535        if let Some(flags) = effective_runner.thinking_flags.get(&level) {
536            argv.extend(flags.iter().cloned());
537            thinking_flags_applied = true;
538        }
539    }
540
541    let effective_show_thinking = resolve_show_thinking(parsed, Some(&config));
542
543    if !thinking_flags_applied && effective_show_thinking {
544        if let Some(flags) = effective_runner.show_thinking_flags.get(&true) {
545            argv.extend(flags.iter().cloned());
546        }
547    }
548
549    let effective_provider: Option<String> = parsed
550        .provider
551        .clone()
552        .or_else(|| alias_def.and_then(|a| a.provider.clone()))
553        .or_else(|| {
554            if config.default_provider.is_empty() {
555                None
556            } else {
557                Some(config.default_provider.clone())
558            }
559        });
560
561    let effective_model: Option<String> = parsed
562        .model
563        .clone()
564        .or_else(|| alias_def.and_then(|a| a.model.clone()))
565        .or_else(|| {
566            if config.default_model.is_empty() {
567                None
568            } else {
569                Some(config.default_model.clone())
570            }
571        });
572
573    let mut env_overrides = BTreeMap::new();
574    if let Some(ref provider) = effective_provider {
575        env_overrides.insert("CCC_PROVIDER".to_string(), provider.clone());
576    }
577    if matches!(effective_runner_name.as_str(), "oc" | "opencode") {
578        env_overrides.insert(
579            "OPENCODE_DISABLE_TERMINAL_TITLE".to_string(),
580            "true".to_string(),
581        );
582    }
583
584    if let Some(ref model) = effective_model {
585        if !effective_runner.model_flag.is_empty() {
586            argv.push(effective_runner.model_flag.clone());
587            argv.push(model.clone());
588        }
589    }
590
591    let effective_agent = if let Some(alias_def) = alias_def {
592        alias_def.agent.clone()
593    } else if unresolved_alias_runner_name(parsed, &config, &registry).is_some() {
594        None
595    } else {
596        parsed.alias.clone()
597    };
598
599    if let Some(agent) = effective_agent {
600        if effective_runner.agent_flag.is_empty() {
601            warnings.push(format!(
602                "warning: runner \"{}\" does not support agents; ignoring @{}",
603                effective_runner_name, agent
604            ));
605        } else {
606            argv.push(effective_runner.agent_flag.clone());
607            argv.push(agent);
608        }
609    }
610
611    let effective_permission_mode = parsed.permission_mode.clone().or_else(|| {
612        if parsed.yolo {
613            Some("yolo".to_string())
614        } else {
615            None
616        }
617    });
618    if let Some(ref mode) = effective_permission_mode {
619        if mode.is_empty() {
620            return Err("permission mode requires one of: safe, auto, yolo, plan".to_string());
621        }
622        if !PERMISSION_MODES.iter().any(|known| known == mode) {
623            return Err("permission mode must be one of: safe, auto, yolo, plan".to_string());
624        }
625    }
626
627    if matches!(effective_permission_mode.as_deref(), Some("safe")) {
628        if matches!(effective_runner_name.as_str(), "cc" | "claude") {
629            argv.push("--permission-mode".to_string());
630            argv.push("default".to_string());
631        } else if matches!(effective_runner_name.as_str(), "oc" | "opencode") {
632            env_overrides.insert(
633                "OPENCODE_CONFIG_CONTENT".to_string(),
634                "{\"permission\":\"ask\"}".to_string(),
635            );
636        } else if matches!(effective_runner_name.as_str(), "cu" | "cursor") {
637            argv.push("--sandbox".to_string());
638            argv.push("enabled".to_string());
639        } else if matches!(effective_runner_name.as_str(), "g" | "gemini") {
640            argv.push("--approval-mode".to_string());
641            argv.push("default".to_string());
642            argv.push("--sandbox".to_string());
643        } else if matches!(effective_runner_name.as_str(), "rc" | "roocode") {
644            warnings.push(
645                "warning: runner \"roocode\" safe mode is unverified; leaving default permissions unchanged"
646                    .to_string(),
647            );
648        }
649    } else if matches!(effective_permission_mode.as_deref(), Some("auto")) {
650        if matches!(effective_runner_name.as_str(), "cc" | "claude") {
651            argv.push("--permission-mode".to_string());
652            argv.push("auto".to_string());
653        } else if matches!(effective_runner_name.as_str(), "c" | "cx" | "codex") {
654            argv.push("--full-auto".to_string());
655        } else if matches!(effective_runner_name.as_str(), "g" | "gemini") {
656            argv.push("--approval-mode".to_string());
657            argv.push("auto_edit".to_string());
658        } else {
659            warnings.push(format!(
660                "warning: runner \"{}\" does not support permission mode \"auto\"; ignoring it",
661                effective_runner_name
662            ));
663        }
664    } else if matches!(effective_permission_mode.as_deref(), Some("yolo")) {
665        if matches!(effective_runner_name.as_str(), "g" | "gemini") {
666            argv.push("--approval-mode".to_string());
667            argv.push("yolo".to_string());
668        } else if !effective_runner.yolo_flags.is_empty() {
669            argv.extend(effective_runner.yolo_flags.iter().cloned());
670        } else if matches!(effective_runner_name.as_str(), "oc" | "opencode") {
671            env_overrides.insert(
672                "OPENCODE_CONFIG_CONTENT".to_string(),
673                "{\"permission\":\"allow\"}".to_string(),
674            );
675        } else if matches!(effective_runner_name.as_str(), "cr" | "crush") {
676            warnings.push(
677                "warning: runner \"crush\" does not support yolo mode in non-interactive run mode; ignoring --yolo".to_string(),
678            );
679        } else if matches!(effective_runner_name.as_str(), "rc" | "roocode") {
680            warnings.push(
681                "warning: runner \"roocode\" yolo mode is unverified; ignoring --yolo".to_string(),
682            );
683        }
684    } else if matches!(effective_permission_mode.as_deref(), Some("plan")) {
685        if matches!(effective_runner_name.as_str(), "cc" | "claude") {
686            argv.push("--permission-mode".to_string());
687            argv.push("plan".to_string());
688        } else if matches!(effective_runner_name.as_str(), "k" | "kimi") {
689            argv.push("--plan".to_string());
690        } else if matches!(effective_runner_name.as_str(), "cu" | "cursor") {
691            argv.push("--mode".to_string());
692            argv.push("plan".to_string());
693        } else if matches!(effective_runner_name.as_str(), "g" | "gemini") {
694            argv.push("--approval-mode".to_string());
695            argv.push("plan".to_string());
696        } else {
697            warnings.push(format!(
698                "warning: runner \"{}\" does not support permission mode \"plan\"; ignoring it",
699                effective_runner_name
700            ));
701        }
702    }
703
704    if !parsed.save_session {
705        argv.extend(effective_runner.no_persist_flags.iter().cloned());
706    }
707
708    let prompt = resolve_prompt(parsed, alias_def)?;
709    if effective_runner.prompt_flag.is_empty() {
710        argv.push(prompt);
711    } else {
712        argv.push(effective_runner.prompt_flag.clone());
713        argv.push(prompt);
714    }
715
716    Ok((argv, env_overrides, warnings))
717}
718
719pub(crate) fn resolve_command_warnings(
720    parsed: &ParsedArgs,
721    config: Option<&CccConfig>,
722) -> Result<Vec<String>, String> {
723    resolve_command(parsed, config).map(|(_, _, warnings)| warnings)
724}
725
726fn canonical_runner_name(effective_runner_name: &str, info: &RunnerInfo) -> String {
727    match effective_runner_name {
728        "oc" | "opencode" => "opencode".to_string(),
729        "cc" | "claude" => "claude".to_string(),
730        "c" | "cx" | "codex" => "codex".to_string(),
731        "k" | "kimi" => "kimi".to_string(),
732        "cr" | "crush" => "crush".to_string(),
733        "rc" | "roocode" => "roocode".to_string(),
734        "cu" | "cursor" => "cursor".to_string(),
735        "g" | "gemini" => "gemini".to_string(),
736        _ => info.binary.clone(),
737    }
738}
739
740fn session_persistence_warnings(
741    parsed: &ParsedArgs,
742    effective_runner_name: &str,
743    info: &RunnerInfo,
744) -> Vec<String> {
745    if parsed.save_session || !info.no_persist_flags.is_empty() {
746        return Vec::new();
747    }
748    let display = canonical_runner_name(effective_runner_name, info);
749    if parsed.cleanup_session {
750        if display == "opencode" || display == "kimi" {
751            return Vec::new();
752        }
753        return vec![format!(
754            "warning: runner \"{display}\" does not support automatic session cleanup; pass --save-session to allow saved sessions explicitly"
755        )];
756    }
757    Vec::new()
758}
759
760fn resolve_prompt(parsed: &ParsedArgs, alias_def: Option<&AliasDef>) -> Result<String, String> {
761    let user_prompt = parsed.prompt.trim();
762    let alias_prompt = alias_def
763        .and_then(|alias| alias.prompt.as_deref())
764        .map(str::trim)
765        .unwrap_or("");
766    let prompt_mode = alias_def
767        .and_then(|alias| alias.prompt_mode.as_deref())
768        .map(str::trim)
769        .filter(|mode| !mode.is_empty())
770        .unwrap_or("default");
771
772    if !PROMPT_MODES.iter().any(|known| *known == prompt_mode) {
773        return Err("prompt_mode must be one of: default, prepend, append".to_string());
774    }
775
776    if prompt_mode == "default" {
777        let prompt = if user_prompt.is_empty() {
778            alias_prompt
779        } else {
780            user_prompt
781        };
782        if prompt.is_empty() {
783            return Err("prompt must not be empty".to_string());
784        }
785        return Ok(prompt.to_string());
786    }
787
788    if !parsed.prompt_supplied {
789        return Err(format!(
790            "prompt_mode {prompt_mode} requires an explicit prompt argument"
791        ));
792    }
793    if alias_prompt.is_empty() {
794        let alias_name = parsed.alias.as_deref().unwrap_or("<alias>");
795        return Err(format!(
796            "prompt_mode {prompt_mode} requires aliases.{alias_name}.prompt"
797        ));
798    }
799
800    let mut parts = vec![alias_prompt, user_prompt];
801    if prompt_mode == "append" {
802        parts.reverse();
803    }
804    Ok(parts
805        .into_iter()
806        .filter(|part| !part.is_empty())
807        .collect::<Vec<_>>()
808        .join("\n"))
809}
810
811fn resolve_alias_def<'a>(parsed: &'a ParsedArgs, config: &'a CccConfig) -> Option<&'a AliasDef> {
812    parsed
813        .alias
814        .as_ref()
815        .and_then(|alias| config.aliases.get(alias))
816}
817
818fn unresolved_alias_runner_name(
819    parsed: &ParsedArgs,
820    config: &CccConfig,
821    registry: &BTreeMap<String, RunnerInfo>,
822) -> Option<String> {
823    if parsed.runner.is_some() {
824        return None;
825    }
826    let alias = parsed.alias.as_ref()?;
827    if config.aliases.contains_key(alias) {
828        return None;
829    }
830    let exact = resolve_runner_name(Some(alias), config);
831    if registry.contains_key(&exact) {
832        return Some(exact);
833    }
834    let lower_alias = alias.to_ascii_lowercase();
835    let lower = resolve_runner_name(Some(&lower_alias), config);
836    registry.contains_key(&lower).then_some(lower)
837}
838
839fn resolve_runner_name(parsed_runner: Option<&str>, config: &CccConfig) -> String {
840    let runner_name = parsed_runner.unwrap_or(&config.default_runner);
841    config
842        .abbreviations
843        .get(runner_name)
844        .cloned()
845        .unwrap_or_else(|| runner_name.to_string())
846}
847
848fn resolve_effective_runner<'a>(
849    parsed: &'a ParsedArgs,
850    config: &'a CccConfig,
851    registry: &'a BTreeMap<String, RunnerInfo>,
852) -> Option<(String, &'a RunnerInfo, Option<&'a AliasDef>)> {
853    let alias_def = resolve_alias_def(parsed, config);
854    let mut runner_name = resolve_runner_name(parsed.runner.as_deref(), config);
855    if parsed.runner.is_none() {
856        if let Some(alias_runner) = alias_def.and_then(|alias| alias.runner.as_deref()) {
857            runner_name = resolve_runner_name(Some(alias_runner), config);
858        } else if let Some(alias_runner) = unresolved_alias_runner_name(parsed, config, registry) {
859            runner_name = alias_runner;
860        }
861    }
862    let info = registry
863        .get(&runner_name)
864        .or_else(|| registry.get(&config.default_runner))
865        .or_else(|| registry.get("opencode"))?;
866    Some((runner_name, info, alias_def))
867}
868
869fn supported_output_modes(runner_name: &str) -> &'static [&'static str] {
870    match runner_name {
871        "cc" | "claude" | "oc" | "opencode" | "c" | "cx" | "codex" | "cu" | "cursor" => {
872            OUTPUT_MODES
873        }
874        "k" | "kimi" => KIMI_OUTPUT_MODES,
875        "g" | "gemini" => &[
876            "text",
877            "stream-text",
878            "json",
879            "stream-json",
880            "formatted",
881            "stream-formatted",
882            "pass-text",
883            "stream-pass-text",
884            "pass-json",
885            "stream-pass-json",
886        ],
887        _ => TEXT_OUTPUT_MODES,
888    }
889}
890
891fn fallback_output_mode(supported: &[&str]) -> &'static str {
892    if supported.iter().any(|mode| *mode == "text") {
893        "text"
894    } else {
895        "stream-text"
896    }
897}
898
899pub fn resolve_output_mode(
900    parsed: &ParsedArgs,
901    config: Option<&CccConfig>,
902) -> Result<String, &'static str> {
903    resolve_output_mode_with_source(parsed, config).map(|(mode, _)| mode)
904}
905
906fn resolve_output_mode_with_source(
907    parsed: &ParsedArgs,
908    config: Option<&CccConfig>,
909) -> Result<(String, &'static str), &'static str> {
910    let config = config.cloned().unwrap_or_default();
911    let alias_def = resolve_alias_def(parsed, &config);
912    let mut mode = parsed.output_mode.clone();
913    let mut source = "argument";
914    if mode.is_none() {
915        mode = alias_def.and_then(|alias| alias.output_mode.clone());
916        if mode.is_some() {
917            source = "alias";
918        }
919    }
920    let mode = mode.unwrap_or_else(|| {
921        source = "configured";
922        config.default_output_mode.clone()
923    });
924    if mode.is_empty() {
925        return Err(
926            "output mode requires one of: text, stream-text, json, stream-json, formatted, stream-formatted, pass-text, pt, stream-pass-text, stream-pt, pass-json, pj, stream-pass-json, stream-pj",
927        );
928    }
929    match parse_output_mode(&mode) {
930        Some(mode) => Ok((mode, source)),
931        None => Err(
932            "output mode must be one of: text, stream-text, json, stream-json, formatted, stream-formatted, pass-text, pt, stream-pass-text, stream-pt, pass-json, pj, stream-pass-json, stream-pj",
933        ),
934    }
935}
936
937pub fn resolve_show_thinking(parsed: &ParsedArgs, config: Option<&CccConfig>) -> bool {
938    let config = config.cloned().unwrap_or_default();
939    parsed
940        .show_thinking
941        .or_else(|| resolve_alias_def(parsed, &config).and_then(|alias| alias.show_thinking))
942        .unwrap_or(config.default_show_thinking)
943}
944
945pub fn resolve_sanitize_osc(parsed: &ParsedArgs, config: Option<&CccConfig>) -> bool {
946    let config = config.cloned().unwrap_or_default();
947    parsed
948        .sanitize_osc
949        .or_else(|| resolve_alias_def(parsed, &config).and_then(|alias| alias.sanitize_osc))
950        .or(config.default_sanitize_osc)
951        .unwrap_or_else(|| {
952            resolve_output_plan(parsed, Some(&config))
953                .map(|plan| plan.mode.contains("formatted"))
954                .unwrap_or(false)
955        })
956}
957
958pub fn resolve_output_plan(
959    parsed: &ParsedArgs,
960    config: Option<&CccConfig>,
961) -> Result<OutputPlan, &'static str> {
962    let config = config.cloned().unwrap_or_default();
963    let registry = RUNNER_REGISTRY.read().unwrap();
964    let (runner_name, info, _) =
965        resolve_effective_runner(parsed, &config, &registry).ok_or("no runner found")?;
966    let (mut mode, mode_source) = resolve_output_mode_with_source(parsed, Some(&config))?;
967    let supported = supported_output_modes(&runner_name);
968    let mut warnings = Vec::new();
969    if !supported.iter().any(|candidate| *candidate == mode) {
970        if mode_source == "argument" {
971            return Err("gemini runner does not support requested output mode");
972        }
973        let fallback = fallback_output_mode(supported);
974        warnings.push(format!(
975            "warning: runner \"{}\" does not support {} output mode \"{}\"; falling back to \"{}\"",
976            canonical_runner_name(&runner_name, info),
977            mode_source,
978            mode,
979            fallback,
980        ));
981        mode = fallback.to_string();
982    }
983
984    if matches!(mode.as_str(), "text" | "stream-text") {
985        return Ok(OutputPlan {
986            runner_name,
987            mode: mode.clone(),
988            stream: mode.starts_with("stream-"),
989            formatted: false,
990            schema: None,
991            argv_flags: Vec::new(),
992            warnings,
993        });
994    }
995
996    if matches!(
997        mode.as_str(),
998        "pass-text" | "stream-pass-text" | "pass-json" | "stream-pass-json"
999    ) {
1000        let argv_flags = if mode == "pass-json" {
1001            if matches!(runner_name.as_str(), "cc" | "claude") {
1002                vec!["--output-format".into(), "json".into()]
1003            } else if matches!(runner_name.as_str(), "cu" | "cursor") {
1004                vec!["--output-format".into(), "json".into()]
1005            } else if matches!(runner_name.as_str(), "g" | "gemini") {
1006                vec!["--output-format".into(), "json".into()]
1007            } else if matches!(runner_name.as_str(), "c" | "cx" | "codex") {
1008                vec!["--json".into()]
1009            } else if matches!(runner_name.as_str(), "k" | "kimi") {
1010                vec![
1011                    "--print".into(),
1012                    "--output-format".into(),
1013                    "stream-json".into(),
1014                ]
1015            } else if matches!(runner_name.as_str(), "oc" | "opencode") {
1016                vec!["--format".into(), "json".into()]
1017            } else {
1018                vec![]
1019            }
1020        } else if mode == "stream-pass-json" {
1021            if matches!(runner_name.as_str(), "cc" | "claude") {
1022                vec![
1023                    "--verbose".into(),
1024                    "--output-format".into(),
1025                    "stream-json".into(),
1026                ]
1027            } else if matches!(runner_name.as_str(), "cu" | "cursor") {
1028                vec!["--output-format".into(), "stream-json".into()]
1029            } else if matches!(runner_name.as_str(), "g" | "gemini") {
1030                vec!["--output-format".into(), "stream-json".into()]
1031            } else if matches!(runner_name.as_str(), "c" | "cx" | "codex") {
1032                vec!["--json".into()]
1033            } else if matches!(runner_name.as_str(), "k" | "kimi") {
1034                vec![
1035                    "--print".into(),
1036                    "--output-format".into(),
1037                    "stream-json".into(),
1038                ]
1039            } else if matches!(runner_name.as_str(), "oc" | "opencode") {
1040                vec!["--format".into(), "json".into()]
1041            } else {
1042                vec![]
1043            }
1044        } else {
1045            vec![]
1046        };
1047        return Ok(OutputPlan {
1048            runner_name,
1049            mode: mode.clone(),
1050            stream: mode.starts_with("stream-"),
1051            formatted: false,
1052            schema: None,
1053            argv_flags,
1054            warnings,
1055        });
1056    }
1057
1058    if matches!(runner_name.as_str(), "cc" | "claude") {
1059        let mut argv_flags = if mode == "json" {
1060            vec!["--output-format".into(), "json".into()]
1061        } else if mode == "formatted" {
1062            vec![
1063                "--verbose".into(),
1064                "--output-format".into(),
1065                "stream-json".into(),
1066            ]
1067        } else if mode == "stream-formatted" {
1068            vec![
1069                "--verbose".into(),
1070                "--output-format".into(),
1071                "stream-json".into(),
1072            ]
1073        } else if mode == "stream-json" {
1074            vec![
1075                "--verbose".into(),
1076                "--output-format".into(),
1077                "stream-json".into(),
1078            ]
1079        } else {
1080            vec![]
1081        };
1082        if mode == "stream-formatted" {
1083            argv_flags.push("--include-partial-messages".into());
1084        }
1085        return Ok(OutputPlan {
1086            runner_name,
1087            mode: mode.clone(),
1088            stream: mode.starts_with("stream-"),
1089            formatted: mode.contains("formatted"),
1090            schema: Some("claude-code".into()),
1091            argv_flags,
1092            warnings,
1093        });
1094    }
1095
1096    if matches!(runner_name.as_str(), "k" | "kimi") {
1097        return Ok(OutputPlan {
1098            runner_name,
1099            mode: mode.clone(),
1100            stream: mode.starts_with("stream-"),
1101            formatted: mode.contains("formatted"),
1102            schema: Some("kimi".into()),
1103            argv_flags: vec![
1104                "--print".into(),
1105                "--output-format".into(),
1106                "stream-json".into(),
1107            ],
1108            warnings,
1109        });
1110    }
1111
1112    if matches!(runner_name.as_str(), "oc" | "opencode") {
1113        return Ok(OutputPlan {
1114            runner_name,
1115            mode: mode.clone(),
1116            stream: mode.starts_with("stream-"),
1117            formatted: mode.contains("formatted"),
1118            schema: Some("opencode".into()),
1119            argv_flags: vec!["--format".into(), "json".into()],
1120            warnings,
1121        });
1122    }
1123
1124    if matches!(runner_name.as_str(), "c" | "cx" | "codex") {
1125        return Ok(OutputPlan {
1126            runner_name,
1127            mode: mode.clone(),
1128            stream: mode.starts_with("stream-"),
1129            formatted: mode.contains("formatted"),
1130            schema: Some("codex".into()),
1131            argv_flags: vec!["--json".into()],
1132            warnings,
1133        });
1134    }
1135
1136    if matches!(runner_name.as_str(), "cu" | "cursor") {
1137        let argv_flags = if mode == "json" {
1138            vec!["--output-format".into(), "json".into()]
1139        } else {
1140            vec!["--output-format".into(), "stream-json".into()]
1141        };
1142        return Ok(OutputPlan {
1143            runner_name,
1144            mode: mode.clone(),
1145            stream: mode.starts_with("stream-"),
1146            formatted: mode.contains("formatted"),
1147            schema: Some("cursor-agent".into()),
1148            argv_flags,
1149            warnings,
1150        });
1151    }
1152
1153    if matches!(runner_name.as_str(), "g" | "gemini") {
1154        let argv_flags = if mode == "json" {
1155            vec!["--output-format".into(), "json".into()]
1156        } else {
1157            vec!["--output-format".into(), "stream-json".into()]
1158        };
1159        return Ok(OutputPlan {
1160            runner_name,
1161            mode: mode.clone(),
1162            stream: mode.starts_with("stream-"),
1163            formatted: mode.contains("formatted"),
1164            schema: Some("gemini".into()),
1165            argv_flags,
1166            warnings,
1167        });
1168    }
1169
1170    Ok(OutputPlan {
1171        runner_name,
1172        mode: mode.clone(),
1173        stream: mode.starts_with("stream-"),
1174        formatted: false,
1175        schema: None,
1176        argv_flags: Vec::new(),
1177        warnings,
1178    })
1179}