fs-librarian 0.3.1

Librarian runs pre-configured commands against a group of files that match a set of filters
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use crate::error;
use std::io::Read;
use std::path::Path;

#[cfg(test)]
mod tests;
// since you only need a few bytes in the start of the file to get
// the magic number used to determine the file type
#[allow(dead_code)]
const MAX_FILE_READ_BYTES: u64 = 10240;

#[derive(Debug)]
pub struct File<'a> {
    path: &'a Path,
}

impl<'a> File<'a> {
    #[allow(dead_code)]
    pub fn new(path: &'a Path) -> File {
        File { path }
    }

    #[allow(dead_code)]
    pub fn get_mime_type(&'a self) -> Result<String, error::Error> {
        let mut file_obj = match std::fs::File::open(self.path) {
            Ok(f) => f,
            Err(e) => {
                return Err(error::Error::new(format!(
                    "Unable to open file '{:?}' to test its type: {}",
                    self.path, e
                )))
            }
        };

        let mut buf = Vec::with_capacity(MAX_FILE_READ_BYTES as usize);
        if let Err(e) = file_obj
            .by_ref()
            .take(MAX_FILE_READ_BYTES)
            .read_to_end(&mut buf)
        {
            return Err(error::Error::new(format!(
                "Unable to read file '{:?}' to test its type: {}",
                self.path, e
            )));
        }

        let mime_type = tree_magic::from_u8(buf.as_slice());

        Ok(mime_type)
    }
}