cargo-build-bpf/cargo-test-bpf now support --workspace/--all

sdk/cargo-build-bpf/src/main.rs

@@ -15,9 +15,9 @@ struct Config {
     bpf_sdk: PathBuf,
     dump: bool,
     features: Vec<String>,
-    manifest_path: Option<PathBuf>,
     no_default_features: bool,
     verbose: bool,
+    workspace: bool,
 }
 
 impl Default for Config {
@@ -32,9 +32,9 @@ impl Default for Config {
             bpf_out_dir: None,
             dump: false,
             features: vec![],
-            manifest_path: None,
             no_default_features: false,
             verbose: false,
+            workspace: false,
         }
     }
 }
@@ -65,27 +65,13 @@ where
     }
 }
 
-fn build_bpf(config: Config) {
-    let mut metadata_command = cargo_metadata::MetadataCommand::new();
-    if let Some(manifest_path) = config.manifest_path {
-        metadata_command.manifest_path(manifest_path);
-    }
-
-    let metadata = metadata_command.exec().unwrap_or_else(|err| {
-        eprintln!("Failed to obtain package metadata: {}", err);
-        exit(1);
-    });
-
-    let root_package = metadata.root_package().unwrap_or_else(|| {
-        eprintln!(
-            "Workspace does not have a root package: {}",
-            metadata.workspace_root.display()
-        );
-        exit(1);
-    });
-
+fn build_bpf_package(
+    config: &Config,
+    target_directory: &PathBuf,
+    package: &cargo_metadata::Package,
+) {
     let program_name = {
-        let cdylib_targets = root_package
+        let cdylib_targets = package
             .targets
             .iter()
             .filter_map(|target| {
@@ -101,7 +87,7 @@ fn build_bpf(config: Config) {
             0 => {
                 println!(
                     "Note: {} crate does not contain a cdylib target",
-                    root_package.name
+                    package.name
                 );
                 None
             }
@@ -109,29 +95,29 @@ fn build_bpf(config: Config) {
             _ => {
                 eprintln!(
                     "{} crate contains multiple cdylib targets: {:?}",
-                    root_package.name, cdylib_targets
+                    package.name, cdylib_targets
                 );
                 exit(1);
             }
         }
     };
 
-    let legacy_program_feature_present = root_package.name == "solana-sdk";
-    let root_package_dir = &root_package.manifest_path.parent().unwrap_or_else(|| {
+    let legacy_program_feature_present = package.name == "solana-sdk";
+    let root_package_dir = &package.manifest_path.parent().unwrap_or_else(|| {
         eprintln!(
             "Unable to get directory of {}",
-            root_package.manifest_path.display()
+            package.manifest_path.display()
         );
         exit(1);
     });
 
     let bpf_out_dir = config
         .bpf_out_dir
-        .unwrap_or_else(|| metadata.target_directory.join("deploy"));
+        .as_ref()
+        .cloned()
+        .unwrap_or_else(|| target_directory.join("deploy"));
 
-    let target_build_directory = metadata
-        .target_directory
-        .join("bpfel-unknown-unknown/release");
+    let target_build_directory = target_directory.join("bpfel-unknown-unknown/release");
 
     env::set_current_dir(&root_package_dir).unwrap_or_else(|err| {
         eprintln!(
@@ -220,6 +206,38 @@ fn build_bpf(config: Config) {
     }
 }
 
+fn build_bpf(config: Config, manifest_path: Option<PathBuf>) {
+    let mut metadata_command = cargo_metadata::MetadataCommand::new();
+    if let Some(manifest_path) = manifest_path {
+        metadata_command.manifest_path(manifest_path);
+    }
+
+    let metadata = metadata_command.exec().unwrap_or_else(|err| {
+        eprintln!("Failed to obtain package metadata: {}", err);
+        exit(1);
+    });
+
+    if let Some(root_package) = metadata.root_package() {
+        if !config.workspace {
+            build_bpf_package(&config, &metadata.target_directory, root_package);
+            return;
+        }
+    }
+
+    let all_bpf_packages = metadata
+        .packages
+        .iter()
+        .filter(|package| {
+            package.manifest_path.with_file_name("Xargo.toml").exists()
+                && metadata.workspace_members.contains(&package.id)
+        })
+        .collect::<Vec<_>>();
+
+    for package in all_bpf_packages {
+        build_bpf_package(&config, &metadata.target_directory, package);
+    }
+}
+
 fn main() {
     let default_config = Config::default();
     let default_bpf_sdk = format!("{}", default_config.bpf_sdk.display());
@@ -285,6 +303,13 @@ fn main() {
                 .takes_value(true)
                 .help("Place final BPF build artifacts in this directory"),
         )
+        .arg(
+            Arg::with_name("workspace")
+                .long("workspace")
+                .takes_value(false)
+                .alias("all")
+                .help("Build all BPF packages in the workspace"),
+        )
         .get_matches_from(args);
 
     let bpf_sdk = value_t_or_exit!(matches, "bpf_sdk", PathBuf);
@@ -312,9 +337,10 @@ fn main() {
         features: values_t!(matches, "features", String)
             .ok()
             .unwrap_or_else(Vec::new),
-        manifest_path: value_t!(matches, "manifest_path", PathBuf).ok(),
         no_default_features: matches.is_present("no_default_features"),
         verbose: matches.is_present("verbose"),
+        workspace: matches.is_present("workspace"),
     };
-    build_bpf(config);
+    let manifest_path = value_t!(matches, "manifest_path", PathBuf).ok();
+    build_bpf(config, manifest_path);
 }