Skip to main content

claude_wrapper/
error.rs

1use std::path::PathBuf;
2
3/// Errors returned by claude-wrapper operations.
4#[derive(Debug, thiserror::Error)]
5pub enum Error {
6    /// The `claude` binary was not found in PATH.
7    #[error("claude binary not found in PATH")]
8    NotFound,
9
10    /// A claude command failed with a non-zero exit code.
11    #[error("claude command failed: {command} (exit code {exit_code}){}{}{}", working_dir.as_ref().map(|d| format!(" (in {})", d.display())).unwrap_or_default(), if stdout.is_empty() { String::new() } else { format!("\nstdout: {stdout}") }, if stderr.is_empty() { String::new() } else { format!("\nstderr: {stderr}") })]
12    CommandFailed {
13        command: String,
14        exit_code: i32,
15        stdout: String,
16        stderr: String,
17        working_dir: Option<PathBuf>,
18    },
19
20    /// An I/O error occurred while spawning or communicating with the process.
21    #[error("io error: {message}{}", working_dir.as_ref().map(|d| format!(" (in {})", d.display())).unwrap_or_default())]
22    Io {
23        message: String,
24        #[source]
25        source: std::io::Error,
26        working_dir: Option<PathBuf>,
27    },
28
29    /// The command timed out.
30    #[error("claude command timed out after {timeout_seconds}s")]
31    Timeout { timeout_seconds: u64 },
32
33    /// JSON parsing failed.
34    #[cfg(feature = "json")]
35    #[error("json parse error: {message}")]
36    Json {
37        message: String,
38        #[source]
39        source: serde_json::Error,
40    },
41
42    /// The installed CLI version does not meet the minimum requirement.
43    #[error("CLI version {found} does not meet minimum requirement {minimum}")]
44    VersionMismatch {
45        found: crate::version::CliVersion,
46        minimum: crate::version::CliVersion,
47    },
48
49    /// Construction of a `dangerous::Client` was attempted without
50    /// the opt-in env-var set. The env-var name is a compile-time
51    /// constant exported from [`crate::dangerous::ALLOW_ENV`].
52    #[error(
53        "dangerous operations are not allowed; set the env var `{env_var}=1` at process start if you really mean it"
54    )]
55    DangerousNotAllowed { env_var: &'static str },
56
57    /// A configured [`BudgetTracker`](crate::budget::BudgetTracker) has
58    /// hit its `max_usd` ceiling. Raised before the next call is
59    /// dispatched, so the CLI is not invoked.
60    #[error("budget exceeded: ${total_usd:.4} spent, ${max_usd:.4} max")]
61    BudgetExceeded { total_usd: f64, max_usd: f64 },
62}
63
64impl From<std::io::Error> for Error {
65    fn from(e: std::io::Error) -> Self {
66        Self::Io {
67            message: e.to_string(),
68            source: e,
69            working_dir: None,
70        }
71    }
72}
73
74/// Result type alias for claude-wrapper operations.
75pub type Result<T> = std::result::Result<T, Error>;
76
77#[cfg(test)]
78mod tests {
79    use super::*;
80
81    fn command_failed(stdout: &str, stderr: &str, working_dir: Option<PathBuf>) -> Error {
82        Error::CommandFailed {
83            command: "/bin/claude --print".to_string(),
84            exit_code: 7,
85            stdout: stdout.to_string(),
86            stderr: stderr.to_string(),
87            working_dir,
88        }
89    }
90
91    #[test]
92    fn command_failed_display_includes_command_and_exit_code() {
93        let e = command_failed("", "", None);
94        let s = e.to_string();
95        assert!(s.contains("/bin/claude --print"));
96        assert!(s.contains("exit code 7"));
97    }
98
99    #[test]
100    fn command_failed_display_omits_empty_stdout_and_stderr() {
101        let s = command_failed("", "", None).to_string();
102        assert!(!s.contains("stdout:"));
103        assert!(!s.contains("stderr:"));
104    }
105
106    #[test]
107    fn command_failed_display_includes_nonempty_stdout() {
108        let s = command_failed("hello", "", None).to_string();
109        assert!(s.contains("stdout: hello"));
110    }
111
112    #[test]
113    fn command_failed_display_includes_nonempty_stderr() {
114        let s = command_failed("", "boom", None).to_string();
115        assert!(s.contains("stderr: boom"));
116    }
117
118    #[test]
119    fn command_failed_display_includes_both_streams_when_present() {
120        let s = command_failed("out", "err", None).to_string();
121        assert!(s.contains("stdout: out"));
122        assert!(s.contains("stderr: err"));
123    }
124
125    #[test]
126    fn command_failed_display_includes_working_dir_when_present() {
127        let s = command_failed("", "", Some(PathBuf::from("/tmp/proj"))).to_string();
128        assert!(s.contains("/tmp/proj"));
129    }
130
131    #[test]
132    fn command_failed_display_omits_working_dir_when_absent() {
133        let s = command_failed("", "", None).to_string();
134        assert!(!s.contains("(in "));
135    }
136
137    #[test]
138    fn timeout_display_formats_seconds() {
139        let s = Error::Timeout {
140            timeout_seconds: 42,
141        }
142        .to_string();
143        assert!(s.contains("42s"));
144    }
145
146    #[test]
147    fn io_error_display_includes_working_dir_when_present() {
148        let e = Error::Io {
149            message: "spawn failed".to_string(),
150            source: std::io::Error::new(std::io::ErrorKind::NotFound, "no file"),
151            working_dir: Some(PathBuf::from("/work")),
152        };
153        let s = e.to_string();
154        assert!(s.contains("spawn failed"));
155        assert!(s.contains("/work"));
156    }
157}