code-graph-cli 3.0.3

pub mod event;
pub mod incremental;

use std::path::Path;
use std::sync::mpsc as std_mpsc;
use std::time::Duration;

use ignore::gitignore::{Gitignore, GitignoreBuilder};
use notify::RecursiveMode;
use notify_debouncer_mini::{DebounceEventResult, new_debouncer};

use event::WatchEvent;

/// Handle to a running watcher. Keeps the debouncer alive (dropping stops watching).
pub struct WatcherHandle {
    /// Keep alive: dropping the debouncer stops the OS watcher.
    _debouncer: notify_debouncer_mini::Debouncer<notify::RecommendedWatcher>,
    /// The bridge thread forwarding events from notify channel to classified events.
    _bridge_thread: std::thread::JoinHandle<()>,
}

/// File extensions we care about for incremental re-index.
const SOURCE_EXTENSIONS: &[&str] = &["ts", "tsx", "js", "jsx", "rs", "py", "go"];

/// File basenames that trigger a full re-index.
/// TypeScript/JS config files and Rust crate root files are all treated as full re-index triggers.
/// __init__.py is included because it defines package structure and re-exports — any change
/// requires re-resolving all Python imports referencing that package.
const FULL_REINDEX_FILES: &[&str] = &[
    "tsconfig.json",
    "package.json",
    "pnpm-workspace.yaml",
    "Cargo.toml",
    "lib.rs",
    "main.rs",
    "mod.rs",
    "__init__.py",
    "go.mod",
    "go.work",
];

/// Rust crate root / module-tree files (subset of FULL_REINDEX_FILES).
/// Any mod.rs change triggers full re-index because it changes module tree structure.
const CRATE_ROOT_FILES: &[&str] = &["Cargo.toml", "lib.rs", "main.rs", "mod.rs"];

/// Build a Gitignore matcher from the project root's .gitignore file.
/// This is the same source of truth used by `walker::walk_project` via `ignore::WalkBuilder`.
/// If no .gitignore exists, returns an empty matcher that matches nothing.
fn build_gitignore_matcher(project_root: &Path) -> Gitignore {
    let mut builder = GitignoreBuilder::new(project_root);
    let gitignore_path = project_root.join(".gitignore");
    if gitignore_path.exists() {
        let _ = builder.add(&gitignore_path);
    }
    // Also check nested .gitignore files are handled — the ignore crate's
    // GitignoreBuilder handles the root .gitignore. For nested dirs, the
    // walker handles them during walk. For the watcher, the root .gitignore
    // covers the vast majority of cases (dist/, build/, *.generated.ts, etc.).
    builder.build().unwrap_or_else(|_| Gitignore::empty())
}

/// Start a debounced file watcher on `watch_root`.
///
/// Returns a `WatcherHandle` (must be kept alive) and a std mpsc receiver
/// that yields classified `WatchEvent`s.
///
/// The watcher:
/// - Debounces at 75ms (within the locked 50-100ms range)
/// - Filters out node_modules and .code-graph paths (hardcoded)
/// - Filters out .gitignore'd paths (same rules as initial indexing)
/// - Classifies events into Modified/Deleted/ConfigChanged/CrateRootChanged
pub fn start_watcher(
    watch_root: &Path,
) -> anyhow::Result<(WatcherHandle, std_mpsc::Receiver<WatchEvent>)> {
    let (notify_tx, notify_rx) = std::sync::mpsc::channel::<DebounceEventResult>();

    // Create debounced watcher with 75ms debounce
    let mut debouncer = new_debouncer(Duration::from_millis(75), move |res| {
        let _ = notify_tx.send(res);
    })?;
    debouncer
        .watcher()
        .watch(watch_root, RecursiveMode::Recursive)?;

    // Build gitignore matcher — same rules as walker::walk_project
    let gitignore = build_gitignore_matcher(watch_root);

    // Channel for classified events
    let (event_tx, event_rx) = std_mpsc::channel::<WatchEvent>();

    // Bridge thread: receive from notify channel, classify, forward as WatchEvent
    let root = watch_root.to_path_buf();
    let bridge_thread = std::thread::spawn(move || {
        let mut consecutive_errors: usize = 0;
        while let Ok(result) = notify_rx.recv() {
            match result {
                Ok(events) => {
                    consecutive_errors = 0;
                    for debounced_event in events {
                        let path = debounced_event.path;
                        if let Some(watch_event) = classify_event(&path, &root, &gitignore)
                            && event_tx.send(watch_event).is_err()
                        {
                            return; // receiver dropped, shutdown
                        }
                    }
                }
                Err(err) => {
                    consecutive_errors += 1;
                    eprintln!("[watcher] error: {:?}", err);
                    if consecutive_errors >= 5 {
                        eprintln!(
                            "[watcher] warning: {} consecutive errors — watcher may be degraded",
                            consecutive_errors
                        );
                    }
                }
            }
        }
    });

    Ok((
        WatcherHandle {
            _debouncer: debouncer,
            _bridge_thread: bridge_thread,
        },
        event_rx,
    ))
}

/// Classify a filesystem event path into a WatchEvent, or None if it should be ignored.
///
/// Filtering order:
/// 1. Hardcoded exclusions: node_modules, .code-graph (always excluded)
/// 2. .gitignore rules via the `gitignore` matcher (same source of truth as initial indexing)
/// 3. Full-reindex trigger detection (FULL_REINDEX_FILES → ConfigChanged or CrateRootChanged)
/// 4. Source extension filter (.ts, .tsx, .js, .jsx, .rs)
/// 5. File existence check (Modified vs Deleted)
fn classify_event(path: &Path, _project_root: &Path, gitignore: &Gitignore) -> Option<WatchEvent> {
    // Filter: skip node_modules (hardcoded, regardless of .gitignore — per CONTEXT.md)
    if path.components().any(|c| c.as_os_str() == "node_modules") {
        return None;
    }
    // Filter: skip .code-graph directory (our own cache writes)
    if path.components().any(|c| c.as_os_str() == ".code-graph") {
        return None;
    }

    // Filter: skip paths matching .gitignore rules (CONTEXT.md locked decision:
    // "Watcher respects same .gitignore rules used during initial indexing")
    let is_dir = path.is_dir();
    if gitignore.matched(path, is_dir).is_ignore() {
        return None;
    }

    // Check if it's a full-reindex trigger file.
    // Rust crate roots (Cargo.toml, lib.rs, main.rs, mod.rs) emit CrateRootChanged.
    // TS/JS config files (tsconfig.json, package.json, pnpm-workspace.yaml) emit ConfigChanged.
    if let Some(file_name) = path.file_name().and_then(|n| n.to_str())
        && FULL_REINDEX_FILES.contains(&file_name)
    {
        if CRATE_ROOT_FILES.contains(&file_name) {
            return Some(WatchEvent::CrateRootChanged(path.to_path_buf()));
        } else {
            return Some(WatchEvent::ConfigChanged);
        }
    }

    // Check if it's a source file we care about
    let ext = path.extension().and_then(|e| e.to_str()).unwrap_or("");
    if !SOURCE_EXTENSIONS.contains(&ext) {
        return None;
    }

    // Classify based on file existence
    if path.exists() {
        // File exists — treat as Modified (notify-debouncer-mini doesn't distinguish
        // create vs modify; incremental pipeline handles both via remove-old + re-parse).
        Some(WatchEvent::Modified(path.to_path_buf()))
    } else {
        Some(WatchEvent::Deleted(path.to_path_buf()))
    }
}