initial commit of smir-pretty

Author: sskeirik

Diff for: .gitignore

@@ -0,0 +1 @@
+target/

Diff for: Cargo.lock

@@ -0,0 +1,17 @@
+[package]
+name = "smir-pretty"
+version = "0.1.0"
+edition = "2021"
+# rust-version = "1.78.0" # I think we get latest available rust by unsetting this and setting rust-toolchain.toml.toolchain.channel = nightly
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+# serde = { version = "1.0.104", features = ["derive"] }
+# serde_cbor = "0.11"
+# serde_json = "1.0"
+# tar = "0.4"
+
+[package.metadata.rust-analyzer]
+# This package uses rustc crates.
+rustc_private=true

Diff for: Cargo.toml

@@ -0,0 +1,2 @@
+[toolchain]
+channel = "nightly"

Diff for: rust-toolchain.toml

@@ -0,0 +1,54 @@
+//! This module provides a compiler driver such that:
+//!
+//! 1.  the rustc compiler context is available
+//! 2.  the rustc `stable_mir` APIs are available
+//!
+//! It exports a single function:
+//!
+//! ```rust
+//! stable_mir_driver(args: &Vec<String>, callback_fn: fn (TyCtxt) -> () )
+//! ```
+//!
+//! Calling this function is essentially equivalent to the following macro call:
+//!
+//! ```rust
+//! rustc_internal::run_with_tcx!( args, callback_fn );
+//! ```
+//!
+//! However, we prefer a non-macro version for clarity and build simplicity.
+
+extern crate rustc_middle;
+extern crate rustc_driver;
+extern crate rustc_interface;
+extern crate rustc_smir;
+use rustc_middle::ty::TyCtxt;
+use rustc_driver::Compilation;
+use rustc_interface::{interface::Compiler, Queries};
+use rustc_smir::rustc_internal;
+
+struct StableMirCallbacks {
+  callback_fn: fn (TyCtxt) -> (),
+}
+
+impl rustc_driver::Callbacks for StableMirCallbacks {
+    fn after_analysis<'tcx>(
+        &mut self,
+        _compiler: &Compiler,
+        queries: &'tcx Queries<'tcx>,
+    ) -> Compilation {
+
+        let _q = queries
+            .global_ctxt()
+            .unwrap()
+            .get_mut()
+            .enter(|tcx| {
+                let _ = rustc_internal::run(tcx, || (self.callback_fn)(tcx));
+            });
+        Compilation::Stop
+    }
+}
+
+pub fn stable_mir_driver(args_outer: &Vec<String>, callback_fn: fn (TyCtxt) -> ()) {
+    let mut callbacks = StableMirCallbacks { callback_fn };
+    rustc_driver::RunCompiler::new(args_outer, &mut callbacks).run().unwrap()
+}

Diff for: src/driver.rs

@@ -0,0 +1,7 @@
+#![feature(rustc_private)]
+pub mod driver;
+pub mod printer;
+pub mod pretty;
+pub use driver::stable_mir_driver;
+pub use printer::*;
+pub use pretty::*;

Diff for: src/lib.rs

@@ -0,0 +1,11 @@
+#![feature(rustc_private)]
+use std::env;
+pub mod driver;
+pub mod printer;
+use driver::stable_mir_driver;
+use printer::print_all_items;
+
+fn main() {
+    let args: Vec<_> = env::args().collect();
+    stable_mir_driver(&args, print_all_items)
+}