introduce expander (#4974)

* introduce expander

* unify towards expand

* bubble up expand

* spelling

Co-authored-by: Andrei Sandu <54316454+sandreim@users.noreply.github.com>

* add newlines

Co-authored-by: Andrei Sandu <54316454+sandreim@users.noreply.github.com>

polkadot/node/overseer/overseer-gen/proc-macro/Cargo.toml

@@ -16,12 +16,13 @@ syn = { version = "1.0.86", features = ["full", "extra-traits"] }
 quote = "1.0.15"
 proc-macro2 = "1.0.36"
 proc-macro-crate = "1.1.2"
+expander = "0.0.4"
 
 [dev-dependencies]
 assert_matches = "1.5.0"
 
 [features]
 default = []
-# write the expanded version to a `overlord-expansion.rs`
-# in the `OUT_DIR` as defined by `cargo`.
-expansion = []
+# write the expanded version to a `overlord-expansion.[a-f0-9]{10}.rs`
+# in the `OUT_DIR` as defined by `cargo` for the `expander` crate.
+expand = []

polkadot/node/overseer/overseer-gen/proc-macro/build.rs

@@ -1,2 +0,0 @@
-/// A dummy build script, so `OUT_DIR` is set.
-fn main() {}

polkadot/node/overseer/overseer-gen/proc-macro/src/lib.rs

@@ -98,36 +98,13 @@ pub(crate) fn impl_overseer_gen(
 	additive.extend(impl_message_wrapper_enum(&info)?);
 	additive.extend(impl_dispatch(&info));
 
-	// Write to a file for expansion, and then use it via `include!()`
-	// in order to obtain better compiler errors when modifying `overlord`.
-	if cfg!(feature = "expansion") {
-		use std::io::Write;
+	let ts = expander::Expander::new("overlord-expansion")
+		.add_comment("Generated overseer code by `#[overlord(..)]`".to_owned())
+		.dry(!cfg!(feature = "expand"))
+		.verbose(false)
+		.fmt(expander::Edition::_2018)
+		.write_to_out_dir(additive)
+		.expect("Expander does not fail due to IO in OUT_DIR. qed");
 
-		let out = env!("OUT_DIR");
-		let out = std::path::PathBuf::from(out);
-		let path = out.join("overlord-expansion.rs");
-		let mut f = std::fs::OpenOptions::new()
-			.write(true)
-			.create(true)
-			.truncate(true)
-			.open(&path)
-			.expect("File exists. qed");
-		f.write_all(
-			&mut format!("// {:?} \n{}", std::time::SystemTime::now(), additive).as_bytes(),
-		)
-		.expect("Got permissions to write to file. qed");
-		std::process::Command::new("rustfmt")
-			.arg("--edition=2018")
-			.arg(&path)
-			.current_dir(out)
-			.spawn()
-			.expect("Running rustfmt works. qed");
-
-		let path = path.display().to_string();
-		Ok(quote! {
-			include!( #path );
-		})
-	} else {
-		Ok(additive)
-	}
+	Ok(ts)
 }