rspw 0.2.0

A command line password generator 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

use clap::Parser;
use rand::distributions::{Alphanumeric, Distribution};
use rand::rngs::OsRng;
use rand::Rng;
use std::ffi::OsString;

pub struct Special;
impl Distribution<u8> for Special {
    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> u8 {
        const CHARSET: &[u8] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZ\
                                 abcdefghijklmnopqrstuvwxyz\
                                 0123456789\
                                 !\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~\
                                 \x20";

        // Using uniform distribution
        return CHARSET[rng.gen_range(0..CHARSET.len())];
    }
}

#[derive(Parser, Debug)]
#[command(author, version, about, arg_required_else_help = true)]
pub struct Arguments {
    /// length of the password [6-64]
    #[arg(short, value_parser = clap::value_parser!(u8).range(6..65))]
    length: u8,

    /// special characters
    #[arg(short, long)]
    special: bool,

    /// Copy password onto the clipboard
    #[arg(short)]
    pub clipboard: bool,
}

impl Arguments {
    /**
      Arguments init functions are just wrapper for `parse()` and
      `parse_from()`, so `clap::Parser` is just used in this file.
    */
    pub fn init() -> Self {
        Arguments::parse()
    }

    pub fn init_daemon<I, T>(itr: I) -> Self
    where
        I: IntoIterator<Item = T>,
        T: Into<OsString> + Clone,
    {
        Arguments::parse_from(itr)
    }
    /**
      Logic/Function to generate the password, considers special
      characters if `-s | --special` flag was specified.
    */
    pub fn generate_passwd(&self) -> Result<String, &'static str> {
        if self.length < 6 || self.length > 64 {
            return Err("Second Argument needs to be a number in range [6, 64] e.g. 'rspw -l 16'.");
        }

        if !self.special {
            Ok(OsRng
                .sample_iter(&Alphanumeric)
                .take(self.length as usize)
                .map(char::from)
                .collect())
        } else {
            Ok(OsRng
                .sample_iter(&Special)
                .take(self.length as usize)
                .map(char::from)
                .collect())
        }
    }
}