From da49b80c3525db7f0a0aae0fe2ac3e0a9a574cd3 Mon Sep 17 00:00:00 2001 From: Ben S Date: Sat, 9 May 2015 16:10:26 +0100 Subject: Clean up filetype-detecting code --- src/filetype.rs | 207 ++++++++++++++++++++++++++++---------------------------- 1 file changed, 104 insertions(+), 103 deletions(-) (limited to 'src/filetype.rs') diff --git a/src/filetype.rs b/src/filetype.rs index cd6a230..a1b39c3 100644 --- a/src/filetype.rs +++ b/src/filetype.rs @@ -5,54 +5,13 @@ use ansi_term::Style; use ansi_term::Style::Plain; use ansi_term::Colour::{Red, Green, Yellow, Blue, Cyan, Fixed}; -#[derive(PartialEq, Debug)] +#[derive(Debug)] pub enum FileType { Normal, Directory, Executable, Immediate, Compiled, Symlink, Special, Image, Video, Music, Lossless, Compressed, Document, Temp, Crypto, } -static IMAGE_TYPES: &'static [&'static str] = &[ - "png", "jpeg", "jpg", "gif", "bmp", "tiff", "tif", - "ppm", "pgm", "pbm", "pnm", "webp", "raw", "arw", - "svg", "stl", "eps", "dvi", "ps", "cbr", - "cbz", "xpm", "ico" ]; - -static VIDEO_TYPES: &'static [&'static str] = &[ - "avi", "flv", "m2v", "mkv", "mov", "mp4", "mpeg", - "mpg", "ogm", "ogv", "vob", "wmv" ]; - -static MUSIC_TYPES: &'static [&'static str] = &[ - "aac", "m4a", "mp3", "ogg", "wma" ]; - -static MUSIC_LOSSLESS: &'static [&'static str] = &[ - "alac", "ape", "flac", "wav" ]; - -static COMPRESSED_TYPES: &'static [&'static str] = &[ - "zip", "tar", "Z", "gz", "bz2", "a", "ar", "7z", - "iso", "dmg", "tc", "rar", "par" ]; - -static DOCUMENT_TYPES: &'static [&'static str] = &[ - "djvu", "doc", "docx", "dvi", "eml", "eps", "fotd", - "odp", "odt", "pdf", "ppt", "pptx", "rtf", - "xls", "xlsx" ]; - -static TEMP_TYPES: &'static [&'static str] = &[ - "tmp", "swp", "swo", "swn", "bak" ]; - -static CRYPTO_TYPES: &'static [&'static str] = &[ - "asc", "gpg", "sig", "signature", "pgp" ]; - -static COMPILED_TYPES: &'static [&'static str] = &[ - "class", "elc", "hi", "o", "pyc" ]; - -static BUILD_TYPES: &'static [&'static str] = &[ - "Makefile", "Cargo.toml", "SConstruct", "CMakeLists.txt", - "build.gradle", "Rakefile", "Gruntfile.js", - "Gruntfile.coffee" ]; - impl FileType { - - /// Get the `ansi_term::Style` that a file of this type should use. pub fn style(&self) -> Style { match *self { Normal => Plain, @@ -74,72 +33,115 @@ impl FileType { } } -pub trait HasType { - /// For a given file, find out what type it has. - fn get_type(&self) -> FileType; +pub fn file_type(file: &File) -> FileType { + match file { + f if f.is_directory() => Directory, + f if f.is_executable_file() => Executable, + f if f.is_link() => Symlink, + f if !f.is_file() => Special, + f if f.is_immediate() => Immediate, + f if f.is_image() => Image, + f if f.is_video() => Video, + f if f.is_music() => Music, + f if f.is_lossless() => Lossless, + f if f.is_crypto() => Crypto, + f if f.is_document() => Document, + f if f.is_compressed() => Compressed, + f if f.is_temp() => Temp, + f if f.is_compiled() => Compiled, + _ => Normal, + } } -impl<'a> HasType for File<'a> { - fn get_type(&self) -> FileType { +trait FileTypes { + fn is_immediate(&self) -> bool; + fn is_image(&self) -> bool; + fn is_video(&self) -> bool; + fn is_music(&self) -> bool; + fn is_lossless(&self) -> bool; + fn is_crypto(&self) -> bool; + fn is_document(&self) -> bool; + fn is_compressed(&self) -> bool; + fn is_temp(&self) -> bool; + fn is_compiled(&self) -> bool; +} - if self.is_directory() { - return Directory; - } - else if self.is_executable_file() { - return Executable; - } - else if self.is_link() { - return Symlink; +impl<'_> FileTypes for File<'_> { + fn is_immediate(&self) -> bool { + self.name.starts_with("README") || self.name_is_one_of( &[ + "Makefile", "Cargo.toml", "SConstruct", "CMakeLists.txt", + "build.gradle", "Rakefile", "Gruntfile.js", + "Gruntfile.coffee", + ]) + } + + fn is_image(&self) -> bool { + self.extension_is_one_of( &[ + "png", "jpeg", "jpg", "gif", "bmp", "tiff", "tif", + "ppm", "pgm", "pbm", "pnm", "webp", "raw", "arw", + "svg", "stl", "eps", "dvi", "ps", "cbr", + "cbz", "xpm", "ico", + ]) + } + + fn is_video(&self) -> bool { + self.extension_is_one_of( &[ + "avi", "flv", "m2v", "mkv", "mov", "mp4", "mpeg", + "mpg", "ogm", "ogv", "vob", "wmv", + ]) + } + + fn is_music(&self) -> bool { + self.extension_is_one_of( &[ + "aac", "m4a", "mp3", "ogg", "wma", + ]) + } + + fn is_lossless(&self) -> bool { + self.extension_is_one_of( &[ + "alac", "ape", "flac", "wav", + ]) + } + + fn is_crypto(&self) -> bool { + self.extension_is_one_of( &[ + "zip", "tar", "Z", "gz", "bz2", "a", "ar", "7z", + "iso", "dmg", "tc", "rar", "par", + ]) + } + + fn is_document(&self) -> bool { + self.extension_is_one_of( &[ + "djvu", "doc", "docx", "dvi", "eml", "eps", "fotd", + "odp", "odt", "pdf", "ppt", "pptx", "rtf", + "xls", "xlsx", + ]) + } + + fn is_compressed(&self) -> bool { + self.extension_is_one_of( &[ + "zip", "tar", "Z", "gz", "bz2", "a", "ar", "7z", + "iso", "dmg", "tc", "rar", "par" + ]) + } + + fn is_temp(&self) -> bool { + self.name.ends_with("~") + || (self.name.starts_with("#") && self.name.ends_with("#")) + || self.extension_is_one_of( &[ "tmp", "swp", "swo", "swn", "bak" ]) + } + + fn is_compiled(&self) -> bool { + if self.extension_is_one_of( &[ "class", "elc", "hi", "o", "pyc" ]) { + true } - else if !self.is_file() { - return Special; + else if let Some(dir) = self.dir { + self.get_source_files().iter().any(|path| dir.contains(path)) } - - if self.name.starts_with("README") || BUILD_TYPES.contains(&&self.name[..]) { - return Immediate; - } - else if let Some(ref ext) = self.ext { - if IMAGE_TYPES.contains(&&ext[..]) { - return Image; - } - else if VIDEO_TYPES.contains(&&ext[..]) { - return Video; - } - else if MUSIC_TYPES.contains(&&ext[..]) { - return Music; - } - else if MUSIC_LOSSLESS.contains(&&ext[..]) { - return Lossless; - } - else if CRYPTO_TYPES.contains(&&ext[..]) { - return Crypto; - } - else if DOCUMENT_TYPES.contains(&&ext[..]) { - return Document; - } - else if COMPRESSED_TYPES.contains(&&ext[..]) { - return Compressed; - } - else if self.is_tmpfile() || TEMP_TYPES.contains(&&ext[..]) { - return Temp; - } - - let source_files = self.get_source_files(); - if source_files.is_empty() { - return Normal; - } - else if let Some(dir) = self.dir { - if source_files.iter().any(|path| dir.contains(path)) { - return Temp; - } - } - else if COMPILED_TYPES.contains(&&ext[..]) { - return Compiled; - } + else { + false } - - return Normal; // no filetype - } + } } #[cfg(broken_test)] @@ -170,5 +172,4 @@ mod test { let file = new_file(dummy_stat(), "/cargo.toml"); assert_eq!(FileType::Normal, file.get_type()) } - } -- cgit v1.2.3