tusks_lib/codegen/handle_matches/arms/
submodule.rs

1use proc_macro2::{Span, TokenStream};
2use quote::quote;
3
4use crate::TusksModule;
5use crate::codegen::module_path::ModulePath;
6use crate::codegen::util::enum_util::to_variant_ident;
7use crate::codegen::util::field_util::is_generated_field;
8
9impl TusksModule {
10    /// Generates a match arm for a submodule. The complete generated pattern:
11    ///
12    /// ```ignore
13    /// Some(cli::Commands::admin { user: p1, sub }) => {
14    ///     let super_parameters = &parameters;
15    ///     let parameters = super::admin::Parameters { user: p1, super_: super_parameters, .. };
16    ///     match sub {
17    ///         // ... nested arms ...
18    ///     }
19    /// }
20    /// ```
21    pub fn build_submodule_match_arm(
22        &self,
23        cli_path: &TokenStream,
24        path: &ModulePath,
25    ) -> TokenStream {
26        let variant_ident = to_variant_ident(&self.name);
27        let param_bindings = self.build_parameter_pattern_bindings();
28        let mut pattern_fields = self.build_pattern_fields(&param_bindings);
29        let has_commands = self.has_commands();
30
31        if has_commands {
32            pattern_fields.push(quote! { sub });
33        }
34
35        let params_init = self.build_parameter_initialization(&param_bindings, path, has_commands);
36        let nested_match = self.build_nested_match_arms(path, has_commands);
37
38        quote! {
39            Some(#cli_path::Commands::#variant_ident { #(#pattern_fields),* }) => {
40                #params_init
41                #nested_match
42            }
43        }
44    }
45
46    fn build_parameter_pattern_bindings(&self) -> Vec<(syn::Ident, syn::Ident)> {
47        let mut bindings = Vec::new();
48        let mut counter = 1;
49
50        if let Some(ref params) = self.parameters {
51            for field in &params.pstruct.fields {
52                if let Some(field_name) = &field.ident
53                    && !is_generated_field(&field_name.to_string()) {
54                        let binding = syn::Ident::new(
55                            &format!("p{}", counter),
56                            Span::call_site(),
57                        );
58                        bindings.push((field_name.clone(), binding));
59                        counter += 1;
60                    }
61            }
62        }
63
64        bindings
65    }
66
67    fn has_commands(&self) -> bool {
68        !self.tusks.is_empty()
69            || !self.submodules.is_empty()
70            || !self.external_modules.is_empty()
71    }
72
73    fn build_parameter_initialization(
74        &self,
75        bindings: &[(syn::Ident, syn::Ident)],
76        path: &ModulePath,
77        has_commands: bool,
78    ) -> TokenStream {
79        if !has_commands || self.parameters.is_none() {
80            return quote! {};
81        }
82
83        let submod_name = &self.name;
84        let params = self.parameters.as_ref().unwrap();
85        let mut field_inits = Vec::new();
86
87        for field in &params.pstruct.fields {
88            if let Some(field_name) = &field.ident {
89                if field_name == "super_" {
90                    field_inits.push(quote! { super_: super_parameters, });
91                } else if field_name == "_phantom_lifetime_marker" {
92                    field_inits.push(quote! {
93                        _phantom_lifetime_marker: ::std::marker::PhantomData,
94                    });
95                } else if let Some((_, binding)) =
96                    bindings.iter().find(|(fname, _)| fname == field_name)
97                {
98                    field_inits.push(quote! { #field_name: #binding, });
99                }
100            }
101        }
102
103        let params_path = path.super_path_to(submod_name);
104
105        quote! {
106            let super_parameters = &parameters;
107            let parameters = #params_path::Parameters {
108                #(#field_inits)*
109            };
110        }
111    }
112
113    fn build_nested_match_arms(
114        &self,
115        path: &ModulePath,
116        has_commands: bool,
117    ) -> TokenStream {
118        if !has_commands {
119            return Self::build_no_command_error(path);
120        }
121
122        let new_path = path.join(&self.name.to_string());
123        let nested_arms = self.build_match_arms_recursive(&new_path);
124
125        quote! {
126            match sub {
127                #(#nested_arms)*
128            }
129        }
130    }
131}
tusks_lib/codegen/handle_matches/arms/submodule.rs

tusks_lib/codegen/handle_matches/arms/
submodule.rs
