//@ run-pass //! Test that users are able to use serialize stable MIR constructs. //@ ignore-stage1 //@ ignore-cross-compile //@ ignore-remote //@ ignore-windows-gnu mingw has troubles with linking https://github.com/rust-lang/rust/pull/116837 //@ edition: 2021 #![feature(rustc_private)] #![feature(assert_matches)] #[macro_use] extern crate rustc_smir; extern crate rustc_driver; extern crate rustc_interface; extern crate rustc_middle; extern crate stable_mir; extern crate serde; extern crate serde_json; use rustc_middle::ty::TyCtxt; use rustc_smir::rustc_internal; use stable_mir::mir::Body; use std::io::{Write, BufWriter}; use std::ops::ControlFlow; use serde_json::to_string; const CRATE_NAME: &str = "input"; fn serialize_to_json(_tcx: TyCtxt<'_>) -> ControlFlow<()> { let path = "output.json"; let mut writer = BufWriter::new(std::fs::File::create(path) .expect("Failed to create path")); let local_crate = stable_mir::local_crate(); let items: Vec = stable_mir::all_local_items() .iter() .map(|item| { item.body() }) .collect(); let crate_data = ( local_crate.name, items ); writer.write_all(to_string(&crate_data) .expect("serde_json failed") .as_bytes()).expect("JSON serialization failed"); ControlFlow::Continue(()) } /// This test will generate and analyze a dummy crate using the stable mir. /// For that, it will first write the dummy crate into a file. /// Then it will create a `StableMir` using custom arguments and then /// it will run the compiler. fn main() { let path = "internal_input.rs"; generate_input(&path).unwrap(); let args = vec![ "rustc".to_string(), "--crate-name".to_string(), CRATE_NAME.to_string(), path.to_string(), ]; run_with_tcx!(args, serialize_to_json).unwrap(); } fn generate_input(path: &str) -> std::io::Result<()> { let mut file = std::fs::File::create(path)?; write!( file, r#" pub fn main() {{ }} "# )?; Ok(()) }