Method extract_function

atomic-semantic/src/parsers/swift.rs:203–236

Extract a function or method declaration.

(
        &self,
        node: &Node,
        source: &str,
        file_path: &str,
        in_type: Option<&str>,
    )

    /// Extract a function or method declaration.
    fn extract_function(
        &self,
        node: &Node,
        source: &str,
        file_path: &str,
        in_type: Option<&str>,
    ) -> Option<Entity> {
        // Try named field first, fall back to first `simple_identifier` child
        let name = if let Some(n) = node.child_by_field_name("name") {
            self.node_text(&n, source)
        } else {
            self.find_first_simple_identifier(node, source)?
        };

        let kind = if in_type.is_some() {
            EntityKind::Method
        } else {
            EntityKind::Function
        };

        let line = node.start_position().row as u32 + 1;
        let end_line = node.end_position().row as u32 + 1;

        let exported = self.is_exported(node, source);
        let signature = self.build_function_signature(node, source);

        let mut entity = Entity::new(name, kind, file_path, line, end_line);
        if let Some(sig) = signature {
            entity = entity.with_signature(sig);
        }
        entity.exported = exported;

        Some(entity)
    }

    /// Extract an `init` declaration.
    ///