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::{EnumDefinition, TreeSitterOracle, extract};

const ENUMS_QUERY: &str = r#"
(enum_item
    name: (type_identifier) @name
    body: (enum_variant_list)? @body)
"#;

impl TreeSitterOracle {
    /// Get all enum definitions in the source.
    pub fn get_enums(&mut self) -> Result<Vec<EnumDefinition>> {
        let result = self.query(ENUMS_QUERY)?;
        result.matches.into_iter().map(enum_definition).collect()
    }
}

fn enum_definition(m: super::AstMatch) -> Result<EnumDefinition> {
    let body = m.captures.get("body").cloned().unwrap_or_default();
    Ok(EnumDefinition {
        name: m.captures.get("name").cloned().unwrap_or_default(),
        variants: extract::enum_variants(&body)?,
        line: m.line,
    })
}