diff --git a/libs/geometry_macros/src/lib.rs b/libs/geometry_macros/src/lib.rs index 0ec6c723..c72e082f 100644 --- a/libs/geometry_macros/src/lib.rs +++ b/libs/geometry_macros/src/lib.rs @@ -10,6 +10,23 @@ use proc_macro_crate::{crate_name, FoundCrate}; use quote::quote; use syn::{parse_macro_input, parse_quote, DeriveInput, Ident}; +pub(crate) fn geometry_ident() -> TokenStream2 { + match crate_name("geometry") { + Ok(FoundCrate::Itself) => quote!(::geometry), + Ok(FoundCrate::Name(name)) => { + let ident = Ident::new(&name, Span::call_site()); + quote!(::#ident) + } + Err(_) => match crate_name("substrate").expect("geometry not found in Cargo.toml") { + FoundCrate::Itself => quote!(::substrate::geometry), + FoundCrate::Name(name) => { + let ident = Ident::new(&name, Span::call_site()); + quote!(::#ident::geometry) + } + }, + } +} + /// Derives `geometry::transform::TranslateMut`. #[proc_macro_derive(TranslateMut)] pub fn derive_translate_mut(input: TokenStream) -> TokenStream { @@ -101,20 +118,3 @@ pub fn derive_transform_ref(input: TokenStream) -> TokenStream { ) .into() } - -pub(crate) fn geometry_ident() -> TokenStream2 { - match crate_name("geometry") { - Ok(FoundCrate::Itself) => quote!(::geometry), - Ok(FoundCrate::Name(name)) => { - let ident = Ident::new(&name, Span::call_site()); - quote!(::#ident) - } - Err(_) => match crate_name("substrate").expect("geometry not found in Cargo.toml") { - FoundCrate::Itself => quote!(::substrate::geometry), - FoundCrate::Name(name) => { - let ident = Ident::new(&name, Span::call_site()); - quote!(::#ident::geometry) - } - }, - } -}