607bfd3b1c
This PR replaces the use of `linkme` with `inventory` for the component preview registration. Release Notes: - N/A
28 lines
898 B
Rust
28 lines
898 B
Rust
use proc_macro::TokenStream;
|
|
use quote::quote;
|
|
use syn::{DeriveInput, parse_macro_input};
|
|
|
|
pub fn derive_register_component(input: TokenStream) -> TokenStream {
|
|
let input = parse_macro_input!(input as DeriveInput);
|
|
let name = input.ident;
|
|
let register_fn_name = syn::Ident::new(
|
|
&format!("__component_registry_internal_register_{}", name),
|
|
name.span(),
|
|
);
|
|
let expanded = quote! {
|
|
const _: () = {
|
|
struct AssertComponent<T: component::Component>(::std::marker::PhantomData<T>);
|
|
let _ = AssertComponent::<#name>(::std::marker::PhantomData);
|
|
};
|
|
|
|
#[allow(non_snake_case)]
|
|
fn #register_fn_name() {
|
|
component::register_component::<#name>();
|
|
}
|
|
|
|
component::__private::inventory::submit! {
|
|
component::ComponentFn::new(#register_fn_name)
|
|
}
|
|
};
|
|
expanded.into()
|
|
}
|