From 4e333d966507b57d84afdd3c121b8246142b0604 Mon Sep 17 00:00:00 2001 From: TheAwiteb Date: Sun, 24 Dec 2023 20:27:50 +0300 Subject: [PATCH] Add generate command --- src/cli/gen_command.rs | 66 ++++++++++++++++++++++++++++++++++++++++++ src/cli/mod.rs | 4 ++- 2 files changed, 69 insertions(+), 1 deletion(-) create mode 100644 src/cli/gen_command.rs diff --git a/src/cli/gen_command.rs b/src/cli/gen_command.rs new file mode 100644 index 0000000..5517aae --- /dev/null +++ b/src/cli/gen_command.rs @@ -0,0 +1,66 @@ +// Local CLI password manager +// Copyright (C) 2024 Awiteb +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program. If not, see . + +use std::num::NonZeroU64; + +use clap::Args; + +use crate::{password::Passwords, PassrsError, PassrsResult, RunCommand}; + +#[derive(Debug, Args)] +#[command(author, version, about, long_about = None)] +pub struct Gen { + /// The password length + #[arg(default_value_t = NonZeroU64::new(18).unwrap())] + length: NonZeroU64, + + /// With uppercase letters (A-Z) + #[arg(short, long)] + uppercase: bool, + /// With lowercase letters (a-z) + #[arg(short, long)] + lowercase: bool, + /// With numbers (0-9) + #[arg(short, long)] + numbers: bool, + /// With symbols (!,# ...) + #[arg(short, long)] + symbols: bool, +} + +impl RunCommand for Gen { + fn run(&self, _password_manager: Passwords) -> PassrsResult<()> { + if self.uppercase || self.lowercase || self.numbers || self.symbols { + println!( + "{}", + passwords::PasswordGenerator::new() + .length(self.length.get() as usize) + .uppercase_letters(self.uppercase) + .lowercase_letters(self.lowercase) + .numbers(self.numbers) + .symbols(self.symbols) + .strict(true) + .generate_one() + .expect("The length cannot be zero") + ); + Ok(()) + } else { + Err(PassrsError::Other( + "You need to enable at least one kind of characters".to_owned(), + )) + } + } +} diff --git a/src/cli/mod.rs b/src/cli/mod.rs index 8090fca..8fe71f2 100644 --- a/src/cli/mod.rs +++ b/src/cli/mod.rs @@ -26,6 +26,7 @@ use crate::{ pub mod add_command; pub mod clean_command; pub mod edit_command; +pub mod gen_command; pub mod list_command; crate::create_commands!( @@ -34,6 +35,7 @@ crate::create_commands!( "List your password and search", List => list_command::List "Clean the password file", Clean => clean_command::Clean "Edit the password content", Edit => edit_command::Edit + "Generate password", Gen => gen_command::Gen // TODO: Remove command // TODO: Export command // TODO: Import command @@ -63,7 +65,7 @@ impl Cli { "Getting password file: {}", passwords_file.to_string_lossy() ); - let password_manager = if matches!(self.command, Commands::Clean(..)) { + let password_manager = if matches!(self.command, Commands::Clean(..) | Commands::Gen(..)) { Passwords { passwords_file, ..Default::default()