[support bundle] Add health checks to support bundles

Cargo.toml

@@ -49,6 +49,7 @@ members = [
     "dev-tools/releng",
     "dev-tools/repl-utils",
     "dev-tools/repo-depot-standalone",
+    "dev-tools/support-bundle-reader-lib",
     "dev-tools/xtask",
     "dns-server",
     "dns-server-api",
@@ -199,6 +200,7 @@ default-members = [
     "dev-tools/releng",
     "dev-tools/repl-utils",
     "dev-tools/repo-depot-standalone",
+    "dev-tools/support-bundle-reader-lib",
     # Do not include xtask in the list of default members, because this causes
     # hakari to not work as well and build times to be longer.
     # See omicron#4392.
@@ -692,6 +694,7 @@ strum = { version = "0.26", features = [ "derive" ] }
 subprocess = "0.2.9"
 subtle = "2.6.1"
 supports-color = "3.0.2"
+support-bundle-reader-lib = { path = "dev-tools/support-bundle-reader-lib" }
 swrite = "0.1.0"
 sync-ptr = "0.1.1"
 libsw = { version = "3.4.0", features = ["tokio"] }

dev-tools/omdb/Cargo.toml

@@ -59,6 +59,7 @@ slog.workspace = true
 slog-error-chain.workspace = true
 steno.workspace = true
 strum.workspace = true
+support-bundle-reader-lib.workspace = true
 supports-color.workspace = true
 tabled.workspace = true
 textwrap.workspace = true

dev-tools/omdb/src/bin/omdb/nexus.rs

@@ -495,19 +495,33 @@ enum SupportBundleCommands {
     Create,
     /// Delete a support bundle
     Delete(SupportBundleDeleteArgs),
+    /// Download an entire support bundle
+    Download(SupportBundleDownloadArgs),
     /// Download the index of a support bundle
     ///
     /// This is a "list of files", from which individual files can be accessed
     GetIndex(SupportBundleIndexArgs),
-    /// View a file within a support bundle
+    /// Download a single file within a support bundle
     GetFile(SupportBundleFileArgs),
+    /// Creates a dashboard for viewing the contents of a support bundle
+    Inspect(SupportBundleInspectArgs),
 }
 
 #[derive(Debug, Args)]
 struct SupportBundleDeleteArgs {
     id: SupportBundleUuid,
 }
 
+#[derive(Debug, Args)]
+struct SupportBundleDownloadArgs {
+    id: SupportBundleUuid,
+
+    /// Optional output path where the file should be written,
+    /// instead of stdout.
+    #[arg(short, long)]
+    output: Option<Utf8PathBuf>,
+}
+
 #[derive(Debug, Args)]
 struct SupportBundleIndexArgs {
     id: SupportBundleUuid,
@@ -523,6 +537,23 @@ struct SupportBundleFileArgs {
     output: Option<Utf8PathBuf>,
 }
 
+#[derive(Debug, Args)]
+struct SupportBundleInspectArgs {
+    /// A specific bundle to inspect.
+    ///
+    /// If none is supplied, the latest active bundle is used.
+    /// Mutually exclusive with "path".
+    #[arg(short, long)]
+    id: Option<SupportBundleUuid>,
+
+    /// A local bundle file to inspect.
+    ///
+    /// If none is supplied, the latest active bundle is used.
+    /// Mutually exclusive with "id".
+    #[arg(short, long)]
+    path: Option<Utf8PathBuf>,
+}
+
 impl NexusArgs {
     /// Run a `omdb nexus` subcommand.
     pub(crate) async fn run_cmd(
@@ -731,12 +762,18 @@ impl NexusArgs {
                 let token = omdb.check_allow_destructive()?;
                 cmd_nexus_support_bundles_delete(&client, args, token).await
             }
+            NexusCommands::SupportBundles(SupportBundleArgs {
+                command: SupportBundleCommands::Download(args),
+            }) => cmd_nexus_support_bundles_download(&client, args).await,
             NexusCommands::SupportBundles(SupportBundleArgs {
                 command: SupportBundleCommands::GetIndex(args),
             }) => cmd_nexus_support_bundles_get_index(&client, args).await,
             NexusCommands::SupportBundles(SupportBundleArgs {
                 command: SupportBundleCommands::GetFile(args),
             }) => cmd_nexus_support_bundles_get_file(&client, args).await,
+            NexusCommands::SupportBundles(SupportBundleArgs {
+                command: SupportBundleCommands::Inspect(args),
+            }) => cmd_nexus_support_bundles_inspect(&client, args).await,
         }
     }
 }
@@ -3832,6 +3869,28 @@ async fn write_stream_to_sink(
     Ok(())
 }
 
+/// Runs `omdb nexus support-bundles download`
+async fn cmd_nexus_support_bundles_download(
+    client: &nexus_client::Client,
+    args: &SupportBundleDownloadArgs,
+) -> Result<(), anyhow::Error> {
+    let stream = client
+        .support_bundle_download(args.id.as_untyped_uuid())
+        .await
+        .with_context(|| format!("downloading support bundle {}", args.id))?
+        .into_inner_stream();
+
+    let sink: Box<dyn std::io::Write> = match &args.output {
+        Some(path) => Box::new(std::fs::File::create(path)?),
+        None => Box::new(std::io::stdout()),
+    };
+
+    write_stream_to_sink(stream, sink)
+        .await
+        .with_context(|| format!("streaming support bundle {}", args.id))?;
+    Ok(())
+}
+
 /// Runs `omdb nexus support-bundles get-index`
 async fn cmd_nexus_support_bundles_get_index(
     client: &nexus_client::Client,
@@ -3880,3 +3939,16 @@ async fn cmd_nexus_support_bundles_get_file(
     })?;
     Ok(())
 }
+
+/// Runs `omdb nexus support-bundles inspect`
+async fn cmd_nexus_support_bundles_inspect(
+    client: &nexus_client::Client,
+    args: &SupportBundleInspectArgs,
+) -> Result<(), anyhow::Error> {
+    support_bundle_reader_lib::run_dashboard(
+        client,
+        args.id,
+        args.path.as_ref(),
+    )
+    .await
+}

dev-tools/support-bundle-reader-lib/Cargo.toml

@@ -0,0 +1,23 @@
+[package]
+name = "support-bundle-reader-lib"
+version = "0.1.0"
+edition = "2021"
+license = "MPL-2.0"
+
+[lints]
+workspace = true
+
+[dependencies]
+anyhow.workspace = true
+async-trait.workspace = true
+bytes.workspace = true
+camino.workspace = true
+crossterm.workspace = true
+futures.workspace = true
+nexus-client.workspace = true
+omicron-workspace-hack.workspace = true
+omicron-uuid-kinds.workspace = true
+ratatui.workspace = true
+reqwest.workspace = true
+tokio.workspace = true
+zip.workspace = true