| //! In-memory document information. |
| |
| use std::mem; |
| |
| use rustc_hash::FxHashMap; |
| use vfs::VfsPath; |
| |
| /// Holds the set of in-memory documents. |
| /// |
| /// For these document, their true contents is maintained by the client. It |
| /// might be different from what's on disk. |
| #[derive(Default, Clone)] |
| pub(crate) struct MemDocs { |
| mem_docs: FxHashMap<VfsPath, DocumentData>, |
| added_or_removed: bool, |
| } |
| |
| impl MemDocs { |
| pub(crate) fn contains(&self, path: &VfsPath) -> bool { |
| self.mem_docs.contains_key(path) |
| } |
| |
| pub(crate) fn insert(&mut self, path: VfsPath, data: DocumentData) -> Result<(), ()> { |
| self.added_or_removed = true; |
| match self.mem_docs.insert(path, data) { |
| Some(_) => Err(()), |
| None => Ok(()), |
| } |
| } |
| |
| pub(crate) fn remove(&mut self, path: &VfsPath) -> Result<(), ()> { |
| self.added_or_removed = true; |
| match self.mem_docs.remove(path) { |
| Some(_) => Ok(()), |
| None => Err(()), |
| } |
| } |
| |
| pub(crate) fn get(&self, path: &VfsPath) -> Option<&DocumentData> { |
| self.mem_docs.get(path) |
| } |
| |
| pub(crate) fn get_mut(&mut self, path: &VfsPath) -> Option<&mut DocumentData> { |
| // NB: don't set `self.added_or_removed` here, as that purposefully only |
| // tracks changes to the key set. |
| self.mem_docs.get_mut(path) |
| } |
| |
| pub(crate) fn iter(&self) -> impl Iterator<Item = &VfsPath> { |
| self.mem_docs.keys() |
| } |
| |
| pub(crate) fn take_changes(&mut self) -> bool { |
| mem::replace(&mut self.added_or_removed, false) |
| } |
| } |
| |
| /// Information about a document that the Language Client |
| /// knows about. |
| /// Its lifetime is driven by the textDocument/didOpen and textDocument/didClose |
| /// client notifications. |
| #[derive(Debug, Clone)] |
| pub(crate) struct DocumentData { |
| pub(crate) version: i32, |
| pub(crate) data: Vec<u8>, |
| } |
| |
| impl DocumentData { |
| pub(crate) fn new(version: i32, data: Vec<u8>) -> Self { |
| DocumentData { version, data } |
| } |
| } |