knot 1.6.2

Codebase Graph + Vector RAG Indexer for Java, TypeScript, JavaScript, Kotlin, Rust, Python, Groovy, C/C++, Build Systems, and HTML/CSS codebases
Documentation 
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

//! CSS and SCSS entity extraction and handling.
//!
//! Handles capture processing for CSS/SCSS specific entity kinds:
//! - CSS classes, IDs, variables
//! - SCSS variables, mixins, functions

use crate::models::EntityKind;
use tree_sitter::Node;

/// Process a CSS or SCSS capture and extract entity information.
///
/// Returns (name, kind, start_line) or None if the capture is not a named entity.
pub(crate) fn handle_css_capture(
    cap_name: &str,
    text: &str,
    node: Node<'_>,
) -> Option<(String, EntityKind, usize)> {
    let start_line = node.start_position().row + 1;

    match cap_name {
        "css.class" => {
            let mut clean_name = text.to_string();
            if clean_name.starts_with('.') {
                clean_name.remove(0);
            }
            Some((clean_name, EntityKind::CssClass, start_line))
        }
        "css.id" => {
            let mut clean_name = text.to_string();
            if clean_name.starts_with('#') {
                clean_name.remove(0);
            }
            Some((clean_name, EntityKind::CssId, start_line))
        }
        "css.variable" => {
            let mut clean_name = text.to_string();
            if clean_name.starts_with("--") {
                clean_name = clean_name[2..].to_string();
            }
            Some((clean_name, EntityKind::CssVariable, start_line))
        }
        "scss.mixin" => Some((text.to_string(), EntityKind::ScssMixin, start_line)),
        "scss.function" => Some((text.to_string(), EntityKind::ScssFunction, start_line)),
        "scss.variable" => {
            let mut clean_name = text.to_string();
            if clean_name.starts_with('$') {
                clean_name.remove(0);
            }
            Some((clean_name, EntityKind::ScssVariable, start_line))
        }
        _ => None,
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use tree_sitter::Parser;

    fn parse_css(source: &str) -> tree_sitter::Tree {
        let mut parser = Parser::new();
        parser
            .set_language(&tree_sitter_css::LANGUAGE.into())
            .expect("failed to load CSS grammar");
        parser.parse(source, None).expect("failed to parse CSS")
    }

    #[test]
    fn test_handle_css_class() {
        let tree = parse_css(".btn-primary {}");
        let node = tree.root_node();
        let result = handle_css_capture("css.class", ".btn-primary", node);

        assert!(result.is_some());
        let (name, kind, start_line) = result.unwrap();
        assert_eq!(name, "btn-primary");
        assert_eq!(kind, EntityKind::CssClass);
        assert_eq!(start_line, 1);
    }

    #[test]
    fn test_handle_css_id() {
        let tree = parse_css("#header {}");
        let node = tree.root_node();
        let result = handle_css_capture("css.id", "#header", node);

        assert!(result.is_some());
        let (name, kind, start_line) = result.unwrap();
        assert_eq!(name, "header");
        assert_eq!(kind, EntityKind::CssId);
        assert_eq!(start_line, 1);
    }

    #[test]
    fn test_handle_scss_mixin() {
        let tree = parse_css("@mixin flex-center {}");
        let node = tree.root_node();
        let result = handle_css_capture("scss.mixin", "flex-center", node);

        assert!(result.is_some());
        let (name, kind, start_line) = result.unwrap();
        assert_eq!(name, "flex-center");
        assert_eq!(kind, EntityKind::ScssMixin);
        assert_eq!(start_line, 1);
    }
}