1use std::sync::atomic::{AtomicBool, Ordering};
2use std::sync::{Arc, Mutex};
3
4use rmcp::ErrorData;
5use rmcp::model::{ContentBlock, Tool};
6use serde_json::{Map, Value, json};
7
8use crate::server::tool_trait::{
9 McpTool, ToolContext, ToolOutput, get_bool, get_f64, get_int, get_str, get_str_array,
10 require_resolved_path,
11};
12use crate::tool_defs::tool_def;
13
14fn per_file_lock(path: &str) -> Arc<Mutex<()>> {
25 crate::core::path_locks::per_file_lock(path)
26}
27
28pub struct CtxReadTool;
29
30impl McpTool for CtxReadTool {
31 fn name(&self) -> &'static str {
32 "ctx_read"
33 }
34
35 fn tool_def(&self) -> Tool {
36 tool_def(
37 "ctx_read",
38 "Read source files. mode recommended — choose by intent (see `mode` below); defaults to auto when omitted.\n\
39 To UNDERSTAND code run ctx_compose FIRST; ctx_read after it identified files.\n\
40 anchored → edit by reference via ctx_patch (no exact-recall).",
41 json!({
42 "type": "object",
43 "properties": {
44 "path": { "type": "string", "description": "Absolute path" },
45 "paths": { "type": "array", "items": { "type": "string" }, "description": "Batch read" },
46 "mode": {
47 "type": "string",
48 "description": "Recommended (defaults to auto). full=verbatim(edit-ready) anchored=full+N:hh|anchors(edit via ctx_patch) raw=exact-bytes signatures=API map=structure auto=smart diff=git-delta lines:N-M=window (comma multi-selects: lines:5,10-20) reference=quotes task=focus"
49 },
50 "raw": { "type": "boolean", "description": "Verbatim (= mode=raw + fresh)" },
51 "start_line": { "type": "integer", "description": "1-based" },
52 "offset": { "type": "integer", "description": "start_line alias" },
53 "limit": { "type": "integer", "description": "Max lines" },
54 "fresh": { "type": "boolean", "description": "Bypass cache" },
55 "aggressiveness": { "type": "number", "description": "0.0–1.0 density (entropy/task)" },
56 "protect": { "type": "array", "items": { "type": "string" }, "description": "Symbols kept verbatim" }
57 },
58 "required": []
59 }),
60 )
61 }
62
63 fn handle(
64 &self,
65 args: &Map<String, Value>,
66 ctx: &ToolContext,
67 ) -> Result<ToolOutput, ErrorData> {
68 if args
71 .get("paths")
72 .and_then(|v| v.as_array())
73 .is_some_and(|a| !a.is_empty())
74 {
75 return super::ctx_multi_read::batch_read(args, ctx);
76 }
77
78 let path = if let Some(repo) = get_str(args, "repo") {
79 let root = crate::core::multi_repo::resolve_repo_root(&repo).ok_or_else(|| {
80 let known = crate::core::multi_repo::known_aliases().join(", ");
81 let known = if known.is_empty() {
82 "none registered — use ctx_multi_repo add_root".to_string()
83 } else {
84 known
85 };
86 ErrorData::invalid_params(
87 format!("unknown repo alias: {repo} (known: {known})"),
88 None,
89 )
90 })?;
91 let rel = get_str(args, "path").unwrap_or_else(|| ".".to_string());
92 crate::core::path_resolve::resolve_tool_path(Some(&root), None, &rel)
93 .map_err(|e| ErrorData::invalid_params(e, None))?
94 } else {
95 require_resolved_path(ctx, args, "path")?
96 };
97
98 match std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
99 self.handle_inner(args, ctx, &path)
100 })) {
101 Ok(result) => result,
102 Err(_) => Err(ErrorData::internal_error(
103 format!(
104 "ctx_read panicked while processing '{path}'. This is a bug — please report it."
105 ),
106 None,
107 )),
108 }
109 }
110}
111
112impl CtxReadTool {
113 #[allow(clippy::unused_self)]
114 fn handle_inner(
115 &self,
116 args: &Map<String, Value>,
117 ctx: &ToolContext,
118 path: &str,
119 ) -> Result<ToolOutput, ErrorData> {
120 let session_lock = ctx
121 .session
122 .as_ref()
123 .ok_or_else(|| ErrorData::internal_error("session not available", None))?;
124 let cache_lock = ctx
125 .cache
126 .as_ref()
127 .ok_or_else(|| ErrorData::internal_error("cache not available", None))?;
128
129 let current_task = {
130 let deadline = std::time::Instant::now() + std::time::Duration::from_secs(8);
131 let mut attempt = 0u32;
132 loop {
133 if let Ok(guard) = session_lock.clone().try_read_owned() {
136 break guard.task.as_ref().map(|t| t.description.clone());
137 }
138 attempt += 1;
139 if std::time::Instant::now() >= deadline {
140 tracing::warn!(
141 "session read-lock timeout after {attempt} attempts in ctx_read for {path}"
142 );
143 break None;
144 }
145 std::thread::sleep(std::time::Duration::from_millis(25));
146 }
147 };
148 let task_ref = current_task.as_deref();
149 let arg_raw = get_bool(args, "raw").unwrap_or(false);
154 let explicit_mode_arg = resolve_raw_alias(arg_raw, get_str(args, "mode"));
155 let explicit_mode = explicit_mode_arg.is_some();
156 if let Some(ref requested) = explicit_mode_arg
164 && (requested.starts_with("lines:") || requested.starts_with("anchored:"))
165 && let Err(e) = requested.parse::<crate::tools::ctx_read::ReadMode>()
166 {
167 return Err(ErrorData::invalid_params(e.user_message(), None));
168 }
169 let configured_mode = (!explicit_mode)
170 .then(crate::core::auto_mode_resolver::configured_default_mode)
171 .flatten();
172 let learned_mode = if !explicit_mode && configured_mode.is_none() {
173 if let Ok(cache) = cache_lock.try_read() {
174 Some(crate::tools::ctx_smart_read::select_mode_with_task(
175 &cache, path, task_ref,
176 ))
177 } else {
178 tracing::debug!(
179 "cache lock contested during auto-mode selection for {path}; \
180 falling back to full"
181 );
182 None
183 }
184 } else {
185 None
186 };
187 let mut mode = crate::core::auto_mode_resolver::resolve_mode_precedence(
188 explicit_mode_arg,
189 configured_mode,
190 learned_mode,
191 "full",
192 );
193 let mut fresh = get_bool(args, "fresh").unwrap_or(false);
194 if arg_raw {
197 fresh = true;
198 }
199 let cache_policy = crate::server::compaction_sync::effective_cache_policy();
200 if cache_policy == "off" {
201 fresh = true;
202 }
203 let aggressiveness =
204 crate::core::aggressiveness::effective(get_f64(args, "aggressiveness"));
205 let protect = get_str_array(args, "protect").unwrap_or_default();
206 if !explicit_mode && let Some(a) = aggressiveness {
210 mode = crate::tools::ctx_read::ReadMode::Density(
213 crate::core::aggressiveness::AggressivenessProfile::from_level(a).density_target,
214 )
215 .to_string();
216 }
217 apply_line_window(
222 &mut mode,
223 &mut fresh,
224 explicit_mode,
225 get_int(args, "start_line"),
226 get_int(args, "offset"),
227 get_int(args, "limit"),
228 );
229
230 let pressure_action = ctx.pressure_snapshot.as_ref().map(|p| &p.recommendation);
231 let resolved_agent_id = ctx.agent_id.as_ref().and_then(|a| match a.try_read() {
232 Ok(guard) => guard.clone(),
233 Err(_) => None,
234 });
235 let gate_result = crate::server::context_gate::pre_dispatch_read_for_agent(
236 path,
237 &mode,
238 task_ref,
239 Some(&ctx.project_root),
240 pressure_action,
241 resolved_agent_id.as_deref(),
242 );
243 if gate_result.budget_blocked {
244 let msg = gate_result
245 .budget_warning
246 .unwrap_or_else(|| "Agent token budget exceeded".to_string());
247 return Err(ErrorData::invalid_params(msg, None));
248 }
249 let budget_warning = gate_result.budget_warning.clone();
250 let mut mode_override_note: Option<String> = None;
253 if mode != "raw"
254 && let Some(overridden) = gate_result.overridden_mode
255 {
256 if explicit_mode {
257 let reason = gate_result.reason.unwrap_or("context-gate");
258 mode_override_note = Some(format!(
259 "[mode overridden: {mode} -> {overridden}, reason={reason}]"
260 ));
261 }
262 mode = overridden;
263 }
264
265 let (instruction_mode, instruction_mode_note) = resolve_instruction_file_mode(path, &mode);
266 let (mut mode, degrade_warning) =
267 if instruction_mode_note.is_some() || instruction_mode != mode {
268 (instruction_mode, None)
269 } else if mode == "raw" || mode.starts_with("anchored") || mode.starts_with("lines:") {
270 (mode, None)
274 } else {
275 auto_degrade_read_mode(&mode)
276 };
277
278 let mut delta_explicit_note: Option<String> = None;
289 if !fresh
290 && explicit_mode
291 && (mode == "full" || mode == "full-compact" || mode.starts_with("lines:"))
292 && crate::core::config::Config::load().delta_explicit_effective()
293 && let Ok(cache) = cache_lock.try_read()
294 {
295 let decision = crate::tools::ctx_read::resolve_explicit_delta_mode(
296 &cache,
297 path,
298 &mode,
299 explicit_mode,
300 fresh,
301 true,
302 );
303 mode = decision.mode;
304 delta_explicit_note = decision.note;
305 }
306
307 if mode.starts_with("lines:") {
308 fresh = true;
309 }
310
311 if crate::core::binary_detect::is_llm_viewable_image(path) {
312 return read_image_file(path);
313 }
314 if crate::core::binary_detect::is_binary_file(path) {
315 let msg = crate::core::binary_detect::binary_file_message(path);
316 return Err(ErrorData::invalid_params(msg, None));
317 }
318 {
319 let cap = crate::core::limits::max_read_bytes() as u64;
320 if let Ok(meta) = std::fs::metadata(path)
321 && meta.len() > cap
322 {
323 let msg = format!(
324 "File too large ({} bytes, limit {} bytes via LCTX_MAX_READ_BYTES). \
325 Use mode=\"lines:1-100\" or start_line+limit for partial reads, \
326 mode=\"anchored\" with start_line+limit for edit-ready windows, \
327 or increase the limit.",
328 meta.len(),
329 cap
330 );
331 return Err(ErrorData::invalid_params(msg, None));
332 }
333 }
334
335 if !fresh
338 && let Ok(data_dir) = crate::core::data_dir::lean_ctx_data_dir()
339 && let Ok(mut cache) = cache_lock.try_write()
340 {
341 crate::server::compaction_sync::sync_if_compacted(&mut cache, &data_dir);
342 }
343
344 let read_timeout = std::time::Duration::from_secs(30);
348 let cancelled = Arc::new(AtomicBool::new(false));
349 let delivery_metadata = crate::core::config::Config::load()
351 .ocla
352 .delivery_enabled()
353 .then(|| crate::tools::ctx_read::file_blake3_prefix(path))
354 .flatten();
355 let (output, resolved_mode, original, is_cache_hit, file_ref, cache_stats) = {
356 let crp_mode = ctx.crp_mode;
357 let fast_result = 'fast: {
358 let file_lock = per_file_lock(path);
359 let Some(_file_guard) = file_lock.try_lock().ok() else {
360 break 'fast None;
361 };
362
363 if !fresh
374 && (mode == "full" || mode == "full-compact" || mode == "auto")
375 && let Ok(cache) = cache_lock.try_read()
376 && let Some(read_output) =
377 crate::tools::ctx_read::try_stub_hit_readonly(&cache, path)
378 {
379 let hit = read_output.is_cache_hit;
380 let content = read_output.content;
381 let rmode = read_output.resolved_mode;
382 let orig = cache.get(path).map_or(0, |e| e.original_tokens);
383 let fref = cache.file_ref_map().get(path).cloned();
384 let stats = cache.get_stats();
385 let stats_snapshot = (stats.total_reads(), stats.cache_hits());
386 break 'fast Some((content, rmode, orig, hit, fref, stats_snapshot));
387 }
388
389 None
393 };
394
395 if let Some(result) = fast_result {
396 result
397 } else {
398 let cache_lock = cache_lock.clone();
399 let mode = mode.clone();
400 let task_owned = current_task.clone();
401 let protect_owned = protect.clone();
402 let path_owned = path.to_string();
403 let cancel_flag = cancelled.clone();
404 let (tx, rx) = std::sync::mpsc::sync_channel(1);
405 std::thread::spawn(move || {
406 let file_lock = per_file_lock(&path_owned);
407
408 let _file_guard = {
409 let deadline =
410 std::time::Instant::now() + std::time::Duration::from_secs(25);
411 loop {
412 if cancel_flag.load(Ordering::Relaxed) {
413 return;
414 }
415 if let Ok(guard) = file_lock.try_lock() {
416 break guard;
417 }
418 if std::time::Instant::now() >= deadline {
419 tracing::error!(
420 "ctx_read: per-file lock timeout after 25s for {path_owned}"
421 );
422 let _ = tx.send((
423 format!("per-file lock contention for {path_owned} — retry in a moment"),
424 "error".to_string(), 0, false, None, (0, 0),
425 ));
426 return;
427 }
428 std::thread::sleep(std::time::Duration::from_millis(50));
429 }
430 };
431
432 if cancel_flag.load(Ordering::Relaxed) {
433 return;
434 }
435
436 if !fresh
443 && (mode == "full" || mode == "full-compact" || mode == "auto")
444 && let Ok(cache) = cache_lock.try_read()
445 && let Some(read_output) =
446 crate::tools::ctx_read::try_stub_hit_readonly(&cache, &path_owned)
447 {
448 let content = read_output.content;
449 let rmode = read_output.resolved_mode;
450 let orig = cache.get(&path_owned).map_or(0, |e| e.original_tokens);
451 let hit = true;
452 let fref = cache.file_ref_map().get(path_owned.as_str()).cloned();
453 let stats = cache.get_stats();
454 let stats_snapshot = (stats.total_reads(), stats.cache_hits());
455 let _ = tx.send((content, rmode, orig, hit, fref, stats_snapshot));
456 return;
457 }
458
459 if !crate::tools::ctx_read::effective_fresh_for_delivery(fresh)
463 && let Some((hash, mtime)) = delivery_metadata
464 && let Some(read_output) = crate::tools::ctx_read::try_cross_agent_stub(
465 &path_owned,
466 &mode,
467 hash,
468 mtime,
469 )
470 {
471 let _ = tx.send((
472 read_output.content,
473 read_output.resolved_mode,
474 0,
475 read_output.is_cache_hit,
476 None,
477 (0, 0),
478 ));
479 return;
480 }
481
482 let preread = crate::tools::ctx_read::read_file_lossy(&path_owned).ok();
484
485 if cancel_flag.load(Ordering::Relaxed) {
486 return;
487 }
488
489 let task_ref = task_owned.as_deref();
499 let tuning =
500 crate::tools::ctx_read::ReadTuning::resolve(aggressiveness, &protect_owned);
501
502 macro_rules! acquire_write {
504 ($deadline_secs:expr, $label:expr) => {{
505 let deadline = std::time::Instant::now()
506 + std::time::Duration::from_secs($deadline_secs);
507 loop {
508 if cancel_flag.load(Ordering::Relaxed) {
509 return;
510 }
511 if let Ok(guard) = cache_lock.try_write() {
512 break guard;
513 }
514 if std::time::Instant::now() >= deadline {
515 tracing::error!(
516 "ctx_read: cache write-lock timeout ({}) for {path_owned}",
517 $label,
518 );
519 let _ = tx.send((
520 format!(
521 "cache lock contention for {path_owned} — retry in a moment"
522 ),
523 "error".into(),
524 0,
525 false,
526 None,
527 (0, 0),
528 ));
529 return;
530 }
531 std::thread::sleep(std::time::Duration::from_millis(50));
532 }
533 }};
534 }
535
536 #[allow(clippy::large_enum_variant)]
540 enum PrepareOutcome {
541 Hit(String, String, usize, bool, Option<String>, (u64, u64)),
542 Compute {
543 file_ref: String,
544 resolved_mode: String,
545 content: String,
546 original_tokens: usize,
547 },
548 }
549
550 let outcome = {
551 let mut cache = acquire_write!(10, "prepare 10s");
552
553 if crate::core::plugins::PluginManager::has_listener("pre_read") {
554 crate::core::plugins::PluginManager::fire_hook_background(
555 crate::core::plugins::executor::HookPoint::PreRead {
556 path: path_owned.clone(),
557 },
558 );
559 }
560 if let Ok(mut bt) = crate::core::bounce_tracker::global().lock() {
561 bt.next_seq();
562 }
563
564 let file_ref = cache.get_file_ref(&path_owned);
565
566 let effective_fresh = fresh
567 || crate::tools::ctx_read::force_fresh_env()
568 || (crate::tools::ctx_read::is_subagent_context()
569 && !crate::core::conversation::scope_enabled());
570
571 let mode_eff = if mode != "raw"
572 && !mode.starts_with("lines:")
573 && crate::core::config::Config::load()
574 .proxy
575 .is_path_compress_protected(&path_owned)
576 {
577 "full".to_string()
578 } else {
579 mode.clone()
580 };
581
582 if effective_fresh {
583 cache.invalidate(&path_owned);
584 }
585
586 if !effective_fresh {
587 let stale = cache.get(&path_owned).is_some_and(|e| {
588 crate::core::cache::is_cache_entry_stale_verified(
589 &path_owned,
590 e.stored_mtime,
591 &e.hash,
592 )
593 });
594 if stale {
595 cache.invalidate(&path_owned);
596 }
597 }
598
599 let snap = cache
600 .get(&path_owned)
601 .map(|e| (e.original_tokens, e.content()));
602
603 if let Some((orig_tok, content_opt)) = snap {
604 let resolved = if mode_eff == "auto" {
605 tuning.auto_density_mode().unwrap_or_else(|| {
606 crate::tools::ctx_read::resolve_auto_mode(
607 Some(&cache),
608 &path_owned,
609 orig_tok,
610 None,
611 task_ref,
612 )
613 })
614 } else {
615 mode_eff
616 };
617
618 if (resolved == "full" || resolved == "full-compact")
619 && let Some(out) = crate::tools::ctx_read::try_stub_hit_readonly(
620 &cache,
621 &path_owned,
622 )
623 {
624 let orig = cache.get(&path_owned).map_or(0, |e| e.original_tokens);
625 let fref = cache.file_ref_map().get(path_owned.as_str()).cloned();
626 let s = cache.get_stats();
627 PrepareOutcome::Hit(
628 out.content,
629 out.resolved_mode,
630 orig,
631 true,
632 fref,
633 (s.total_reads(), s.cache_hits()),
634 )
635 } else if crate::tools::ctx_read::is_cacheable_mode(&resolved) {
636 let ck = crate::tools::ctx_read::compressed_cache_key(
637 &resolved,
638 crp_mode,
639 task_ref,
640 tuning.aggressiveness,
641 tuning.protect,
642 );
643 if let Some(hit) = cache.get_compressed(&path_owned, &ck).cloned() {
644 crate::core::auto_mode_resolver::count_source(
645 "compressed_cache_hit",
646 );
647 let hit = crate::core::redaction::redact_text_if_enabled(&hit);
648 let orig =
649 cache.get(&path_owned).map_or(0, |e| e.original_tokens);
650 let fref =
651 cache.file_ref_map().get(path_owned.as_str()).cloned();
652 let s = cache.get_stats();
653 PrepareOutcome::Hit(
654 hit,
655 resolved,
656 orig,
657 true,
658 fref,
659 (s.total_reads(), s.cache_hits()),
660 )
661 } else {
662 let c = content_opt
663 .or_else(|| preread.as_deref().map(String::from));
664 PrepareOutcome::Compute {
665 file_ref,
666 resolved_mode: resolved,
667 content: c.unwrap_or_default(),
668 original_tokens: orig_tok,
669 }
670 }
671 } else {
672 let c =
673 content_opt.or_else(|| preread.as_deref().map(String::from));
674 PrepareOutcome::Compute {
675 file_ref,
676 resolved_mode: resolved,
677 content: c.unwrap_or_default(),
678 original_tokens: orig_tok,
679 }
680 }
681 } else {
682 let raw = preread.unwrap_or_else(|| {
683 crate::tools::ctx_read::read_file_lossy(&path_owned)
684 .unwrap_or_default()
685 });
686 let sr = cache.store(&path_owned, &raw);
687 let resolved = if mode_eff == "auto" {
688 tuning.auto_density_mode().unwrap_or_else(|| {
689 crate::tools::ctx_read::resolve_auto_mode(
690 None,
691 &path_owned,
692 sr.original_tokens,
693 Some(sr.line_count),
694 task_ref,
695 )
696 })
697 } else {
698 mode_eff
699 };
700 PrepareOutcome::Compute {
701 file_ref,
702 resolved_mode: resolved,
703 content: raw,
704 original_tokens: sr.original_tokens,
705 }
706 }
707 }; if let PrepareOutcome::Hit(c, rm, orig, hit, fref, ss) = outcome {
710 let mut cache = acquire_write!(10, "hit last_mode 10s");
713 if let Some(entry) = cache.get_mut(&path_owned) {
714 entry.last_mode.clone_from(&rm);
715 }
716 let _ = tx.send((c, rm, orig, hit, fref, ss));
717 return;
718 }
719 let PrepareOutcome::Compute {
720 file_ref,
721 resolved_mode,
722 content: compute_content,
723 original_tokens,
724 } = outcome
725 else {
726 unreachable!()
727 };
728
729 if cancel_flag.load(Ordering::Relaxed) {
730 return;
731 }
732
733 let short = crate::core::protocol::shorten_path(&path_owned);
738 let ext_s = std::path::Path::new(&*path_owned)
739 .extension()
740 .and_then(|e| e.to_str())
741 .unwrap_or("");
742
743 let (mut computed, rmode) = if resolved_mode == "full"
744 || resolved_mode == "full-compact"
745 {
746 if resolved_mode == "full-compact" {
747 let (out, _) = crate::tools::ctx_read::format_full_compact_output(
748 &compute_content,
749 );
750 (out, "full-compact".to_string())
751 } else {
752 let lc = compute_content.lines().count();
753 let (out, _) = crate::tools::ctx_read::format_full_output(
754 &file_ref,
755 &short,
756 ext_s,
757 &compute_content,
758 original_tokens,
759 lc,
760 task_ref,
761 );
762 let ft = crate::core::tokens::count_tokens(&out);
763 let out = crate::tools::ctx_read::cap_to_raw(
764 out,
765 ft,
766 &compute_content,
767 original_tokens,
768 );
769 (out, "full".to_string())
770 }
771 } else {
772 let (out, _) = crate::tools::ctx_read::process_mode_tuned(
773 &compute_content,
774 &resolved_mode,
775 &file_ref,
776 &short,
777 ext_s,
778 original_tokens,
779 crp_mode,
780 &path_owned,
781 task_ref,
782 tuning,
783 );
784 let out = if crate::tools::ctx_read::mode_allows_raw_cap(&resolved_mode) {
785 let ft = crate::core::tokens::count_tokens(&out);
786 crate::tools::ctx_read::cap_to_raw(
787 out,
788 ft,
789 &compute_content,
790 original_tokens,
791 )
792 } else {
793 out
794 };
795 (out, resolved_mode)
796 };
797
798 computed = crate::core::redaction::redact_text_if_enabled(&computed);
799
800 if cancel_flag.load(Ordering::Relaxed) {
801 return;
802 }
803
804 {
809 let deadline =
810 std::time::Instant::now() + std::time::Duration::from_secs(5);
811 let cache_guard = loop {
812 if cancel_flag.load(Ordering::Relaxed) {
813 return;
814 }
815 if let Ok(g) = cache_lock.try_write() {
816 break Some(g);
817 }
818 if std::time::Instant::now() >= deadline {
819 tracing::warn!(
820 "ctx_read: store-lock timeout (5s) for {path_owned}, returning without caching"
821 );
822 break None;
823 }
824 std::thread::sleep(std::time::Duration::from_millis(50));
825 };
826
827 if let Some(mut cache) = cache_guard {
828 if crate::tools::ctx_read::is_cacheable_mode(&rmode) {
829 let ck = crate::tools::ctx_read::compressed_cache_key(
830 &rmode,
831 crp_mode,
832 task_ref,
833 tuning.aggressiveness,
834 tuning.protect,
835 );
836 cache.set_compressed(&path_owned, &ck, computed.clone());
837 }
838 if rmode == "full" || rmode == "full-compact" {
839 cache.mark_full_delivered(&path_owned);
840 }
841 if let Some(entry) = cache.get_mut(&path_owned) {
842 entry.last_mode.clone_from(&rmode);
843 }
844 if let Ok(mut bt) = crate::core::bounce_tracker::global().lock() {
845 bt.record_read(
846 &path_owned,
847 &rmode,
848 crate::core::tokens::count_tokens(&computed),
849 original_tokens,
850 );
851 }
852 let orig = cache.get(&path_owned).map_or(0, |e| e.original_tokens);
853 let fref = cache.file_ref_map().get(path_owned.as_str()).cloned();
854 let s = cache.get_stats();
855 let _ = tx.send((
856 computed,
857 rmode,
858 orig,
859 false,
860 fref,
861 (s.total_reads(), s.cache_hits()),
862 ));
863 } else {
864 let _ =
865 tx.send((computed, rmode, original_tokens, false, None, (0, 0)));
866 }
867 }
868 });
869 if let Ok(result) = rx.recv_timeout(read_timeout) {
870 result
871 } else {
872 cancelled.store(true, Ordering::Relaxed);
873 tracing::error!("ctx_read timed out after {read_timeout:?} for {path}");
874 let msg = format!(
875 "ERROR: ctx_read timed out after {}s reading {path}. \
876 The file may be very large or a blocking I/O issue occurred. \
877 Try mode=\"lines:1-100\" for a partial read.",
878 read_timeout.as_secs()
879 );
880 return Err(ErrorData::internal_error(msg, None));
881 }
882 } };
884
885 if resolved_mode == "error" {
886 return Err(ErrorData::invalid_params(output, None));
887 }
888
889 let output_tokens = crate::core::tokens::count_tokens(&output);
890 let saved = original.saturating_sub(output_tokens);
891
892 if !is_cache_hit {
893 if let Some((hash, mtime)) = delivery_metadata {
894 crate::tools::ctx_read::record_cross_agent_delivery(
895 path,
896 hash,
897 mtime,
898 0,
899 output_tokens,
900 None,
901 None,
902 );
903 }
904 }
905
906 let mut ensured_root: Option<String> = None;
908 let mut traversal_working_set: Vec<String> = Vec::new();
909 let project_root_snapshot;
910 {
911 let deadline = std::time::Instant::now() + std::time::Duration::from_secs(10);
912 let session_guard = loop {
913 if let Ok(g) = session_lock.clone().try_write_owned() {
914 break Some(g);
915 }
916 if std::time::Instant::now() >= deadline {
917 break None;
918 }
919 std::thread::sleep(std::time::Duration::from_millis(25));
920 };
921 if let Some(mut session) = session_guard {
922 session.touch_file(path, file_ref.as_deref(), &resolved_mode, original);
923 traversal_working_set =
926 crate::core::tool_lifecycle::recent_working_set(&session, path);
927 let file_summary = extract_file_summary(&output, path);
928 if !file_summary.is_empty() {
929 session.set_file_summary(path, &file_summary);
930 }
931 if is_cache_hit {
932 session.record_cache_hit();
933 }
934 if session.active_structured_intent.is_none() && session.files_touched.len() >= 2 {
935 let touched: Vec<String> = session
936 .files_touched
937 .iter()
938 .map(|f| f.path.clone())
939 .collect();
940 let inferred =
941 crate::core::intent_engine::StructuredIntent::from_file_patterns(&touched);
942 if inferred.confidence >= 0.4 {
943 session.active_structured_intent = Some(inferred);
944 }
945 }
946 if session.task.is_none() && session.stats.files_read % 5 == 0 {
947 session.auto_infer_task();
948 }
949 let root_missing = session
950 .project_root
951 .as_deref()
952 .is_none_or(|r| r.trim().is_empty());
953 if root_missing && let Some(root) = crate::core::protocol::detect_project_root(path)
954 {
955 session.project_root = Some(root.clone());
956 ensured_root = Some(root);
957 }
958 project_root_snapshot = session
959 .project_root
960 .clone()
961 .unwrap_or_else(|| ".".to_string());
962 } else {
963 tracing::warn!(
964 "session write-lock timeout (5s) in ctx_read post-update for {path}"
965 );
966 project_root_snapshot = ctx.project_root.clone();
967 }
968 }
969 if let Some(root) = ensured_root.as_deref() {
970 crate::core::index_orchestrator::ensure_all_background(root);
971 }
972
973 {
979 let path_bg = path.to_string();
980 let resolved_mode_bg = resolved_mode.clone();
981 let project_root_bg = project_root_snapshot.clone();
982 let (turns, hits) = cache_stats;
983 let ledger_cache = (crate::core::savings_ledger::ledger_family()
989 != crate::core::tokens::TokenizerFamily::O200kBase)
990 .then(|| cache_lock.clone());
991 let ledger_output = ledger_cache.as_ref().map(|_| output.clone());
992 std::thread::spawn(move || {
993 let _ = std::panic::catch_unwind(std::panic::AssertUnwindSafe(move || {
996 crate::core::heatmap::record_file_access(&path_bg, original, saved);
997
998 {
1003 use crate::core::savings_ledger as ledger;
1004 let (lbase, lsaved) = match (&ledger_cache, &ledger_output) {
1005 (Some(cl), Some(out)) => match cl.try_read().ok().and_then(|c| {
1006 c.get(&path_bg)
1007 .and_then(crate::core::cache::CacheEntry::content)
1008 }) {
1009 Some(raw) => {
1010 let lo = ledger::count_for_ledger(&raw);
1011 (lo, lo.saturating_sub(ledger::count_for_ledger(out)))
1012 }
1013 None => (original, saved),
1014 },
1015 _ => (original, saved),
1016 };
1017 ledger::record_read_event(lbase, lsaved, None, None);
1018 }
1019
1020 if let Some(root) =
1023 crate::core::tool_lifecycle::usable_root(Some(project_root_bg.as_str()))
1024 {
1025 crate::core::cooccurrence::record_focus_access(
1026 root,
1027 &path_bg,
1028 &traversal_working_set,
1029 );
1030 }
1031 let sig =
1032 crate::core::mode_predictor::FileSignature::from_path(&path_bg, original);
1033 let density = if output_tokens > 0 {
1034 original as f64 / output_tokens as f64
1035 } else {
1036 1.0
1037 };
1038 let outcome = crate::core::mode_predictor::ModeOutcome {
1039 mode: resolved_mode_bg,
1040 tokens_in: original,
1041 tokens_out: output_tokens,
1042 density: density.min(1.0),
1043 };
1044 let mut predictor = crate::core::mode_predictor::ModePredictor::new();
1045 predictor.set_project_root(&project_root_bg);
1046 predictor.record(sig, outcome);
1047 predictor.save();
1048
1049 let ext = std::path::Path::new(&path_bg)
1050 .extension()
1051 .and_then(|e| e.to_str())
1052 .unwrap_or("")
1053 .to_string();
1054 let thresholds =
1055 crate::core::adaptive_thresholds::thresholds_for_path(&path_bg);
1056 let feedback_outcome = crate::core::feedback::CompressionOutcome {
1057 session_id: format!("{}", std::process::id()),
1058 language: ext,
1059 entropy_threshold: thresholds.bpe_entropy,
1060 jaccard_threshold: thresholds.jaccard,
1061 total_turns: turns as u32,
1062 tokens_saved: saved as u64,
1063 tokens_original: original as u64,
1064 cache_hits: hits as u32,
1065 total_reads: turns as u32,
1066 task_completed: crate::core::bounce_tracker::global()
1074 .lock()
1075 .ok()
1076 .and_then(|bt| bt.bounce_rate_for_extension(&path_bg))
1077 .is_none_or(|rate| rate < 0.30),
1078 timestamp: chrono::Local::now().to_rfc3339(),
1079 };
1080 let mut store = crate::core::feedback::FeedbackStore::load();
1081 store.project_root = Some(project_root_bg);
1082 store.record_outcome(feedback_outcome);
1083 }));
1084 });
1085 }
1086 if let Some(aid) = resolved_agent_id.as_deref() {
1087 crate::core::agent_budget::record_consumption(aid, output_tokens);
1088 }
1089
1090 let graph_hint = if !is_cache_hit
1094 && !resolved_mode.starts_with("lines:")
1095 && crate::core::profiles::active_profile()
1096 .output_hints
1097 .related_hint()
1098 {
1099 crate::tools::ctx_read::graph_related_hint(path)
1100 } else {
1101 None
1102 };
1103
1104 let hints_suffix = if crate::core::profiles::active_profile()
1108 .output_hints
1109 .cross_source_hint()
1110 {
1111 let graph_db =
1112 crate::core::property_graph::graph_dir(&ctx.project_root).join("graph.db");
1113 let graph = graph_db
1114 .exists()
1115 .then(|| crate::core::property_graph::CodeGraph::open(&ctx.project_root))
1116 .transpose()
1117 .ok()
1118 .flatten();
1119 graph.map_or_else(String::new, |graph| {
1120 let edges = graph.all_cross_source_edges();
1121 if edges.is_empty() {
1122 String::new()
1123 } else {
1124 let ranges = scoped_read_ranges(&resolved_mode);
1125 let relative_path =
1126 crate::core::graph_index::graph_relative_key(path, &ctx.project_root);
1127 let hints = crate::core::cross_source_hints::hints_for_file_matching(
1128 path,
1129 &edges,
1130 &ctx.project_root,
1131 |hint| {
1132 ranges.as_ref().is_none_or(|ranges| {
1133 hint_intersects_ranges(hint, ranges, &graph, &relative_path)
1134 })
1135 },
1136 );
1137 crate::core::cross_source_hints::format_hints(&hints)
1138 }
1139 })
1140 } else {
1141 String::new()
1142 };
1143
1144 let rules_suffix = {
1148 let client_id = ctx
1149 .client_name
1150 .as_ref()
1151 .map(|c| c.blocking_read().clone())
1152 .unwrap_or_default();
1153 crate::core::rule_discovery::rules_suffix_for_read(path, &ctx.project_root, &client_id)
1154 };
1155 let mut warnings = Vec::new();
1156 if let Some(ref w) = budget_warning {
1157 warnings.push(w.as_str());
1158 }
1159 if let Some(ref w) = degrade_warning {
1160 warnings.push(w.as_str());
1161 }
1162 if let Some(ref w) = delta_explicit_note {
1163 warnings.push(w.as_str());
1164 }
1165 if let Some(ref w) = mode_override_note {
1166 warnings.push(w.as_str());
1167 }
1168 if let Some(ref w) = instruction_mode_note {
1169 warnings.push(w.as_str());
1170 }
1171 let graph_suffix = graph_hint.map(|h| format!("\n{h}")).unwrap_or_default();
1172 let final_output = if !warnings.is_empty() {
1175 format!(
1176 "{}\n\n{output}{hints_suffix}{graph_suffix}{rules_suffix}",
1177 warnings.join("\n")
1178 )
1179 } else if hints_suffix.is_empty() && graph_suffix.is_empty() && rules_suffix.is_empty() {
1180 output
1181 } else {
1182 format!("{output}{hints_suffix}{graph_suffix}{rules_suffix}")
1183 };
1184 let final_output = if crate::core::profiles::active_profile()
1188 .output_hints
1189 .proactive_context()
1190 {
1191 let proactive_query = format!(
1192 "ctx_read path={path} mode={resolved_mode} task={}",
1193 task_ref.unwrap_or_default()
1194 );
1195 if let Some(block) =
1196 crate::core::relevance_tracker::proactive_context_for_path(&proactive_query, path)
1197 {
1198 format!("{final_output}{block}")
1199 } else {
1200 final_output
1201 }
1202 } else {
1203 final_output
1204 };
1205
1206 let final_tokens = crate::core::tokens::count_tokens(&final_output);
1213 let verified_saved = original.saturating_sub(final_tokens);
1214
1215 Ok(ToolOutput {
1216 text: final_output,
1217 original_tokens: original,
1218 saved_tokens: verified_saved,
1219 mode: Some(resolved_mode),
1220 path: Some(path.to_string()),
1221 changed: false,
1222 shell_outcome: None,
1223 content_blocks: None,
1224 })
1225 }
1226}
1227
1228fn resolve_line_window(
1235 start_line: Option<i64>,
1236 offset: Option<i64>,
1237 limit: Option<i64>,
1238) -> Option<(i64, Option<i64>)> {
1239 let start = start_line.or(offset).map(|v| v.max(1));
1240 let limit = limit.filter(|&l| l > 0);
1241 match (start, limit) {
1242 (Some(s), l) => Some((s, l)),
1243 (None, Some(_)) => Some((1, limit)),
1244 (None, None) => None,
1245 }
1246}
1247
1248fn lines_mode(start: i64, limit: Option<i64>) -> String {
1251 match limit {
1252 Some(l) => format!("lines:{start}-{}", start + l - 1),
1253 None => format!("lines:{start}-999999"),
1254 }
1255}
1256
1257fn anchored_lines_mode(start: i64, limit: Option<i64>) -> String {
1261 match limit {
1262 Some(l) => format!("anchored:{start}-{}", start + l - 1),
1263 None => format!("anchored:{start}-999999"),
1264 }
1265}
1266fn resolve_instruction_file_mode(path: &str, mode: &str) -> (String, Option<String>) {
1267 if !crate::tools::ctx_read::is_instruction_file(path)
1268 || matches!(mode, "full" | "raw" | "anchored")
1269 || mode.starts_with("anchored:")
1270 || mode.starts_with("lines:")
1271 {
1272 return (mode.to_string(), None);
1273 }
1274
1275 (
1276 "full".to_string(),
1277 Some(format!(
1278 "[mode overridden: {mode} -> full, reason=instruction file requires complete content]"
1279 )),
1280 )
1281}
1282
1283fn scoped_read_ranges(mode: &str) -> Option<Vec<crate::tools::ctx_read::mode::LineRange>> {
1284 use crate::tools::ctx_read::{ReadMode, mode::LineRange};
1285
1286 match mode.parse::<ReadMode>().ok()? {
1287 ReadMode::Lines(range) | ReadMode::Anchored(Some(range)) => Some(vec![range]),
1288 ReadMode::LinesMulti(payload) => Some(
1289 payload
1290 .split(',')
1291 .filter_map(|part| {
1292 let (start, end) = part.split_once('-').unwrap_or((part, part));
1293 Some(LineRange::new(start.parse().ok()?, end.parse().ok()?))
1294 })
1295 .collect(),
1296 ),
1297 _ => None,
1298 }
1299}
1300
1301fn hint_intersects_ranges(
1302 hint: &crate::core::cross_source_hints::CrossSourceHint,
1303 ranges: &[crate::tools::ctx_read::mode::LineRange],
1304 graph: &crate::core::property_graph::CodeGraph,
1305 relative_path: &str,
1306) -> bool {
1307 if hint.relation != "health_hotspot" {
1308 return false;
1309 }
1310 let Some((_, symbol)) = hint.source_uri.rsplit_once('#') else {
1311 return false;
1312 };
1313 let Ok(Some(node)) = graph.get_node_by_symbol(symbol, relative_path) else {
1314 return false;
1315 };
1316 let (Some(start), Some(end)) = (node.line_start, node.line_end) else {
1317 return false;
1318 };
1319 ranges
1320 .iter()
1321 .any(|range| start <= range.end as usize && end >= range.start as usize)
1322}
1323
1324fn apply_line_window(
1329 mode: &mut String,
1330 fresh: &mut bool,
1331 explicit_mode: bool,
1332 start_line: Option<i64>,
1333 offset: Option<i64>,
1334 limit: Option<i64>,
1335) {
1336 let preserve_explicit_window = explicit_mode
1337 && start_line.is_none()
1338 && offset.is_none()
1339 && limit.is_some_and(|value| value > 0)
1340 && matches!(
1341 mode.parse::<crate::tools::ctx_read::ReadMode>(),
1342 Ok(crate::tools::ctx_read::ReadMode::Lines(_)
1343 | crate::tools::ctx_read::ReadMode::Anchored(Some(_)))
1344 );
1345 if preserve_explicit_window {
1346 return;
1347 }
1348
1349 let Some((start, limit)) = resolve_line_window(start_line, offset, limit) else {
1350 return;
1351 };
1352 if start <= 1 && limit.is_none() {
1353 return;
1354 }
1355 *fresh = true;
1356 if mode == "anchored" {
1360 *mode = anchored_lines_mode(start, limit);
1361 } else {
1362 *mode = lines_mode(start, limit);
1363 }
1364}
1365
1366fn resolve_raw_alias(arg_raw: bool, mode_arg: Option<String>) -> Option<String> {
1373 if arg_raw {
1374 Some("raw".to_string())
1375 } else {
1376 mode_arg
1377 }
1378}
1379
1380fn apply_verdict(
1381 mode: &str,
1382 verdict: crate::core::degradation_policy::DegradationVerdictV1,
1383) -> (String, bool) {
1384 use crate::core::degradation_policy::DegradationVerdictV1;
1385 match verdict {
1386 DegradationVerdictV1::Ok => (mode.to_string(), false),
1387 DegradationVerdictV1::Warn => match mode {
1388 "full" => ("map".to_string(), true),
1389 other => (other.to_string(), false),
1390 },
1391 DegradationVerdictV1::Throttle => match mode {
1392 "full" | "map" => ("signatures".to_string(), true),
1393 other => (other.to_string(), false),
1394 },
1395 DegradationVerdictV1::Block => {
1396 if mode == "signatures" {
1397 ("signatures".to_string(), false)
1398 } else {
1399 ("signatures".to_string(), true)
1400 }
1401 }
1402 }
1403}
1404
1405fn auto_degrade_read_mode(mode: &str) -> (String, Option<String>) {
1406 if crate::core::config::Config::load().no_degrade_effective() {
1407 return (mode.to_string(), None);
1408 }
1409 let profile = crate::core::profiles::active_profile();
1410 if !profile.degradation.enforce_effective() {
1411 return (mode.to_string(), None);
1412 }
1413 let policy = crate::core::degradation_policy::evaluate_v1_for_tool("ctx_read", None);
1414 let (new_mode, degraded) = apply_verdict(mode, policy.decision.verdict);
1415 let warning = if degraded {
1416 Some(format!(
1417 "⚠ Context pressure: mode={mode} was downgraded to mode={new_mode} \
1418 (verdict: {:?}). Use start_line=1 to bypass, or run ctx_compress to free budget.",
1419 policy.decision.verdict
1420 ))
1421 } else {
1422 None
1423 };
1424 (new_mode, warning)
1425}
1426
1427fn extract_file_summary(output: &str, path: &str) -> String {
1428 let hint = crate::core::auto_findings::extract_content_hint(output);
1429 if !hint.is_empty() {
1430 return hint;
1431 }
1432 let ext = std::path::Path::new(path)
1433 .extension()
1434 .and_then(|e| e.to_str())
1435 .unwrap_or("");
1436 let line_count = output.lines().count();
1437 if line_count > 5 {
1438 format!("{ext} file, {line_count} lines")
1439 } else {
1440 String::new()
1441 }
1442}
1443
1444#[cfg(test)]
1446#[path = "ctx_read_inline_tests.rs"]
1447mod tests;
1448
1449fn read_image_file(path: &str) -> Result<ToolOutput, ErrorData> {
1454 use crate::core::binary_detect::{IMAGE_MAX_BYTES, image_mime_type};
1455 use base64::Engine;
1456
1457 let metadata = std::fs::metadata(path)
1458 .map_err(|e| ErrorData::invalid_params(format!("Cannot read image: {e}"), None))?;
1459
1460 if metadata.len() > IMAGE_MAX_BYTES {
1461 return Err(ErrorData::invalid_params(
1462 format!(
1463 "Image too large ({:.1} MB, limit {:.0} MB). Resize or use a smaller image.",
1464 metadata.len() as f64 / 1024.0 / 1024.0,
1465 IMAGE_MAX_BYTES as f64 / 1024.0 / 1024.0,
1466 ),
1467 None,
1468 ));
1469 }
1470
1471 let mime_type = image_mime_type(path)
1472 .ok_or_else(|| ErrorData::invalid_params("Unsupported image format".to_string(), None))?;
1473
1474 let bytes = std::fs::read(path)
1475 .map_err(|e| ErrorData::invalid_params(format!("Cannot read image: {e}"), None))?;
1476
1477 let base64_data = base64::prelude::BASE64_STANDARD.encode(&bytes);
1478 let short_name = std::path::Path::new(path)
1479 .file_name()
1480 .and_then(|n| n.to_str())
1481 .unwrap_or(path);
1482
1483 let text_block = ContentBlock::text(format!(
1484 "[Image: {} ({} KB, {})]",
1485 short_name,
1486 bytes.len() / 1024,
1487 mime_type
1488 ));
1489 let image_block = ContentBlock::image(base64_data, mime_type);
1490
1491 Ok(ToolOutput::image(
1492 vec![text_block, image_block],
1493 path.to_string(),
1494 ))
1495}
1496
1497#[cfg(test)]
1498#[path = "ctx_read_repo_param_tests.rs"]
1499mod repo_param_tests;