aft/commands/
trace_data.rs

1use std::path::Path;
2
3use crate::context::AppContext;
4use crate::protocol::{RawRequest, Response};
5
6/// Handle a `trace_data` request.
7///
8/// Traces how an expression flows through variable assignments within a
9/// function body and across function boundaries via argument-to-parameter
10/// matching. Destructuring, spread, and unresolved calls produce approximate
11/// hops and stop tracking.
12///
13/// Expects:
14/// - `file` (string, required) — path to the source file containing the symbol
15/// - `symbol` (string, required) — name of the function containing the expression
16/// - `expression` (string, required) — the expression/variable name to track
17/// - `depth` (number, optional, default 5) — maximum cross-file hop depth
18///
19/// Returns `TraceDataResult` with fields: `expression`, `origin_file`,
20/// `origin_symbol`, `hops` (array of DataFlowHop), `depth_limited`.
21///
22/// Returns error if:
23/// - required params missing
24/// - call graph not initialized (configure not called)
25/// - symbol not found in the file
26pub fn handle_trace_data(req: &RawRequest, ctx: &AppContext) -> Response {
27    let file = match req.params.get("file").and_then(|v| v.as_str()) {
28        Some(f) => f,
29        None => {
30            return Response::error(
31                &req.id,
32                "invalid_request",
33                "trace_data: missing required param 'file'",
34            );
35        }
36    };
37
38    let symbol = match req.params.get("symbol").and_then(|v| v.as_str()) {
39        Some(s) => s,
40        None => {
41            return Response::error(
42                &req.id,
43                "invalid_request",
44                "trace_data: missing required param 'symbol'",
45            );
46        }
47    };
48
49    let expression = match req.params.get("expression").and_then(|v| v.as_str()) {
50        Some(e) => e,
51        None => {
52            return Response::error(
53                &req.id,
54                "invalid_request",
55                "trace_data: missing required param 'expression'",
56            );
57        }
58    };
59
60    let depth = req
61        .params
62        .get("depth")
63        .and_then(|v| v.as_u64())
64        .unwrap_or(5) as usize;
65
66    let mut cg_ref = ctx.callgraph().borrow_mut();
67    let graph = match cg_ref.as_mut() {
68        Some(g) => g,
69        None => {
70            return Response::error(
71                &req.id,
72                "not_configured",
73                "trace_data: project not configured — send 'configure' first",
74            );
75        }
76    };
77
78    let file_path = Path::new(file);
79
80    // Build file data first to check if the symbol exists
81    match graph.build_file(file_path) {
82        Ok(data) => {
83            let has_symbol = data.calls_by_symbol.contains_key(symbol)
84                || data.exported_symbols.contains(&symbol.to_string())
85                || data.symbol_metadata.contains_key(symbol);
86            if !has_symbol {
87                return Response::error(
88                    &req.id,
89                    "symbol_not_found",
90                    format!("trace_data: symbol '{}' not found in {}", symbol, file),
91                );
92            }
93        }
94        Err(e) => {
95            return Response::error(&req.id, e.code(), e.to_string());
96        }
97    }
98
99    match graph.trace_data(file_path, symbol, expression, depth) {
100        Ok(result) => {
101            let result_json = serde_json::to_value(&result).unwrap_or_default();
102            Response::success(&req.id, result_json)
103        }
104        Err(e) => Response::error(&req.id, e.code(), e.to_string()),
105    }
106}
aft/commands/trace_data.rs

aft/commands/
trace_data.rs
