use tree_sitter::{Node, Range};

pub fn named_children<'a>(node: &'a Node) -> impl Iterator<Item = Node<'a>> {
    (0..node.child_count()).map(move |i| node.child(i).unwrap())
}

pub fn shrink_to_range<'a>(root_node: Node<'a>, range: &Range) -> Node<'a> {
    let mut node = root_node;
    'outer: loop {
        let parent = node;
        let mut cursor = parent.walk();
        for child in parent.children(&mut cursor) {
            if child.range().start_byte <= range.start_byte
                && range.end_byte <= child.range().end_byte
                && child.is_named()
            {
                node = child;
                continue 'outer;
            }
        }
        return highest_node_of_same_range(node);
    }
}

fn highest_node_of_same_range<'a>(current_node: Node<'a>) -> Node<'a> {
    let start_byte = current_node.start_byte();
    let end_byte = current_node.end_byte();
    let mut node = current_node;
    while let Some(parent) = node.parent().and_then(|parent| {
        if parent.start_byte() < start_byte || end_byte < parent.end_byte() {
            None
        } else {
            Some(parent)
        }
    }) {
        node = parent
    }
    node
}