rename all things

1 files changed, 110 insertions, 0 deletions

diff --git a/crates/ra_lsp_server/src/path_map.rs b/crates/ra_lsp_server/src/path_map.rs
new file mode 100644
index 000000000..19c3b1d3b
--- /dev/null
+++ b/crates/ra_lsp_server/src/path_map.rs
@@ -0,0 +1,110 @@
+use std::path::{PathBuf, Path, Component};
+use im;
+use relative_path::RelativePath;
+use ra_analysis::{FileId, FileResolver};
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum Root {
+    Workspace, Lib
+}
+
+#[derive(Debug, Default, Clone)]
+pub struct PathMap {
+    next_id: u32,
+    path2id: im::HashMap<PathBuf, FileId>,
+    id2path: im::HashMap<FileId, PathBuf>,
+    id2root: im::HashMap<FileId, Root>,
+}
+
+impl PathMap {
+    pub fn new() -> PathMap {
+        Default::default()
+    }
+    pub fn get_or_insert(&mut self, path: PathBuf, root: Root) -> FileId {
+        self.path2id.get(path.as_path())
+            .map(|&id| id)
+            .unwrap_or_else(|| {
+                let id = self.new_file_id();
+                self.insert(path, id, root);
+                id
+            })
+    }
+    pub fn get_id(&self, path: &Path) -> Option<FileId> {
+        self.path2id.get(path).map(|&id| id)
+    }
+    pub fn get_path(&self, file_id: FileId) -> &Path {
+        self.id2path.get(&file_id)
+            .unwrap()
+            .as_path()
+    }
+    pub fn get_root(&self, file_id: FileId) -> Root {
+        self.id2root[&file_id]
+    }
+    fn insert(&mut self, path: PathBuf, file_id: FileId, root: Root) {
+        self.path2id.insert(path.clone(), file_id);
+        self.id2path.insert(file_id, path.clone());
+        self.id2root.insert(file_id, root);
+    }
+
+    fn new_file_id(&mut self) -> FileId {
+        let id = FileId(self.next_id);
+        self.next_id += 1;
+        id
+    }
+}
+
+impl FileResolver for PathMap {
+    fn file_stem(&self, file_id: FileId) -> String {
+        self.get_path(file_id).file_stem().unwrap().to_str().unwrap().to_string()
+    }
+
+    fn resolve(&self, file_id: FileId, path: &RelativePath) -> Option<FileId> {
+        let path = path.to_path(&self.get_path(file_id));
+        let path = normalize(&path);
+        self.get_id(&path)
+    }
+}
+
+fn normalize(path: &Path) -> PathBuf {
+    let mut components = path.components().peekable();
+    let mut ret = if let Some(c @ Component::Prefix(..)) = components.peek().cloned() {
+        components.next();
+        PathBuf::from(c.as_os_str())
+    } else {
+        PathBuf::new()
+    };
+
+    for component in components {
+        match component {
+            Component::Prefix(..) => unreachable!(),
+            Component::RootDir => {
+                ret.push(component.as_os_str());
+            }
+            Component::CurDir => {}
+            Component::ParentDir => {
+                ret.pop();
+            }
+            Component::Normal(c) => {
+                ret.push(c);
+            }
+        }
+    }
+    ret
+}
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[test]
+    fn test_resolve() {
+        let mut m = PathMap::new();
+        let id1 = m.get_or_insert(PathBuf::from("/foo"), Root::Workspace);
+        let id2 = m.get_or_insert(PathBuf::from("/foo/bar.rs"), Root::Workspace);
+        assert_eq!(
+            m.resolve(id1, &RelativePath::new("bar.rs")),
+            Some(id2),
+        )
+    }
+}
+