Cargo Fmt on Generated Rust Files

crates/cli/src/subcommands/generate/mod.rs

@@ -4,6 +4,7 @@ use std::path::{Path, PathBuf};
 use clap::Arg;
 use clap::ArgAction::SetTrue;
 use convert_case::{Case, Casing};
+use duct::cmd;
 use spacetimedb_lib::sats::{AlgebraicType, Typespace};
 use spacetimedb_lib::{bsatn, MiscModuleExport, ModuleDef, ReducerDef, TableDef, TypeAlias};
 use wasmtime::{AsContext, Caller, ExternType};
@@ -110,10 +111,15 @@ Failed to compile module {:?}. See cargo errors above for more details.",
         ));
     }
 
+    let mut paths = vec![];
     for (fname, code) in generate(&wasm_file, lang, namespace.as_str())?.into_iter() {
-        fs::write(out_dir.join(fname), code)?;
+        let path = out_dir.join(fname);
+        paths.push(path.clone());
+        fs::write(path, code)?;
     }
 
+    format_files(paths, lang)?;
+
     println!("Generate finished successfully.");
     Ok(())
 }
@@ -433,3 +439,19 @@ impl Memory {
             .get(..len as usize)
     }
 }
+
+fn format_files(generated_files: Vec<PathBuf>, lang: Language) -> anyhow::Result<()> {
+    match lang {
+        Language::Rust => {
+            cmd!("rustup", "component", "add", "rustfmt").run()?;
+            for path in generated_files {
+                cmd!("rustfmt", path.to_str().unwrap()).run()?;
+            }
+        }
+        Language::Csharp => {}
+        Language::TypeScript => {}
+        Language::Python => {}
+    }
+
+    Ok(())
+}