foil/src/main.rs

use docopt::Docopt;
use log::debug;
use rand::rngs::OsRng;
use rand::seq::IteratorRandom;
use rand::seq::SliceRandom;
use serde::Deserialize;
use std::iter::FromIterator;

static USAGE: &'static str = "
foil

Usage:
  foil set [--db=<db>]
  foil get [--db=<db>]
  foil list [--db=<db>]
  foil generate <spec>
  foil (-h | --help)

Options:
  --db=<db>     The path to the sqlite database [default: db.sqlite3].
  -h --help     Show this screen.
  --version     Show version.
";

#[derive(Debug, Deserialize)]
struct Args {
    cmd_set: bool,
    cmd_get: bool,
    cmd_list: bool,
    cmd_generate: bool,
    flag_db: Option<String>,
    arg_spec: Option<String>,
}

fn generate(spec: &str) {
    let (len, rest) = {
        let mut separated = spec.splitn(2, ",");
        let len = separated.next().unwrap();
        let rest = separated.next().unwrap();
        (len, rest)
    };
    let mut rnd = OsRng::new().unwrap();

    let mut password_vec: Vec<char> = rest
        .chars()
        .cycle()
        .map(|c| match c {
            'a' => "abcdefghijklmnopqrstuvwxyz"
                .chars()
                .choose(&mut rnd)
                .expect("Could not get random element"),
            'A' => "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
                .chars()
                .choose(&mut rnd)
                .expect("Could not get random element"),
            '1' => "1234567890"
                .chars()
                .choose(&mut rnd)
                .expect("Could not get random element"),
            '.' => ".,/?-=#%^"
                .chars()
                .choose(&mut rnd)
                .expect("Could not get random element"),
            _ => panic!("Unknown character spec: {}", c),
        })
        .take(len.parse().unwrap())
        .collect();

    password_vec.shuffle(&mut rnd);
    let shuffled: String = String::from_iter(password_vec.into_iter());
    println!("{}", shuffled);
}

fn main() {
    pretty_env_logger::init();
    let args: Args = Docopt::new(USAGE)
        .and_then(|dopt| dopt.deserialize())
        .unwrap_or_else(|e| e.exit());
    debug!("{:?}", args);

    if args.cmd_generate {
        generate(&args.arg_spec.unwrap());
        return;
    }
}