1 files changed, 101 insertions, 15 deletions

diff --git a/crates/ra_lsp_server/src/project_model.rs b/crates/ra_lsp_server/src/project_model.rs
index 9f429c9a1..6fbaba7d9 100644
--- a/crates/ra_lsp_server/src/project_model.rs
+++ b/crates/ra_lsp_server/src/project_model.rs
@@ -1,4 +1,7 @@
-use std::path::{Path, PathBuf};
+use std::{
+    path::{Path, PathBuf},
+    process::Command,
+};
 
 use cargo_metadata::{metadata_run, CargoOpt};
 use ra_syntax::SmolStr;
@@ -9,6 +12,36 @@ use thread_worker::{WorkerHandle, Worker};
 
 use crate::Result;
 
+#[derive(Debug, Clone)]
+pub struct ProjectWorkspace {
+    pub(crate) cargo: CargoWorkspace,
+    pub(crate) sysroot: Sysroot,
+}
+
+impl ProjectWorkspace {
+    pub fn discover(path: &Path) -> Result<ProjectWorkspace> {
+        let cargo_toml = find_cargo_toml(path)?;
+        let cargo = CargoWorkspace::from_cargo_metadata(&cargo_toml)?;
+        let sysroot = sysroot_info(&cargo_toml)?;
+        let res = ProjectWorkspace { cargo, sysroot };
+        Ok(res)
+    }
+}
+
+pub fn workspace_loader() -> (Worker<PathBuf, Result<ProjectWorkspace>>, WorkerHandle) {
+    thread_worker::spawn::<PathBuf, Result<ProjectWorkspace>, _>(
+        "workspace loader",
+        1,
+        |input_receiver, output_sender| {
+            input_receiver
+                .into_iter()
+                .map(|path| ProjectWorkspace::discover(path.as_path()))
+                .try_for_each(|it| output_sender.send(it))
+                .unwrap()
+        },
+    )
+}
+
 /// `CargoWorksapce` represents the logical structure of, well, a Cargo
 /// workspace. It pretty closely mirrors `cargo metadata` output.
 ///
@@ -63,6 +96,11 @@ pub enum TargetKind {
     Other,
 }
 
+#[derive(Debug, Clone)]
+pub(crate) struct Sysroot {
+    crates: FxHashMap<SmolStr, PathBuf>,
+}
+
 impl Package {
     pub fn name(self, ws: &CargoWorkspace) -> &str {
         ws.packages[self].name.as_str()
@@ -160,6 +198,68 @@ impl CargoWorkspace {
     }
 }
 
+fn sysroot_info(cargo_toml: &Path) -> Result<Sysroot> {
+    let rustc_output = Command::new("rustc")
+        .current_dir(cargo_toml.parent().unwrap())
+        .args(&["--print", "sysroot"])
+        .output()?;
+    if !rustc_output.status.success() {
+        failure::bail!("failed to locate sysroot")
+    }
+    let stdout = String::from_utf8(rustc_output.stdout)?;
+    let sysroot_path = Path::new(stdout.trim());
+    let src = sysroot_path.join("lib/rustlib/src/rust/src");
+
+    let crates: &[(&str, &[&str])] = &[
+        (
+            "std",
+            &[
+                "alloc_jemalloc",
+                "alloc_system",
+                "panic_abort",
+                "rand",
+                "compiler_builtins",
+                "unwind",
+                "rustc_asan",
+                "rustc_lsan",
+                "rustc_msan",
+                "rustc_tsan",
+                "build_helper",
+            ],
+        ),
+        ("core", &[]),
+        ("alloc", &[]),
+        ("collections", &[]),
+        ("libc", &[]),
+        ("panic_unwind", &[]),
+        ("proc_macro", &[]),
+        ("rustc_unicode", &[]),
+        ("std_unicode", &[]),
+        ("test", &[]),
+        // Feature gated
+        ("alloc_jemalloc", &[]),
+        ("alloc_system", &[]),
+        ("compiler_builtins", &[]),
+        ("getopts", &[]),
+        ("panic_unwind", &[]),
+        ("panic_abort", &[]),
+        ("rand", &[]),
+        ("term", &[]),
+        ("unwind", &[]),
+        // Dependencies
+        ("build_helper", &[]),
+        ("rustc_asan", &[]),
+        ("rustc_lsan", &[]),
+        ("rustc_msan", &[]),
+        ("rustc_tsan", &[]),
+        ("syntax", &[]),
+    ];
+
+    Ok(Sysroot {
+        crates: FxHashMap::default(),
+    })
+}
+
 fn find_cargo_toml(path: &Path) -> Result<PathBuf> {
     if path.ends_with("Cargo.toml") {
         return Ok(path.to_path_buf());
@@ -190,17 +290,3 @@ impl TargetKind {
         TargetKind::Other
     }
 }
-
-pub fn workspace_loader() -> (Worker<PathBuf, Result<CargoWorkspace>>, WorkerHandle) {
-    thread_worker::spawn::<PathBuf, Result<CargoWorkspace>, _>(
-        "workspace loader",
-        1,
-        |input_receiver, output_sender| {
-            input_receiver
-                .into_iter()
-                .map(|path| CargoWorkspace::from_cargo_metadata(path.as_path()))
-                .try_for_each(|it| output_sender.send(it))
-                .unwrap()
-        },
-    )
-}