Merge 3e1aea8 into 03f349e

vercel · Dec 22, 2022 · c5d25c9 · c5d25c9
2 parents 03f349e + 3e1aea8
commit c5d25c9
Show file tree

Hide file tree

Showing 4 changed files with 303 additions and 114 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/xtask/Cargo.toml b/xtask/Cargo.toml
@@ -20,6 +20,7 @@ chrono = "0.4.22"
 clap = { version = "4.0.18", features = ["derive"] }
 indexmap = { workspace = true }
 inquire = { version = "0.5.1" }
+num-format = "0.4.4"
 owo-colors = "3"
 plotters = { version = "0.3.4", features = ["full_palette"] }
 semver = "1"

diff --git a/xtask/src/main.rs b/xtask/src/main.rs
@@ -68,7 +68,8 @@ To generate the summary json file:
 Visualizations generated by this command will appear in a sibling directory to the summary data \
                      file.",
                 )
-                .arg(arg!(<PATH_TO_SUMMARY_JSON> "the path to the benchmark summary json file")),
+                .arg(arg!(<PATH_TO_SUMMARY_JSON> "the path to the benchmark summary json file"))
+                .arg(arg!(--bundlers <BUNDLERS> "comma separated list of bundlers to include in the visualization")),
         )
 }
 
@@ -166,9 +167,13 @@ fn main() {
             let path = sub_matches
                 .get_one::<String>("PATH_TO_SUMMARY_JSON")
                 .expect("PATH_TO_SUMMARY_JSON is required");
+            let bundlers: Option<HashSet<&str>> = sub_matches
+                .get_one::<String>("bundlers")
+                .map(|s| s.split(',').collect());
+
             let path = PathBuf::from(path);
             let path = path.canonicalize().unwrap();
-            visualize_bundler_bench::generate(path).unwrap();
+            visualize_bundler_bench::generate(path, bundlers).unwrap();
         }
         _ => {
             panic!("Unknown command {:?}", matches.subcommand().map(|c| c.0));