xtask: Allow filtering examples

This commit is contained in:
Henrik Tjäder 2023-02-04 15:47:23 +01:00
parent f7651911d7
commit 4124fbdd61

View file

@ -26,18 +26,24 @@ struct Options {
/// For which ARM target to build: v7 or v6 /// For which ARM target to build: v7 or v6
/// ///
/// The permissible targets are: /// The permissible targets are:
/// * all
/// ///
/// * thumbv6m-none-eabi /// * thumbv6m-none-eabi
/// ///
/// * thumbv7m-none-eabi /// * thumbv7m-none-eabi
#[structopt(short, long)] #[structopt(short, long)]
target: String, target: String,
/// Example to run, by default all examples are run
///
/// Example: `cargo xtask --target <..> --example complex`
#[structopt(short, long)]
example: Option<String>,
/// Enables also running `cargo size` on the selected examples /// Enables also running `cargo size` on the selected examples
/// ///
/// To pass options to `cargo size`, add `--` and then the following /// To pass options to `cargo size`, add `--` and then the following
/// arguments will be passed on /// arguments will be passed on
/// ///
/// Example: `cargo xtask --target thumbv7m-none-eabi -s -- -A` /// Example: `cargo xtask --target <..> -s -- -A`
#[structopt(short, long)] #[structopt(short, long)]
size: bool, size: bool,
/// Options to pass to `cargo size` /// Options to pass to `cargo size`
@ -110,7 +116,7 @@ fn main() -> anyhow::Result<()> {
let targets = [ARMV7M, ARMV6M]; let targets = [ARMV7M, ARMV6M];
let examples: Vec<_> = std::fs::read_dir("./examples")? let mut examples: Vec<_> = std::fs::read_dir("./examples")?
.filter_map(|p| p.ok()) .filter_map(|p| p.ok())
.map(|p| p.path()) .map(|p| p.path())
.filter(|p| p.display().to_string().ends_with(".rs")) .filter(|p| p.display().to_string().ends_with(".rs"))
@ -123,7 +129,22 @@ fn main() -> anyhow::Result<()> {
let target = &opts.target; let target = &opts.target;
let check_size = opts.size; let check_size = opts.size;
let size_arguments = &opts.sizearguments; let size_arguments = &opts.sizearguments;
let example = opts.example;
if let Some(example) = example {
if examples.contains(&example) {
println!("\nTesting example: {example}");
// If we managed to filter, set the examples to test to only this one
examples = vec![example]
} else {
eprintln!(
"\nThe example you specified is not available. Available examples are:\
\n{examples:#?}\n\
By default all examples are tested.",
);
process::exit(1);
}
}
init_build_dir()?; init_build_dir()?;
if target == "all" { if target == "all" {