neti 1.6.6

High-integrity code quality governance and transactional change management for AI-assisted development
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

//! AST Visitor for analysis. Extracts high-level structure (Scopes/Methods).

use super::scope::Scope;
use super::extract::RustExtractor; // Updated import
use crate::lang::Lang;
use std::collections::HashMap;
use tree_sitter::Node;

pub struct AstVisitor<'a> {
    source: &'a str,
    lang: Lang,
}

impl<'a> AstVisitor<'a> {
    #[must_use]
    pub fn new(source: &'a str, lang: Lang) -> Self {
        Self { source, lang }
    }

    /// Extracts all scopes (classes/structs/enums) from the AST.
    #[must_use]
    pub fn extract_scopes(&self, root: Node) -> HashMap<String, Scope> {
        let mut scopes = HashMap::new();
        if self.lang == Lang::Rust {
            RustExtractor::extract_scopes(self.source, root, &mut scopes);
        }
        scopes
    }
}