context_builder/tree_sitter/
mod.rs

1//! Tree-sitter integration for intelligent code parsing.
2//!
3//! This module provides:
4//! - Signature extraction (function/class signatures without bodies)
5//! - Smart truncation (truncate at AST boundaries)
6//! - Structure extraction (imports, exports, symbol counts)
7//!
8//! Feature-gated: Only compiled when one of the tree-sitter-* features is enabled.
9
10#[cfg(feature = "tree-sitter-base")]
11pub mod language_support;
12
13#[cfg(feature = "tree-sitter-base")]
14pub mod signatures;
15
16#[cfg(feature = "tree-sitter-base")]
17pub mod structure;
18
19#[cfg(feature = "tree-sitter-base")]
20pub mod truncation;
21
22#[cfg(feature = "tree-sitter-base")]
23pub mod languages;
24
25#[cfg(feature = "tree-sitter-base")]
26use std::path::Path;
27
28#[cfg(feature = "tree-sitter-base")]
29pub use language_support::{CodeStructure, LanguageSupport, Signature, SignatureKind, Visibility};
30
31#[cfg(feature = "tree-sitter-base")]
32pub use signatures::extract_signatures;
33
34#[cfg(feature = "tree-sitter-base")]
35pub use structure::extract_structure;
36
37#[cfg(feature = "tree-sitter-base")]
38pub use truncation::find_truncation_point;
39
40/// Check if tree-sitter is available for a given file extension.
41#[cfg(feature = "tree-sitter-base")]
42pub fn is_supported_extension(ext: &str) -> bool {
43    languages::get_language_support(ext).is_some()
44}
45
46#[cfg(not(feature = "tree-sitter-base"))]
47pub fn is_supported_extension(_ext: &str) -> bool {
48    false
49}
50
51/// Extract file extension from a path.
52#[cfg(feature = "tree-sitter-base")]
53fn get_extension(path: &Path) -> Option<String> {
54    path.extension()
55        .and_then(|e| e.to_str())
56        .map(|s| s.to_lowercase())
57}
58
59/// Get language support for a file path.
60#[cfg(feature = "tree-sitter-base")]
61pub fn get_language_for_path(path: &Path) -> Option<&'static dyn LanguageSupport> {
62    let ext = get_extension(path)?;
63    languages::get_language_support(&ext)
64}
65
66/// Extract signatures from source code for a given file extension.
67#[cfg(feature = "tree-sitter-base")]
68pub fn extract_signatures_for_file(
69    source: &str,
70    ext: &str,
71    visibility_filter: Visibility,
72) -> Option<Vec<Signature>> {
73    let support = languages::get_language_support(ext)?;
74    Some(extract_signatures(source, support, visibility_filter))
75}
76
77/// Extract structure from source code for a given file extension.
78#[cfg(feature = "tree-sitter-base")]
79pub fn extract_structure_for_file(source: &str, ext: &str) -> Option<CodeStructure> {
80    let support = languages::get_language_support(ext)?;
81    Some(extract_structure(source, support))
82}
83
84/// Find a smart truncation point for a given file extension.
85#[cfg(feature = "tree-sitter-base")]
86pub fn find_smart_truncation_point(source: &str, max_bytes: usize, ext: &str) -> Option<usize> {
87    let support = languages::get_language_support(ext)?;
88    Some(find_truncation_point(source, max_bytes, support))
89}
90
91#[cfg(not(feature = "tree-sitter-base"))]
92pub fn extract_signatures_for_file(
93    _source: &str,
94    _ext: &str,
95    _visibility_filter: (),
96) -> Option<()> {
97    None
98}
99
100#[cfg(not(feature = "tree-sitter-base"))]
101pub fn extract_structure_for_file(_source: &str, _ext: &str) -> Option<()> {
102    None
103}
104
105#[cfg(not(feature = "tree-sitter-base"))]
106pub fn find_smart_truncation_point(_source: &str, _max_bytes: usize, _ext: &str) -> Option<usize> {
107    None
108}
109
110#[cfg(not(feature = "tree-sitter-base"))]
111pub fn get_language_for_path(_path: &std::path::Path) -> Option<()> {
112    None
113}
114
115#[cfg(test)]
116mod tests {
117    use super::*;
118
119    #[test]
120    #[cfg(feature = "tree-sitter-base")]
121    fn test_is_supported_extension() {
122        #[cfg(feature = "tree-sitter-rust")]
123        assert!(is_supported_extension("rs"));
124        #[cfg(feature = "tree-sitter-python")]
125        assert!(is_supported_extension("py"));
126        #[cfg(feature = "tree-sitter-js")]
127        assert!(is_supported_extension("js"));
128        assert!(!is_supported_extension("xyz"));
129    }
130
131    #[test]
132    #[cfg(not(feature = "tree-sitter-base"))]
133    fn test_no_tree_sitter_support() {
134        assert!(!is_supported_extension("rs"));
135        assert!(!is_supported_extension("py"));
136    }
137}
context_builder/tree_sitter/mod.rs

context_builder/tree_sitter/
mod.rs
