1extern crate proc_macro;
4
5use proc_macro::TokenStream;
6use quote::quote;
7use syn::{DeriveInput, parse_macro_input};
8
9#[proc_macro_derive(BindTyCtx, attributes(bind))]
21pub fn bind_ty_ctx(input: TokenStream) -> TokenStream {
22 let input = parse_macro_input!(input as DeriveInput);
24
25 let expanded = match input.data {
27 syn::Data::Struct(..) => {
28 let name = &input.ident;
29 let bind_name = input
30 .attrs
31 .iter()
32 .find_map(|attr| {
33 if attr.path().is_ident("bind") {
34 Some(attr.parse_args::<syn::Expr>().unwrap())
35 } else {
36 None
37 }
38 })
39 .unwrap_or_else(|| {
40 let t = syn::Ident::new("tyctx", input.ident.span());
41 syn::parse_quote!(#t)
42 });
43 let (impl_generics, ty_generics, where_clause) = input.generics.split_for_impl();
44
45 quote! {
46 impl #impl_generics TyCtx for #name #ty_generics #where_clause {
47 fn global_bounds(&self, var: &Interned<TypeVar>, pol: bool) -> Option<DynTypeBounds> {
48 self.#bind_name.global_bounds(var, pol)
49 }
50 fn local_bind_of(&self, var: &Interned<TypeVar>) -> Option<Ty> {
51 self.#bind_name.local_bind_of(var)
52 }
53 }
54 }
55 }
56 _ => panic!("only structs are supported"),
57 };
58
59 TokenStream::from(expanded)
61}
62
63#[proc_macro_derive(DeclEnum)]
75pub fn gen_decl_enum(input: TokenStream) -> TokenStream {
76 let input = parse_macro_input!(input as DeriveInput);
86
87 let variants = match input.data {
88 syn::Data::Enum(data) => data.variants,
89 _ => panic!("only enums are supported"),
90 };
91
92 let names = variants.iter().map(|v| &v.ident).collect::<Vec<_>>();
93
94 let input_name = &input.ident;
95
96 let expanded = quote! {
97 impl #input_name {
98 pub fn name(&self) -> &Interned<str> {
100 match self {
101 #(Self::#names(x) => x.name()),*
102 }
103 }
104 pub fn span(&self) -> Span {
106 match self {
107 #(Self::#names(x) => x.span()),*
108 }
109 }
110 }
111
112 impl fmt::Debug for Decl {
113 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
114 match self {
115 #(Self::#names(x) => write!(f, concat!(stringify!(#names), "({:?})"), x)),*
116 }
117 }
118 }
119
120 };
121
122 TokenStream::from(expanded)
123}
124
125#[proc_macro_derive(TypliteAttr)]
136pub fn gen_typlite_element(input: TokenStream) -> TokenStream {
137 let input = parse_macro_input!(input as DeriveInput);
139
140 let field_parsers = match &input.data {
142 syn::Data::Struct(data) => match &data.fields {
143 syn::Fields::Named(fields) => fields
144 .named
145 .iter()
146 .map(|f| {
147 let name = f.ident.as_ref().unwrap();
148
149 let ty = &f.ty;
150
151 quote! {
152 md_attr::#name => {
153 let value = <#ty>::parse_attr(content)?;
154 result.#name = value;
155 }
156 }
157 })
158 .collect::<Vec<_>>(),
159 syn::Fields::Unnamed(_) => panic!("unnamed fields are not supported"),
160 syn::Fields::Unit => panic!("unit structs are not supported"),
161 },
162 _ => panic!("only structs are supported"),
163 };
164
165 let input_name = &input.ident;
166
167 let expanded = quote! {
169 impl TypliteAttrsParser for #input_name {
170 fn parse(attrs: &HtmlAttrs) -> Result<Self> {
171 let mut result = Self::default();
172 for (name, content) in attrs.0.iter() {
173 match *name {
174 #(#field_parsers)*
175 _ => {
176 return Err(format!("unknown attribute: {name}").into());
177 }
178 }
179 }
180
181 Ok(result)
182 }
183 }
184 };
185
186 TokenStream::from(expanded)
187}