Now users can change their databases password.

Author: replydev

src/argument_functions.rs

@@ -16,6 +16,7 @@ pub fn help(){
     println!("  -j,--json                                            | Print results in json format");
     println!("  -s,--single                                          | Print OTP codes in single mode");
     println!("  -in,--info [ID]                                      | Print info of choosen OTP code");
+    println!("  -chpw,--change-password                              | Change the database password");
     println!("  -h,--help                                            | Print this help");
 }
 
@@ -177,6 +178,28 @@ pub fn info(args: Vec<String>){
         }
     }
     else{
-        println!("Invalid arguments, type cotp --info");
+        println!("Invalid arguments, type cotp --info [ID]");
     }
-}
+}
+
+pub fn change_password(args: Vec<String>) {
+    if args.len() == 2{
+        let old_password = &cryptograpy::prompt_for_passwords("Old password: ", 8, false);
+        let decrypted_text = database_loader::read_decrypted_text(old_password);
+        match decrypted_text{
+            Ok(s) => {
+                let new_password = &cryptograpy::prompt_for_passwords("New password: ", 8, true);
+                match database_loader::overwrite_database_json(&s, new_password) {
+                    Ok(()) => println!("Password changed"),
+                    Err(e) => eprintln!("An error has occurred: {}",e),
+                }
+            },
+            Err(e) => {
+                eprintln!("An error has occurred: {}",e);
+            }
+        }
+    }
+    else{
+        println!("Invalid arguments, type cotp --change-password");
+    }
+} 

src/database_loader.rs

@@ -6,12 +6,14 @@ use utils::{get_db_path,check_elements};
 use crate::cryptograpy;
 use crate::otp::otp_element::OTPElement;
 
-
-pub fn read_from_file(password: &str) -> Result<Vec<OTPElement>,String>{
+pub fn read_decrypted_text(password: &str) -> Result<String,String>{
     let encrypted_contents = read_to_string(&get_db_path()).unwrap();
     //rust close files at the end of the function
-    let contents = cryptograpy::decrypt_string(&encrypted_contents, password);
-    match contents {
+    cryptograpy::decrypt_string(&encrypted_contents, password)
+}
+
+pub fn read_from_file(password: &str) -> Result<Vec<OTPElement>,String>{
+    match read_decrypted_text(password) {
         Ok(contents) => {
             let vector: Vec<OTPElement> = serde_json::from_str(&contents).unwrap();
             return Ok(vector);

src/main.rs

@@ -121,6 +121,7 @@ fn args_parser(args: Vec<String>) -> bool {
         "-j" | "--json" => argument_functions::json(args),
         "-s" | "--single" => argument_functions::single(args),
         "-in" | "--info" => argument_functions::info(args),
+        "-chpw" | "--change-password" => argument_functions::change_password(args),
         _=>{
             println!("Invalid argument: {}, type cotp -h to get command options", args[1]);
             return true;