feat: add useful utilities

Cargo.lock

@@ -409,7 +409,7 @@ dependencies = [
  "atty",
  "bitflags",
  "clap_derive",
- "clap_lex",
+ "clap_lex 0.2.4",
  "indexmap",
  "once_cell",
  "strsim",
@@ -417,6 +417,20 @@ dependencies = [
  "textwrap",
 ]
 
+[[package]]
+name = "clap"
+version = "4.0.23"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0eb41c13df48950b20eb4cd0eefa618819469df1bffc49d11e8487c4ba0037e5"
+dependencies = [
+ "atty",
+ "bitflags",
+ "clap_lex 0.3.0",
+ "once_cell",
+ "strsim",
+ "termcolor",
+]
+
 [[package]]
 name = "clap_derive"
 version = "3.2.18"
@@ -439,6 +453,15 @@ dependencies = [
  "os_str_bytes",
 ]
 
+[[package]]
+name = "clap_lex"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0d4198f73e42b4936b35b5bb248d81d2b595ecb170da0bac7655c54eedfa8da8"
+dependencies = [
+ "os_str_bytes",
+]
+
 [[package]]
 name = "codespan-reporting"
 version = "0.11.1"
@@ -1893,6 +1916,18 @@ dependencies = [
  "tokio",
 ]
 
+[[package]]
+name = "portfolio-recovery"
+version = "0.1.0"
+dependencies = [
+ "clap 4.0.23",
+ "portfolio-core",
+ "portfolio-entity",
+ "sea-orm",
+ "tokio",
+ "url",
+]
+
 [[package]]
 name = "ppv-lite86"
 version = "0.2.17"
@@ -2339,7 +2374,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "882af0d4cd3d6cc2f427d14a48e9f468b37c563b405ea486fd314ba18ca334d0"
 dependencies = [
  "chrono",
- "clap",
+ "clap 3.2.23",
  "dotenvy",
  "regex",
  "sea-schema",
@@ -2368,7 +2403,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "86fe6e594b078712e1e797b951b9b56e55d3cfa04aac8ea76eb4bed7c94c5910"
 dependencies = [
  "async-trait",
- "clap",
+ "clap 3.2.23",
  "dotenvy",
  "sea-orm",
  "sea-orm-cli",

recovery/Cargo.toml

@@ -9,6 +9,7 @@ url = "^2.3"
 clap = { version = "^4.0", features = ["cargo"] }
 
 portfolio-entity = { path = "../entity" }
+portfolio-core = { path = "../core" }
 
 [dependencies.tokio]
 version = "^1.21"

recovery/src/main.rs

@@ -1,16 +1,22 @@
 use std::path::PathBuf;
 use url::Url;
 
-use clap::{arg, command, value_parser};
+use clap::{arg, command, value_parser, Command, ArgAction};
 use sea_orm::{Database, DatabaseConnection};
 
-use sea_orm::*;
 use ::entity::candidate::Entity as Candidate;
 use ::entity::parent::Entity as Parent;
+use sea_orm::*;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let clap = command!()
+        .propagate_version(true)
+        .subcommand_required(true)
+        .arg_required_else_help(true)
+        .subcommand(
+            Command::new("portfolio")
+                .about("Database & Portfolio operations")
                 .arg(
                     arg!(
                         -d --database <URL> "URL to the database or sql file with postgres:// or sqlite://"
@@ -32,9 +38,45 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                     )
                     .required(true),
                 )
+        )
+        .subcommand(
+            Command::new("hash")
+                .about("Hash operations")
+                .arg(
+                    arg!(
+                        -i --input <INPUT> "Plaintext to hash"
+                    )
+                    .required(true)
+                ),
+        )
+        .subcommand(
+            Command::new("encryption")
+                .about("Encryption operations")
+                .arg(
+                    arg!(
+                        -d --decrypt ... "Decrypt flag"
+                    )
+                    .action(ArgAction::SetFalse)
+                    .required(false)
+                )
+                .arg(
+                    arg!(
+                        -i --input <INPUT> "Plaintext to encrypt/decrypt"
+                    )
+                    .required(true)
+                )
+                .arg(
+                    arg!(
+                        -k --key <KEY> "Key for encryption/decryption"
+                    )
+                    .required(true),
+                )
+        )
         .get_matches();
 
-    let sqlite_url = clap.get_one::<Url>("database").unwrap();
+    match clap.subcommand() {
+        Some(("portfolio", sub_matches)) => {
+            let sqlite_url = sub_matches.get_one::<Url>("database").unwrap();
 
             println!("Connecting to {:?}", sqlite_url);
 
@@ -55,6 +97,31 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 .all(&db)
                 .await?;
 
-    Ok(())
+            println!("Found {} entries", entries.len());
-    
+        }
+        Some(("hash", sub_matches)) => {
+            let input = sub_matches.get_one::<String>("input").unwrap();
+
+            let hash = portfolio_core::crypto::hash_password(input.to_string()).await?;
+
+            println!("{}", hash);
+        }
+        Some(("encryption", 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();
+
+
+            let result = if !*decrypt {
+                portfolio_core::crypto::decrypt_password(input.to_string(), key.to_string()).await?
+            } else {
+                portfolio_core::crypto::encrypt_password(input.to_string(), key.to_string()).await?
+            };
+
+            println!("{}", result);
+        }
+        _ => unreachable!("Exhausted list of subcommands and subcommand_required prevents `None`"),
+    }
+
+    Ok(())
 }