1
0
Fork 0
mirror of https://gitlab.com/msrd0/gotham-restful.git synced 2025-07-16 01:01:28 +00:00

add derive macro for resource

This commit is contained in:
Dominic 2019-10-06 15:03:30 +02:00
parent 75c399d97a
commit 0cf7c9aa3a
Signed by: msrd0
GPG key ID: DCC8C247452E98F9
9 changed files with 132 additions and 15 deletions

View file

@ -7,6 +7,7 @@ use syn::{
ReturnType,
parse_macro_input
};
use std::str::FromStr;
pub enum Method
{
@ -19,9 +20,27 @@ pub enum Method
Delete
}
impl FromStr for Method
{
type Err = String;
fn from_str(str : &str) -> Result<Self, Self::Err>
{
match str {
"ReadAll" | "read_all" => Ok(Self::ReadAll),
"Read" | "read" => Ok(Self::Read),
"Create" | "create" => Ok(Self::Create),
"UpdateAll" | "update_all" => Ok(Self::UpdateAll),
"Update" | "update" => Ok(Self::Update),
"DeleteAll" | "delete_all" => Ok(Self::DeleteAll),
"Delete" | "delete" => Ok(Self::Delete),
_ => Err("unknown method".to_string())
}
}
}
impl Method
{
fn trait_ident(&self) -> Ident
pub fn trait_ident(&self) -> Ident
{
use Method::*;
@ -37,7 +56,7 @@ impl Method
format_ident!("Resource{}", name)
}
fn fn_ident(&self) -> Ident
pub fn fn_ident(&self) -> Ident
{
use Method::*;
@ -52,6 +71,11 @@ impl Method
};
format_ident!("{}", name)
}
pub fn setup_ident(&self) -> Ident
{
format_ident!("{}_setup_impl", self.fn_ident())
}
}
pub fn expand_method(method : Method, attrs : TokenStream, item : TokenStream) -> TokenStream
@ -87,6 +111,7 @@ pub fn expand_method(method : Method, attrs : TokenStream, item : TokenStream) -
let trait_ident = method.trait_ident();
let fn_ident = method.fn_ident();
let setup_ident = method.setup_ident();
let output = quote! {
impl ::gotham_restful::#trait_ident<#(#generics),*> for #ident
@ -98,6 +123,12 @@ pub fn expand_method(method : Method, attrs : TokenStream, item : TokenStream) -
#ret_stmt
}
}
#[deny(dead_code)]
fn #setup_ident<D : ::gotham_restful::DrawResourceRoutes>(route : &mut D)
{
route.#fn_ident::<#ident, #(#generics),*>();
}
};
output.into()
}