filetype.rs 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
  1. //! Tests for various types of file (video, image, compressed, etc).
  2. //!
  3. //! Currently this is dependent on the file’s name and extension, because
  4. //! those are the only metadata that we have access to without reading the
  5. //! file’s contents.
  6. use fs::File;
  7. impl<'a> File<'a> {
  8. /// An “immediate” file is something that can be run or activated somehow
  9. /// in order to kick off the build of a project. It’s usually only present
  10. /// in directories full of source code.
  11. pub fn is_immediate(&self) -> bool {
  12. self.name.starts_with("README") || self.name_is_one_of( &[
  13. "Makefile", "Cargo.toml", "SConstruct", "CMakeLists.txt",
  14. "build.gradle", "Rakefile", "Gruntfile.js",
  15. "Gruntfile.coffee",
  16. ])
  17. }
  18. pub fn is_image(&self) -> bool {
  19. self.extension_is_one_of( &[
  20. "png", "jpeg", "jpg", "gif", "bmp", "tiff", "tif",
  21. "ppm", "pgm", "pbm", "pnm", "webp", "raw", "arw",
  22. "svg", "stl", "eps", "dvi", "ps", "cbr",
  23. "cbz", "xpm", "ico", "cr2",
  24. ])
  25. }
  26. pub fn is_video(&self) -> bool {
  27. self.extension_is_one_of( &[
  28. "avi", "flv", "m2v", "mkv", "mov", "mp4", "mpeg",
  29. "mpg", "ogm", "ogv", "vob", "wmv",
  30. ])
  31. }
  32. pub fn is_music(&self) -> bool {
  33. self.extension_is_one_of( &[
  34. "aac", "m4a", "mp3", "ogg", "wma",
  35. ])
  36. }
  37. // Lossless music, rather than any other kind of data...
  38. pub fn is_lossless(&self) -> bool {
  39. self.extension_is_one_of( &[
  40. "alac", "ape", "flac", "wav",
  41. ])
  42. }
  43. pub fn is_crypto(&self) -> bool {
  44. self.extension_is_one_of( &[
  45. "asc", "enc", "gpg", "pgp", "sig", "signature", "pfx", "p12",
  46. ])
  47. }
  48. pub fn is_document(&self) -> bool {
  49. self.extension_is_one_of( &[
  50. "djvu", "doc", "docx", "dvi", "eml", "eps", "fotd",
  51. "odp", "odt", "pdf", "ppt", "pptx", "rtf",
  52. "xls", "xlsx",
  53. ])
  54. }
  55. pub fn is_compressed(&self) -> bool {
  56. self.extension_is_one_of( &[
  57. "zip", "tar", "Z", "gz", "bz2", "a", "ar", "7z",
  58. "iso", "dmg", "tc", "rar", "par", "tgz",
  59. ])
  60. }
  61. pub fn is_temp(&self) -> bool {
  62. self.name.ends_with('~')
  63. || (self.name.starts_with('#') && self.name.ends_with('#'))
  64. || self.extension_is_one_of( &[ "tmp", "swp", "swo", "swn", "bak" ])
  65. }
  66. pub fn is_compiled(&self) -> bool {
  67. if self.extension_is_one_of( &[ "class", "elc", "hi", "o", "pyc" ]) {
  68. true
  69. }
  70. else if let Some(dir) = self.parent_dir {
  71. self.get_source_files().iter().any(|path| dir.contains(path))
  72. }
  73. else {
  74. false
  75. }
  76. }
  77. }
  78. #[cfg(broken_test)]
  79. mod test {
  80. use file::test::{dummy_stat, new_file};
  81. #[test]
  82. fn lowercase() {
  83. let file = new_file(dummy_stat(), "/barracks.wav");
  84. assert_eq!(FileType::Lossless, file.get_type())
  85. }
  86. #[test]
  87. fn uppercase() {
  88. let file = new_file(dummy_stat(), "/BARRACKS.WAV");
  89. assert_eq!(FileType::Lossless, file.get_type())
  90. }
  91. #[test]
  92. fn cargo() {
  93. let file = new_file(dummy_stat(), "/Cargo.toml");
  94. assert_eq!(FileType::Immediate, file.get_type())
  95. }
  96. #[test]
  97. fn not_cargo() {
  98. let file = new_file(dummy_stat(), "/cargo.toml");
  99. assert_eq!(FileType::Normal, file.get_type())
  100. }
  101. }