run cargo fmt after code generation

services/autorust/azure-autorust/src/main.rs

@@ -5,7 +5,7 @@ use autorust_codegen::{
     crates::{list_crate_names, list_dirs},
     gen, get_mgmt_readmes, get_svc_readmes,
     jinja::{CargoToml, CheckAllServicesYml, PublishSdksYml, PublishServicesYml},
-    Result, RunConfig,
+    Error, ErrorKind, Result, RunConfig,
 };
 use clap::Parser;
 
@@ -18,6 +18,10 @@ struct Args {
     /// Specify specific package to generate. Multiple accepted.
     #[clap(long = "package", short = 'p')]
     package: Vec<String>,
+
+    /// Run `cargo fmt` after generating the code
+    #[clap(long, default_value = "true", action = clap::ArgAction::Set)]
+    fmt: bool,
 }
 
 impl Args {
@@ -38,6 +42,9 @@ fn main() -> Result<()> {
             gen_workflow_publish_services()?;
         }
     }
+    if args.fmt {
+        fmt(&args.packages())?;
+    }
     Ok(())
 }
 
@@ -122,3 +129,22 @@ fn gen_workflow_publish_services() -> Result<()> {
     yml.create("../../.github/workflows/publish-services.yml")?;
     Ok(())
 }
+
+/// Run `cargo fmt` on the services workspace or a subset of packages.
+fn fmt(only_packages: &Vec<&str>) -> Result<()> {
+    let services_dir = "../";
+    let mut args = vec!["fmt"];
+    if !only_packages.is_empty() {
+        args.push("-p");
+        for package in only_packages {
+            args.push(package);
+        }
+    }
+    let out = std::process::Command::new("cargo").current_dir(services_dir).args(args).output()?;
+    if !out.status.success() {
+        println!("cargo fmt failed");
+        println!("{}", std::str::from_utf8(&out.stderr)?);
+        return Err(Error::new(ErrorKind::Io, "cargo fmt failed"));
+    }
+    Ok(())
+}