feat: implement password reading from stdin

src/main.rs

@@ -9,7 +9,7 @@ use interface::ui::Tui;
 use otp::otp_element::{OTPDatabase, CURRENT_DATABASE_VERSION};
 use ratatui::prelude::CrosstermBackend;
 use ratatui::Terminal;
-use reading::{get_elements, ReadResult};
+use reading::{get_elements_from_input, get_elements_from_stdin, ReadResult};
 use std::{io, vec};
 use zeroize::Zeroize;
 
@@ -25,7 +25,7 @@ mod path;
 mod reading;
 mod utils;
 
-fn init() -> color_eyre::Result<ReadResult> {
+fn init(read_password_from_stdin: bool) -> color_eyre::Result<ReadResult> {
     match utils::init_app() {
         Ok(first_run) => {
             if first_run {
@@ -39,8 +39,10 @@ fn init() -> color_eyre::Result<ReadResult> {
                 let save_result = database.save_with_pw(&pw);
                 pw.zeroize();
                 save_result.map(|(key, salt)| (database, key, salt.to_vec()))
+            } else if read_password_from_stdin {
+                get_elements_from_stdin()
             } else {
-                get_elements()
+                get_elements_from_input()
             }
         }
         Err(()) => Err(eyre!("An error occurred during database creation")),
@@ -50,8 +52,8 @@ fn init() -> color_eyre::Result<ReadResult> {
 fn main() -> AppResult<()> {
     color_eyre::install()?;
 
-    let cotp_args = CotpArgs::parse();
-    let (database, mut key, salt) = match init() {
+    let cotp_args: CotpArgs = CotpArgs::parse();
+    let (database, mut key, salt) = match init(cotp_args.password_from_stdin) {
         Ok(v) => v,
         Err(e) => {
             println!("{e}");

src/reading.rs

@@ -4,15 +4,26 @@ use crate::path::get_db_path;
 use crate::utils;
 use color_eyre::eyre::{eyre, ErrReport};
 use std::fs::read_to_string;
-use std::io;
+use std::io::{self, BufRead};
 use zeroize::Zeroize;
 
 pub type ReadResult = (OTPDatabase, Vec<u8>, Vec<u8>);
 
-pub fn get_elements() -> color_eyre::Result<ReadResult> {
-    let mut pw = utils::password("Password: ", 8);
-    let (elements, key, salt) = read_from_file(&pw)?;
-    pw.zeroize();
+pub fn get_elements_from_input() -> color_eyre::Result<ReadResult> {
+    let pw = utils::password("Password: ", 8);
+    get_elements_with_password(pw)
+}
+
+pub fn get_elements_from_stdin() -> color_eyre::Result<ReadResult> {
+    for password in io::stdin().lock().lines() {
+        return get_elements_with_password(password?);
+    }
+    Err(eyre!("Failure during stdin reading"))
+}
+
+fn get_elements_with_password(mut password: String) -> color_eyre::Result<ReadResult> {
+    let (elements, key, salt) = read_from_file(&password)?;
+    password.zeroize();
     Ok((elements, key, salt))
 }