dan/Portfolio
1
0
Fork 0
mirror of https://github.com/danbulant/Portfolio synced 2026-06-20 06:51:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add asymetric & fix default value for decrypt flags

Browse source
This commit is contained in:
EETagent 2022-11-11 20:41:40 +01:00
parent 693426f041
commit 1039b80518
1 changed files with 41 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
recovery/src/main.rs
View file

@ -1,7 +1,7 @@
use std::path::PathBuf;
use url::Url;
use clap::{arg, command, value_parser, Command, ArgAction};
use clap::{arg, command, value_parser, ArgAction, Command};
use sea_orm::{Database, DatabaseConnection};
use ::entity::candidate::Entity as Candidate;
@ -56,7 +56,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
arg!(
-d --decrypt ... "Decrypt flag"
)
.action(ArgAction::SetFalse)
.action(ArgAction::SetTrue)
.required(false)
)
.arg(
@ -72,6 +72,29 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
.required(true),
)
)
.subcommand(
Command::new("asymetric")
.about("Asymetric encryption operations")
.arg(
arg!(
-d --decrypt ... "Decrypt flag"
)
.action(ArgAction::SetTrue)
.required(false)
)
.arg(
arg!(
-i --input <INPUT> "Plaintext to encrypt/decrypt"
)
.required(true)
)
.arg(
arg!(
-k --key <KEY> "Public key / Private key"
)
.required(true),
)
)
.get_matches();
match clap.subcommand() {
@ -111,11 +134,25 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let input = sub_matches.get_one::<String>("input").unwrap();
let key = sub_matches.get_one::<String>("key").unwrap();
let result = if !*decrypt {
portfolio_core::crypto::encrypt_password(input.to_string(), key.to_string()).await?
} else {
portfolio_core::crypto::decrypt_password(input.to_string(), key.to_string()).await?
};
println!("{}", result);
}
Some(("asymetric", sub_matches)) => {
let decrypt = sub_matches.get_one::<bool>("decrypt").unwrap();
let input = sub_matches.get_one::<String>("input").unwrap();
let key = sub_matches.get_one::<String>("key").unwrap();
println!("Decrypt: {}", decrypt);
let result = if !*decrypt {
portfolio_core::crypto::decrypt_password(input.to_string(), key.to_string()).await?
portfolio_core::crypto::encrypt_password_with_recipients(input, &vec![key]).await?
} else {
portfolio_core::crypto::encrypt_password(input.to_string(), key.to_string()).await?
portfolio_core::crypto::decrypt_password_with_private_key(input, key).await.map_err(|e| e.to_string())?
};
println!("{}", result);