file_endec 1.0.1

Secure file encryption and decryption from the command line.
Documentation 
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

use ::std::env;

use crate::FedResult;
use ::lazy_static::lazy_static;
use std::ffi::OsStr;
use std::fs;
use std::path::{Path, PathBuf};

/// Recursively get all the files in a directory that have '.enc' extension.
pub fn get_enc_files_direct(dir: &Path) -> FedResult<Vec<PathBuf>> {
    let mut matches = vec![];
    match fs::read_dir(dir) {
        Ok(content) => {
            for path in content {
                match path {
                    Ok(path) => {
                        let path = path.path();
                        if !path.is_file() {
                            continue;
                        }
                        if let Some("enc") = path.extension().and_then(OsStr::to_str) {
                            matches.push(path.to_owned());
                        }
                    }
                    Err(err) => {
                        return Err(format!(
                            "Failed on entry in directory '{}' because '{}'",
                            dir.to_string_lossy(),
                            err
                        ))
                    }
                }
            }
        }
        Err(err) => {
            return Err(format!(
                "Failed to read directory '{}' because '{}'",
                dir.to_string_lossy(),
                err
            ))
        }
    }
    Ok(matches)
}

#[cfg(test)]
lazy_static! {
    pub static ref TEST_FILE_DIR: PathBuf = {
        // Try to find relative to target dir.
        let mut test_files_dir: PathBuf = {
            let mut p = std::env::current_exe().unwrap();
            p.pop();
            p.pop();
            p.pop();
            p.pop();
            //TODO: note that file structure depends on target, it's different for musl, so this is not great...
            p.push("test_files");
            p
        };
        let mut original_file = test_files_dir.clone();
        original_file.push("original.png");
        if !original_file.is_file() {
            // Perhaps the target dir is not in the default location, try something else.
            #[allow(clippy::match_wild_err_arm)]
            match env::var("ENDEC_TEST_FILE_DIR") {
                Ok(test_file_dir_env) => {
                    test_files_dir = PathBuf::from(test_file_dir_env);
                    original_file = test_files_dir.clone();
                    original_file.push("original.png");
                    if !original_file.is_file() {
                        panic!(format!("Expected test files at '{}' based on environment variable 'ENDEC_TEST_FILE_DIR', but the files were not found.", test_files_dir.to_string_lossy()));
                    }
                },
                Err(_) => panic!(format!("Expected test files at '{}' but they were not found; set the environment variable 'ENDEC_TEST_FILE_DIR' to the test file location.", test_files_dir.to_string_lossy())),
            }
        }
        test_files_dir
    };
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn find_files() {
        let files = get_enc_files_direct(&*TEST_FILE_DIR).unwrap();
        assert!(!files.is_empty(), "no .enc files found");
    }
}