61 lines
2.0 KiB
Rust
61 lines
2.0 KiB
Rust
use quote::quote;
|
|
use syn::{parse_macro_input, parse_quote, DeriveInput, GenericParam, Generics};
|
|
|
|
#[proc_macro_derive(SharedFunctor)]
|
|
pub fn derive_shared_functor(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
|
|
let input = parse_macro_input!(input as DeriveInput);
|
|
let name = input.ident;
|
|
let generics = add_clone_bounds(input.generics);
|
|
let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
|
|
let expanded = quote! {
|
|
impl #impl_generics SharedFunctor for #name #ty_generics #where_clause {
|
|
type Shared<'a, A: 'a + Clone> = Self::F<'a, A>
|
|
where
|
|
Self: 'a;
|
|
|
|
fn share<'a, A: 'a + Clone>(fa: Self::F<'a, A>) -> Self::Shared<'a, A>
|
|
where
|
|
Self: 'a,
|
|
{
|
|
fa
|
|
}
|
|
|
|
fn unshare<'a, A: 'a + Clone>(sa: Self::Shared<'a, A>) -> Self::F<'a, A>
|
|
where
|
|
Self: 'a,
|
|
{
|
|
sa
|
|
}
|
|
}
|
|
};
|
|
proc_macro::TokenStream::from(expanded)
|
|
}
|
|
|
|
fn add_clone_bounds(mut generics: Generics) -> Generics {
|
|
for param in &mut generics.params {
|
|
if let GenericParam::Type(ref mut type_param) = *param {
|
|
type_param.bounds.push(parse_quote!(Clone));
|
|
}
|
|
}
|
|
generics
|
|
}
|
|
|
|
#[proc_macro_derive(CovariantFunctor)]
|
|
pub fn derive_covariant_functor(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
|
|
let input = parse_macro_input!(input as DeriveInput);
|
|
let name = input.ident;
|
|
let generics = input.generics;
|
|
let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
|
|
let expanded = quote! {
|
|
impl #impl_generics CovariantFunctor for #name #ty_generics #where_clause {
|
|
fn variate<'a: 'b, 'b, A: 'a>(fa: Self::F<'a, A>) -> Self::F<'b, A>
|
|
where
|
|
Self: 'a,
|
|
{
|
|
fa
|
|
}
|
|
}
|
|
};
|
|
proc_macro::TokenStream::from(expanded)
|
|
}
|