Added encrpyt and decrypt using the main thread

Author: Mohaned2023

src/commands/mod.rs

@@ -73,17 +73,19 @@ pub fn commands() -> Vec<Command> {
                     .alias("end")
                     .about("Encrypt directory.")
                     .args([
-                        arg!(<PATH>   "Directory path (e.g. \"/home/user/important\")."),
-                        arg!(--delete "Delete the origin files in the directory."),
-                        arg!(--key    "Use custom key.")
+                        arg!(<PATH>         "Directory path (e.g. \"/home/user/important\")."),
+                        arg!(--delete       "Delete the origin files in the directory."),
+                        arg!(--"no-threads" "Encrypt directory using the main thread only."),
+                        arg!(--key          "Use custom key.")
                     ]),
                 Command::new("decrypt-dir")
                     .alias("ded")
                     .about("Decrypt directory.")
                     .args([
-                        arg!(<PATH>   "Directory path (e.g. \"/home/user/important\")."),
-                        arg!(--delete "Delete the origin files in the directory."),
-                        arg!(--xpmv1  "Decrypt XPManager v1.0 directory.")
+                        arg!(<PATH>         "Directory path (e.g. \"/home/user/important\")."),
+                        arg!(--delete       "Delete the origin files in the directory."),
+                        arg!(--"no-threads" "Decrypt directory using the main thread only."),
+                        arg!(--xpmv1        "Decrypt XPManager v1.0 directory.")
                     ]),
                 Command::new("encode")
                     .alias("enc")

src/encryption_manager/decrypt_dir.rs

@@ -77,6 +77,7 @@ pub fn main(command: &ArgMatches) {
     let mut files_paths: Vec<PathBuf> = vec![];
     let is_xpmv1 = *command.get_one::<bool>("xpmv1").unwrap_or(&false);
     let is_delete = *command.get_one::<bool>("delete").unwrap_or(&false);
+    let no_threads = *command.get_one::<bool>("no-threads").unwrap_or(&false);
     let key = utilities::input("Enter your key: ");
     logger.start();
     filelib::dir_files_tree(
@@ -96,18 +97,30 @@ pub fn main(command: &ArgMatches) {
     }
     logger.info("directory listed successfully.");
 
-    // Distribute files over the number of threads 
-    let distributed_paths: Vec<Vec<PathBuf>> = utilities::distribute_paths(files_paths.clone());
-
-    // Run the threads
-    distributed_paths.par_iter().for_each(|paths| {
+    if no_threads {
+        logger.info("start the decryption using the main thread.");
         decrypt(
-            paths, 
+            &files_paths, 
             key.clone(), 
             is_delete, 
             is_xpmv1
         );
-    });
+    } else {
+        logger.info("start the decryption with the max number of threads.");
+        // Distribute files over the number of threads 
+        let distributed_paths: Vec<Vec<PathBuf>> = utilities::distribute_paths(files_paths.clone());
+    
+        // Run the threads
+        distributed_paths.par_iter().for_each(|paths| {
+            decrypt(
+                paths, 
+                key.clone(), 
+                is_delete, 
+                is_xpmv1
+            );
+        });
+    }
+
     logger.info("directory decrypted successfully.");
     displaylib::key::display(key);
 }

src/encryption_manager/encrypt_dir.rs

@@ -68,6 +68,7 @@ pub fn main(command: &ArgMatches) {
     let mut logger = loglib::Logger::new("encrypt-dir");
     let path = command.get_one::<String>("PATH").unwrap();
     let is_delete = *command.get_one::<bool>("delete").unwrap_or(&false);
+    let no_threads = *command.get_one::<bool>("no-threads").unwrap_or(&false);
     let mut files_paths: Vec<PathBuf> = vec![];
     let key = if *command.get_one::<bool>("key")
         .unwrap_or(&false) {
@@ -102,18 +103,30 @@ pub fn main(command: &ArgMatches) {
     logger.start();
     let log_db_path = filelib::log::get_log_db_path();
 
-    // Distribute files over the number of threads 
-    let distributed_paths: Vec<Vec<PathBuf>> = utilities::distribute_paths(files_paths.clone());
-
-    // Run the threads
-    distributed_paths.par_iter().for_each(|paths| {
+    if no_threads {
+        logger.info("start the encryption using the main thread.");
         encrypt(
-            paths, 
+            &files_paths, 
             key.clone(), 
             is_delete, 
             log_db_path.clone()
         );
-    });
+    } else {
+        logger.info("start the encryption with the max number of threads.");
+        // Distribute files over the number of threads 
+        let distributed_paths: Vec<Vec<PathBuf>> = utilities::distribute_paths(files_paths.clone());
+
+        // Run the threads
+        distributed_paths.par_iter().for_each(|paths| {
+            encrypt(
+                paths, 
+                key.clone(), 
+                is_delete, 
+                log_db_path.clone()
+            );
+        });
+    }
+
     logger.info("directory encrypted successfully.");
     displaylib::key::display(key);
     dblib::log::register(