feat: add --cpu-prof-flamegraph to generate interactive SVG flamegraphs

cli/args/flags.rs

@@ -269,9 +269,10 @@ pub struct CpuProfFlags {
   pub name: Option<String>,
   pub interval: Option<u32>,
   pub md: bool,
+  pub flamegraph: bool,
 }
 
-#[derive(Clone, Debug, Eq, PartialEq)]
+#[derive(Clone, Default, Debug, Eq, PartialEq)]
 pub struct EvalFlags {
   pub print: bool,
   pub code: String,
@@ -5548,6 +5549,7 @@ fn cpu_prof_args(cmd: Command) -> Command {
     .arg(cpu_prof_name_arg())
     .arg(cpu_prof_interval_arg())
     .arg(cpu_prof_md_arg())
+    .arg(cpu_prof_flamegraph_arg())
 }
 
 fn cpu_prof_parse(matches: &mut ArgMatches) -> Option<CpuProfFlags> {
@@ -5556,12 +5558,20 @@ fn cpu_prof_parse(matches: &mut ArgMatches) -> Option<CpuProfFlags> {
   let name = matches.remove_one::<String>("cpu-prof-name");
   let interval = matches.remove_one::<u32>("cpu-prof-interval");
   let md = matches.get_flag("cpu-prof-md");
-  if enabled || dir.is_some() || name.is_some() || interval.is_some() || md {
+  let flamegraph = matches.get_flag("cpu-prof-flamegraph");
+  if enabled
+    || dir.is_some()
+    || name.is_some()
+    || interval.is_some()
+    || md
+    || flamegraph
+  {
     Some(CpuProfFlags {
       dir,
       name,
       interval,
       md,
+      flamegraph,
     })
   } else {
     None
@@ -5607,6 +5617,13 @@ fn cpu_prof_md_arg() -> Arg {
     .action(ArgAction::SetTrue)
 }
 
+fn cpu_prof_flamegraph_arg() -> Arg {
+  Arg::new("cpu-prof-flamegraph")
+    .long("cpu-prof-flamegraph")
+    .help("Generate an SVG flamegraph alongside the CPU profile")
+    .action(ArgAction::SetTrue)
+}
+
 fn permit_no_files_arg() -> Arg {
   Arg::new("permit-no-files")
     .long("permit-no-files")

cli/args/mod.rs

@@ -1003,6 +1003,10 @@ impl CliOptions {
     self.flags.cpu_prof.as_ref().is_some_and(|f| f.md)
   }
 
+  pub fn cpu_prof_flamegraph(&self) -> bool {
+    self.flags.cpu_prof.as_ref().is_some_and(|f| f.flamegraph)
+  }
+
   pub fn enable_testing_features(&self) -> bool {
     self.flags.enable_testing_features
   }

cli/factory.rs

@@ -1130,6 +1130,7 @@ impl CliFactory {
         name: cli_options.cpu_prof_name(),
         interval: cli_options.cpu_prof_interval(),
         md: cli_options.cpu_prof_md(),
+        flamegraph: cli_options.cpu_prof_flamegraph(),
       });
 
     let lib_main_worker_factory = LibMainWorkerFactory::new(
@@ -1238,6 +1239,7 @@ impl CliFactory {
         name: cli_options.cpu_prof_name(),
         interval: cli_options.cpu_prof_interval(),
         md: cli_options.cpu_prof_md(),
+        flamegraph: cli_options.cpu_prof_flamegraph(),
       });
 
     let initial_cwd =

cli/worker.rs

@@ -288,6 +288,7 @@ impl CliMainWorker {
       filename,
       config.interval,
       config.md,
+      config.flamegraph,
     );
     cpu_profiler.start_profiling();