franklinblanco/dev-macros
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added extend

Browse Source
This commit is contained in:
Franklin 2023-01-24 17:48:32 -04:00
parent b255a01774
commit 6733e07a4f
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/lib.rs
View File

@ -4,8 +4,8 @@ mod auth;
use proc_macro::TokenStream; use proc_macro::TokenStream;
#[proc_macro_attribute] #[proc_macro_attribute]
pub fn authenticated_route(_: TokenStream, _: TokenStream) -> TokenStream { pub fn authenticated_route(_: TokenStream, mut input: TokenStream) -> TokenStream {
TokenStream::from(quote::quote!{ let tt_to_add = TokenStream::from(quote::quote!{
let __header_conversion_result: dev_dtos::dtos::user::user_dtos::UserAuthHeader = match serde_json::from_str(match request.headers().get("authentication") { let __header_conversion_result: dev_dtos::dtos::user::user_dtos::UserAuthHeader = match serde_json::from_str(match request.headers().get("authentication") {
Some(auth_header) => match auth_header.to_str() { Some(auth_header) => match auth_header.to_str() {
Ok(string) => string, Ok(string) => string,
@ -20,5 +20,7 @@ pub fn authenticated_route(_: TokenStream, _: TokenStream) -> TokenStream {
Ok(authed_user) => authed_user, Ok(authed_user) => authed_user,
Err(error) => return actix_web_utils::extensions::typed_response::TypedHttpResponse::return_standard_error(401, err::MessageResource::new_from_string(error.to_string())), Err(error) => return actix_web_utils::extensions::typed_response::TypedHttpResponse::return_standard_error(401, err::MessageResource::new_from_string(error.to_string())),
}; };
}) });
input.extend(tt_to_add);
input
} }