Method walk_tree

atomic-semantic/src/parser.rs:545–628

Walk the AST and extract entities

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

    /// Walk the AST and extract entities
    fn walk_tree(
        &self,
        node: &Node,
        source: &str,
        file_path: &str,
        entities: &mut Vec<Entity>,
        in_export: bool,
    ) {
        // Check if this node is an export statement
        let is_export = node.kind() == "export_statement";
        let child_in_export = in_export || is_export;

        match node.kind() {
            // Function declarations
            "function_declaration" => {
                if let Some(entity) =
                    self.extract_function(node, source, file_path, child_in_export)
                {
                    entities.push(entity);
                }
            }

            // Arrow functions assigned to variables (const foo = () => {})
            "lexical_declaration" | "variable_declaration" => {
                entities.extend(self.extract_variable_declarations(
                    node,
                    source,
                    file_path,
                    child_in_export,
                ));
            }

            // Class declarations
            "class_declaration" => {
                if let Some(entity) = self.extract_class(node, source, file_path, child_in_export) {
                    entities.push(entity);
                }
                // Also extract methods from the class
                self.extract_class_members(node, source, file_path, entities);
            }

            // Interface declarations (TypeScript)
            "interface_declaration" => {
                if let Some(entity) =
                    self.extract_interface(node, source, file_path, child_in_export)
                {
                    entities.push(entity);
                }
            }

            // Type alias declarations (TypeScript)
            "type_alias_declaration" => {
                if let Some(entity) =
                    self.extract_type_alias(node, source, file_path, child_in_export)
                {
                    entities.push(entity);
                }
            }