fix: validate digits input value from command line

replydev · Jun 19, 2024 · d03d4b8 · d03d4b8
1 parent 95cf5a2
commit d03d4b8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/arguments/add.rs b/src/arguments/add.rs
@@ -1,4 +1,4 @@
-use clap::Args;
+use clap::{value_parser, Args};
 use color_eyre::eyre::{eyre, ErrReport};
 
 use zeroize::Zeroize;
@@ -39,7 +39,8 @@ pub struct AddArgs {
         short,
         long,
         default_value_t = 6,
-        default_value_if("type", "STEAM", "5")
+        default_value_if("type", "STEAM", "5"),
+        value_parser=value_parser!(u64).range(0..=9)
     )]
     pub digits: u64,
 

diff --git a/src/arguments/edit.rs b/src/arguments/edit.rs
@@ -1,4 +1,4 @@
-use clap::Args;
+use clap::{value_parser, Args};
 use color_eyre::eyre::eyre;
 
 use crate::otp::{otp_algorithm::OTPAlgorithm, otp_element::OTPDatabase};
@@ -24,7 +24,7 @@ pub struct EditArgs {
     pub algorithm: Option<OTPAlgorithm>,
 
     /// Code digits
-    #[arg(short, long)]
+    #[arg(short, long, value_parser=value_parser!(u64).range(0..=9))]
     pub digits: Option<u64>,
 
     /// Code period