Update totp_command.rb

adds a rescue round where totp parse the value. If its been entered incorrectly, will just display ?? instead of crapping out

1 files changed, 5 insertions, 1 deletions

diff --git a/lib/tfa/totp_command.rb b/lib/tfa/totp_command.rb
index bba1eb9..8dd0125 100644
--- a/lib/tfa/totp_command.rb
+++ b/lib/tfa/totp_command.rb
@@ -12,7 +12,11 @@ module TFA
     private
 
     def password_for(secret)
-      ::ROTP::TOTP.new(secret).now
+      begin
+        ::ROTP::TOTP.new(secret).now
+      rescue
+        "???"
+      end  
     end
 
     def all_passwords