tasks: add docs and bench commands

Author: nyonson

tasks/src/bench.rs

@@ -0,0 +1,31 @@
+//! Benchmark testing tasks.
+
+use crate::environment::{get_crate_dirs, quiet_println};
+use crate::quiet_cmd;
+use crate::toolchain::{check_toolchain, Toolchain};
+use xshell::Shell;
+
+/// Run benchmark tests for all crates in the workspace.
+pub fn run(sh: &Shell) -> Result<(), Box<dyn std::error::Error>> {
+    check_toolchain(sh, Toolchain::Nightly)?;
+
+    let crate_dirs = get_crate_dirs(sh)?;
+
+    quiet_println(&format!(
+        "Running bench tests for {} crates",
+        crate_dirs.len()
+    ));
+
+    for crate_dir in &crate_dirs {
+        quiet_println(&format!("Running bench tests in: {}", crate_dir));
+
+        // Use pushd pattern to change and restore directory.
+        let _dir = sh.push_dir(crate_dir);
+
+        quiet_cmd!(sh, "cargo bench")
+            .env("RUSTFLAGS", "--cfg=bench")
+            .run()?;
+    }
+
+    Ok(())
+}

tasks/src/docs.rs

@@ -0,0 +1,30 @@
+//! Documentation building tasks.
+
+use crate::quiet_cmd;
+use crate::toolchain::{check_toolchain, Toolchain};
+use xshell::Shell;
+
+/// Build documentation with stable toolchain.
+pub fn build_docs_stable(sh: &Shell) -> Result<(), Box<dyn std::error::Error>> {
+    check_toolchain(sh, Toolchain::Stable)?;
+
+    quiet_cmd!(sh, "cargo doc --all-features")
+        .env("RUSTDOCFLAGS", "-D warnings")
+        .run()?;
+
+    Ok(())
+}
+
+/// Build documentation with nightly toolchain for docs.rs.
+pub fn build_docs_nightly(sh: &Shell) -> Result<(), Box<dyn std::error::Error>> {
+    check_toolchain(sh, Toolchain::Nightly)?;
+
+    quiet_cmd!(sh, "cargo doc --all-features")
+        .env(
+            "RUSTDOCFLAGS",
+            "--cfg docsrs -D warnings -D rustdoc::broken-intra-doc-links",
+        )
+        .run()?;
+
+    Ok(())
+}

tasks/src/main.rs

@@ -1,3 +1,5 @@
+mod bench;
+mod docs;
 mod environment;
 mod lint;
 mod toolchain;
@@ -20,6 +22,12 @@ struct Cli {
 enum Commands {
     /// Run the linter (clippy) for workspace and all crates.
     Lint,
+    /// Build documentation with stable toolchain.
+    Docs,
+    /// Build documentation with nightly toolchain for docs.rs.
+    Docsrs,
+    /// Run benchmark tests for all crates.
+    Bench,
 }
 
 fn main() {
@@ -35,5 +43,23 @@ fn main() {
                 process::exit(1);
             }
         }
+        Commands::Docs => {
+            if let Err(e) = docs::build_docs_stable(&sh) {
+                eprintln!("Error building docs: {}", e);
+                process::exit(1);
+            }
+        }
+        Commands::Docsrs => {
+            if let Err(e) = docs::build_docs_nightly(&sh) {
+                eprintln!("Error building docs.rs docs: {}", e);
+                process::exit(1);
+            }
+        }
+        Commands::Bench => {
+            if let Err(e) = bench::run(&sh) {
+                eprintln!("Error running bench tests: {}", e);
+                process::exit(1);
+            }
+        }
     }
 }