zed_util/
fs.rs

1use crate::ResultExt;
2use anyhow::{Result, bail};
3use async_fs as fs;
4use futures_lite::StreamExt;
5use std::path::{Path, PathBuf};
6
7/// Removes all files and directories matching the given predicate
8pub async fn remove_matching<F>(dir: &Path, predicate: F)
9where
10    F: Fn(&Path) -> bool,
11{
12    if let Some(mut entries) = fs::read_dir(dir).await.log_err() {
13        while let Some(entry) = entries.next().await {
14            if let Some(entry) = entry.log_err() {
15                let entry_path = entry.path();
16                if predicate(entry_path.as_path())
17                    && let Ok(metadata) = fs::metadata(&entry_path).await
18                {
19                    if metadata.is_file() {
20                        fs::remove_file(&entry_path).await.log_err();
21                    } else {
22                        fs::remove_dir_all(&entry_path).await.log_err();
23                    }
24                }
25            }
26        }
27    }
28}
29
30pub async fn collect_matching<F>(dir: &Path, predicate: F) -> Vec<PathBuf>
31where
32    F: Fn(&Path) -> bool,
33{
34    let mut matching = vec![];
35
36    if let Some(mut entries) = fs::read_dir(dir).await.log_err() {
37        while let Some(entry) = entries.next().await {
38            if let Some(entry) = entry.log_err()
39                && predicate(entry.path().as_path())
40            {
41                matching.push(entry.path());
42            }
43        }
44    }
45
46    matching
47}
48
49pub async fn find_file_name_in_dir<F>(dir: &Path, predicate: F) -> Option<PathBuf>
50where
51    F: Fn(&str) -> bool,
52{
53    if let Some(mut entries) = fs::read_dir(dir).await.log_err() {
54        while let Some(entry) = entries.next().await {
55            if let Some(entry) = entry.log_err() {
56                let entry_path = entry.path();
57
58                if let Some(file_name) = entry_path
59                    .file_name()
60                    .map(|file_name| file_name.to_string_lossy())
61                    && predicate(&file_name)
62                {
63                    return Some(entry_path);
64                }
65            }
66        }
67    }
68
69    None
70}
71
72pub async fn move_folder_files_to_folder<P: AsRef<Path>>(
73    source_path: P,
74    target_path: P,
75) -> Result<()> {
76    if !target_path.as_ref().is_dir() {
77        bail!("Folder not found or is not a directory");
78    }
79
80    let mut entries = fs::read_dir(source_path.as_ref()).await?;
81    while let Some(entry) = entries.next().await {
82        let entry = entry?;
83        let old_path = entry.path();
84        let new_path = target_path.as_ref().join(entry.file_name());
85
86        fs::rename(&old_path, &new_path).await?;
87    }
88
89    fs::remove_dir(source_path).await?;
90
91    Ok(())
92}
93
94#[cfg(unix)]
95/// Set the permissions for the given path so that the file becomes executable.
96/// This is a noop for non-unix platforms.
97pub async fn make_file_executable(path: &Path) -> std::io::Result<()> {
98    fs::set_permissions(
99        path,
100        <fs::Permissions as fs::unix::PermissionsExt>::from_mode(0o755),
101    )
102    .await
103}
104
105#[cfg(not(unix))]
106#[allow(clippy::unused_async)]
107/// Set the permissions for the given path so that the file becomes executable.
108/// This is a noop for non-unix platforms.
109pub async fn make_file_executable(_path: &Path) -> std::io::Result<()> {
110    Ok(())
111}
zed_util/fs.rs

zed_util/
fs.rs
