frigg 0.9.2

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
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

//! Symbol extraction, structural search, heuristic references, and graph registration.
//!
//! Coordinates tree-sitter extraction, structural inspection, and heuristic relation resolution
//! into symbol inventories the graph layer and retrieval projections can consume after indexing.

use std::collections::{BTreeMap, HashMap};
use std::fs;
use std::path::{Path, PathBuf};

use crate::domain::{FriggError, FriggResult};
use crate::graph::{HeuristicConfidence, SymbolGraph, SymbolNode};
use crate::languages::{
    SymbolLanguage, collect_blade_symbols_from_source, parser_for_path, symbol_from_node,
    tree_sitter_language_for_path,
};
use blake3::Hasher;
use serde::{Deserialize, Serialize};
use tree_sitter::{Node, Query, QueryCursor, StreamingIterator, Tree};

mod extraction;
mod heuristics;
mod inspection;
mod model;
mod spans;

pub use extraction::{
    STABLE_SYMBOL_ID_ALGORITHM_VERSION, extract_symbols_for_paths,
    extract_symbols_for_paths_with_root, extract_symbols_from_file, extract_symbols_from_source,
    extract_symbols_from_source_with_identity_path,
};
pub(crate) use extraction::{
    assign_repository_relative_symbol_identities, assign_symbol_identity_path,
    push_symbol_definition,
};
pub use heuristics::{HeuristicReferenceResolver, resolve_heuristic_references};
pub use inspection::{
    generated_follow_up_structural_at_location_in_source, inspect_syntax_tree_in_source,
    inspect_syntax_tree_with_follow_up_in_source, search_structural_grouped_in_source,
    search_structural_grouped_with_follow_up_in_source, search_structural_in_source,
    search_structural_with_follow_up_in_source,
};
#[cfg(test)]
pub use inspection::{
    generated_follow_up_structural_for_focus, generated_follow_up_structural_for_location_in_source,
};
pub use model::*;
pub(crate) use spans::{
    byte_offset_for_line_column, line_column_for_offset, source_span, source_span_from_offsets,
};

/// Registers extracted symbol definitions into the repository symbol graph.
pub fn register_symbol_definitions(
    graph: &mut SymbolGraph,
    repository_id: &str,
    symbols: &[SymbolDefinition],
) {
    graph.register_symbols(symbols.iter().map(|symbol| {
        SymbolNode::new(
            symbol.stable_id.clone(),
            repository_id.to_owned(),
            symbol.name.clone(),
            symbol.kind.as_str().to_owned(),
            symbol.path.to_string_lossy().into_owned(),
            symbol.line,
        )
    }));
}

/// Ranks how closely a symbol definition matches a navigation query string.
pub fn navigation_symbol_target_rank(symbol: &SymbolDefinition, symbol_query: &str) -> Option<u8> {
    if symbol.stable_id == symbol_query {
        return Some(0);
    }
    if symbol.name == symbol_query {
        return Some(1);
    }
    if symbol.name.eq_ignore_ascii_case(symbol_query) {
        return Some(2);
    }

    None
}