codetether-rlm 0.1.0

Recursive Language Model processing for CodeTether
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

use anyhow::Result;

use super::{FunctionSignature, TreeSitterOracle};

const FUNCTIONS_QUERY: &str = r#"
(function_item
    name: (identifier) @name
    parameters: (parameters) @params
    return_type: (_)? @return_type)
"#;

impl TreeSitterOracle {
    /// Get all function signatures in the source.
    pub fn get_functions(&mut self) -> Result<Vec<FunctionSignature>> {
        let result = self.query(FUNCTIONS_QUERY)?;
        Ok(result.matches.into_iter().map(function_signature).collect())
    }
}

fn function_signature(m: super::AstMatch) -> FunctionSignature {
    FunctionSignature {
        name: m.captures.get("name").cloned().unwrap_or_default(),
        params: m.captures.get("params").cloned().unwrap_or_default(),
        return_type: m.captures.get("return_type").cloned(),
        line: m.line,
    }
}