Option to generate a dot file representing the smir-json

Cargo.toml

@@ -7,6 +7,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+dot-writer = "0.1.4"
 tracing = "0.1"
 # serde = { version = "=1.0.202", features = ["derive"] }
 # serde_cbor = "0.11"

src/lib.rs

@@ -1,5 +1,6 @@
 #![feature(rustc_private)]
 pub mod driver;
+pub mod mk_graph;
 pub mod printer;
 pub use driver::stable_mir_driver;
 pub use printer::*;

src/main.rs

@@ -4,8 +4,23 @@ pub mod driver;
 pub mod printer;
 use driver::stable_mir_driver;
 use printer::emit_smir;
+use smir_pretty::mk_graph::emit_dotfile;
 
 fn main() {
-    let args: Vec<_> = env::args().collect();
-    stable_mir_driver(&args, emit_smir)
+    let mut args: Vec<String> = env::args().collect();
+
+    match args.get(1) {
+        None =>
+            stable_mir_driver(&args, emit_smir), // backward compatibility
+        Some(arg) if arg == "--json" => {
+            args.remove(1);
+            stable_mir_driver(  &args, emit_smir)
+        }
+        Some(arg) if arg == "--dot" => {
+            args.remove(1);
+            stable_mir_driver(&args, emit_dotfile)
+        }
+        Some(_other) =>
+            stable_mir_driver(&args, emit_smir), // backward compatibility
+    }
 }