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 zeroable;
22
23#[proc_macro_attribute]
24pub fn pin_data(args: TokenStream, input: TokenStream) -> TokenStream {
25 let args = parse_macro_input!(args);
26 let input = parse_macro_input!(input);
27 DiagCtxt::with(|dcx| pin_data::pin_data(args, input, dcx)).into()
28}
29
30#[proc_macro_attribute]
31pub fn pinned_drop(args: TokenStream, input: TokenStream) -> TokenStream {
32 let args = parse_macro_input!(args);
33 let input = parse_macro_input!(input);
34 DiagCtxt::with(|dcx| pinned_drop::pinned_drop(args, input, dcx)).into()
35}
36
37#[proc_macro_derive(Zeroable)]
38pub fn derive_zeroable(input: TokenStream) -> TokenStream {
39 let input = parse_macro_input!(input);
40 DiagCtxt::with(|dcx| zeroable::derive(input, dcx)).into()
41}
42
43#[proc_macro_derive(MaybeZeroable)]
44pub fn maybe_derive_zeroable(input: TokenStream) -> TokenStream {
45 let input = parse_macro_input!(input);
46 DiagCtxt::with(|dcx| zeroable::maybe_derive(input, dcx)).into()
47}
48#[proc_macro]
49pub fn init(input: TokenStream) -> TokenStream {
50 let input = parse_macro_input!(input);
51 DiagCtxt::with(|dcx| init::expand(input, Some("::core::convert::Infallible"), false, dcx))
52 .into()
53}
54
55#[proc_macro]
56pub fn pin_init(input: TokenStream) -> TokenStream {
57 let input = parse_macro_input!(input);
58 DiagCtxt::with(|dcx| init::expand(input, Some("::core::convert::Infallible"), true, dcx)).into()
59}