mirror of
https://github.com/lifegpc/pixiv_downloader.git
synced 2026-07-05 20:01:23 +08:00
108 lines
3.6 KiB
Rust
108 lines
3.6 KiB
Rust
use proc_macro::TokenStream;
|
|
use quote::quote;
|
|
use syn::parse_macro_input;
|
|
use syn::Ident;
|
|
use syn::ItemFn;
|
|
use syn::LitInt;
|
|
|
|
#[proc_macro]
|
|
pub fn define_struct_reader_fn(item: TokenStream) -> TokenStream {
|
|
let i = parse_macro_input!(item as Ident);
|
|
let lefname = format!("read_le_{}", i);
|
|
let lefname = Ident::new(&lefname, i.span());
|
|
let befname = format!("read_be_{}", i);
|
|
let befname = Ident::new(&befname, i.span());
|
|
let stream = quote! {
|
|
#[doc = concat!("Read [", stringify!(#i), "] in little endian.")]
|
|
fn #lefname(&mut self) -> Result<#i, Self::Error>;
|
|
#[doc = concat!("Read [", stringify!(#i), "] in big endian.")]
|
|
fn #befname(&mut self) -> Result<#i, Self::Error>;
|
|
};
|
|
stream.into()
|
|
}
|
|
|
|
#[proc_macro]
|
|
pub fn impl_struct_reader_read(item: TokenStream) -> TokenStream {
|
|
let i = parse_macro_input!(item as Ident);
|
|
let lefname = format!("read_le_{}", i);
|
|
let lefname = Ident::new(&lefname, i.span());
|
|
let befname = format!("read_be_{}", i);
|
|
let befname = Ident::new(&befname, i.span());
|
|
let stream = quote! {
|
|
fn #lefname(&mut self) -> Result<#i, Self::Error> {
|
|
let mut buf = [0u8; std::mem::size_of::<#i>()];
|
|
self.read_exact(&mut buf)?;
|
|
Ok(<#i>::from_le_bytes(buf))
|
|
}
|
|
fn #befname(&mut self) -> Result<#i, Self::Error> {
|
|
let mut buf = [0u8; std::mem::size_of::<#i>()];
|
|
self.read_exact(&mut buf)?;
|
|
Ok(<#i>::from_be_bytes(buf))
|
|
}
|
|
};
|
|
stream.into()
|
|
}
|
|
|
|
#[proc_macro]
|
|
pub fn define_struct_writer_fn(item: TokenStream) -> TokenStream {
|
|
let i = parse_macro_input!(item as Ident);
|
|
let lefname = format!("write_le_{}", i);
|
|
let lefname = Ident::new(&lefname, i.span());
|
|
let befname = format!("write_be_{}", i);
|
|
let befname = Ident::new(&befname, i.span());
|
|
let stream = quote! {
|
|
#[doc = concat!("Write [", stringify!(#i), "] in little endian.")]
|
|
fn #lefname(&mut self, data: #i) -> Result<(), Self::Error>;
|
|
#[doc = concat!("Write [", stringify!(#i), "] in big endian.")]
|
|
fn #befname(&mut self, data: #i) -> Result<(), Self::Error>;
|
|
};
|
|
stream.into()
|
|
}
|
|
|
|
#[proc_macro]
|
|
pub fn impl_struct_writer_write(item: TokenStream) -> TokenStream {
|
|
let i = parse_macro_input!(item as Ident);
|
|
let lefname = format!("write_le_{}", i);
|
|
let lefname = Ident::new(&lefname, i.span());
|
|
let befname = format!("write_be_{}", i);
|
|
let befname = Ident::new(&befname, i.span());
|
|
let stream = quote! {
|
|
#[inline]
|
|
fn #lefname(&mut self, data: #i) -> Result<(), Self::Error> {
|
|
self.write_all(&data.to_le_bytes())
|
|
}
|
|
#[inline]
|
|
fn #befname(&mut self, data: #i) -> Result<(), Self::Error> {
|
|
self.write_all(&data.to_be_bytes())
|
|
}
|
|
};
|
|
stream.into()
|
|
}
|
|
|
|
#[proc_macro_attribute]
|
|
pub fn async_timeout_test(attr: TokenStream, item: TokenStream) -> TokenStream {
|
|
let ItemFn {
|
|
attrs,
|
|
vis,
|
|
sig,
|
|
block,
|
|
} = parse_macro_input!(item as ItemFn);
|
|
let dura = parse_duration::parse(attr.to_string().as_str()).unwrap();
|
|
let secs = LitInt::new(format!("{}", dura.as_secs()).as_str(), sig.ident.span());
|
|
let nanos = LitInt::new(
|
|
format!("{}", dura.subsec_nanos()).as_str(),
|
|
sig.ident.span(),
|
|
);
|
|
let stmts = &block.stmts;
|
|
let stream = quote! {
|
|
#(#attrs)* #vis #sig {
|
|
let dura = std::time::Duration::new(#secs, #nanos);
|
|
let f = async {
|
|
#(#stmts)*
|
|
};
|
|
tokio::time::timeout(dura, f).await.unwrap();
|
|
}
|
|
};
|
|
stream.into()
|
|
}
|