diff options
Diffstat (limited to 'crates/ra_project_model/src/sysroot.rs')
-rw-r--r-- | crates/ra_project_model/src/sysroot.rs | 138 |
1 files changed, 138 insertions, 0 deletions
diff --git a/crates/ra_project_model/src/sysroot.rs b/crates/ra_project_model/src/sysroot.rs new file mode 100644 index 000000000..18824dbe5 --- /dev/null +++ b/crates/ra_project_model/src/sysroot.rs | |||
@@ -0,0 +1,138 @@ | |||
1 | use std::{ | ||
2 | path::{Path, PathBuf}, | ||
3 | process::Command, | ||
4 | }; | ||
5 | |||
6 | use smol_str::SmolStr; | ||
7 | |||
8 | use ra_arena::{Arena, RawId, impl_arena_id}; | ||
9 | |||
10 | use crate::Result; | ||
11 | |||
12 | #[derive(Debug, Clone)] | ||
13 | pub struct Sysroot { | ||
14 | crates: Arena<SysrootCrate, SysrootCrateData>, | ||
15 | } | ||
16 | |||
17 | #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] | ||
18 | pub struct SysrootCrate(RawId); | ||
19 | impl_arena_id!(SysrootCrate); | ||
20 | |||
21 | #[derive(Debug, Clone)] | ||
22 | struct SysrootCrateData { | ||
23 | name: SmolStr, | ||
24 | root: PathBuf, | ||
25 | deps: Vec<SysrootCrate>, | ||
26 | } | ||
27 | |||
28 | impl Sysroot { | ||
29 | pub fn std(&self) -> Option<SysrootCrate> { | ||
30 | self.by_name("std") | ||
31 | } | ||
32 | |||
33 | pub fn crates<'a>(&'a self) -> impl Iterator<Item = SysrootCrate> + 'a { | ||
34 | self.crates.iter().map(|(id, _data)| id) | ||
35 | } | ||
36 | |||
37 | pub fn discover(cargo_toml: &Path) -> Result<Sysroot> { | ||
38 | let rustc_output = Command::new("rustc") | ||
39 | .current_dir(cargo_toml.parent().unwrap()) | ||
40 | .args(&["--print", "sysroot"]) | ||
41 | .output()?; | ||
42 | if !rustc_output.status.success() { | ||
43 | failure::bail!("failed to locate sysroot") | ||
44 | } | ||
45 | let stdout = String::from_utf8(rustc_output.stdout)?; | ||
46 | let sysroot_path = Path::new(stdout.trim()); | ||
47 | let src = sysroot_path.join("lib/rustlib/src/rust/src"); | ||
48 | if !src.exists() { | ||
49 | failure::bail!( | ||
50 | "can't load standard library from sysroot\n\ | ||
51 | {:?}\n\ | ||
52 | try running `rustup component add rust-src`", | ||
53 | src, | ||
54 | ); | ||
55 | } | ||
56 | |||
57 | let mut sysroot = Sysroot { crates: Arena::default() }; | ||
58 | for name in SYSROOT_CRATES.trim().lines() { | ||
59 | let root = src.join(format!("lib{}", name)).join("lib.rs"); | ||
60 | if root.exists() { | ||
61 | sysroot.crates.alloc(SysrootCrateData { | ||
62 | name: name.into(), | ||
63 | root, | ||
64 | deps: Vec::new(), | ||
65 | }); | ||
66 | } | ||
67 | } | ||
68 | if let Some(std) = sysroot.std() { | ||
69 | for dep in STD_DEPS.trim().lines() { | ||
70 | if let Some(dep) = sysroot.by_name(dep) { | ||
71 | sysroot.crates[std].deps.push(dep) | ||
72 | } | ||
73 | } | ||
74 | } | ||
75 | Ok(sysroot) | ||
76 | } | ||
77 | |||
78 | fn by_name(&self, name: &str) -> Option<SysrootCrate> { | ||
79 | self.crates.iter().find(|(_id, data)| data.name == name).map(|(id, _data)| id) | ||
80 | } | ||
81 | } | ||
82 | |||
83 | impl SysrootCrate { | ||
84 | pub fn name(self, sysroot: &Sysroot) -> &SmolStr { | ||
85 | &sysroot.crates[self].name | ||
86 | } | ||
87 | pub fn root(self, sysroot: &Sysroot) -> &Path { | ||
88 | sysroot.crates[self].root.as_path() | ||
89 | } | ||
90 | pub fn root_dir(self, sysroot: &Sysroot) -> &Path { | ||
91 | self.root(sysroot).parent().unwrap() | ||
92 | } | ||
93 | pub fn deps<'a>(self, sysroot: &'a Sysroot) -> impl Iterator<Item = SysrootCrate> + 'a { | ||
94 | sysroot.crates[self].deps.iter().map(|&it| it) | ||
95 | } | ||
96 | } | ||
97 | |||
98 | const SYSROOT_CRATES: &str = " | ||
99 | std | ||
100 | core | ||
101 | alloc | ||
102 | collections | ||
103 | libc | ||
104 | panic_unwind | ||
105 | proc_macro | ||
106 | rustc_unicode | ||
107 | std_unicode | ||
108 | test | ||
109 | alloc_jemalloc | ||
110 | alloc_system | ||
111 | compiler_builtins | ||
112 | getopts | ||
113 | panic_unwind | ||
114 | panic_abort | ||
115 | rand | ||
116 | term | ||
117 | unwind | ||
118 | build_helper | ||
119 | rustc_asan | ||
120 | rustc_lsan | ||
121 | rustc_msan | ||
122 | rustc_tsan | ||
123 | syntax"; | ||
124 | |||
125 | const STD_DEPS: &str = " | ||
126 | alloc | ||
127 | alloc_jemalloc | ||
128 | alloc_system | ||
129 | core | ||
130 | panic_abort | ||
131 | rand | ||
132 | compiler_builtins | ||
133 | unwind | ||
134 | rustc_asan | ||
135 | rustc_lsan | ||
136 | rustc_msan | ||
137 | rustc_tsan | ||
138 | build_helper"; | ||