rstructor_derive 0.2.10

Derive macros for the rstructor library
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

use syn::Variant;

/// Represents parsed variant attributes
pub struct VariantAttributes {
    pub description: Option<String>,
    /// Variant rename from #[serde(rename = "...")]
    pub serde_rename: Option<String>,
}

/// Parse a single enum variant's llm and serde attributes
pub fn parse_variant_attributes(variant: &Variant) -> VariantAttributes {
    let mut description = None;
    let mut serde_rename = None;

    // Extract attributes
    for attr in &variant.attrs {
        // Parse serde attributes for rename
        if attr.path().is_ident("serde") {
            let _ = attr.parse_nested_meta(|meta| {
                if meta.path.is_ident("rename") {
                    let value = meta.value()?;
                    let content: syn::LitStr = value.parse()?;
                    serde_rename = Some(content.value());
                }
                Ok(())
            });
        }

        if attr.path().is_ident("llm") {
            // Parse attribute arguments
            let _result = attr.parse_nested_meta(|meta| {
                if meta.path.is_ident("description") {
                    let value = meta.value()?;
                    let content: syn::LitStr = value.parse()?;
                    description = Some(content.value());
                }
                Ok(())
            });
        }
    }

    VariantAttributes {
        description,
        serde_rename,
    }
}