Struct TreeSitterContextAnalyzer 

pub struct TreeSitterContextAnalyzer;

Tree-sitter based context analyzer for scope detection

This analyzer uses tree-sitter to parse code and detect scopes, available symbols, and expected types. It supports Rust, TypeScript, and Python.

Supported Languages

• Rust: Full support with scope detection and symbol extraction
• TypeScript/JavaScript: Full support with scope detection and symbol extraction
• Python: Full support with scope detection and symbol extraction

Example

use ricecoder_completion::context::TreeSitterContextAnalyzer;
use ricecoder_completion::types::Position;

let analyzer = TreeSitterContextAnalyzer;
let context = analyzer.analyze_context(
    "fn main() { let x = ",
    Position::new(0, 20),
    "rust",
).await?;

Implementations

impl TreeSitterContextAnalyzer

pub fn infer_variable_type( code: &str, position: Position, language: &str, ) -> Option<Type>

Infer variable type from assignments and declarations

pub fn infer_function_return_type( code: &str, position: Position, language: &str, ) -> Option<Type>

Infer function return type from signature

Trait Implementations

impl ContextAnalyzer for TreeSitterContextAnalyzer

fn analyze_context<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, code: &'life1 str, position: Position, language: &'life2 str, ) -> Pin<Box<dyn Future<Output = CompletionResult<CompletionContext>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Analyze the code context at the given position

fn get_available_symbols( &self, context: &CompletionContext, code: &str, ) -> Vec<Symbol>

Get available symbols in the given context

fn infer_expected_type(&self, context: &CompletionContext) -> Option<Type>

Infer the expected type at the given context