src/tools/error_index_generator/build.rs

   1 use std::path::PathBuf;
   2 use std::{env, fs};
   3 use walkdir::WalkDir;
   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
  11     let error_codes_path = "../../../compiler/rustc_error_codes/src/error_codes.rs";
  12
  13     println!("cargo:rerun-if-changed={}", error_codes_path);
  14     let file = fs::read_to_string(error_codes_path)
  15         .unwrap()
  16         .replace(": include_str!(\"./error_codes/", ": include_str!(\"./");
  17     let contents = format!("(|| {{\n{}\n}})()", file);
  18     fs::write(&out_dir.join("all_error_codes.rs"), &contents).unwrap();
  19
  20     // We copy the md files as well to the target directory.
  21     for entry in WalkDir::new("../../../compiler/rustc_error_codes/src/error_codes") {
  22         let entry = entry.unwrap();
  23         match entry.path().extension() {
  24             Some(s) if s == "md" => {}
  25             _ => continue,
  26         }
  27         println!("cargo:rerun-if-changed={}", entry.path().to_str().unwrap());
  28         let md_content = fs::read_to_string(entry.path()).unwrap();
  29         fs::write(&out_dir.join(entry.file_name()), &md_content).unwrap();
  30     }
  31 }