sarah 0.1.9

Count tokens, lines and function sizes per file to help AI agents decide what to refactor
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
30
31

use tree_sitter::Node;

use super::lang::{Grammar, Lang};

/// One function definition in its parsed file: what every per-function metric walks.
#[derive(Clone, Copy)]
pub(super) struct Scope<'a> {
    pub lang: Lang,
    pub node: Node<'a>,
    pub source: &'a str,
}

impl<'a> Scope<'a> {
    pub(super) fn grammar(&self) -> &'static Grammar {
        self.lang.grammar()
    }

    pub(super) fn kind_of(&self, node: Node<'a>) -> &'a str {
        self.lang.kind_of(node, self.source)
    }

    pub(super) fn text(&self, node: Node) -> &'a str {
        self.source.get(node.byte_range()).unwrap_or_default()
    }

    /// `else if` continues a chain; it is not one level deeper.
    pub(super) fn is_else_if(child: Node, parent: Node) -> bool {
        child.kind().contains("if")
            && (parent.kind() == child.kind() || matches!(parent.kind(), "else_clause" | "else"))
    }
}