easy-utils 0.0.1

Multiple utils written in rust.
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

use rand::{Rng};
use rand::distributions::Alphanumeric;

/// Generate random String with given charset of len length
///
/// # Example
///
/// ```
/// const CHARSET: &[u8] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZ\
///                     abcdefghijklmnopqrstuvwxyz\
///                     0123456789)(*&^%$#@!~";
/// const PASSWORD_LEN: usize = 30;
/// let password = rust_utils::random::get_random_charset_string(PASSWORD_LEN, CHARSET);
/// assert_eq!(password.len(), PASSWORD_LEN);
/// for x in password.bytes() {
///     assert!(CHARSET.contains(&x));
/// }
/// ```
pub fn get_random_charset_string(len: usize, charset: &[u8]) -> String {
    (0..len)
        .map(|_| {
            let idx = rand::thread_rng().gen_range(0..charset.len());
            // It's safe, because `idx` is in `CHARSET`'s range.
            char::from(unsafe { *charset.get_unchecked(idx) })
        })
        .collect()
}

/// Generate random String with alphanumeric(uppercase, lowercase & number) charset of len length
///
/// # Example
///
/// ```
///const GEN_ASCII_STR_CHARSET: &[u8] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZ\
///         abcdefghijklmnopqrstuvwxyz\
///         0123456789";
/// const PASSWORD_LEN: usize = 30;
/// let password = rust_utils::random::get_random_alphanumeric_string(PASSWORD_LEN);
/// assert_eq!(password.len(), PASSWORD_LEN);
/// for x in password.bytes() {
///     assert!(GEN_ASCII_STR_CHARSET.contains(&x));
/// }
/// ```
pub fn get_random_alphanumeric_string(len: usize) -> String {
    rand::thread_rng()
        .sample_iter(&Alphanumeric)
        .take(len)
        .map(char::from)
        .collect()
}

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

    #[test]
    fn get_random_charset_string_test() {
        const CHARSET: &[u8] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZ\
                            abcdefghijklmnopqrstuvwxyz\
                            0123456789)(*&^%$#@!~";
        const PASSWORD_LEN: usize = 30;

        let password = get_random_charset_string(PASSWORD_LEN, CHARSET);

        assert_eq!(password.len(), PASSWORD_LEN);
        for x in password.bytes() {
            assert!(CHARSET.contains(&x));
        }
    }

    #[test]
    fn get_random_alphanumeric_string_test() {
        const GEN_ASCII_STR_CHARSET: &[u8] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZ\
                abcdefghijklmnopqrstuvwxyz\
                0123456789";
        const PASSWORD_LEN: usize = 30;

        let password = get_random_alphanumeric_string(PASSWORD_LEN);
        assert_eq!(password.len(), PASSWORD_LEN);
        for x in password.bytes() {
            assert!(GEN_ASCII_STR_CHARSET.contains(&x));
        }
    }
}