Apply string casing to method params while deriving proxies and services

Author: Eligioo

derive/src/lib.rs

@@ -5,7 +5,7 @@ use darling::FromMeta;
 use heck::{ToKebabCase, ToLowerCamelCase, ToShoutySnakeCase, ToSnakeCase, ToUpperCamelCase};
 use proc_macro2::{Literal, TokenStream};
 use quote::{format_ident, quote};
-use syn::{Attribute, FnArg, Ident, Pat, Signature, Type};
+use syn::{spanned::Spanned, Attribute, FnArg, Ident, Pat, Signature, Type};
 
 use proxy::proxy_macro;
 use service::service_macro;
@@ -51,7 +51,7 @@ impl MethodAttributes {
 
 pub(crate) struct RpcMethod<'a> {
     signature: &'a Signature,
-    args: Vec<(&'a Ident, &'a Type)>,
+    args: Vec<(Ident, &'a Type)>,
     method_name: String,
     method_name_literal: Literal,
     args_struct_ident: Ident,
@@ -75,7 +75,13 @@ impl<'a> RpcMethod<'a> {
                 }
                 FnArg::Typed(pat_type) => {
                     let ident = match &*pat_type.pat {
-                        Pat::Ident(ty) => &ty.ident,
+                        Pat::Ident(ty) => {
+                            let fn_arg = rename_all
+                                .as_ref()
+                                .map(|r| r.rename(&ty.ident.to_string()))
+                                .unwrap_or(ty.ident.to_string());
+                            Ident::new(&fn_arg, ty.span().clone())
+                        }
                         _ => panic!("Arguments must not be patterns."),
                     };
                     args.push((ident, &*pat_type.ty));
@@ -120,6 +126,7 @@ impl<'a> RpcMethod<'a> {
         let tokens = quote! {
             #[derive(Debug, ::serde::Serialize, ::serde::Deserialize)]
             #[allow(non_camel_case_types)]
+            #[allow(non_snake_case)]
             struct #args_struct_ident {
                 #(#struct_fields)*
             }
@@ -213,6 +220,7 @@ impl<'a> RpcMethod<'a> {
         };
 
         quote! {
+            #[allow(non_snake_case)]
             async fn #method_ident(&mut self, #(#method_args),*) #output {
                 let args = #args_struct_ident {
                     #(#struct_fields),*