mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2025-01-12 21:32:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

Implement InMemoryFs::rename

Browse source
This commit is contained in:
Antonio Scandurra 2021-07-09 17:02:42 +02:00
parent 63e76e426a
commit 3d799fe8e7
1 changed files with 34 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
zed/src/worktree.rs
View file

@ -204,14 +204,15 @@ struct InMemoryFsState {
impl InMemoryFsState {
fn validate_path(&self, path: &Path) -> Result<()> {
if path
.parent()
.and_then(|path| self.entries.get(path))
.map_or(false, |e| e.is_dir)
if path.is_absolute()
&& path
.parent()
.and_then(|path| self.entries.get(path))
.map_or(false, |e| e.is_dir)
{
Ok(())
} else {
Err(anyhow!("invalid "))
Err(anyhow!("invalid path {:?}", path))
}
}
@ -277,23 +278,37 @@ impl InMemoryFs {
pub async fn remove(&self, path: &Path) -> Result<()> {
let mut state = self.state.write().await;
state.validate_path(path)?;
let mut paths = Vec::new();
state.entries.retain(|path, _| {
if path.starts_with(path) {
paths.push(path.to_path_buf());
false
} else {
true
}
});
for path in paths {
state.emit_event(&path).await;
}
state.entries.retain(|path, _| !path.starts_with(path));
state.emit_event(&path).await;
Ok(())
}
pub async fn rename(&self, source: &Path, target: &Path) -> Result<()> {
let mut state = self.state.write().await;
state.validate_path(source)?;
state.validate_path(target)?;
if state.entries.contains_key(target) {
Err(anyhow!("target path already exists"))
} else {
let mut removed = Vec::new();
state.entries.retain(|path, entry| {
if let Ok(relative_path) = path.strip_prefix(source) {
removed.push((relative_path.to_path_buf(), entry.clone()));
false
} else {
true
}
});
for (relative_path, entry) in removed {
let new_path = target.join(relative_path);
state.entries.insert(new_path, entry);
}
Ok(())
}
}
pub async fn events(&self) -> broadcast::Receiver<fsevent::Event> {
self.state.read().await.events_tx.subscribe()
}