meta-ast 0.7.0

//! Data/flow node extraction dispatcher.
//!
//! Extracts `DataNode` (value-bearing entities) and `FlowEdge`
//! (def-use transitions) from parse trees.
//!
//! Each language implements its own extraction in its respective module
//! (e.g. `rust::extract_rust_dataflow`). This module dispatches via
//! exhaustive `match` on `LangId` — no trait objects.
//!
//! ## Current status
//!
//! | Language          | Status      |
//! |-------------------|-------------|
//! | Rust              | Implemented |
//! | Python            | Implemented |
//! | JavaScript        | Implemented |
//! | TypeScript        | Implemented |
//! | TSX               | Implemented |
//! | Ruby              | TODO        |
//! | Go                | TODO        |
//! | C/C++             | TODO        |
//!
//! ## Design
//!
//! Extraction follows a two-phase approach per file:
//! 1. **Definition capture**: tree-sitter queries identify `let` bindings,
//!    function parameters, and other binding constructs.
//! 2. **Usage matching**: identifier references in expression context are
//!    matched to the nearest preceding definition of the same name within
//!    the same function scope, creating `DefUse` flow edges.
//!
//! This is a conservative intra-procedural analysis. Inter-procedural
//! (cross-function) def-use and full type-based resolution are deferred.

use crate::language::LangId;
use crate::model::{DataNode, DataNodeId, FlowEdge, IdGenerator};

/// Extract data nodes and flow edges from a parse tree.
///
/// Language-specific dispatch. Returns empty vectors for languages
/// without dataflow extraction implemented yet.
pub fn extract_dataflow(
    lang: LangId,
    tree: &tree_sitter::Tree,
    source: &[u8],
    id_gen: &IdGenerator<DataNodeId>,
) -> (Vec<DataNode>, Vec<FlowEdge>) {
    match lang {
        LangId::Rust => crate::language::rust::extract_rust_dataflow(tree, source, id_gen),
        LangId::Python => crate::language::python::extract_python_dataflow(tree, source, id_gen),
        LangId::JavaScript => {
            crate::language::javascript::extract_javascript_dataflow(tree, source, id_gen)
        }
        LangId::TypeScript => {
            crate::language::typescript::extract_typescript_dataflow(tree, source, id_gen)
        }
        LangId::Tsx => crate::language::tsx::extract_tsx_dataflow(tree, source, id_gen),
        // TODO: Implement Go dataflow extraction
        LangId::Go => (Vec::new(), Vec::new()),
        // TODO: Implement C/C++ dataflow extraction
        LangId::C | LangId::Cpp => (Vec::new(), Vec::new()),
        // TODO: Implement Ruby dataflow extraction
        LangId::Ruby => (Vec::new(), Vec::new()),
    }
}

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

    #[test]
    fn unsupported_lang_returns_empty() {
        // Ruby is the canonical "unsupported" LangId in the dispatcher.
        let source = b"x = 1\n";
        let lang = LangId::Ruby;
        let tree = crate::parser::parse_tree(lang, source).unwrap();
        let id_gen = IdGenerator::new();
        let (nodes, edges) = extract_dataflow(lang, &tree, source, &id_gen);
        assert!(nodes.is_empty());
        assert!(edges.is_empty());
    }

    #[test]
    fn javascript_dispatch_returns_dataflow() {
        let source = b"function f() { let x = 1; return x; }\n";
        let lang = LangId::JavaScript;
        let tree = crate::parser::parse_tree(lang, source).unwrap();
        let id_gen = IdGenerator::new();
        let (nodes, edges) = extract_dataflow(lang, &tree, source, &id_gen);
        assert!(!nodes.is_empty(), "JS dispatch should produce data nodes");
        assert!(!edges.is_empty(), "JS dispatch should produce flow edges");
    }
}