src/tools/error_index_generator/build.rs

   1 use walkdir::WalkDir;
   2 use std::path::PathBuf;
   3 use std::{env, fs};
   4
   5 fn main() {
   6     // The src directory (we are in src/tools/error_index_generator)
   7     // Note that we could skip one of the .. but this ensures we at least loosely find the right
   8     // directory.
   9     let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
  10     let dest = out_dir.join("error_codes.rs");
  11
  12     let error_codes_path = "../../../src/librustc_error_codes/error_codes.rs";
  13
  14     println!("cargo:rerun-if-changed={}", error_codes_path);
  15     let file = fs::read_to_string(error_codes_path).unwrap()
  16                   .replace("crate::register_diagnostics!", "register_diagnostics!")
  17                   .replace(": include_str!(\"./error_codes/", ": include_str!(\"./");
  18     let contents = format!("(|| {{\n{}\n}})()", file);
  19     fs::write(&out_dir.join("all_error_codes.rs"), &contents).unwrap();
  20
  21     // We copy the md files as well to the target directory.
  22     for entry in WalkDir::new("../../../src/librustc_error_codes/error_codes") {
  23         let entry = entry.unwrap();
  24         match entry.path().extension() {
  25             Some(s) if s == "md" => {}
  26             _ => continue,
  27         }
  28         println!("cargo:rerun-if-changed={}", entry.path().to_str().unwrap());
  29         let md_content = fs::read_to_string(entry.path()).unwrap();
  30         fs::write(&out_dir.join(entry.file_name()), &md_content).unwrap();
  31     }
  32
  33     let mut all = String::new();
  34     all.push_str(r###"
  35 fn register_all() -> Vec<(&'static str, Option<&'static str>)> {
  36     let mut long_codes: Vec<(&'static str, Option<&'static str>)> = Vec::new();
  37     macro_rules! register_diagnostics {
  38         ($($ecode:ident: $message:expr,)*) => (
  39             register_diagnostics!{$($ecode:$message,)* ;}
  40         );
  41
  42         ($($ecode:ident: $message:expr,)* ; $($code:ident,)*) => (
  43             $(
  44                 {long_codes.extend([
  45                     (stringify!($ecode), Some($message)),
  46                 ].iter());}
  47             )*
  48             $(
  49                 {long_codes.extend([
  50                     stringify!($code),
  51                 ].iter().cloned().map(|s| (s, None)).collect::<Vec<_>>());}
  52             )*
  53         )
  54     }
  55 "###);
  56     all.push_str(r#"include!(concat!(env!("OUT_DIR"), "/all_error_codes.rs"));"#);
  57     all.push_str("\nlong_codes\n");
  58     all.push_str("}\n");
  59
  60     fs::write(&dest, all).unwrap();
  61 }