lean_ctx/tools/

ctx_intent.rs

use std::path::Path;

use crate::core::cache::SessionCache;
use crate::core::tokens::count_tokens;
use crate::tools::CrpMode;

#[derive(Debug)]
enum Intent {
    FixBug { area: String },
    AddFeature { area: String },
    Refactor { area: String },
    Understand { area: String },
    Test { area: String },
    Config,
    Deploy,
    Unknown,
}

pub fn handle(
    cache: &mut SessionCache,
    query: &str,
    project_root: &str,
    crp_mode: CrpMode,
) -> String {
    let intent = classify_intent(query);
    let strategy = build_strategy(&intent, project_root);

    let mut result = Vec::new();
    result.push(format!("Intent: {:?}", intent));
    result.push(format!(
        "Strategy: {} files, modes: {}",
        strategy.len(),
        strategy
            .iter()
            .map(|(_, m)| m.as_str())
            .collect::<Vec<_>>()
            .join(", ")
    ));
    result.push(String::new());

    for (path, mode) in &strategy {
        if !Path::new(path).exists() {
            continue;
        }
        let file_result = crate::tools::ctx_read::handle(cache, path, mode, crp_mode);
        result.push(file_result);
        result.push("---".to_string());
    }

    let output = result.join("\n");
    let tokens = count_tokens(&output);
    format!("{output}\n\n[ctx_intent: {tokens} tok]")
}

fn classify_intent(query: &str) -> Intent {
    let q = query.to_lowercase();

    let area = extract_area(&q);

    if q.contains("fix")
        || q.contains("bug")
        || q.contains("error")
        || q.contains("broken")
        || q.contains("crash")
        || q.contains("fail")
    {
        return Intent::FixBug { area };
    }
    if q.contains("add")
        || q.contains("create")
        || q.contains("implement")
        || q.contains("new")
        || q.contains("feature")
    {
        return Intent::AddFeature { area };
    }
    if q.contains("refactor")
        || q.contains("clean")
        || q.contains("restructure")
        || q.contains("rename")
        || q.contains("move")
    {
        return Intent::Refactor { area };
    }
    if q.contains("understand")
        || q.contains("how")
        || q.contains("what")
        || q.contains("explain")
        || q.contains("where")
    {
        return Intent::Understand { area };
    }
    if q.contains("test") || q.contains("spec") || q.contains("coverage") {
        return Intent::Test { area };
    }
    if q.contains("config") || q.contains("setup") || q.contains("env") || q.contains("install") {
        return Intent::Config;
    }
    if q.contains("deploy") || q.contains("release") || q.contains("publish") || q.contains("ship")
    {
        return Intent::Deploy;
    }

    Intent::Unknown
}

fn extract_area(query: &str) -> String {
    let keywords: Vec<&str> = query
        .split_whitespace()
        .filter(|w| {
            w.len() > 3
                && !matches!(
                    *w,
                    "the"
                        | "this"
                        | "that"
                        | "with"
                        | "from"
                        | "into"
                        | "have"
                        | "please"
                        | "could"
                        | "would"
                        | "should"
                )
        })
        .collect();

    let file_refs: Vec<&&str> = keywords
        .iter()
        .filter(|w| w.contains('.') || w.contains('/') || w.contains('\\'))
        .collect();

    if let Some(path) = file_refs.first() {
        return path.to_string();
    }

    let code_terms: Vec<&&str> = keywords
        .iter()
        .filter(|w| {
            w.chars().any(|c| c.is_uppercase())
                || w.contains('_')
                || matches!(
                    **w,
                    "auth"
                        | "login"
                        | "api"
                        | "database"
                        | "db"
                        | "server"
                        | "client"
                        | "user"
                        | "admin"
                        | "router"
                        | "handler"
                        | "middleware"
                        | "controller"
                        | "model"
                        | "view"
                        | "component"
                        | "service"
                        | "repository"
                        | "cache"
                        | "queue"
                        | "worker"
                )
        })
        .collect();

    if let Some(term) = code_terms.first() {
        return term.to_string();
    }

    keywords.last().unwrap_or(&"").to_string()
}

fn build_strategy(intent: &Intent, root: &str) -> Vec<(String, String)> {
    let mut files = Vec::new();
    let graph = crate::core::graph_index::ProjectIndex::load(root);

    match intent {
        Intent::FixBug { area } => {
            if let Some(paths) = find_files_for_area(area, root) {
                for path in paths.iter().take(3) {
                    files.push((path.clone(), "full".to_string()));
                }
                if let Some(ref idx) = graph {
                    enrich_with_graph(&mut files, &paths, idx, root, "map", 5);
                }
                for path in paths.iter().skip(3).take(5) {
                    if !files.iter().any(|(f, _)| f == path) {
                        files.push((path.clone(), "map".to_string()));
                    }
                }
            }
            if let Some(test_files) = find_test_files(area, root) {
                for path in test_files.iter().take(2) {
                    if !files.iter().any(|(f, _)| f == path) {
                        files.push((path.clone(), "signatures".to_string()));
                    }
                }
            }
        }
        Intent::AddFeature { area } => {
            if let Some(paths) = find_files_for_area(area, root) {
                for path in paths.iter().take(2) {
                    files.push((path.clone(), "signatures".to_string()));
                }
                if let Some(ref idx) = graph {
                    enrich_with_graph(&mut files, &paths, idx, root, "map", 5);
                }
                for path in paths.iter().skip(2).take(5) {
                    if !files.iter().any(|(f, _)| f == path) {
                        files.push((path.clone(), "map".to_string()));
                    }
                }
            }
        }
        Intent::Refactor { area } => {
            if let Some(paths) = find_files_for_area(area, root) {
                for path in paths.iter().take(5) {
                    files.push((path.clone(), "full".to_string()));
                }
                if let Some(ref idx) = graph {
                    enrich_with_graph(&mut files, &paths, idx, root, "full", 5);
                }
            }
        }
        Intent::Understand { area } => {
            if let Some(paths) = find_files_for_area(area, root) {
                for path in &paths {
                    files.push((path.clone(), "map".to_string()));
                }
                if let Some(ref idx) = graph {
                    enrich_with_graph(&mut files, &paths, idx, root, "map", 8);
                }
            }
        }
        Intent::Test { area } => {
            if let Some(test_files) = find_test_files(area, root) {
                for path in test_files.iter().take(3) {
                    files.push((path.clone(), "full".to_string()));
                }
            }
            if let Some(src_files) = find_files_for_area(area, root) {
                for path in src_files.iter().take(3) {
                    if !files.iter().any(|(f, _)| f == path) {
                        files.push((path.clone(), "signatures".to_string()));
                    }
                }
            }
        }
        Intent::Config => {
            for name in &[
                "Cargo.toml",
                "package.json",
                "pyproject.toml",
                "go.mod",
                "tsconfig.json",
                "docker-compose.yml",
            ] {
                let path = format!("{root}/{name}");
                if Path::new(&path).exists() {
                    files.push((path, "full".to_string()));
                }
            }
        }
        Intent::Deploy => {
            for name in &[
                "Dockerfile",
                "docker-compose.yml",
                "Makefile",
                ".github/workflows",
            ] {
                let path = format!("{root}/{name}");
                if Path::new(&path).exists() {
                    files.push((path, "full".to_string()));
                }
            }
        }
        Intent::Unknown => {}
    }

    files
}

fn enrich_with_graph(
    files: &mut Vec<(String, String)>,
    seed_paths: &[String],
    index: &crate::core::graph_index::ProjectIndex,
    root: &str,
    mode: &str,
    max: usize,
) {
    let mut added = 0;
    for seed in seed_paths {
        let rel = seed
            .strip_prefix(root)
            .unwrap_or(seed)
            .trim_start_matches('/');

        for related in index.get_related(rel, 2) {
            if added >= max {
                return;
            }
            let abs = format!("{root}/{related}");
            if !files.iter().any(|(f, _)| *f == abs || *f == related) && Path::new(&abs).exists() {
                files.push((abs, mode.to_string()));
                added += 1;
            }
        }
    }
}

fn find_files_for_area(area: &str, root: &str) -> Option<Vec<String>> {
    let mut matches = Vec::new();
    let search_term = area.to_lowercase();

    ignore::WalkBuilder::new(root)
        .hidden(true)
        .git_ignore(true)
        .git_global(true)
        .git_exclude(true)
        .max_depth(Some(6))
        .build()
        .filter_map(|e| e.ok())
        .filter(|e| e.file_type().is_some_and(|ft| ft.is_file()))
        .filter(|e| {
            let name = e.file_name().to_string_lossy().to_lowercase();
            name.contains(&search_term)
                || e.path()
                    .to_string_lossy()
                    .to_lowercase()
                    .contains(&search_term)
        })
        .take(10)
        .for_each(|e| {
            let path = e.path().to_string_lossy().to_string();
            if !matches.contains(&path) {
                matches.push(path);
            }
        });

    if matches.is_empty() {
        None
    } else {
        Some(matches)
    }
}

fn find_test_files(area: &str, root: &str) -> Option<Vec<String>> {
    let search_term = area.to_lowercase();
    let mut matches = Vec::new();

    ignore::WalkBuilder::new(root)
        .hidden(true)
        .git_ignore(true)
        .git_global(true)
        .git_exclude(true)
        .max_depth(Some(6))
        .build()
        .filter_map(|e| e.ok())
        .filter(|e| e.file_type().is_some_and(|ft| ft.is_file()))
        .filter(|e| {
            let name = e.file_name().to_string_lossy().to_lowercase();
            (name.contains("test") || name.contains("spec"))
                && (name.contains(&search_term)
                    || e.path()
                        .to_string_lossy()
                        .to_lowercase()
                        .contains(&search_term))
        })
        .take(5)
        .for_each(|e| {
            matches.push(e.path().to_string_lossy().to_string());
        });

    if matches.is_empty() {
        None
    } else {
        Some(matches)
    }
}