asfa 0.10.0

Avoid sending file attachments by uploading via SSH to a remote site with non-guessable (hash-based) prefix and print URLs.
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
91
92
93
94
95
96
97
98
99
100
101
102

use anyhow::{bail, Result};
use log::error;
use std::fs::File;
use std::io::prelude::*;
use std::io::BufReader;
use std::path::Path;
use thiserror::Error;
use yaml_rust::{yaml, Yaml};

#[derive(Debug, Error)]
#[error("Invalid Yaml type found for key: {key}")]
pub struct InvalidYamlTypeError {
    key: String,
}

/// Helper function static yaml strings.
pub fn yaml_string(s: &str) -> Yaml {
    Yaml::String(String::from(s))
}

/// Get hash digest of given file with chosen length
pub fn get_hash(path: &Path, length: u8) -> Result<String> {
    let hash = if length == 0 {
        bail!("Length cannot be zero!");
    } else if length <= 32 {
        get_explicit_hash::<sha2::Sha256>(path)?
    } else if length <= 64 {
        get_explicit_hash::<sha2::Sha512>(path)?
    } else {
        bail!("Length should be equal to or smaller than 64.");
    };
    Ok(hash[..length as usize].to_string())
}

fn get_explicit_hash<Hasher: sha2::Digest>(path: &Path) -> Result<String> {
    let mut hash = Hasher::new();
    let mut reader = BufReader::new(File::open(path)?);
    loop {
        let buf = &reader.fill_buf()?;
        let to_write = buf.len();
        if to_write > 0 {
            hash.update(buf);
            reader.consume(to_write);
        } else {
            break;
        }
    }
    Ok(base64::encode_config(hash.finalize(), base64::URL_SAFE))
}

macro_rules! make_yaml_getter {
    ($function_name:ident, $variant:ident, $return_type:ty) => {
        #[allow(dead_code)]
        pub fn $function_name<'a>(
            dict: &'a yaml::Hash,
            name: &str,
        ) -> Result<Option<&'a $return_type>, InvalidYamlTypeError> {
            match dict.get(&yaml_string(name)) {
                Some(Yaml::$variant(v)) => Ok(Some(&v)),
                Some(_) => Err(InvalidYamlTypeError {
                    key: name.to_string(),
                }),
                None => Ok(None),
            }
        }
    };
}

make_yaml_getter! {get_string_from, String, String}
make_yaml_getter! {get_bool_from, Boolean, bool}
make_yaml_getter! {get_dict_from, Hash, yaml::Hash}
make_yaml_getter! {get_array_from, Array, yaml::Array}
make_yaml_getter! {get_int_from, Integer, i64}
make_yaml_getter! {get_real_from, Real, String}

/// Raise an error if the required key does not exist.
pub fn get_required<
    'a,
    T,
    F: Fn(&'a yaml::Hash, &str) -> Result<Option<&'a T>, InvalidYamlTypeError>,
>(
    dict: &'a yaml::Hash,
    name: &str,
    getter: F,
) -> Result<&'a T> {
    match getter(dict, name)? {
        Some(v) => Ok(v),
        None => bail!("Required key '{}' not defined!", name),
    }
}

#[cfg(test)]
pub mod test {
    use anyhow::{Context, Result};
    /// Perform test initialization that needs to be called prior to any test.
    pub fn init() -> Result<()> {
        simple_logger::SimpleLogger::new()
            .with_level(log::LevelFilter::Trace)
            .init()
            .context("Failed to set up logger for tests.")
    }
}