aft/
lib.rs

1#![allow(
2    clippy::collapsible_if,
3    clippy::double_ended_iterator_last,
4    clippy::int_plus_one,
5    clippy::large_enum_variant,
6    clippy::len_without_is_empty,
7    clippy::let_and_return,
8    clippy::manual_contains,
9    clippy::manual_pattern_char_comparison,
10    clippy::manual_repeat_n,
11    clippy::manual_strip,
12    clippy::manual_unwrap_or_default,
13    clippy::map_clone,
14    clippy::needless_borrow,
15    clippy::needless_borrows_for_generic_args,
16    clippy::needless_range_loop,
17    clippy::new_without_default,
18    clippy::obfuscated_if_else,
19    clippy::ptr_arg,
20    clippy::question_mark,
21    clippy::same_item_push,
22    clippy::should_implement_trait,
23    clippy::single_match,
24    clippy::too_many_arguments,
25    clippy::type_complexity,
26    clippy::unnecessary_cast,
27    clippy::unnecessary_lazy_evaluations,
28    clippy::unnecessary_map_or
29)]
30
31// ## Note on `.unwrap()` / `.expect()` usage
32//
33// The remaining `.unwrap()` and `.expect()` calls in `src/` are in:
34// - **Tree-sitter query operations** (parser.rs, zoom.rs, extract.rs, inline.rs,
35//   outline.rs): These operate on AFT's own compiled grammars and query patterns, which
36//   are compile-time constants. Pattern captures and node kinds are guaranteed to exist.
37// - **Checkpoint serialization** (checkpoint.rs): serde_json::to_value on known-good
38//   HashMap<PathBuf, String> types cannot fail.
39// - **lib.rs main loop**: JSON parsing of stdin lines — a malformed line is logged and
40//   skipped, not unwrapped.
41//
42// All production command handlers that process user/agent input return Result or
43// Response::error instead of panicking. Confirmed zero .unwrap()/.expect() in
44// production error paths as of v0.6.3 audit.
45
46pub mod ast_grep_lang;
47pub mod backup;
48pub mod bash_background;
49pub mod bash_permissions;
50pub mod bash_rewrite;
51pub mod callgraph;
52pub mod calls;
53pub mod checkpoint;
54pub mod commands;
55pub mod compress;
56pub mod config;
57pub mod context;
58pub mod edit;
59pub mod error;
60pub mod extract;
61pub mod format;
62pub mod fuzzy_match;
63pub mod imports;
64pub mod indent;
65pub mod language;
66pub mod log_ctx;
67pub mod lsp;
68pub mod lsp_hints;
69pub mod parser;
70pub mod protocol;
71pub mod search_index;
72pub mod semantic_index;
73pub mod symbol_cache_disk;
74pub mod symbols;
75
76#[cfg(test)]
77mod tests {
78    use super::*;
79    use config::Config;
80    use error::AftError;
81    use protocol::{RawRequest, Response};
82
83    // --- Protocol serialization ---
84
85    #[test]
86    fn raw_request_deserializes_ping() {
87        let json = r#"{"id":"1","command":"ping"}"#;
88        let req: RawRequest = serde_json::from_str(json).unwrap();
89        assert_eq!(req.id, "1");
90        assert_eq!(req.command, "ping");
91        assert!(req.lsp_hints.is_none());
92    }
93
94    #[test]
95    fn raw_request_deserializes_echo_with_params() {
96        let json = r#"{"id":"2","command":"echo","message":"hello"}"#;
97        let req: RawRequest = serde_json::from_str(json).unwrap();
98        assert_eq!(req.id, "2");
99        assert_eq!(req.command, "echo");
100        // "message" is captured in the flattened params
101        assert_eq!(req.params["message"], "hello");
102    }
103
104    #[test]
105    fn raw_request_preserves_unknown_fields() {
106        let json = r#"{"id":"3","command":"ping","future_field":"abc","nested":{"x":1}}"#;
107        let req: RawRequest = serde_json::from_str(json).unwrap();
108        assert_eq!(req.params["future_field"], "abc");
109        assert_eq!(req.params["nested"]["x"], 1);
110    }
111
112    #[test]
113    fn raw_request_with_lsp_hints() {
114        let json = r#"{"id":"4","command":"ping","lsp_hints":{"completions":["foo","bar"]}}"#;
115        let req: RawRequest = serde_json::from_str(json).unwrap();
116        assert!(req.lsp_hints.is_some());
117        let hints = req.lsp_hints.unwrap();
118        assert_eq!(hints["completions"][0], "foo");
119    }
120
121    #[test]
122    fn response_success_round_trip() {
123        let resp = Response::success("42", serde_json::json!({"command": "pong"}));
124        let json_str = serde_json::to_string(&resp).unwrap();
125        let v: serde_json::Value = serde_json::from_str(&json_str).unwrap();
126        assert_eq!(v["id"], "42");
127        assert_eq!(v["success"], true);
128        assert_eq!(v["command"], "pong");
129    }
130
131    #[test]
132    fn response_error_round_trip() {
133        let resp = Response::error("99", "unknown_command", "unknown command: foo");
134        let json_str = serde_json::to_string(&resp).unwrap();
135        let v: serde_json::Value = serde_json::from_str(&json_str).unwrap();
136        assert_eq!(v["id"], "99");
137        assert_eq!(v["success"], false);
138        assert_eq!(v["code"], "unknown_command");
139        assert_eq!(v["message"], "unknown command: foo");
140    }
141
142    // --- Error formatting ---
143
144    #[test]
145    fn error_display_symbol_not_found() {
146        let err = AftError::SymbolNotFound {
147            name: "foo".into(),
148            file: "bar.rs".into(),
149        };
150        assert_eq!(err.to_string(), "symbol 'foo' not found in bar.rs");
151        assert_eq!(err.code(), "symbol_not_found");
152    }
153
154    #[test]
155    fn error_display_ambiguous_symbol() {
156        let err = AftError::AmbiguousSymbol {
157            name: "Foo".into(),
158            candidates: vec!["a.rs:10".into(), "b.rs:20".into()],
159        };
160        let s = err.to_string();
161        assert!(s.contains("Foo"));
162        assert!(s.contains("a.rs:10, b.rs:20"));
163    }
164
165    #[test]
166    fn error_display_parse_error() {
167        let err = AftError::ParseError {
168            message: "unexpected token".into(),
169        };
170        assert_eq!(err.to_string(), "parse error: unexpected token");
171    }
172
173    #[test]
174    fn error_display_file_not_found() {
175        let err = AftError::FileNotFound {
176            path: "/tmp/missing.rs".into(),
177        };
178        assert_eq!(err.to_string(), "file not found: /tmp/missing.rs");
179    }
180
181    #[test]
182    fn error_display_invalid_request() {
183        let err = AftError::InvalidRequest {
184            message: "missing field".into(),
185        };
186        assert_eq!(err.to_string(), "invalid request: missing field");
187    }
188
189    #[test]
190    fn error_display_checkpoint_not_found() {
191        let err = AftError::CheckpointNotFound {
192            name: "pre-refactor".into(),
193        };
194        assert_eq!(err.to_string(), "checkpoint not found: pre-refactor");
195        assert_eq!(err.code(), "checkpoint_not_found");
196    }
197
198    #[test]
199    fn error_display_no_undo_history() {
200        let err = AftError::NoUndoHistory {
201            path: "src/main.rs".into(),
202        };
203        assert_eq!(err.to_string(), "no undo history for: src/main.rs");
204        assert_eq!(err.code(), "no_undo_history");
205    }
206
207    #[test]
208    fn error_display_ambiguous_match() {
209        let err = AftError::AmbiguousMatch {
210            pattern: "TODO".into(),
211            count: 5,
212        };
213        assert_eq!(
214            err.to_string(),
215            "pattern 'TODO' matches 5 occurrences, expected exactly 1"
216        );
217        assert_eq!(err.code(), "ambiguous_match");
218    }
219
220    #[test]
221    fn error_display_project_too_large() {
222        let err = AftError::ProjectTooLarge {
223            count: 20001,
224            max: 20000,
225        };
226        assert_eq!(
227            err.to_string(),
228            "project has 20001 source files, exceeding max_callgraph_files=20000. Call-graph operations (callers, trace_to, trace_data, impact) are disabled for this root. Open a specific subdirectory or raise max_callgraph_files in config."
229        );
230        assert_eq!(err.code(), "project_too_large");
231    }
232
233    #[test]
234    fn error_to_json_has_code_and_message() {
235        let err = AftError::FileNotFound { path: "/x".into() };
236        let j = err.to_error_json();
237        assert_eq!(j["code"], "file_not_found");
238        assert!(j["message"].as_str().unwrap().contains("/x"));
239    }
240
241    // --- Config defaults ---
242
243    #[test]
244    fn config_default_values() {
245        let cfg = Config::default();
246        assert!(cfg.project_root.is_none());
247        assert_eq!(cfg.validation_depth, 1);
248        assert_eq!(cfg.checkpoint_ttl_hours, 24);
249        assert_eq!(cfg.max_symbol_depth, 10);
250        assert_eq!(cfg.formatter_timeout_secs, 10);
251        assert_eq!(cfg.type_checker_timeout_secs, 30);
252        assert_eq!(cfg.max_callgraph_files, 5_000);
253    }
254}
aft/lib.rs

aft/
lib.rs
