1248a639ee
There are cases where we want an enum variant name. Right now the only way to do that with rust's std is to derive Debug, but this will also print out the variant's fields. This creates the unfortunate situation where we need to manually write out each variant's string name (ex: in #1963), which is both boilerplate-ey and error-prone. Crates such as `strum` exist for this reason, but it includes a lot of code and complexity that we don't need. This adds a dead-simple `EnumVariantMeta` derive that exposes `enum_variant_index` and `enum_variant_name` functions. This allows us to make cases like #1963 much cleaner (see the second commit). We might also be able to reuse this logic for `bevy_reflect` enum derives.
66 lines
1.9 KiB
Rust
66 lines
1.9 KiB
Rust
use find_crate::Manifest;
|
|
use proc_macro::TokenStream;
|
|
use syn::{Attribute, Path};
|
|
|
|
#[derive(Debug)]
|
|
pub struct Modules {
|
|
pub bevy_render: String,
|
|
pub bevy_asset: String,
|
|
pub bevy_core: String,
|
|
pub bevy_utils: String,
|
|
pub bevy_app: String,
|
|
}
|
|
|
|
impl Modules {
|
|
pub fn meta(name: &str) -> Modules {
|
|
Modules {
|
|
bevy_asset: format!("{}::asset", name),
|
|
bevy_render: format!("{}::render", name),
|
|
bevy_core: format!("{}::core", name),
|
|
bevy_utils: format!("{}::utils", name),
|
|
bevy_app: format!("{}::app", name),
|
|
}
|
|
}
|
|
|
|
pub fn external() -> Modules {
|
|
Modules {
|
|
bevy_asset: "bevy_asset".to_string(),
|
|
bevy_render: "bevy_render".to_string(),
|
|
bevy_core: "bevy_core".to_string(),
|
|
bevy_utils: "bevy_utils".to_string(),
|
|
bevy_app: "bevy_app".to_string(),
|
|
}
|
|
}
|
|
}
|
|
|
|
fn get_meta() -> Option<Modules> {
|
|
let manifest = Manifest::new().unwrap();
|
|
if let Some(package) = manifest.find(|name| name == "bevy") {
|
|
Some(Modules::meta(&package.name))
|
|
} else if let Some(package) = manifest.find(|name| name == "bevy_internal") {
|
|
Some(Modules::meta(&package.name))
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
const AS_CRATE_ATTRIBUTE_NAME: &str = "as_crate";
|
|
|
|
pub fn get_modules(attributes: &[Attribute]) -> Modules {
|
|
let mut modules = get_meta().unwrap_or_else(Modules::external);
|
|
for attribute in attributes.iter() {
|
|
if *attribute.path.get_ident().as_ref().unwrap() == AS_CRATE_ATTRIBUTE_NAME {
|
|
let value = attribute.tokens.to_string();
|
|
if value[1..value.len() - 1] == modules.bevy_render {
|
|
modules.bevy_render = "crate".to_string();
|
|
}
|
|
}
|
|
}
|
|
|
|
modules
|
|
}
|
|
|
|
pub fn get_path(path_str: &str) -> Path {
|
|
syn::parse(path_str.parse::<TokenStream>().unwrap()).unwrap()
|
|
}
|