gin-tonic 0.5.2

main gin-tonic crate - rust protobuf with gin and tonic
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use protox::prost_reflect::EnumDescriptor;

use crate::codegen::{case, module, Context};

/// generate code for enumerations
pub(crate) fn generate(
    ctx: &Context,
    parent: &mut module::Module,
    module_path: &str,
    ty: EnumDescriptor,
) {
    let qualified_name = ty.full_name();

    // external types are not generated
    if ctx.resolve_ident(qualified_name).is_some() {
        return;
    }
    if !ctx.filter(qualified_name) {
        return;
    }

    tracing::info!("generating enumeration: {qualified_name}");

    let attributes = ctx.attributes(qualified_name);

    let module = module::create_child(parent, module_path);

    let ty_name = case::convert(ty.name(), case::Case::Pascal);
    let name = quote::format_ident!("{}", ty_name);

    let mut body = quote::quote!();

    for value in ty.values() {
        let tag = value.number();
        let (_package, value) = value
            .full_name()
            .rsplit_once('.')
            .expect("Enum values should be in a package");
        if value.ends_with("UNSPECIFIED") {
            // TODO: add support for UNSPECIFIED variant
            continue;
        }
        let value = case::convert(value, case::Case::Pascal);
        let value = value
            .strip_prefix(ty_name.as_ref())
            .unwrap_or(value.as_ref());

        let value_name = quote::format_ident!("{}", value);

        body.extend(quote::quote! {
            #[gin(tag = #tag)]
            #value_name,
        });
    }

    let item: syn::ItemEnum = syn::parse_quote! {
        #[derive(Clone, Copy, Debug, Eq, PartialEq, Enumeration)]
        #attributes
        pub enum #name {
            #body
        }
    };

    module.extend(quote::quote! {
        #item
    });
}