3 use std::ffi::OsString;
6 use std::process::Command;
8 use super::rustc_info::{get_file_name, get_rustc_path, get_rustc_version};
9 use super::utils::{copy_dir_recursively, spawn_and_wait};
11 pub(crate) fn prepare() {
14 eprintln!("[INSTALL] hyperfine");
15 Command::new("cargo").arg("install").arg("hyperfine").spawn().unwrap().wait().unwrap();
17 clone_repo_shallow_github(
21 "0f933f9c7176e53b2a3c7952ded484e1783f0bf1",
23 apply_patches("rand", Path::new("rand"));
25 clone_repo_shallow_github(
29 "341f207c1071f7290e3f228c710817c280c8dca1",
32 clone_repo_shallow_github(
36 "b8d6b6844602f80af79cd96401339ec594d472d8",
38 apply_patches("portable-simd", Path::new("portable-simd"));
40 clone_repo_shallow_github(
44 "804a7a21b9e673a482797aa289a18ed480e4d813",
47 eprintln!("[LLVM BUILD] simple-raytracer");
48 let mut build_cmd = Command::new("cargo");
49 build_cmd.arg("build").env_remove("CARGO_TARGET_DIR").current_dir("simple-raytracer");
50 spawn_and_wait(build_cmd);
52 Path::new("simple-raytracer/target/debug").join(get_file_name("main", "bin")),
53 // FIXME use get_file_name here too once testing is migrated to rust
54 "simple-raytracer/raytracer_cg_llvm",
59 fn prepare_sysroot() {
60 let rustc_path = get_rustc_path();
61 let sysroot_src_orig = rustc_path.parent().unwrap().join("../lib/rustlib/src/rust");
62 let sysroot_src = env::current_dir().unwrap().join("build_sysroot").join("sysroot_src");
64 assert!(sysroot_src_orig.exists());
66 if sysroot_src.exists() {
67 fs::remove_dir_all(&sysroot_src).unwrap();
69 fs::create_dir_all(sysroot_src.join("library")).unwrap();
70 eprintln!("[COPY] sysroot src");
71 copy_dir_recursively(&sysroot_src_orig.join("library"), &sysroot_src.join("library"));
73 let rustc_version = get_rustc_version();
74 fs::write(Path::new("build_sysroot").join("rustc_version"), &rustc_version).unwrap();
76 eprintln!("[GIT] init");
77 let mut git_init_cmd = Command::new("git");
78 git_init_cmd.arg("init").arg("-q").current_dir(&sysroot_src);
79 spawn_and_wait(git_init_cmd);
81 init_git_repo(&sysroot_src);
83 apply_patches("sysroot", &sysroot_src);
85 clone_repo_shallow_github(
86 "build_sysroot/compiler-builtins",
91 apply_patches("compiler-builtins", Path::new("build_sysroot/compiler-builtins"));
95 fn clone_repo(target_dir: &str, repo: &str, rev: &str) {
96 eprintln!("[CLONE] {}", repo);
97 // Ignore exit code as the repo may already have been checked out
98 Command::new("git").arg("clone").arg(repo).arg(target_dir).spawn().unwrap().wait().unwrap();
100 let mut clean_cmd = Command::new("git");
101 clean_cmd.arg("checkout").arg("--").arg(".").current_dir(target_dir);
102 spawn_and_wait(clean_cmd);
104 let mut checkout_cmd = Command::new("git");
105 checkout_cmd.arg("checkout").arg("-q").arg(rev).current_dir(target_dir);
106 spawn_and_wait(checkout_cmd);
109 fn clone_repo_shallow_github(target_dir: &str, username: &str, repo: &str, rev: &str) {
111 // Older windows doesn't have tar or curl by default. Fall back to using git.
112 clone_repo(target_dir, &format!("https://github.com/{}/{}.git", username, repo), rev);
116 let archive_url = format!("https://github.com/{}/{}/archive/{}.tar.gz", username, repo, rev);
117 let archive_file = format!("{}.tar.gz", rev);
118 let archive_dir = format!("{}-{}", repo, rev);
120 eprintln!("[DOWNLOAD] {}/{} from {}", username, repo, archive_url);
122 // Remove previous results if they exists
123 let _ = std::fs::remove_file(&archive_file);
124 let _ = std::fs::remove_dir_all(&archive_dir);
125 let _ = std::fs::remove_dir_all(target_dir);
127 // Download zip archive
128 let mut download_cmd = Command::new("curl");
129 download_cmd.arg("--location").arg("--output").arg(&archive_file).arg(archive_url);
130 spawn_and_wait(download_cmd);
132 // Unpack tar archive
133 let mut unpack_cmd = Command::new("tar");
134 unpack_cmd.arg("xf").arg(&archive_file);
135 spawn_and_wait(unpack_cmd);
137 // Rename unpacked dir to the expected name
138 std::fs::rename(archive_dir, target_dir).unwrap();
140 init_git_repo(Path::new(target_dir));
143 std::fs::remove_file(archive_file).unwrap();
146 fn init_git_repo(repo_dir: &Path) {
147 let mut git_init_cmd = Command::new("git");
148 git_init_cmd.arg("init").arg("-q").current_dir(repo_dir);
149 spawn_and_wait(git_init_cmd);
151 let mut git_add_cmd = Command::new("git");
152 git_add_cmd.arg("add").arg(".").current_dir(repo_dir);
153 spawn_and_wait(git_add_cmd);
155 let mut git_commit_cmd = Command::new("git");
156 git_commit_cmd.arg("commit").arg("-m").arg("Initial commit").arg("-q").current_dir(repo_dir);
157 spawn_and_wait(git_commit_cmd);
160 fn get_patches(crate_name: &str) -> Vec<OsString> {
161 let mut patches: Vec<_> = fs::read_dir("patches")
163 .map(|entry| entry.unwrap().path())
164 .filter(|path| path.extension() == Some(OsStr::new("patch")))
165 .map(|path| path.file_name().unwrap().to_owned())
166 .filter(|file_name| {
167 file_name.to_str().unwrap().split_once("-").unwrap().1.starts_with(crate_name)
174 fn apply_patches(crate_name: &str, target_dir: &Path) {
175 for patch in get_patches(crate_name) {
176 eprintln!("[PATCH] {:?} <- {:?}", target_dir.file_name().unwrap(), patch);
177 let patch_arg = env::current_dir().unwrap().join("patches").join(patch);
178 let mut apply_patch_cmd = Command::new("git");
179 apply_patch_cmd.arg("am").arg(patch_arg).arg("-q").current_dir(target_dir);
180 spawn_and_wait(apply_patch_cmd);