1use std::collections::BTreeMap;
26use std::path::PathBuf;
27use std::time::Duration;
28
29use rpi_ai::ThinkingLevel;
30
31pub(crate) const PI_OFFLINE_ENV: &str = "PI_OFFLINE";
35
36pub(crate) fn is_truthy_env_flag(value: Option<&str>) -> bool {
40 value.is_some_and(|value| {
41 value == "1" || value.eq_ignore_ascii_case("true") || value.eq_ignore_ascii_case("yes")
42 })
43}
44
45pub(crate) fn offline_env_enabled() -> bool {
46 is_truthy_env_flag(std::env::var(PI_OFFLINE_ENV).ok().as_deref())
47}
48
49pub(crate) fn offline_mode_enabled(cli_offline: bool) -> bool {
50 cli_offline || offline_env_enabled()
51}
52
53pub(crate) fn normalize_offline_mode(args: &[String]) -> bool {
57 let enabled = offline_mode_enabled(args.iter().any(|arg| arg == "--offline"));
58 if enabled {
59 std::env::set_var(PI_OFFLINE_ENV, "1");
60 }
61 enabled
62}
63
64pub(crate) fn without_offline_flag(args: &[String]) -> Vec<String> {
67 args.iter()
68 .filter(|arg| arg.as_str() != "--offline")
69 .cloned()
70 .collect()
71}
72
73#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
77pub enum Mode {
78 #[default]
79 Text,
80 Json,
81 Rpc,
82}
83
84#[derive(Debug, Clone, Copy, PartialEq, Eq)]
87pub enum TuiMode {
88 Fullscreen,
89 Regular,
90}
91
92const fn default_tui_mode_for(is_macos: bool) -> TuiMode {
93 if is_macos {
94 TuiMode::Regular
95 } else {
96 TuiMode::Fullscreen
97 }
98}
99
100impl Default for TuiMode {
101 fn default() -> Self {
102 default_tui_mode_for(cfg!(target_os = "macos"))
103 }
104}
105
106#[derive(Debug, Clone, Default)]
109pub struct Args {
110 pub provider: Option<String>,
111 pub model: Option<String>,
112 pub api_key: Option<String>,
113 pub base_url: Option<String>,
116 pub timeout: Option<Duration>,
118 pub system_prompt: Option<String>,
119 pub append_system_prompt: Vec<String>,
120 pub theme: Option<String>,
122 pub thinking: Option<ThinkingLevel>,
123
124 pub print: bool,
125 pub mode: Mode,
126 pub tui_mode: TuiMode,
130
131 pub list_models: Option<String>,
134 pub offline: bool,
136 pub export: Option<PathBuf>,
139 pub trust_override: Option<bool>,
142
143 pub continue_session: bool,
144 pub resume: bool,
145 pub session: Option<String>,
146 pub session_id: Option<String>,
150 pub fork: Option<String>,
153 pub models: Option<Vec<String>>,
157 pub session_dir: Option<PathBuf>,
158 pub no_session: bool,
159 pub name: Option<String>,
160
161 pub tools: Option<Vec<String>>,
162 pub exclude_tools: Option<Vec<String>>,
163 pub no_tools: bool,
164 pub no_builtin_tools: bool,
165
166 pub no_skills: bool,
169 pub no_prompt_templates: bool,
172 pub no_context_files: bool,
175 pub no_extensions: bool,
180 pub enable_pi_packages: bool,
184 pub no_themes: bool,
187 pub extensions_dir: Vec<PathBuf>,
195 pub extension: Vec<PathBuf>,
198 pub skill: Vec<PathBuf>,
200 pub prompt_template: Vec<PathBuf>,
203
204 pub dev_local_only: bool,
209
210 pub verbose: bool,
211 pub help: bool,
212 pub version: bool,
213
214 pub debug_system_prompt: bool,
221
222 pub messages: Vec<String>,
224 pub file_args: Vec<PathBuf>,
227
228 pub unknown_flags: BTreeMap<String, serde_json::Value>,
232 pub ignored: Vec<String>,
235 pub errors: Vec<String>,
238}
239
240pub const VALID_THINKING_LEVELS: &[&str] =
243 &["off", "minimal", "low", "medium", "high", "xhigh", "max"];
244
245pub fn parse_thinking_level(s: &str) -> Option<ThinkingLevel> {
247 Some(match s {
248 "off" => ThinkingLevel::Off,
249 "minimal" => ThinkingLevel::Minimal,
250 "low" => ThinkingLevel::Low,
251 "medium" => ThinkingLevel::Medium,
252 "high" => ThinkingLevel::High,
253 "xhigh" => ThinkingLevel::Xhigh,
254 "max" => ThinkingLevel::Max,
255 _ => return None,
256 })
257}
258
259fn file_arg(arg: &str) -> Option<PathBuf> {
262 if let Some(rest) = arg.strip_prefix('@') {
263 if rest.is_empty() {
265 None
266 } else {
267 Some(PathBuf::from(rest))
268 }
269 } else {
270 None
271 }
272}
273
274pub fn parse_args(args: &[String]) -> Args {
280 let mut result = Args::default();
281 result.offline = offline_env_enabled();
282 if let Ok(raw) = std::env::var("RPI_EXTENSIONS_DIR") {
286 if !raw.is_empty() {
287 let sep = if cfg!(windows) { ';' } else { ':' };
288 for part in raw.split(sep) {
289 let trimmed = part.trim();
290 if !trimmed.is_empty() {
291 result.extensions_dir.push(PathBuf::from(trimmed));
292 }
293 }
294 }
295 }
296 let mut i = 0;
297 while i < args.len() {
298 let arg = args[i].clone();
299 let (flag_key, inline) = if arg.starts_with("--") {
303 match arg.find('=') {
304 Some(eq) => (arg[..eq].to_string(), Some(arg[eq + 1..].to_string())),
305 None => (arg.clone(), None),
306 }
307 } else {
308 (arg.clone(), None)
309 };
310
311 let mut take_value = |result: &mut Args, _flag: &str| -> Option<String> {
315 if let Some(v) = inline.clone() {
316 return Some(v);
317 }
318 if i + 1 < args.len() {
319 let next = &args[i + 1];
320 if !next.starts_with('-') || next == "-" {
321 i += 1;
322 return Some(args[i].clone());
323 }
324 }
325 result.errors.push(format!("{flag_key} requires a value"));
326 None
327 };
328
329 match flag_key.as_str() {
330 "--help" | "-h" => result.help = true,
331 "--version" | "-v" => result.version = true,
332 "--print" | "-p" => {
333 result.print = true;
334 if i + 1 < args.len() {
339 let next = &args[i + 1];
340 if !next.starts_with('@') && !next.starts_with('-') {
341 i += 1;
342 result.messages.push(args[i].clone());
343 }
344 }
345 }
346 "--mode" => {
347 if let Some(v) = take_value(&mut result, "--mode") {
348 result.mode = match v.as_str() {
349 "text" => Mode::Text,
350 "json" => Mode::Json,
351 "rpc" => Mode::Rpc,
352 other => {
353 result.errors.push(format!(
354 "Invalid --mode \"{other}\". Valid: text, json, rpc"
355 ));
356 Mode::Text
357 }
358 };
359 }
360 }
361 "--tui-mode" => {
362 if let Some(v) = take_value(&mut result, "--tui-mode") {
363 result.tui_mode = match v.to_ascii_lowercase().as_str() {
364 "regular" => TuiMode::Regular,
365 "fullscreen" => TuiMode::Fullscreen,
366 other => {
367 result.errors.push(format!(
368 "Invalid --tui-mode \"{other}\". Valid: regular, fullscreen"
369 ));
370 TuiMode::Fullscreen
371 }
372 };
373 }
374 }
375 "--continue" | "-c" => result.continue_session = true,
376 "--resume" | "-r" => result.resume = true,
377 "--no-session" => result.no_session = true,
378 "--no-tools" | "-nt" => result.no_tools = true,
379 "--no-builtin-tools" | "-nbt" => result.no_builtin_tools = true,
380 "--no-skills" | "-ns" => result.no_skills = true,
381 "--no-prompt-templates" | "-np" => result.no_prompt_templates = true,
382 "--no-context-files" | "-nc" => result.no_context_files = true,
383 "--no-extensions" | "-ne" => result.no_extensions = true,
384 "--enable-pi-packages" => result.enable_pi_packages = true,
385 "--extensions-dir" | "-ed" => {
386 if let Some(v) = take_value(&mut result, &flag_key) {
387 result.extensions_dir.push(PathBuf::from(v));
388 }
389 }
390 "--verbose" => result.verbose = true,
391 "--debug-system-prompt" => result.debug_system_prompt = true,
392 "--provider" => result.provider = take_value(&mut result, "--provider"),
393 "--model" => result.model = take_value(&mut result, "--model"),
394 "--api-key" => result.api_key = take_value(&mut result, "--api-key"),
395 "--base-url" => result.base_url = take_value(&mut result, "--base-url"),
396 "--timeout" => {
397 if let Some(v) = take_value(&mut result, "--timeout") {
398 match v.parse::<u64>() {
399 Ok(seconds) if seconds > 0 => {
400 result.timeout = Some(Duration::from_secs(seconds));
401 }
402 _ => result.errors.push(format!(
403 "Invalid --timeout \"{v}\". Expected a positive integer number of seconds"
404 )),
405 }
406 }
407 }
408 "--system-prompt" => result.system_prompt = take_value(&mut result, "--system-prompt"),
409 "--append-system-prompt" => {
410 if let Some(v) = take_value(&mut result, "--append-system-prompt") {
411 result.append_system_prompt.push(v);
412 }
413 }
414 "--name" | "-n" => result.name = take_value(&mut result, "--name"),
415 "--session" => result.session = take_value(&mut result, "--session"),
416 "--session-id" => result.session_id = take_value(&mut result, "--session-id"),
417 "--fork" => result.fork = take_value(&mut result, "--fork"),
418 "--models" => {
419 if let Some(v) = take_value(&mut result, &flag_key) {
420 result.models = Some(split_csv(&v));
421 }
422 }
423 "--extension" | "-e" => {
424 if let Some(v) = take_value(&mut result, &flag_key) {
425 result.extension.push(PathBuf::from(v));
426 }
427 }
428 "--skill" => {
429 if let Some(v) = take_value(&mut result, &flag_key) {
430 result.skill.push(PathBuf::from(v));
431 }
432 }
433 "--prompt-template" => {
434 if let Some(v) = take_value(&mut result, &flag_key) {
435 result.prompt_template.push(PathBuf::from(v));
436 }
437 }
438 "--session-dir" => {
439 if let Some(v) = take_value(&mut result, "--session-dir") {
440 result.session_dir = Some(PathBuf::from(v));
441 }
442 }
443 "--thinking" => {
444 if let Some(v) = take_value(&mut result, "--thinking") {
445 match parse_thinking_level(&v) {
446 Some(lvl) => result.thinking = Some(lvl),
447 None => result.ignored.push(format!(
448 "Invalid --thinking \"{v}\". Valid: {}",
449 VALID_THINKING_LEVELS.join(", ")
450 )),
451 }
452 }
453 }
454 "--tools" | "-t" => {
455 if let Some(v) = take_value(&mut result, &flag_key) {
456 result.tools = Some(split_csv(&v));
457 }
458 }
459 "--exclude-tools" | "-xt" => {
460 if let Some(v) = take_value(&mut result, &flag_key) {
461 result.exclude_tools = Some(split_csv(&v));
462 }
463 }
464 "--list-models" => {
465 let mut search = inline.clone().unwrap_or_default();
468 if inline.is_none()
469 && i + 1 < args.len()
470 && !args[i + 1].starts_with('-')
471 && !args[i + 1].starts_with('@')
472 {
473 i += 1;
474 search = args[i].clone();
475 }
476 result.list_models = Some(search);
477 }
478 "--offline" => result.offline = true,
479 "--export" => {
480 if let Some(value) = take_value(&mut result, &flag_key) {
481 result.export = Some(PathBuf::from(value));
482 }
483 }
484 "--approve" | "-a" => result.trust_override = Some(true),
485 "--no-approve" | "-na" => result.trust_override = Some(false),
486 other if matches!(other, "--models") => {
497 if inline.is_none()
500 && i + 1 < args.len()
501 && !args[i + 1].starts_with('-')
502 && !args[i + 1].starts_with('@')
503 {
504 i += 1;
505 }
506 result
507 .ignored
508 .push(format!("{other} is not supported in v1 (ignored)"));
509 }
510 "--theme" => {
511 result.theme = take_value(&mut result, "--theme");
512 }
513 "--no-themes" => result.no_themes = true,
514 other if other.starts_with("--") => {
519 let name = &flag_key;
520 let value = if let Some(value) = inline {
521 serde_json::Value::String(value)
522 } else if i + 1 < args.len()
523 && !args[i + 1].starts_with('-')
524 && !args[i + 1].starts_with('@')
525 {
526 i += 1;
527 serde_json::Value::String(args[i].clone())
528 } else {
529 serde_json::Value::Bool(true)
530 };
531 result.unknown_flags.insert(name[2..].to_string(), value);
532 }
533 other if other.starts_with('-') && other.len() > 1 => {
535 result.errors.push(format!("Unknown option: {other}"));
536 }
537 other => {
538 if let Some(path) = file_arg(other) {
539 result.file_args.push(path);
540 } else {
541 result.messages.push(other.to_string());
542 }
543 }
544 }
545 i += 1;
546 }
547
548 result
552}
553
554fn split_csv(v: &str) -> Vec<String> {
556 v.split(',')
557 .map(|s| s.trim().to_string())
558 .filter(|s| !s.is_empty())
559 .collect()
560}
561
562pub fn resolve_mode(parsed: &Args, stdin_is_tty: bool, stdout_is_tty: bool) -> RunMode {
567 if parsed.mode == Mode::Rpc {
568 return RunMode::Rpc;
569 }
570 if parsed.mode == Mode::Json {
571 return RunMode::Json;
572 }
573 if parsed.print || !stdin_is_tty || !stdout_is_tty {
574 RunMode::Print
575 } else {
576 RunMode::Interactive
577 }
578}
579
580#[derive(Debug, Clone, Copy, PartialEq, Eq)]
585pub enum RunMode {
586 Interactive,
587 Print,
588 Json,
589 Rpc,
590}
591
592pub fn print_help() {
594 let builtin = "read, bash, edit, write, docs";
595 println!(
596 "{name} - AI coding assistant with read, bash, edit, write, docs tools
597
598{u}Usage:{r}
599 {name} [options] [@files...] [messages...]
600
601{u}Options:{r}
602 --provider <name> Provider name (anthropic, openai-completions, openai-responses, or models.json id)
603 --model <pattern> Model pattern or ID (supports \"provider/id\" and optional \":<thinking>\")
604 --api-key <key> API key override for the selected provider
605 --base-url <url> Override the selected model endpoint
606 --timeout <seconds> LLM API request timeout (default: 600)
607 --system-prompt <text> Replace the default system prompt
608 --append-system-prompt <text> Append text to the system prompt (repeatable)
609 --thinking <level> off, minimal, low, medium, high, xhigh, max
610 --mode <mode> Output mode: text (default), json, or rpc
611 --tui-mode <mode> TUI buffer: regular or fullscreen (macOS default: regular)
612 --list-models [search] List available models (with optional fuzzy search)
613 --offline Disable startup network operations (same as PI_OFFLINE=1)
614 --export <file> Export a JSONL session to HTML and exit
615 --approve, -a Force-enable current-project resources
616 --no-approve, -na Disable current-project resources
617 --print, -p Non-interactive: process prompt(s) and exit
618 --continue, -c Continue the most recent session
619 --resume, -r Browse and select a session to resume
620 --session <id|path> Use a specific session (partial UUID or file)
621 --session-dir <dir> Directory for session storage
622 --no-session Ephemeral mode (do not persist the session)
623 --name, -n <name> Set the session display name
624 --tools, -t <list> Comma-separated allowlist of tool names to enable
625 --exclude-tools, -xt <list> Comma-separated denylist of tool names to disable
626 --no-tools, -nt Disable all tools
627 --no-builtin-tools, -nbt Disable the built-in tools (read, bash, edit, write, docs)
628 --no-skills, -ns Skip skill discovery (no <available_skills> block)
629 --no-prompt-templates, -np Skip prompt-template discovery (/expand templates)
630 --no-context-files, -nc Skip AGENTS.md/CLAUDE.md discovery (no <project_context>)
631 --no-extensions, -ne Skip Rust cdylib and JS/TS extension loading
632 --enable-pi-packages Enable configured Pi JS/TS packages (starts Node)
633 --extensions-dir, -ed <dir> Extra dir to scan for plugins (.dll/.so/.dylib); repeatable
634 (also via RPI_EXTENSIONS_DIR env: ';' on Windows, ':' on Unix)
635 --debug-system-prompt Print the resolved system-prompt sections to stderr (verification)
636 --verbose Show startup warnings (e.g. ignored flags)
637 --help, -h Show this help
638 --version, -v Show version
639
640{u}Subcommands:{r}
641 update Update the rpi CLI from crates.io
642 auth login|check|logout Manage persisted credentials in ~/.rpi/auth.json
643 (see `rpi auth --help`)
644 package list|add|remove|update Update Rust-native packages and manage Pi package settings
645 (see `rpi package --help`)
646 install <crate> Build and install a Rust cdylib extension
647 (see `rpi install --help`)
648 install-pi <spec> Install an npm/git/local Pi package
649 (see `rpi install-pi --help`)
650 pi-package update Update configured Pi npm/Git packages
651 uninstall <crate> Remove an installed Rust cdylib extension
652 (use `rpi uninstall pi <spec>` for Pi packages)
653 uninstall-pi <spec> Remove an installed npm/git/local Pi package
654 (see `rpi uninstall-pi --help`)
655 dev [options] Build, watch, and hot-reload a Rust extension
656 (see `rpi dev --help`)
657 dev-local [options] Debug only the current Rust extension
658 (shortcut for `rpi dev --local-only`)
659
660{u}Built-in Tools:{r}
661 {builtin} (enabled by default)
662
663{u}Examples:{r}
664 # Interactive with an initial prompt
665 {name} \"List all .rs files in src/\"
666
667 # Single-shot print mode
668 {name} -p \"Summarize this project\"
669
670 # Include a file in the initial message
671 {name} @README.md \"What does this project do?\"
672
673 # Continue the previous session
674 {name} -c \"What did we discuss?\"
675
676 # Use a specific model + thinking level
677 {name} --model claude-sonnet-5 --thinking high \"Refactor this\"
678
679 # JSON event stream (one JSON object per line on stdout)
680 {name} --mode json -p \"Inspect the code\"
681
682 # Read-only: no file-modifying tools
683 {name} --tools read,bash -p \"Review the code in src/\"
684
685{u}Environment:{r}
686 ANTHROPIC_API_KEY Anthropic API key (x-api-key) — fallback when no stored credential
687 ANTHROPIC_AUTH_TOKEN Bearer token (Authorization: Bearer) for third-party gateways
688 ANTHROPIC_BASE_URL Override the Anthropic endpoint (e.g. a compatible proxy)
689 OPENAI_API_KEY Bearer token for openai-completions/responses
690 PI_OFFLINE Disable startup network operations when set to 1/true/yes
691 RPI_CODING_AGENT_DIR Override the ~/.rpi config directory (auth.json + models.json)
692
693{u}Notes:{r}
694 Supported HTTP protocols are Anthropic Messages and OpenAI Chat Completions.
695 Define custom model catalogs and provider apiKey values in
696 ~/.rpi/agent/models.json. The interactive TUI, Rust and JS/TS extensions,
697 opt-in Pi package resources, skills, prompt templates, themes, model cycling, session
698 fork/export, and trust commands are
699 available in the current build. OAuth, RPC, and full model cycling remain
700 outside the current implementation.
701",
702 name = crate::APP_NAME,
703 builtin = builtin,
704 u = "\x1b[1m",
705 r = "\x1b[0m",
706 );
707}
708
709pub fn print_version() {
711 println!("{} {}", crate::APP_NAME, crate::VERSION);
712}
713
714#[cfg(test)]
715mod tests {
716 use super::*;
717
718 struct RestoreOfflineEnv(Option<std::ffi::OsString>);
719
720 impl Drop for RestoreOfflineEnv {
721 fn drop(&mut self) {
722 match self.0.take() {
723 Some(value) => std::env::set_var(PI_OFFLINE_ENV, value),
724 None => std::env::remove_var(PI_OFFLINE_ENV),
725 }
726 }
727 }
728
729 fn s(args: &[&str]) -> Vec<String> {
730 args.iter().map(|a| a.to_string()).collect()
731 }
732
733 #[test]
734 fn parses_basic_prompt() {
735 let a = parse_args(&s(&["hello", "world"]));
736 assert_eq!(a.messages, vec!["hello".to_string(), "world".to_string()]);
737 assert!(!a.help);
738 }
739
740 #[test]
741 fn parses_help_and_version() {
742 let a = parse_args(&s(&["--help"]));
743 assert!(a.help);
744 let a = parse_args(&s(&["-v"]));
745 assert!(a.version);
746 }
747
748 #[test]
749 fn print_consumes_following_positional() {
750 let a = parse_args(&s(&["-p", "summarize"]));
751 assert!(a.print);
752 assert_eq!(a.messages, vec!["summarize".to_string()]);
753 }
754
755 #[test]
756 fn print_does_not_consume_file_or_flag() {
757 let a = parse_args(&s(&["-p", "@file.md"]));
758 assert!(a.print);
759 assert!(a.messages.is_empty());
760 assert_eq!(a.file_args, vec![PathBuf::from("file.md")]);
761 }
762
763 #[test]
764 fn model_and_thinking() {
765 let a = parse_args(&s(&["--model", "claude-sonnet-5", "--thinking", "high"]));
766 assert_eq!(a.model.as_deref(), Some("claude-sonnet-5"));
767 assert_eq!(a.thinking, Some(ThinkingLevel::High));
768 }
769
770 #[test]
771 fn model_with_thinking_shorthand() {
772 let a = parse_args(&s(&["--model", "claude-sonnet-5:high"]));
773 assert_eq!(a.model.as_deref(), Some("claude-sonnet-5:high"));
775 }
776
777 #[test]
778 fn tools_split_csv() {
779 let a = parse_args(&s(&["--tools", "read, bash ,write"]));
780 assert_eq!(
781 a.tools.as_deref(),
782 Some(&["read".to_string(), "bash".to_string(), "write".to_string()][..])
783 );
784 }
785
786 #[test]
787 fn unknown_short_flag_errors() {
788 let a = parse_args(&s(&["-Z"]));
789 assert!(!a.errors.is_empty());
790 }
791
792 #[test]
793 fn unknown_long_flag_is_retained_for_extensions() {
794 let a = parse_args(&s(&["--frobnicate", "value"]));
795 assert!(a.errors.is_empty());
796 assert_eq!(
797 a.unknown_flags.get("frobnicate"),
798 Some(&serde_json::Value::String("value".into()))
799 );
800 assert!(a.ignored.is_empty());
801 }
802
803 #[test]
804 fn unknown_long_boolean_flag_is_retained() {
805 let a = parse_args(&s(&["--server"]));
806 assert_eq!(
807 a.unknown_flags.get("server"),
808 Some(&serde_json::Value::Bool(true))
809 );
810 }
811
812 #[test]
813 fn unknown_long_flags_keep_string_and_equals_values() {
814 let a = parse_args(&s(&["--port", "8080", "--bind=127.0.0.1"]));
815 assert_eq!(
816 a.unknown_flags.get("port"),
817 Some(&serde_json::Value::String("8080".into()))
818 );
819 assert_eq!(
820 a.unknown_flags.get("bind"),
821 Some(&serde_json::Value::String("127.0.0.1".into()))
822 );
823 }
824
825 #[test]
826 fn models_flag_parses_csv() {
827 let a = parse_args(&s(&["--models", "a,b,c"]));
828 assert!(a.errors.is_empty());
829 assert!(a.ignored.is_empty(), "--models is implemented");
830 assert_eq!(
831 a.models.as_deref(),
832 Some(&["a".to_string(), "b".to_string(), "c".to_string()][..])
833 );
834 assert!(a.messages.is_empty());
836 }
837
838 #[test]
839 fn list_models_accepts_bare_and_search_forms() {
840 let bare = parse_args(&s(&["--list-models"]));
841 assert_eq!(bare.list_models.as_deref(), Some(""));
842 assert!(bare.ignored.is_empty());
843 assert!(bare.messages.is_empty());
844
845 let search = parse_args(&s(&["--list-models", "claude"]));
846 assert_eq!(search.list_models.as_deref(), Some("claude"));
847 assert!(search.messages.is_empty());
848
849 let inline = parse_args(&s(&["--list-models=gpt"]));
850 assert_eq!(inline.list_models.as_deref(), Some("gpt"));
851 }
852
853 #[test]
854 fn offline_flag_is_honored_without_warning() {
855 let args = parse_args(&s(&["--offline"]));
856 assert!(args.offline);
857 assert!(args.ignored.is_empty());
858 }
859
860 #[test]
861 fn timeout_parses_seconds_in_separate_and_equals_forms() {
862 let separate = parse_args(&s(&["--timeout", "45"]));
863 assert!(separate.errors.is_empty());
864 assert_eq!(separate.timeout, Some(Duration::from_secs(45)));
865
866 let inline = parse_args(&s(&["--timeout=90"]));
867 assert!(inline.errors.is_empty());
868 assert_eq!(inline.timeout, Some(Duration::from_secs(90)));
869 }
870
871 #[test]
872 fn timeout_rejects_zero_and_invalid_values() {
873 for value in ["0", "1.5", "forever", "18446744073709551616"] {
874 let args = parse_args(&s(&[&format!("--timeout={value}")]));
875 assert_eq!(args.errors.len(), 1, "value: {value}");
876 assert!(args.timeout.is_none(), "value: {value}");
877 }
878 }
879
880 #[test]
881 fn native_pi_offline_truthy_values_are_case_insensitive() {
882 for value in [
883 Some("1"),
884 Some("true"),
885 Some("TRUE"),
886 Some("Yes"),
887 Some("yEs"),
888 ] {
889 assert!(is_truthy_env_flag(value), "value={value:?}");
890 }
891 for value in [
892 None,
893 Some(""),
894 Some("0"),
895 Some("false"),
896 Some("no"),
897 Some(" true "),
898 ] {
899 assert!(!is_truthy_env_flag(value), "value={value:?}");
900 }
901 }
902
903 #[test]
904 fn pi_offline_env_and_cli_flag_share_one_normalized_gate() {
905 let _guard = crate::config::test_support::env_lock().lock().unwrap();
906 let _restore = RestoreOfflineEnv(std::env::var_os(PI_OFFLINE_ENV));
907
908 std::env::set_var(PI_OFFLINE_ENV, "YeS");
909 assert!(parse_args(&[]).offline);
910 assert!(normalize_offline_mode(&[]));
911 assert_eq!(std::env::var(PI_OFFLINE_ENV).as_deref(), Ok("1"));
912
913 std::env::set_var(PI_OFFLINE_ENV, "0");
914 assert!(!parse_args(&[]).offline);
915 let argv = s(&["package", "update", "--offline"]);
916 assert!(normalize_offline_mode(&argv));
917 assert_eq!(std::env::var(PI_OFFLINE_ENV).as_deref(), Ok("1"));
918 assert_eq!(without_offline_flag(&argv), s(&["package", "update"]));
919 }
920
921 #[test]
922 fn project_trust_flags_are_honored_without_warning() {
923 let approved = parse_args(&s(&["--approve"]));
924 assert_eq!(approved.trust_override, Some(true));
925 assert!(approved.ignored.is_empty());
926 let denied = parse_args(&s(&["--no-approve"]));
927 assert_eq!(denied.trust_override, Some(false));
928 assert!(denied.ignored.is_empty());
929 }
930
931 #[test]
932 fn export_flag_captures_input_and_output_position() {
933 let args = parse_args(&s(&["--export", "session.jsonl", "transcript.html"]));
934 assert_eq!(args.export, Some(PathBuf::from("session.jsonl")));
935 assert_eq!(args.messages, vec!["transcript.html".to_string()]);
936 assert!(args.ignored.is_empty());
937 }
938
939 #[test]
940 fn session_id_and_fork_flags_parse() {
941 let a = parse_args(&s(&["--session-id", "01abc", "--fork", "xyz"]));
942 assert!(a.errors.is_empty());
943 assert_eq!(a.session_id.as_deref(), Some("01abc"));
944 assert_eq!(a.fork.as_deref(), Some("xyz"));
945 let a = parse_args(&s(&[
946 "-e",
947 "plugin.dll",
948 "--skill",
949 "s",
950 "--prompt-template",
951 "t.md",
952 ]));
953 assert_eq!(a.extension.len(), 1);
954 assert_eq!(a.skill.len(), 1);
955 assert_eq!(a.prompt_template.len(), 1);
956 }
957
958 #[test]
959 fn no_skills_flag_honored() {
960 let a = parse_args(&s(&["-ns"]));
961 assert!(a.errors.is_empty());
962 assert!(a.no_skills);
963 assert!(a.ignored.is_empty());
965 }
966
967 #[test]
968 fn no_prompt_templates_flag_honored() {
969 let a = parse_args(&s(&["--no-prompt-templates"]));
970 assert!(a.no_prompt_templates);
971 assert!(a.ignored.is_empty());
972 }
973
974 #[test]
975 fn no_context_files_flag_honored() {
976 let a = parse_args(&s(&["-nc"]));
977 assert!(a.no_context_files);
978 assert!(a.ignored.is_empty());
979 }
980
981 #[test]
982 fn no_extensions_flag_honored() {
983 let a = parse_args(&s(&["--no-extensions"]));
985 assert!(a.no_extensions);
986 assert!(a.ignored.is_empty());
987 }
988
989 #[test]
990 fn pi_packages_are_disabled_by_default_and_explicitly_enabled() {
991 let a = parse_args(&s(&[]));
992 assert!(!a.enable_pi_packages);
993 assert!(a.ignored.is_empty());
994
995 let a = parse_args(&s(&["--enable-pi-packages"]));
996 assert!(a.enable_pi_packages);
997 assert!(a.ignored.is_empty());
998 }
999
1000 #[test]
1001 fn extensions_dir_flag_collects_dirs() {
1002 let a = parse_args(&s(&["--extensions-dir", "/a/b", "-ed", "/c/d"]));
1003 assert_eq!(
1004 a.extensions_dir,
1005 vec![PathBuf::from("/a/b"), PathBuf::from("/c/d")]
1006 );
1007 assert!(a.ignored.is_empty());
1008 }
1009
1010 #[test]
1011 fn extensions_dir_inline_equals_form() {
1012 let a = parse_args(&s(&["--extensions-dir=/x/y"]));
1013 assert_eq!(a.extensions_dir, vec![PathBuf::from("/x/y")]);
1014 }
1015
1016 #[test]
1017 fn extensions_dir_env_is_merged() {
1018 let a = parse_args(&s(&["--extensions-dir", "/flag/only"]));
1024 assert!(a
1025 .extensions_dir
1026 .iter()
1027 .any(|p| p == &PathBuf::from("/flag/only")));
1028 }
1029
1030 #[test]
1031 fn file_args_stripped() {
1032 let a = parse_args(&s(&["@a.txt", "@b.md", "hi"]));
1033 assert_eq!(
1034 a.file_args,
1035 vec![PathBuf::from("a.txt"), PathBuf::from("b.md")]
1036 );
1037 assert_eq!(a.messages, vec!["hi".to_string()]);
1038 }
1039
1040 #[test]
1041 fn equals_form_supported() {
1042 let a = parse_args(&s(&["--model=claude-sonnet-5", "--thinking=low"]));
1043 assert_eq!(a.model.as_deref(), Some("claude-sonnet-5"));
1044 assert_eq!(a.thinking, Some(ThinkingLevel::Low));
1045 }
1046
1047 #[test]
1048 fn theme_flag_is_honored() {
1049 let a = parse_args(&s(&["--theme", "ocean.json"]));
1050 assert_eq!(a.theme.as_deref(), Some("ocean.json"));
1051 assert!(a.ignored.is_empty());
1052 }
1053
1054 #[test]
1055 fn no_themes_is_honored() {
1056 let a = parse_args(&s(&["--no-themes"]));
1057 assert!(a.no_themes);
1058 assert!(a.ignored.is_empty());
1059 }
1060
1061 #[test]
1062 fn tui_mode_parses_and_validates() {
1063 assert_eq!(
1064 parse_args(&s(&["--tui-mode", "regular"])).tui_mode,
1065 TuiMode::Regular
1066 );
1067 assert_eq!(
1068 parse_args(&s(&["--tui-mode=fullscreen"])).tui_mode,
1069 TuiMode::Fullscreen
1070 );
1071 let invalid = parse_args(&s(&["--tui-mode", "split"]));
1072 assert!(!invalid.errors.is_empty());
1073 }
1074
1075 #[test]
1076 fn tui_mode_default_preserves_native_macos_scrollback() {
1077 assert_eq!(default_tui_mode_for(true), TuiMode::Regular);
1078 assert_eq!(default_tui_mode_for(false), TuiMode::Fullscreen);
1079 }
1080
1081 #[test]
1082 fn resolve_mode_interactive_when_tty() {
1083 let a = Args {
1084 print: true,
1085 ..Args::default()
1086 };
1087 assert_eq!(resolve_mode(&a, true, true), RunMode::Print);
1088 let a = Args::default();
1089 assert_eq!(resolve_mode(&a, true, true), RunMode::Interactive);
1090 let a = Args {
1091 mode: Mode::Json,
1092 ..Args::default()
1093 };
1094 assert_eq!(resolve_mode(&a, true, true), RunMode::Json);
1095 let a = Args {
1096 mode: Mode::Rpc,
1097 ..Args::default()
1098 };
1099 assert_eq!(resolve_mode(&a, true, true), RunMode::Rpc);
1100 }
1101
1102 #[test]
1103 fn piped_stdout_forces_print() {
1104 let a = Args::default();
1105 assert_eq!(resolve_mode(&a, true, false), RunMode::Print);
1107 }
1108}