frigg 0.9.0

Frigg gives AI agents local, source-backed code search and navigation without sending whole repositories through every prompt.
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

//! Lua language adapter for tree-sitter symbol extraction.
//!
//! Registered through the language registry for indexer symbol corpora and MCP structural tools;
//! extension-gated parsing skips unrelated paths.

use std::path::Path;

use tree_sitter::Node;

use crate::indexer::SymbolKind;

use super::registry::{node_field_text, node_name_text};

pub(crate) fn is_lua_path(path: &Path) -> bool {
    path.extension()
        .and_then(|extension| extension.to_str())
        .is_some_and(|extension| extension.eq_ignore_ascii_case("lua"))
}

pub(super) fn symbol_from_node(source: &str, node: Node<'_>) -> Option<(SymbolKind, String)> {
    if node.kind() != "function_declaration" {
        return None;
    }

    let name_node = node.child_by_field_name("name")?;
    match name_node.kind() {
        "method_index_expression" => {
            node_field_text(name_node, source, "method").map(|name| (SymbolKind::Method, name))
        }
        "dot_index_expression" => {
            node_field_text(name_node, source, "field").map(|name| (SymbolKind::Function, name))
        }
        _ => node_name_text(name_node, source)
            .or_else(|| node_name_text(node, source))
            .map(|name| (SymbolKind::Function, name)),
    }
}