forbid enable multiple stm32 features

This commit is contained in:
eZio Pan 2024-02-19 16:59:36 +08:00
parent 6097928f72
commit 1dc88ab4fa

View File

@ -2,14 +2,40 @@ use std::env;
#[cfg(any(feature = "rt", feature = "memory-x"))] #[cfg(any(feature = "rt", feature = "memory-x"))]
use std::path::PathBuf; use std::path::PathBuf;
enum GetOneError {
None,
Multiple,
}
trait IteratorExt: Iterator {
fn get_one(self) -> Result<Self::Item, GetOneError>;
}
impl<T: Iterator> IteratorExt for T {
fn get_one(mut self) -> Result<Self::Item, GetOneError> {
match self.next() {
None => Err(GetOneError::None),
Some(res) => match self.next() {
Some(_) => Err(GetOneError::Multiple),
None => Ok(res),
},
}
}
}
fn main() { fn main() {
#[cfg(any(feature = "rt", feature = "memory-x"))] #[cfg(any(feature = "rt", feature = "memory-x"))]
let crate_dir = PathBuf::from(env::var_os("CARGO_MANIFEST_DIR").unwrap()); let crate_dir = PathBuf::from(env::var_os("CARGO_MANIFEST_DIR").unwrap());
let chip_core_name = env::vars_os() let chip_core_name = match env::vars()
.map(|(a, _)| a.to_string_lossy().to_string()) .map(|(a, _)| a)
.find(|x| x.starts_with("CARGO_FEATURE_STM32")) .filter(|x| x.starts_with("CARGO_FEATURE_STM32"))
.expect("No stm32xx Cargo feature enabled") .get_one()
{
Ok(x) => x,
Err(GetOneError::None) => panic!("No stm32xx Cargo feature enabled"),
Err(GetOneError::Multiple) => panic!("Multiple stm32xx Cargo features enabled"),
}
.strip_prefix("CARGO_FEATURE_") .strip_prefix("CARGO_FEATURE_")
.unwrap() .unwrap()
.to_ascii_lowercase() .to_ascii_lowercase()