Method walk_tree

atomic-semantic/src/parsers/rust.rs:45–144

Walk the AST and extract entities.

(
        &self,
        node: &Node,
        source: &str,
        file_path: &str,
        entities: &mut Vec<Entity>,
        in_impl: Option<&str>,
    )

    /// Walk the AST and extract entities.
    fn walk_tree(
        &self,
        node: &Node,
        source: &str,
        file_path: &str,
        entities: &mut Vec<Entity>,
        in_impl: Option<&str>,
    ) {
        match node.kind() {
            "function_item" => {
                if let Some(entity) = self.extract_function(node, source, file_path, in_impl) {
                    entities.push(entity);
                }
            }
            "struct_item" => {
                if let Some(entity) = self.extract_struct(node, source, file_path) {
                    entities.push(entity);
                }
            }
            "enum_item" => {
                if let Some(entity) = self.extract_enum(node, source, file_path) {
                    entities.push(entity);
                }
            }
            "trait_item" => {
                if let Some(entity) = self.extract_trait(node, source, file_path) {
                    let trait_name = entity.name.clone();
                    entities.push(entity);

                    // Walk trait body for method signatures
                    if let Some(body) = node.child_by_field_name("body") {
                        let mut cursor = body.walk();
                        for child in body.children(&mut cursor) {
                            self.walk_tree(&child, source, file_path, entities, Some(&trait_name));
                        }
                    }
                    return;
                }
            }
            "impl_item" => {
                let impl_name = self.extract_impl_name(node, source);
                if let Some(ref name) = impl_name {
                    // Create an entity for the impl block itself
                    let line = node.start_position().row as u32 + 1;
                    let end_line = node.end_position().row as u32 + 1;
                    let sig = self.build_impl_signature(node, source);

                    let mut entity =
                        Entity::new(name.clone(), EntityKind::Module, file_path, line, end_line);
                    if let Some(s) = sig {
                        entity = entity.with_signature(s);
                    }
                    entities.push(entity);

                    // Walk impl body for methods
                    if let Some(body) = node.child_by_field_name("body") {
                        let mut cursor = body.walk();
                        for child in body.children(&mut cursor) {