]> git.lizzy.rs Git - rust.git/blob - src/bootstrap/clean.rs
Rollup merge of #56470 - llogiq:process-termination-doctest, r=GuillaumeGomez
[rust.git] / src / bootstrap / clean.rs
1 //! Implementation of `make clean` in rustbuild.
2 //!
3 //! Responsible for cleaning out a build directory of all old and stale
4 //! artifacts to prepare for a fresh build. Currently doesn't remove the
5 //! `build/cache` directory (download cache) or the `build/$target/llvm`
6 //! directory unless the `--all` flag is present.
7
8 use std::fs;
9 use std::io::{self, ErrorKind};
10 use std::path::Path;
11
12 use crate::Build;
13
14 pub fn clean(build: &Build, all: bool) {
15     rm_rf("tmp".as_ref());
16
17     if all {
18         rm_rf(&build.out);
19     } else {
20         rm_rf(&build.out.join("tmp"));
21         rm_rf(&build.out.join("dist"));
22
23         for host in &build.hosts {
24             let entries = match build.out.join(host).read_dir() {
25                 Ok(iter) => iter,
26                 Err(_) => continue,
27             };
28
29             for entry in entries {
30                 let entry = t!(entry);
31                 if entry.file_name().to_str() == Some("llvm") {
32                     continue
33                 }
34                 let path = t!(entry.path().canonicalize());
35                 rm_rf(&path);
36             }
37         }
38     }
39 }
40
41 fn rm_rf(path: &Path) {
42     match path.symlink_metadata() {
43         Err(e) => {
44             if e.kind() == ErrorKind::NotFound {
45                 return;
46             }
47             panic!("failed to get metadata for file {}: {}", path.display(), e);
48         },
49         Ok(metadata) => {
50             if metadata.file_type().is_file() || metadata.file_type().is_symlink() {
51                 do_op(path, "remove file", |p| fs::remove_file(p));
52                 return;
53             }
54
55             for file in t!(fs::read_dir(path)) {
56                 rm_rf(&t!(file).path());
57             }
58             do_op(path, "remove dir", |p| fs::remove_dir(p));
59         },
60     };
61 }
62
63 fn do_op<F>(path: &Path, desc: &str, mut f: F)
64     where F: FnMut(&Path) -> io::Result<()>
65 {
66     match f(path) {
67         Ok(()) => {}
68         // On windows we can't remove a readonly file, and git will often clone files as readonly.
69         // As a result, we have some special logic to remove readonly files on windows.
70         // This is also the reason that we can't use things like fs::remove_dir_all().
71         Err(ref e) if cfg!(windows) &&
72                       e.kind() == ErrorKind::PermissionDenied => {
73             let mut p = t!(path.symlink_metadata()).permissions();
74             p.set_readonly(false);
75             t!(fs::set_permissions(path, p));
76             f(path).unwrap_or_else(|e| {
77                 panic!("failed to {} {}: {}", desc, path.display(), e);
78             })
79         }
80         Err(e) => {
81             panic!("failed to {} {}: {}", desc, path.display(), e);
82         }
83     }
84 }