edgedb_tokio_ext_derive/
lib.rs

1use proc_macro::TokenStream;
2use syn::{parse_macro_input, DeriveInput, ItemFn, LitStr};
3
4mod shape;
5mod tx_variant;
6
7#[proc_macro_derive(Shape, attributes(shape))]
8pub fn derive_shape_macro(input: TokenStream) -> TokenStream {
9    let input = parse_macro_input!(input as DeriveInput);
10    shape::derive_shape(input).into()
11}
12
13#[proc_macro]
14pub fn shaped_query(input: TokenStream) -> TokenStream {
15    let input = parse_macro_input!(input as LitStr);
16    shape::derive_shaped_query(input).into()
17}
18
19#[proc_macro_attribute]
20pub fn tx_variant(_attr: TokenStream, input: TokenStream) -> TokenStream {
21    let input = parse_macro_input!(input as ItemFn);
22    tx_variant::derive_tx_variant(input).into()
23}