src/cli/gen_command.rs

@@ -44,7 +44,7 @@ pub struct Gen {
 
 impl LprsCommand for Gen {
     fn run(self, _vault_manager: Vaults) -> LprsResult<()> {
-        println!(
+        print!(
             "{}",
             passwords::PasswordGenerator::new()
                 .length(self.length.get() as usize)
@@ -52,7 +52,7 @@ impl LprsCommand for Gen {
                 .lowercase_letters(self.lowercase)
                 .numbers(self.numbers)
                 .symbols(self.symbols)
-                .strict(true)
+                .strict(false)
                 .generate_one()
                 .expect("The length cannot be zero")
         );

src/cli/get_command.rs

@@ -0,0 +1,140 @@
+// Lprs - A local CLI vault manager
+// Copyright (C) 2024  Awiteb <a@4rs.nl>
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/gpl-3.0.html>.
+
+use std::str::FromStr;
+
+use clap::Args;
+
+use crate::{
+    vault::{Vault, Vaults},
+    LprsCommand,
+    LprsError,
+    LprsResult,
+};
+
+#[derive(Debug, Clone, Eq, PartialEq)]
+enum VaultGetField {
+    Index,
+    Name,
+    Username,
+    Password,
+    Service,
+    Note,
+    Custom(String),
+}
+
+impl FromStr for VaultGetField {
+    type Err = LprsError;
+
+    fn from_str(input: &str) -> Result<Self, Self::Err> {
+        Ok(match input.to_lowercase().as_str() {
+            "index" => Self::Index,
+            "name" => Self::Name,
+            "username" => Self::Username,
+            "password" => Self::Password,
+            "service" => Self::Service,
+            "note" => Self::Note,
+            _ => Self::Custom(input.to_owned()),
+        })
+    }
+}
+
+impl VaultGetField {
+    /// Returns the field from the vault
+    pub fn get_from_vault<'a>(&self, vault: &'a Vault) -> Option<&'a str> {
+        match self {
+            Self::Index => None,
+            Self::Name => Some(&vault.name),
+            Self::Username => vault.username.as_deref(),
+            Self::Password => vault.password.as_deref(),
+            Self::Service => vault.service.as_deref(),
+            Self::Note => vault.note.as_deref(),
+            Self::Custom(custom_field) => vault.custom_fields.get(custom_field).map(|x| x.as_str()),
+        }
+    }
+
+    /// Returns the field as `&str`
+    pub fn as_str(&self) -> &str {
+        match self {
+            Self::Index => "index",
+            Self::Name => "name",
+            Self::Username => "username",
+            Self::Password => "password",
+            Self::Service => "service",
+            Self::Note => "note",
+            Self::Custom(field) => field.as_str(),
+        }
+    }
+}
+
+#[derive(Debug, Args)]
+#[command(author, version, about, long_about = None)]
+/// Command to get a entire vault or single field from it
+pub struct Get {
+    /// Whether the index of the vault or its name
+    #[arg(value_name = "INDEX-or-NAME")]
+    location: String,
+    /// A Specific field to get.
+    ///
+    /// Can be [name,username,password,service,note,"string"] where the string
+    /// means a custom field
+    #[arg(value_parser = VaultGetField::from_str)]
+    field:    Option<VaultGetField>,
+}
+
+impl LprsCommand for Get {
+    fn run(self, vault_manager: Vaults) -> LprsResult<()> {
+        let parsed_index = self.location.trim().parse::<usize>();
+        let Some((index, vault)) = (if let Ok(index) = parsed_index {
+            vault_manager.vaults.get(index - 1).map(|v| (index, v))
+        } else {
+            vault_manager
+                .vaults
+                .iter()
+                .enumerate()
+                .find(|(_, v)| v.name == self.location)
+        }) else {
+            return Err(LprsError::Other(format!(
+                "There is no vault with the given {} `{}`",
+                if parsed_index.is_ok() {
+                    "index"
+                } else {
+                    "name"
+                },
+                self.location.trim(),
+            )));
+        };
+
+        if let Some(field) = self.field {
+            if field == VaultGetField::Index {
+                print!("{index}");
+                return Ok(());
+            }
+
+            if let Some(value) = field.get_from_vault(vault) {
+                print!("{value}")
+            } else {
+                return Err(LprsError::Other(format!(
+                    "There is no value for `{}`",
+                    field.as_str()
+                )));
+            }
+        } else {
+            println!("{vault}");
+        }
+        Ok(())
+    }
+}

src/cli/list_command.rs

@@ -14,8 +14,6 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <https://www.gnu.org/licenses/gpl-3.0.html>.
 
-use std::num::NonZeroU64;
-
 use clap::Args;
 use inquire::Select;
 
@@ -25,9 +23,6 @@ use crate::{vault::Vaults, LprsCommand, LprsError, LprsResult};
 #[command(author, version, about, long_about = None)]
 /// List command, used to list the vaults and search
 pub struct List {
-    /// Return the password with spesifc index
-    #[arg(short, long, value_name = "INDEX")]
-    get:    Option<NonZeroU64>,
     /// Filter the select list
     #[arg(short, long, value_name = "TEXT")]
     filter: Option<String>,
@@ -43,21 +38,7 @@ impl LprsCommand for List {
                 "Looks like there is no vaults to list".to_owned(),
             ));
         }
-        if let Some(user_vault_index) = self.get.map(|n| (n.get() - 1) as usize) {
+
-            log::info!("Getting the vault at index: {user_vault_index}");
-            if user_vault_index >= vault_manager.vaults.len() {
-                return Err(LprsError::Other(
-                    "The `--get` index is great then the vaults length".to_owned(),
-                ));
-            }
-            println!(
-                "{}",
-                vault_manager
-                    .vaults
-                    .get(user_vault_index)
-                    .expect("The index is correct")
-            );
-        } else {
         let pattern = if self.regex || self.filter.is_none() {
             self.filter.unwrap_or_else(|| ".".to_owned())
         } else {
@@ -116,7 +97,7 @@ impl LprsCommand for List {
                 .get(vault_idx - 1)
                 .expect("The index is correct")
         );
-        }
+
         Ok(())
     }
 
@@ -126,11 +107,6 @@ impl LprsCommand for List {
                 "You cannot use the `--regex` flag if you did not use the search option".to_owned(),
             ));
         }
-        if self.filter.is_some() && self.get.is_some() {
-            return Err(LprsError::Other(
-                "You cannot search while you want a vault with a specific index".to_owned(),
-            ));
-        }
         Ok(())
     }
 }

src/cli/mod.rs

@@ -31,6 +31,8 @@ pub mod edit_command;
 pub mod export_command;
 /// Generate command, used to generate a password
 pub mod gen_command;
+/// Command to get a entire vault or single field from it
+pub mod get_command;
 /// Import command, used to import vaults from the exported files, `lprs` or
 /// `BitWarden`
 pub mod import_command;
@@ -56,13 +58,15 @@ pub enum Commands {
     Edit(edit_command::Edit),
     /// Generate a password
     Gen(gen_command::Gen),
+    /// Get a entire vault or single field from it
+    Get(get_command::Get),
     /// Export the vaults
     Export(export_command::Export),
     /// Import vaults
     Import(import_command::Import),
 }
 
-impl_commands!(Commands, Add Remove List Clean Edit Gen Export Import);
+impl_commands!(Commands, Add Remove List Clean Edit Gen Get Export Import);
 
 #[derive(Parser, Debug)]
 #[command(author, version, about, long_about = None)]