schedule-parser-rusted/actix-macros/src/lib.rs

extern crate proc_macro;

use proc_macro::TokenStream;

mod shared {
    use quote::{ToTokens, quote};
    use syn::{Attribute, DeriveInput};

    pub fn find_status_code(attrs: &Vec<Attribute>) -> Option<proc_macro2::TokenStream> {
        attrs
            .iter()
            .find_map(|attr| -> Option<proc_macro2::TokenStream> {
                if !attr.path().is_ident("status_code") {
                    return None;
                }

                let meta = attr.meta.require_name_value().ok()?;

                let code = meta.value.to_token_stream().to_string();
                let trimmed_code = code.trim_matches('"');

                if let Ok(numeric_code) = trimmed_code.parse::<u16>() {
                    Some(quote! { actix_web::http::StatusCode::from_u16(#numeric_code).unwrap() })
                } else {
                    let string_code: proc_macro2::TokenStream =
                        trimmed_code.to_string().parse().unwrap();

                    Some(quote! { #string_code })
                }
            })
    }

    pub fn get_arms(ast: &DeriveInput) -> Vec<proc_macro2::TokenStream> {
        let name = &ast.ident;

        let variants = if let syn::Data::Enum(data) = &ast.data {
            &data.variants
        } else {
            panic!("Only enums are supported");
        };

        let mut status_code_arms: Vec<proc_macro2::TokenStream> = variants
            .iter()
            .map(|v| -> Option<proc_macro2::TokenStream> {
                let status_code = find_status_code(&v.attrs)?;
                let variant_name = &v.ident;

                Some(quote! { #name::#variant_name => #status_code, })
            })
            .filter(|v| v.is_some())
            .map(|v| v.unwrap())
            .collect();

        if status_code_arms.len() < variants.len() {
            let status_code = find_status_code(&ast.attrs)
                .unwrap_or_else(|| quote! { ::actix_web::http::StatusCode::INTERNAL_SERVER_ERROR });

            status_code_arms.push(quote! { _ => #status_code });
        }

        status_code_arms
    }
}

mod response_error_message {
    use proc_macro::TokenStream;
    use quote::quote;

    pub fn fmt(ast: &syn::DeriveInput) -> TokenStream {
        let name = &ast.ident;

        let status_code_arms = super::shared::get_arms(ast);

        TokenStream::from(quote! {
            impl ::actix_web::ResponseError for #name {
                fn status_code(&self) -> ::actix_web::http::StatusCode {
                    match self {
                        #(#status_code_arms)*
                    }
                }

                fn error_response(&self) -> ::actix_web::HttpResponse<BoxBody> {
                    ::actix_web::HttpResponse::build(self.status_code())
                        .json(crate::utility::error::ResponseErrorMessage::new(self.clone()))
                }
            }
        })
    }
}

mod status_code {
    use proc_macro::TokenStream;
    use quote::quote;

    pub fn fmt(ast: &syn::DeriveInput) -> TokenStream {
        let name = &ast.ident;

        let status_code_arms = super::shared::get_arms(ast);

        TokenStream::from(quote! {
            impl crate::routes::schema::PartialStatusCode for #name {
                fn status_code(&self) -> ::actix_web::http::StatusCode {
                    match self {
                        #(#status_code_arms)*
                    }
                }
            }
        })
    }
}

mod responder_json {
    use proc_macro::TokenStream;
    use quote::quote;

    pub fn fmt(ast: &syn::DeriveInput) -> TokenStream {
        let name = &ast.ident;

        TokenStream::from(quote! {
            impl ::actix_web::Responder for #name {
                type Body = ::actix_web::body::EitherBody<::actix_web::body::BoxBody>;

                fn respond_to(self, _: &::actix_web::HttpRequest) -> ::actix_web::HttpResponse<Self::Body> {
                    ::actix_web::HttpResponse::Ok()
                            .json(self)
                            .map_into_left_body()
                }
            }
        })
    }
}

mod into_response_error {
    use proc_macro::TokenStream;
    use quote::quote;

    pub fn fmt(ast: &syn::DeriveInput) -> TokenStream {
        let name = &ast.ident;

        TokenStream::from(quote! {
            impl ::core::convert::Into<crate::routes::schema::ResponseError<#name>> for #name {
                fn into(self) -> crate::routes::schema::ResponseError<#name> {
                    crate::routes::schema::ResponseError {
                        code: self,
                        message: ::core::option::Option::None,
                    }
                }
            }

            impl<T> crate::routes::schema::IntoResponseAsError<T> for #name
            where
                T: ::serde::ser::Serialize + ::utoipa::PartialSchema {}
        })
    }

    pub fn fmt_named(ast: &syn::DeriveInput) -> TokenStream {
        let name = &ast.ident;

        TokenStream::from(quote! {
            impl ::core::convert::Into<crate::routes::schema::ResponseError<#name>> for #name {
                fn into(self) -> crate::routes::schema::ResponseError<#name> {
                    crate::routes::schema::ResponseError {
                        message: ::core::option::Option::Some(format!("{}", self)),
                        code: self,
                    }
                }
            }

            impl<T> crate::routes::schema::IntoResponseAsError<T> for #name
            where
                T: ::serde::ser::Serialize + ::utoipa::PartialSchema {}
        })
    }
}

#[proc_macro_derive(ResponseErrorMessage, attributes(status_code))]
pub fn rem_derive(input: TokenStream) -> TokenStream {
    let ast = syn::parse(input).unwrap();

    response_error_message::fmt(&ast)
}

#[proc_macro_derive(ResponderJson)]
pub fn responser_json_derive(input: TokenStream) -> TokenStream {
    let ast = syn::parse(input).unwrap();

    responder_json::fmt(&ast)
}

#[proc_macro_derive(IntoResponseError)]
pub fn into_response_error_derive(input: TokenStream) -> TokenStream {
    let ast = syn::parse(input).unwrap();

    into_response_error::fmt(&ast)
}

#[proc_macro_derive(IntoResponseErrorNamed)]
pub fn into_response_error_named_derive(input: TokenStream) -> TokenStream {
    let ast = syn::parse(input).unwrap();

    into_response_error::fmt_named(&ast)
}

#[proc_macro_derive(StatusCode, attributes(status_code))]
pub fn status_code_derive(input: TokenStream) -> TokenStream {
    let ast = syn::parse(input).unwrap();

    status_code::fmt(&ast)
}