StNicolay/pass_manager
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Now creating password on 1 blocking task because the generation doesn't take that long, removed back slash from the password chars

Browse Source
This commit is contained in:
StNicolay 2023-05-12 20:29:44 +03:00
parent 522194c708
commit 0b92bcd9f3
Signed by: StNicolay
GPG Key ID: 9693D04DCD962B0D
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/handlers/commands/gen_password.rs
View File

@ -1,11 +1,11 @@
use crate::handlers::markups::deletion_markup; use crate::handlers::markups::deletion_markup;
use arrayvec::{ArrayString, ArrayVec}; use arrayvec::{ArrayString, ArrayVec};
use rand::{rngs::OsRng, seq::SliceRandom}; use rand::{rngs::OsRng, seq::SliceRandom};
use std::{iter, str::from_utf8_unchecked}; use std::str::from_utf8_unchecked;
use teloxide::{adaptors::Throttle, prelude::*, types::ParseMode}; use teloxide::{adaptors::Throttle, prelude::*, types::ParseMode};
use tokio::task::JoinSet; use tokio::task::spawn_blocking;
const CHARS: &[u8] = br##"!"#$%&\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[]^_abcdefghijklmnopqrstuvwxyz{|}~"##; const CHARS: &[u8] = br##"!"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[]^_abcdefghijklmnopqrstuvwxyz{|}~"##;
/// Returns true if the generated master password is valid. /// Returns true if the generated master password is valid.
/// It checks that it has at least one lowercase, one lowercase and one punctuation char /// It checks that it has at least one lowercase, one lowercase and one punctuation char
@ -28,32 +28,30 @@ fn check_generated_password(password: &[u8]) -> bool {
} }
/// Continuously generates the password until it passes the checks /// Continuously generates the password until it passes the checks
fn generete_password() -> ArrayString<34> { fn generete_passwords() -> ArrayVec<ArrayString<34>, 10> {
loop { let mut passwords = ArrayVec::new_const();
let password: ArrayVec<u8, 32> = iter::repeat_with(|| *CHARS.choose(&mut OsRng).unwrap()) while !passwords.is_full() {
.take(32) let password: ArrayVec<u8, 32> = (0..32)
.map(|_| *CHARS.choose(&mut OsRng).unwrap())
.collect(); .collect();
if check_generated_password(&password) { if check_generated_password(&password) {
let mut string = ArrayString::<34>::new_const(); let mut string = ArrayString::<34>::new_const();
string.push('`'); string.push('`');
unsafe { string.push_str(from_utf8_unchecked(&password)) }; unsafe { string.push_str(from_utf8_unchecked(&password)) };
string.push('`'); string.push('`');
return string; passwords.push(string)
} }
} }
passwords
} }
/// Handles /gen_password command by generating 10 copyable passwords and sending them to the user /// Handles /gen_password command by generating 10 copyable passwords and sending them to the user
pub async fn gen_password(bot: Throttle<Bot>, msg: Message) -> crate::Result<()> { pub async fn gen_password(bot: Throttle<Bot>, msg: Message) -> crate::Result<()> {
let mut message = ArrayString::<{ 11 + 35 * 10 }>::new(); let mut message: ArrayString<{ 10 + 35 * 10 }> = "Passwords:".try_into().unwrap();
message.push_str("Passwords:\n"); let passwords = spawn_blocking(|| generete_passwords()).await?;
let mut join_set = JoinSet::new(); for password in passwords {
for _ in 0..10 { message.push('\n');
join_set.spawn_blocking(generete_password); message.push_str(&password)
}
while let Some(password) = join_set.join_next().await {
message.push_str(password?.as_str());
message.push('\n')
} }
bot.send_message(msg.chat.id, message.as_str()) bot.send_message(msg.chat.id, message.as_str())
.parse_mode(ParseMode::MarkdownV2) .parse_mode(ParseMode::MarkdownV2)