mirror of https://codeberg.org/topola/topola.git
41 lines
987 B
Rust
41 lines
987 B
Rust
use proc_macro::TokenStream;
|
|
use syn::{DeriveInput, Attribute, LitStr};
|
|
|
|
mod read;
|
|
mod write;
|
|
|
|
#[proc_macro_derive(ReadDsn, attributes(opt, anon, vec, anon_vec))]
|
|
pub fn derive_read(input: TokenStream)
|
|
-> TokenStream
|
|
{
|
|
let input = syn::parse_macro_input!(input as DeriveInput);
|
|
read::impl_read(&input).into()
|
|
}
|
|
|
|
#[proc_macro_derive(WriteDsn, attributes(anon))]
|
|
pub fn derive_write(input: TokenStream)
|
|
-> TokenStream
|
|
{
|
|
let input = syn::parse_macro_input!(input as DeriveInput);
|
|
write::impl_write(&input).into()
|
|
}
|
|
|
|
fn attr_present(attrs: &Vec<Attribute>, name: &str) -> bool {
|
|
attrs
|
|
.iter()
|
|
.find(|attr| attr.path().is_ident(name))
|
|
.is_some()
|
|
}
|
|
|
|
fn attr_content(attrs: &Vec<Attribute>, name: &str) -> Option<String> {
|
|
attrs
|
|
.iter()
|
|
.find(|attr| attr.path().is_ident(name))
|
|
.and_then(|attr| Some(attr
|
|
.parse_args::<LitStr>()
|
|
.expect("string literal")
|
|
.value()
|
|
))
|
|
}
|
|
|