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

use anyhow::Result;

use super::{StructDefinition, TreeSitterOracle, extract};

const STRUCTS_QUERY: &str = r#"
(struct_item
    name: (type_identifier) @name
    body: (field_declaration_list)? @body)
"#;

impl TreeSitterOracle {
    /// Get all struct definitions in the source.
    pub fn get_structs(&mut self) -> Result<Vec<StructDefinition>> {
        let result = self.query(STRUCTS_QUERY)?;
        result.matches.into_iter().map(struct_definition).collect()
    }
}

fn struct_definition(m: super::AstMatch) -> Result<StructDefinition> {
    let body = m.captures.get("body").cloned().unwrap_or_default();
    Ok(StructDefinition {
        name: m.captures.get("name").cloned().unwrap_or_default(),
        fields: extract::struct_fields(&body)?,
        line: m.line,
    })
}