ast-bro 3.0.0

//! CLI handler functions called from `src/main.rs`. Each one mirrors a
//! single subcommand and returns the process exit code (0 success, 2
//! user error, 1 internal error).

use std::path::{Path, PathBuf};

use crate::deps::render;
use crate::deps::scc;
use crate::deps::traverse;
use crate::deps::DepGraph;
use crate::graph_cache;
use crate::project_root::{find_root_for, relative_posix, resolve_home, Marker};
use std::sync::Arc;

pub fn run_deps(
    file: &Path,
    depth: usize,
    include_external: bool,
    json: bool,
    pretty: bool,
    rebuild: bool,
) -> i32 {
    if file.is_dir() {
        eprintln!(
            "# note: `deps` expects a file, not a directory.\n  \
             Use `ast-bro graph <dir>` to visualize the full dependency graph."
        );
        return 2;
    }
    let root = match find_root_for_with_cache(file) {
        Ok(r) => r,
        Err(e) => {
            eprintln!("# note: {}", e);
            return 2;
        }
    };
    let unified = match load_unified(&root, rebuild) {
        Ok(g) => g,
        Err(e) => {
            eprintln!("# note: {}", e);
            return 1;
        }
    };
    let graph = unified.deps.clone();
    let canonical = match canonicalise_in_root(file, &graph) {
        Some(p) => p,
        None => {
            eprintln!(
                "# note: {} is not part of the dep graph (excluded by .gitignore or unsupported language?)",
                file.display()
            );
            return 2;
        }
    };
    let hits = traverse::forward(&graph, &canonical, depth.max(1));
    if json {
        println!(
            "{}",
            render::render_deps_json(&graph, &canonical, &hits, include_external, pretty)
        );
    } else {
        print!(
            "{}",
            render::render_deps_text(&graph, &canonical, &hits, include_external)
        );
    }
    0
}

#[allow(clippy::too_many_arguments)]
pub fn run_reverse_deps(
    file: &Path,
    depth: usize,
    limit: usize,
    tests: bool,
    exclude_tests: bool,
    json: bool,
    pretty: bool,
    rebuild: bool,
) -> i32 {
    if file.is_dir() {
        eprintln!(
            "# note: `reverse-deps` expects a file, not a directory.\n  \
             Use `ast-bro graph <dir>` to visualize the full dependency graph."
        );
        return 2;
    }
    let root = match find_root_for_with_cache(file) {
        Ok(r) => r,
        Err(e) => {
            eprintln!("# note: {}", e);
            return 2;
        }
    };
    let unified = match load_unified(&root, rebuild) {
        Ok(g) => g,
        Err(e) => {
            eprintln!("# note: {}", e);
            return 1;
        }
    };
    let graph = unified.deps.clone();
    let canonical = match canonicalise_in_root(file, &graph) {
        Some(p) => p,
        None => {
            eprintln!(
                "# note: {} is not part of the dep graph",
                file.display()
            );
            return 2;
        }
    };
    let hits = traverse::reverse(&graph, &canonical, depth.max(1), limit, |e| {
        if !tests && !exclude_tests {
            return true;
        }
        let is_test = crate::file_filter::is_test_file(&e.target, &root);
        if exclude_tests {
            !is_test
        } else {
            is_test
        }
    });
    if json {
        println!(
            "{}",
            render::render_reverse_deps_json(&graph, &canonical, &hits, pretty)
        );
    } else {
        print!(
            "{}",
            render::render_reverse_deps_text(&graph, &canonical, &hits)
        );
    }
    0
}

pub fn run_cycles(
    path: &Path,
    min_size: usize,
    json: bool,
    pretty: bool,
    rebuild: bool,
) -> i32 {
    let cwd = current_dir_or_dot();
    let (root, scope) = match resolve_dir_root_and_scope(path, &cwd) {
        Ok(rs) => rs,
        Err(e) => {
            eprintln!("# note: {}", e);
            return 2;
        }
    };
    let unified = match load_unified(&root, rebuild) {
        Ok(g) => g,
        Err(e) => {
            eprintln!("# note: {}", e);
            return 1;
        }
    };
    let graph = unified.deps.clone();
    // Detect cycles on the full cached graph, then keep only those whose
    // members are entirely within the user's requested scope.
    let mut cycles = scc::detect(&graph, min_size);
    if !scope.is_empty() {
        cycles.retain(|c| c.members.iter().all(|m| graph.in_scope(m, &scope)));
    }
    if json {
        println!(
            "{}",
            render::render_cycles_json(&graph, &cycles, pretty)
        );
    } else {
        print!("{}", render::render_cycles_text(&graph, &cycles));
    }
    if cycles.is_empty() {
        0
    } else {
        // Non-zero exit so this can be wired into CI gates.
        3
    }
}

pub fn run_graph(
    path: &Path,
    json: bool,
    include_external: bool,
    pretty: bool,
    rebuild: bool,
) -> i32 {
    // `graph` expects a directory (repo root), not a file.
    // Give a helpful hint if the user passes a file path.
    if path.is_file() {
        eprintln!(
            "# note: `graph` expects a directory, not a file.\n\
             For per-file dependency analysis, use:\n  \
             ast-bro deps <file>       # what this file imports\n  \
             ast-bro reverse-deps <file>  # who imports this file"
        );
        return 2;
    }
    let cwd = current_dir_or_dot();
    let (root, scope) = match resolve_dir_root_and_scope(path, &cwd) {
        Ok(rs) => rs,
        Err(e) => {
            eprintln!("# note: {}", e);
            return 2;
        }
    };
    let unified = match load_unified(&root, rebuild) {
        Ok(g) => g,
        Err(e) => {
            eprintln!("# note: {}", e);
            return 1;
        }
    };
    let full = unified.deps.clone();
    let graph = if scope.is_empty() {
        full
    } else {
        full.subgraph_for_scope(&scope)
    };
    if json {
        println!("{}", render::render_graph_json(&graph, include_external, pretty));
    } else {
        print!("{}", render::render_graph_text(&graph, include_external));
    }
    0
}

/// Resolve `(graph_root, scope)` for directory-arg subcommands (cycles,
/// graph). Walks up from `path` looking for an existing `.ast-bro/deps/`
/// (capped at `cwd`); falls back to building at `cwd` when `path` is under
/// `cwd`, else `path` itself. `scope` is `path` expressed relative to the
/// resolved root (POSIX, "" = whole root).
fn resolve_dir_root_and_scope(path: &Path, cwd: &Path) -> Result<(PathBuf, String), String> {
    if !path.exists() {
        return Err(format!("path not found: {}", path.display()));
    }
    let (home, _) = resolve_home(path, cwd, Marker::DepsCache);
    let scope = relative_posix(path, &home).unwrap_or_default();
    Ok((home, scope))
}

fn current_dir_or_dot() -> PathBuf {
    std::env::current_dir().unwrap_or_else(|_| PathBuf::from("."))
}

/// Prefer an existing `.ast-bro/deps/` cache between `file` and CWD.
/// Falls back to the manifest walk (`find_root_for`) when no cache exists,
/// preserving the historical behaviour for users running `deps` for the
/// first time.
fn find_root_for_with_cache(file: &Path) -> Result<PathBuf, String> {
    if !file.exists() {
        return Err(format!("file not found: {}", file.display()));
    }
    let cwd = current_dir_or_dot();
    let (home, found) = resolve_home(file, &cwd, Marker::DepsCache);
    if found {
        return Ok(home);
    }
    find_root_for(file)
}

/// Load (or build) the unified graph for `root`, going through the shared
/// in-memory `Arc` so repeated calls within one process — most importantly
/// `ast-bro mcp` — reuse the same parsed graph instead of re-deserialising.
fn load_unified(root: &Path, force_rebuild: bool) -> std::io::Result<Arc<crate::graph_cache::UnifiedGraph>> {
    if force_rebuild {
        graph_cache::shared::rebuild(root)
    } else {
        graph_cache::get_or_init(root)
    }
}

fn canonicalise_in_root(file: &Path, graph: &DepGraph) -> Option<PathBuf> {
    let abs = file.canonicalize().ok()?;
    if graph.forward.contains_key(&abs) {
        return Some(abs);
    }
    // Try matching by suffix — user may have passed a relative path.
    let target_str = abs.to_string_lossy();
    for known in graph.forward.keys() {
        if known.to_string_lossy() == target_str {
            return Some(known.clone());
        }
    }
    None
}