leyoda/cosmic-settings
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

perf(wallpaper): ignore files that are not images

Browse source
This commit is contained in:
Michael Aaron Murphy 2023-12-14 04:16:27 +01:00
parent 8000a67442
commit 2be76211e8
No known key found for this signature in database
GPG key ID: B2732D4240C9212C
3 changed files with 39 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pages/desktop/src/wallpaper.rs
View file

@ -93,8 +93,11 @@ pub fn load_each_from_path(path: PathBuf) -> Receiver<(PathBuf, RgbaImage, RgbaI
let (tx, rx) = mpsc::channel(1);
tokio::task::spawn(async move {
// Scratch space for storing images into.
let mut buffer = Vec::new();
// Directories to search recursively.
let mut paths = vec![path];
// Discovered image files that will be loaded as wallpapers.
let mut wallpapers = BTreeSet::new();
while let Some(path) = paths.pop() {
@ -106,10 +109,17 @@ pub fn load_each_from_path(path: PathBuf) -> Receiver<(PathBuf, RgbaImage, RgbaI
let path = entry.path();
// Recursively search directories, while storing only image files.
if file_type.is_dir() {
paths.push(path);
} else if file_type.is_file() {
wallpapers.insert(path);
let Ok(Some(kind)) = infer::get_from_path(&path) else {
continue
};
if infer::MatcherType::Image == kind.matcher_type() {
wallpapers.insert(path);
}
}
}
}