lean-ctx 3.8.16

use std::collections::HashMap;

use crate::core::cache::SessionCache;
use crate::tools::{CrpMode, ToolCallRecord};

pub fn handle(cache: &SessionCache, tool_calls: &[ToolCallRecord], crp_mode: CrpMode) -> String {
    let cache_stats = cache.get_stats();
    let refs = cache.file_ref_map();

    let total_original: u64 = tool_calls.iter().map(|c| c.original_tokens as u64).sum();
    let total_saved: u64 = tool_calls.iter().map(|c| c.saved_tokens as u64).sum();
    let total_sent = total_original.saturating_sub(total_saved);
    let pct = if total_original > 0 {
        total_saved as f64 / total_original as f64 * 100.0
    } else {
        0.0
    };

    let mut out = Vec::new();
    let env_model = std::env::var("LEAN_CTX_MODEL")
        .or_else(|_| std::env::var("LCTX_MODEL"))
        .ok();
    let pricing = crate::core::gain::model_pricing::ModelPricing::load();
    let quote = pricing.quote(env_model.as_deref());

    if crp_mode.is_tdd() {
        out.push("§metrics".to_string());
        out.push("═".repeat(40));

        out.push(format!(
            "files:{} reads:{} hits:{} ({:.0}%)",
            cache_stats.files_tracked(),
            cache_stats.total_reads(),
            cache_stats.cache_hits(),
            cache_stats.hit_rate()
        ));

        out.push(format!(
            "tok: {}→{} | saved:{} ({:.1}%)",
            format_tokens(total_original),
            format_tokens(total_sent),
            format_tokens(total_saved),
            pct
        ));

        let cost_saved = total_saved as f64 / 1_000_000.0 * quote.cost.input_per_m;
        let cost_without = total_original as f64 / 1_000_000.0 * quote.cost.input_per_m;
        out.push(format!(
            "cost: ${:.4}→${:.4} | -${:.4}",
            cost_without,
            cost_without - cost_saved,
            cost_saved
        ));
        if let Some(line) = cache_adjusted_line(&quote.cost, total_saved, true) {
            out.push(line);
        }
    } else {
        out.push("lean-ctx session metrics".to_string());
        out.push("═".repeat(50));

        out.push(format!(
            "Files tracked: {} | Reads: {} | Cache hits: {} ({:.0}%)",
            cache_stats.files_tracked(),
            cache_stats.total_reads(),
            cache_stats.cache_hits(),
            cache_stats.hit_rate()
        ));

        out.push(format!(
            "Input tokens:  {} original → {} sent | {} saved ({:.1}%)",
            format_tokens(total_original),
            format_tokens(total_sent),
            format_tokens(total_saved),
            pct
        ));

        let cost_saved = total_saved as f64 / 1_000_000.0 * quote.cost.input_per_m;
        let cost_without = total_original as f64 / 1_000_000.0 * quote.cost.input_per_m;
        let cost_with = total_sent as f64 / 1_000_000.0 * quote.cost.input_per_m;
        out.push(format!(
            "Cost estimate: ${cost_without:.4} without → ${cost_with:.4} with lean-ctx | ${cost_saved:.4} saved"
        ));
        if let Some(line) = cache_adjusted_line(&quote.cost, total_saved, false) {
            out.push(line);
        }
    }

    if let Ok(bt) = crate::core::bounce_tracker::global().lock() {
        let bounces = bt.total_bounces();
        let wasted = bt.total_wasted_tokens();
        if bounces > 0 {
            let adjusted = bt.adjusted_savings(total_saved as usize);
            out.push(String::new());
            if crp_mode.is_tdd() {
                out.push("§bounce".to_string());
            } else {
                out.push("Bounce Detection:".to_string());
            }
            out.push(format!(
                "  bounces: {bounces} | wasted: {} tok",
                format_tokens(wasted as u64)
            ));
            out.push(format!(
                "  adjusted savings: {} tok ({:.1}%)",
                format_tokens(adjusted.max(0) as u64),
                if total_original > 0 {
                    adjusted.max(0) as f64 / total_original as f64 * 100.0
                } else {
                    0.0
                }
            ));
        }
    }

    // Online-learned threshold deltas (#538): show which extensions have
    // shifted away from the static base table and by how much.
    let learned = crate::core::threshold_learning::report();
    if !learned.is_empty() {
        out.push(String::new());
        if crp_mode.is_tdd() {
            out.push("§learned-thresholds".to_string());
        } else {
            out.push("Learned Thresholds (quality loop):".to_string());
        }
        out.extend(learned);
    }

    // LITM placement calibration (#539): observed begin/end hit rates and the
    // calibrated begin-share per client profile.
    let litm_cal = crate::core::litm_calibration::report();
    if !litm_cal.is_empty() {
        out.push(String::new());
        if crp_mode.is_tdd() {
            out.push("§litm-calibration".to_string());
        } else {
            out.push("LITM Placement Calibration:".to_string());
        }
        out.extend(litm_cal);
    }

    // Learning efficacy (#549): proof the learning loops work — bounce-rate
    // trend, placement-hit movement, prevented duplicate work, playbook
    // survival. Capturing here keeps the daily snapshot ring fresh without
    // any timer.
    crate::core::efficacy::capture();
    let efficacy = crate::core::efficacy::report();
    if !efficacy.is_empty() {
        out.push(String::new());
        if crp_mode.is_tdd() {
            out.push("§learning-efficacy".to_string());
        } else {
            out.push("Learning Efficacy (is the adaptation working?):".to_string());
        }
        out.extend(efficacy);
    }

    // Embedding engine status (#551): semantic features are self-activating;
    // surface where the engine currently stands so "why no semantics?" is
    // never a mystery.
    out.push(String::new());
    out.push(format!(
        "{} {}",
        if crp_mode.is_tdd() {
            "§embeddings"
        } else {
            "Embedding engine:"
        },
        crate::tools::ctx_knowledge::embeddings::engine_status_line()
    ));

    if !tool_calls.is_empty() {
        out.push(String::new());

        let sep_w = if crp_mode.is_tdd() { 40 } else { 50 };
        if crp_mode.is_tdd() {
            out.push(format!(
                "{:<12} {:>4} {:>7} {:>7} {:>4}",
                "tool", "n", "orig", "saved", "%"
            ));
        } else {
            out.push("By Tool:".to_string());
            out.push(format!(
                "{:<14} {:>5}  {:>8}  {:>8}  {:>5}",
                "Tool", "Calls", "Original", "Saved", "Avg%"
            ));
        }
        out.push("─".repeat(sep_w));

        let mut by_tool: HashMap<&str, ToolStats> = HashMap::new();
        for call in tool_calls {
            let entry = by_tool.entry(&call.tool).or_default();
            entry.calls += 1;
            entry.original += call.original_tokens;
            entry.saved += call.saved_tokens;
        }

        let mut sorted: Vec<_> = by_tool
            .iter()
            .filter(|(_, ts)| ts.original > 0 || ts.saved > 0)
            .collect();
        sorted.sort_by_key(|x| std::cmp::Reverse(x.1.saved));

        for (tool, ts) in &sorted {
            let avg = if ts.original > 0 {
                ts.saved as f64 / ts.original as f64 * 100.0
            } else {
                0.0
            };
            if crp_mode.is_tdd() {
                out.push(format!(
                    "{:<12} {:>4} {:>7} {:>7} {:>3.0}%",
                    tool,
                    ts.calls,
                    format_tokens(ts.original as u64),
                    format_tokens(ts.saved as u64),
                    avg
                ));
            } else {
                out.push(format!(
                    "{:<14} {:>5}  {:>8}  {:>8}  {:>4.0}%",
                    tool,
                    ts.calls,
                    format_tokens(ts.original as u64),
                    format_tokens(ts.saved as u64),
                    avg
                ));
            }
        }

        let mut by_mode: HashMap<&str, ModeStats> = HashMap::new();
        for call in tool_calls {
            if let Some(ref mode) = call.mode {
                let entry = by_mode.entry(mode).or_default();
                entry.calls += 1;
                entry.saved += call.saved_tokens;
            }
        }

        if !by_mode.is_empty() {
            out.push(String::new());
            if crp_mode.is_tdd() {
                out.push(format!("{:<12} {:>4} {:>7}", "mode", "n", "saved"));
            } else {
                out.push("By Mode:".to_string());
                out.push(format!("{:<14} {:>5}  {:>8}", "Mode", "Calls", "Saved"));
            }
            out.push("─".repeat(if crp_mode.is_tdd() { 28 } else { 30 }));

            let mut sorted_modes: Vec<_> = by_mode.iter().collect();
            sorted_modes.sort_by_key(|x| std::cmp::Reverse(x.1.saved));

            for (mode, ms) in &sorted_modes {
                if crp_mode.is_tdd() {
                    out.push(format!(
                        "{:<12} {:>4} {:>7}",
                        mode,
                        ms.calls,
                        format_tokens(ms.saved as u64)
                    ));
                } else {
                    out.push(format!(
                        "{:<14} {:>5}  {:>8}",
                        mode,
                        ms.calls,
                        format_tokens(ms.saved as u64)
                    ));
                }
            }
        }
    }

    if !refs.is_empty() {
        out.push(String::new());
        if crp_mode.is_tdd() {
            out.push("§refs:".to_string());
        } else {
            out.push("File Refs:".to_string());
        }
        let mut ref_list: Vec<_> = refs.iter().collect();
        ref_list.sort_by_key(|(_, r)| (*r).clone());
        for (path, r) in &ref_list {
            let short = crate::core::protocol::shorten_path(path);
            if let Some(entry) = cache.get(path) {
                out.push(format!(
                    "  {r}={short} [{}L {}t r:{}]",
                    entry.line_count,
                    entry.original_tokens,
                    entry.read_count()
                ));
            } else {
                out.push(format!("  {r}={short}"));
            }
        }
    }

    let projected_session =
        total_saved as f64 / 1_000_000.0 * (quote.cost.input_per_m + quote.cost.output_per_m * 0.3);
    if projected_session > 0.001 {
        out.push(String::new());
        if crp_mode.is_tdd() {
            out.push(format!(
                "∴ session savings (incl. thinking): ${projected_session:.3}"
            ));
        } else {
            out.push(format!(
                "Projected session savings (incl. thinking): ${projected_session:.3}"
            ));
        }
    }

    let cep = compute_cep_compliance(cache, tool_calls);
    out.push(String::new());
    if crp_mode.is_tdd() {
        out.push("§CEP compliance".to_string());
    } else {
        out.push("CEP Compliance:".to_string());
    }
    out.push(format!(
        "  Cache utilization: {:.0}%  (hit rate for repeated files)",
        cep.cache_utilization * 100.0
    ));
    out.push(format!(
        "  Mode diversity:    {:.0}%  (using optimal modes per file)",
        cep.mode_diversity * 100.0
    ));
    out.push(format!(
        "  Compression rate:  {:.0}%  (overall token reduction)",
        cep.compression_rate * 100.0
    ));
    // Output-echo statistic (#501): how much of recent replies re-quoted
    // content that was already delivered into context.
    let echo_stats = crate::core::output_echo::load_stats();
    if !echo_stats.reports.is_empty() {
        out.push(format!(
            "  Output echo:       {:.0}%  (replies re-quoting delivered content, last {})",
            echo_stats.avg_ratio(50) * 100.0,
            echo_stats.reports.len()
        ));
    }
    out.push(format!(
        "  CEP Score:         {:.0}/100",
        cep.overall_score * 100.0
    ));

    let complexity = crate::core::adaptive::classify_from_context(cache);
    out.push(format!("  Task complexity:   {complexity:?}"));

    // Which signals decided auto-mode this session — makes the learning
    // loops (bounce memory, heatmap, predictor, policy) observable (#496).
    let sources = crate::core::auto_mode_resolver::source_counts();
    if !sources.is_empty() {
        let line = sources
            .iter()
            .take(6)
            .map(|(s, n)| format!("{s}={n}"))
            .collect::<Vec<_>>()
            .join(" ");
        out.push(format!("  Auto-mode sources: {line}"));
    }

    // Quality loop (#494): edit-failure rates per (ext × read-mode), risky
    // pairs currently penalized to full, and served read escalations.
    let eq = crate::core::edit_quality::metrics_snapshot();
    let eq_pairs = eq["pairs"].as_array().cloned().unwrap_or_default();
    let escalations = eq["escalations_served"].as_u64().unwrap_or(0);
    if !eq_pairs.is_empty() || escalations > 0 {
        out.push("\nEdit quality (compression-correlated):".to_string());
        for p in eq_pairs.iter().take(5) {
            let risky = if p["risky"].as_bool().unwrap_or(false) {
                "  [risky -> full]"
            } else {
                ""
            };
            out.push(format!(
                "  {}: {} fail / {} ok ({:.0}%){}",
                p["pair"].as_str().unwrap_or("?"),
                p["fails"].as_u64().unwrap_or(0),
                p["successes"].as_u64().unwrap_or(0),
                p["fail_rate"].as_f64().unwrap_or(0.0) * 100.0,
                risky
            ));
        }
        out.push(format!(
            "  Read escalations served after edit-fails: {escalations} (pending: {} full, {} anchored)",
            eq["pending_escalations"].as_u64().unwrap_or(0),
            eq["pending_anchored_escalations"].as_u64().unwrap_or(0)
        ));
    }

    out.join("\n")
}

/// Honest cache-adjusted economics (GL #573): with provider prompt caching,
/// context that *stays* in the prompt is re-billed at the cache-read rate on
/// every later turn, so tokens lean-ctx removed save the full input price once
/// and the cache-read price per repeat turn. Both figures come straight from
/// the pricing table — no invented hit rate. Returns `None` when the model has
/// no cache-read pricing or nothing was saved.
fn cache_adjusted_line(
    cost: &crate::core::gain::model_pricing::ModelCost,
    total_saved: u64,
    tdd: bool,
) -> Option<String> {
    if total_saved == 0 || cost.cache_read_per_m <= 0.0 || cost.input_per_m <= 0.0 {
        return None;
    }
    let per_repeat_turn = total_saved as f64 / 1_000_000.0 * cost.cache_read_per_m;
    let pct = cost.cache_read_per_m / cost.input_per_m * 100.0;
    Some(if tdd {
        format!(
            "cache-adj: repeat turns -${per_repeat_turn:.4}/turn (cache-read {pct:.0}% of input)"
        )
    } else {
        format!(
            "Cache-adjusted: each repeat turn re-bills retained context at cache-read rate ({pct:.0}% of input) — saved tokens avoid ${per_repeat_turn:.4} per repeat turn"
        )
    })
}

struct CepCompliance {
    cache_utilization: f64,
    mode_diversity: f64,
    compression_rate: f64,
    overall_score: f64,
}

fn compute_cep_compliance(cache: &SessionCache, tool_calls: &[ToolCallRecord]) -> CepCompliance {
    let stats = cache.get_stats();

    let cache_utilization = stats.hit_rate() / 100.0;

    let modes_used: std::collections::HashSet<&str> = tool_calls
        .iter()
        .filter_map(|c| c.mode.as_deref())
        .collect();
    let possible_modes = crate::core::budgets::READ_MODE_COUNT;
    let mode_diversity = (modes_used.len() as f64 / possible_modes).min(1.0);

    let total_original: u64 = tool_calls.iter().map(|c| c.original_tokens as u64).sum();
    let total_saved: u64 = tool_calls.iter().map(|c| c.saved_tokens as u64).sum();
    let compression_rate = if total_original > 0 {
        total_saved as f64 / total_original as f64
    } else {
        0.0
    };

    let overall_score = cache_utilization * 0.3 + mode_diversity * 0.2 + compression_rate * 0.5;

    CepCompliance {
        cache_utilization,
        mode_diversity,
        compression_rate,
        overall_score,
    }
}

fn format_tokens(n: u64) -> String {
    if n >= 1_000_000_000_000 {
        format!("{:.2}T", n as f64 / 1_000_000_000_000.0)
    } else if n >= 1_000_000_000 {
        format!("{:.2}B", n as f64 / 1_000_000_000.0)
    } else if n >= 1_000_000 {
        format!("{:.1}M", n as f64 / 1_000_000.0)
    } else if n >= 1_000 {
        format!("{:.1}K", n as f64 / 1_000.0)
    } else {
        format!("{n}")
    }
}

#[derive(Default)]
struct ToolStats {
    calls: u32,
    original: usize,
    saved: usize,
}

#[derive(Default)]
struct ModeStats {
    calls: u32,
    saved: usize,
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_cep_compliance_section_present_tdd() {
        let cache = SessionCache::new();
        let calls = vec![ToolCallRecord {
            tool: "ctx_read".to_string(),
            original_tokens: 1000,
            saved_tokens: 300,
            mode: Some("full".to_string()),
            duration_ms: 0,
            timestamp: String::new(),
        }];
        let output = handle(&cache, &calls, CrpMode::Tdd);
        assert!(
            output.contains("§CEP compliance"),
            "TDD output must contain CEP compliance section"
        );
        assert!(output.contains("Cache utilization:"));
        assert!(output.contains("Mode diversity:"));
        assert!(output.contains("Compression rate:"));
        assert!(output.contains("CEP Score:"));
        assert!(output.contains("Task complexity:"));
    }

    #[test]
    fn test_cep_compliance_section_present_normal() {
        let cache = SessionCache::new();
        let calls = vec![];
        let output = handle(&cache, &calls, CrpMode::Off);
        assert!(
            output.contains("CEP Compliance:"),
            "Normal output must contain CEP Compliance section"
        );
        assert!(output.contains("Task complexity:"));
    }

    #[test]
    fn test_cep_scores_zero_with_no_calls() {
        let cache = SessionCache::new();
        let calls = vec![];
        let output = handle(&cache, &calls, CrpMode::Tdd);
        assert!(output.contains("CEP Score:         0/100"));
        assert!(output.contains("Cache utilization: 0%"));
    }

    #[test]
    fn test_format_tokens_units() {
        assert_eq!(format_tokens(500), "500");
        assert_eq!(format_tokens(1500), "1.5K");
        assert_eq!(format_tokens(1_500_000), "1.5M");
    }
}