1#![allow(missing_docs)]
11
12use proc_macro::TokenStream;
13use syn::parse_macro_input;
14
15use crate::diagnostics::DiagCtxt;
16
17mod diagnostics;
18mod init;
19mod pin_data;
20mod pinned_drop;
21mod util;
22mod zeroable;
23
24#[proc_macro_attribute]
25pub fn pin_data(args: TokenStream, input: TokenStream) -> TokenStream {
26 let args = parse_macro_input!(args);
27 let input = parse_macro_input!(input);
28 DiagCtxt::with(|dcx| pin_data::pin_data(args, input, dcx)).into()
29}
30
31#[proc_macro_attribute]
32pub fn pinned_drop(args: TokenStream, input: TokenStream) -> TokenStream {
33 let args = parse_macro_input!(args);
34 let input = parse_macro_input!(input);
35 DiagCtxt::with(|dcx| pinned_drop::pinned_drop(args, input, dcx)).into()
36}
37
38#[proc_macro_derive(Zeroable)]
39pub fn derive_zeroable(input: TokenStream) -> TokenStream {
40 let input = parse_macro_input!(input);
41 DiagCtxt::with(|dcx| zeroable::derive(input, dcx)).into()
42}
43
44#[proc_macro_derive(MaybeZeroable)]
45pub fn maybe_derive_zeroable(input: TokenStream) -> TokenStream {
46 let input = parse_macro_input!(input);
47 DiagCtxt::with(|dcx| zeroable::maybe_derive(input, dcx)).into()
48}
49#[proc_macro]
50pub fn init(input: TokenStream) -> TokenStream {
51 let input = parse_macro_input!(input);
52 DiagCtxt::with(|dcx| {
53 init::expand_with_cfg(input, Some("::core::convert::Infallible"), false, dcx)
54 })
55 .into()
56}
57
58#[proc_macro]
59pub fn pin_init(input: TokenStream) -> TokenStream {
60 let input = parse_macro_input!(input);
61 DiagCtxt::with(|dcx| {
62 init::expand_with_cfg(input, Some("::core::convert::Infallible"), true, dcx)
63 })
64 .into()
65}