xtask: Add "cargo xtask build --feature-permutations"

The `--feature-permutations` flag runs a bunch of builds of just the
`uefi` and `uefi-services` crates to test that the build works no matter
what features are selected.

Author: nicholasbishop

xtask/Cargo.toml

@@ -11,6 +11,7 @@ clap = { version = "4.0.4", features = ["derive"] }
 fatfs = { git = "https://github.com/rafalh/rust-fatfs.git", rev = "87fc1ed5074a32b4e0344fcdde77359ef9e75432" }
 fs-err = "2.6.0"
 heck = "0.4.0"
+itertools = "0.10.5"
 # Use git as a temporary workaround for https://github.com/rust-osdev/uefi-rs/issues/573.
 mbrman = { git = "https://github.com/cecton/mbrman.git", rev = "78565860e55c272488d94480e72a4e2cd159ad8e" }
 nix = "0.25.0"

xtask/src/cargo.rs

@@ -45,24 +45,55 @@ impl Package {
 
 #[derive(Clone, Copy, Debug)]
 pub enum Feature {
-    GlobalAllocator,
+    // `uefi` features.
     Alloc,
+    GlobalAllocator,
     Logger,
+    PanicOnLoggerErrors,
+    Unstable,
 
+    // `uefi-services` features.
+    PanicHandler,
+    Qemu,
+    ServicesLogger,
+
+    // `uefi-test-runner` features.
     Ci,
 }
 
 impl Feature {
     fn as_str(&self) -> &'static str {
         match self {
-            Self::GlobalAllocator => "global_allocator",
-            Self::Alloc => "alloc",
-            Self::Logger => "logger",
+            Self::Alloc => "uefi/alloc",
+            Self::GlobalAllocator => "uefi/global_allocator",
+            Self::Logger => "uefi/logger",
+            Self::PanicOnLoggerErrors => "uefi/panic-on-logger-errors",
+            Self::Unstable => "uefi/unstable",
+
+            Self::PanicHandler => "uefi-services/panic_handler",
+            Self::Qemu => "uefi-services/qemu",
+            Self::ServicesLogger => "uefi-services/logger",
 
             Self::Ci => "uefi-test-runner/ci",
         }
     }
 
+    /// Get the features for the given package.
+    pub fn package_features(package: Package) -> Vec<Self> {
+        match package {
+            Package::Uefi => vec![
+                Self::Alloc,
+                Self::GlobalAllocator,
+                Self::Logger,
+                Self::PanicOnLoggerErrors,
+                Self::Unstable,
+            ],
+            Package::UefiServices => vec![Self::PanicHandler, Self::Qemu, Self::ServicesLogger],
+            Package::UefiTestRunner => vec![Self::Ci],
+            _ => vec![],
+        }
+    }
+
     /// Set of features that enables more code in the root uefi crate.
     pub fn more_code() -> Vec<Self> {
         vec![Self::GlobalAllocator, Self::Alloc, Self::Logger]

xtask/src/main.rs

@@ -12,12 +12,39 @@ mod util;
 use anyhow::Result;
 use cargo::{fix_nested_cargo_env, Cargo, CargoAction, Feature, Package, TargetTypes};
 use clap::Parser;
+use itertools::Itertools;
 use opt::{Action, BuildOpt, ClippyOpt, DocOpt, Opt, QemuOpt};
 use std::process::Command;
 use tempfile::TempDir;
 use util::{command_to_string, run_cmd};
 
+fn build_feature_permutations(opt: &BuildOpt) -> Result<()> {
+    for package in [Package::Uefi, Package::UefiServices] {
+        let all_package_features = Feature::package_features(package);
+        for features in all_package_features.iter().powerset() {
+            let features = features.iter().map(|f| **f).collect();
+
+            let cargo = Cargo {
+                action: CargoAction::Build,
+                features,
+                packages: vec![package],
+                release: opt.build_mode.release,
+                target: Some(*opt.target),
+                warnings_as_errors: true,
+                target_types: TargetTypes::BinsExamplesLib,
+            };
+            run_cmd(cargo.command()?)?;
+        }
+    }
+
+    Ok(())
+}
+
 fn build(opt: &BuildOpt) -> Result<()> {
+    if opt.feature_permutations {
+        return build_feature_permutations(opt);
+    }
+
     let cargo = Cargo {
         action: CargoAction::Build,
         features: Feature::more_code(),

xtask/src/opt.rs

@@ -62,6 +62,11 @@ pub struct BuildOpt {
 
     #[clap(flatten)]
     pub build_mode: BuildModeOpt,
+
+    /// Build multiple times to check that different feature
+    /// combinations work.
+    #[clap(long, action)]
+    pub feature_permutations: bool,
 }
 
 /// Run clippy on all the packages.