2019-10-02 10:59:25 +02:00
|
|
|
use proc_macro::TokenStream;
|
2019-10-14 02:17:25 +02:00
|
|
|
use proc_macro2::TokenStream as TokenStream2;
|
|
|
|
use quote::quote;
|
2019-10-02 10:59:25 +02:00
|
|
|
|
2020-04-12 19:58:05 +00:00
|
|
|
mod util;
|
|
|
|
|
2019-10-20 14:49:53 +00:00
|
|
|
mod from_body;
|
|
|
|
use from_body::expand_from_body;
|
2019-10-05 00:21:41 +02:00
|
|
|
mod method;
|
|
|
|
use method::{expand_method, Method};
|
2019-10-20 14:49:53 +00:00
|
|
|
mod request_body;
|
|
|
|
use request_body::expand_request_body;
|
2019-10-06 15:03:30 +02:00
|
|
|
mod resource;
|
|
|
|
use resource::expand_resource;
|
2019-10-02 10:59:25 +02:00
|
|
|
#[cfg(feature = "openapi")]
|
|
|
|
mod openapi_type;
|
|
|
|
|
2020-04-08 22:18:06 +02:00
|
|
|
#[inline]
|
|
|
|
fn print_tokens(tokens : TokenStream) -> TokenStream
|
|
|
|
{
|
|
|
|
//eprintln!("{}", tokens);
|
|
|
|
tokens
|
|
|
|
}
|
|
|
|
|
2019-10-14 02:17:25 +02:00
|
|
|
fn krate() -> TokenStream2
|
|
|
|
{
|
2019-10-14 02:37:50 +02:00
|
|
|
quote!(::gotham_restful)
|
2019-10-14 02:17:25 +02:00
|
|
|
}
|
|
|
|
|
2019-10-20 14:49:53 +00:00
|
|
|
#[proc_macro_derive(FromBody)]
|
|
|
|
pub fn derive_from_body(tokens : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_from_body(tokens))
|
2019-10-20 14:49:53 +00:00
|
|
|
}
|
|
|
|
|
2019-10-02 10:59:25 +02:00
|
|
|
#[cfg(feature = "openapi")]
|
2020-04-07 20:44:02 +00:00
|
|
|
#[proc_macro_derive(OpenapiType, attributes(openapi))]
|
2019-10-02 10:59:25 +02:00
|
|
|
pub fn derive_openapi_type(tokens : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(openapi_type::expand(tokens))
|
2019-10-02 10:59:25 +02:00
|
|
|
}
|
2019-10-05 00:21:41 +02:00
|
|
|
|
2019-10-20 14:49:53 +00:00
|
|
|
#[proc_macro_derive(RequestBody, attributes(supported_types))]
|
|
|
|
pub fn derive_request_body(tokens : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_request_body(tokens))
|
2019-10-20 14:49:53 +00:00
|
|
|
}
|
|
|
|
|
2019-10-06 15:03:30 +02:00
|
|
|
#[proc_macro_derive(Resource, attributes(rest_resource))]
|
|
|
|
pub fn derive_resource(tokens : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_resource(tokens))
|
2019-10-06 15:03:30 +02:00
|
|
|
}
|
|
|
|
|
2019-10-05 00:21:41 +02:00
|
|
|
#[proc_macro_attribute]
|
|
|
|
pub fn rest_read_all(attr : TokenStream, item : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_method(Method::ReadAll, attr, item))
|
2019-10-05 00:21:41 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
#[proc_macro_attribute]
|
|
|
|
pub fn rest_read(attr : TokenStream, item : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_method(Method::Read, attr, item))
|
2019-10-05 00:21:41 +02:00
|
|
|
}
|
|
|
|
|
2019-10-13 17:43:42 +02:00
|
|
|
#[proc_macro_attribute]
|
|
|
|
pub fn rest_search(attr : TokenStream, item : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_method(Method::Search, attr, item))
|
2019-10-13 17:43:42 +02:00
|
|
|
}
|
|
|
|
|
2019-10-05 00:21:41 +02:00
|
|
|
#[proc_macro_attribute]
|
|
|
|
pub fn rest_create(attr : TokenStream, item : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_method(Method::Create, attr, item))
|
2019-10-05 00:21:41 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
#[proc_macro_attribute]
|
|
|
|
pub fn rest_update_all(attr : TokenStream, item : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_method(Method::UpdateAll, attr, item))
|
2019-10-05 00:21:41 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
#[proc_macro_attribute]
|
|
|
|
pub fn rest_update(attr : TokenStream, item : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_method(Method::Update, attr, item))
|
2019-10-05 00:21:41 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
#[proc_macro_attribute]
|
|
|
|
pub fn rest_delete_all(attr : TokenStream, item : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_method(Method::DeleteAll, attr, item))
|
2019-10-05 00:21:41 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
#[proc_macro_attribute]
|
|
|
|
pub fn rest_delete(attr : TokenStream, item : TokenStream) -> TokenStream
|
|
|
|
{
|
2020-04-08 22:18:06 +02:00
|
|
|
print_tokens(expand_method(Method::Delete, attr, item))
|
2019-10-05 00:21:41 +02:00
|
|
|
}
|