vikarti.anatra/polaris-mirror
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Only list music files in browse/flatten

Browse source
This commit is contained in:
Antoine Gersant 2016-08-29 22:44:33 -07:00
parent df1c2b8b59
commit 1e2ed34dd1
1 changed files with 34 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
src/collection.rs
View file

@ -15,7 +15,7 @@ pub struct Song {
} }
impl Song { impl Song {
pub fn read(collection: &Collection, path: &Path) -> Result<Song, PError> { fn read(collection: &Collection, path: &Path) -> Result<Song, PError> {
let virtual_path = try!(collection.vfs.real_to_virtual(path)); let virtual_path = try!(collection.vfs.real_to_virtual(path));
let path_string = try!(virtual_path.to_str().ok_or(PError::PathDecoding)); let path_string = try!(virtual_path.to_str().ok_or(PError::PathDecoding));
@ -28,6 +28,24 @@ impl Song {
display_name: display_name, display_name: display_name,
}) })
} }
fn is_song(path: &Path) -> bool {
let extension = match path.extension() {
Some(e) => e,
_ => return false,
};
let extension = match extension.to_str() {
Some(e) => e,
_ => return false,
};
match extension {
"mp3" => return true,
"ogg" => return true,
"m4a" => return true,
"flac" => return true,
_ => return false,
}
}
} }
#[derive(Debug, RustcEncodable)] #[derive(Debug, RustcEncodable)]
@ -37,7 +55,7 @@ pub struct Directory {
} }
impl Directory { impl Directory {
pub fn read(collection: &Collection, path: &Path) -> Result<Directory, PError> { fn read(collection: &Collection, path: &Path) -> Result<Directory, PError> {
let virtual_path = try!(collection.vfs.real_to_virtual(path)); let virtual_path = try!(collection.vfs.real_to_virtual(path));
let path_string = try!(virtual_path.to_str().ok_or(PError::PathDecoding)); let path_string = try!(virtual_path.to_str().ok_or(PError::PathDecoding));
@ -154,17 +172,20 @@ impl Collection {
for file in try!(fs::read_dir(full_path)) { for file in try!(fs::read_dir(full_path)) {
let file = try!(file); let file = try!(file);
let file_meta = try!(file.metadata()); let file_meta = try!(file.metadata());
let file_path = file.path();
let file_path = file_path.as_path();
if file_meta.is_file() { if file_meta.is_file() {
let song = try!(Song::read(self, file.path().as_path())); if Song::is_song( file_path ) {
let song = try!(Song::read(self, file_path));
out.push(CollectionFile::Song(song)); out.push(CollectionFile::Song(song));
}
} else if file_meta.is_dir() { } else if file_meta.is_dir() {
let directory = try!(Directory::read(self, file.path().as_path())); let directory = try!(Directory::read(self, file_path));
out.push(CollectionFile::Directory(directory)); out.push(CollectionFile::Directory(directory));
} }
} }
} }
Ok(out) Ok(out)
} }
@ -174,13 +195,15 @@ impl Collection {
let mut acc = try!(acc); let mut acc = try!(acc);
let file: fs::DirEntry = try!(file); let file: fs::DirEntry = try!(file);
let file_meta = try!(file.metadata()); let file_meta = try!(file.metadata());
let file_path = file.path();
let file_path = file_path.as_path();
if file_meta.is_file() { if file_meta.is_file() {
let song = try!(Song::read(self, file.path().as_path())); if Song::is_song(file_path) {
let song = try!(Song::read(self, file_path));
acc.push(song); acc.push(song);
}
} else { } else {
let explore_path = file.path(); let mut explore_content = try!(self.flatten_internal(file_path));
let explore_path = explore_path.as_path();
let mut explore_content = try!(self.flatten_internal(explore_path));
acc.append(&mut explore_content); acc.append(&mut explore_content);
} }
Ok(acc) Ok(acc)