Add digest parameter to CLI.

Author: tristanmorgan

lib/rotp/arguments.rb

@@ -68,6 +68,10 @@ def parser
         parser.on_tail('-h', '--help', 'Show this message') do
           options!.mode = :help
         end
+
+        parser.on('-d', '--digest [ALGORITHM]', 'Use algorithm for the digest (default sha1)') do |digest|
+          options!.digest = digest
+        end
       end
     end
 

lib/rotp/cli.rb

@@ -31,9 +31,9 @@ def output
       return arguments.to_s if options.mode == :help
 
       if options.mode == :time
-        ROTP::TOTP.new(options.secret).now
+        ROTP::TOTP.new(options.secret, options).now
       elsif options.mode == :hmac
-        ROTP::HOTP.new(options.secret).at options.counter
+        ROTP::HOTP.new(options.secret, options).at options.counter
       end
     end
 

spec/lib/rotp/cli_spec.rb

@@ -18,6 +18,14 @@
     end
   end
 
+  context 'generating a TOTP with sha256 digest' do
+    let(:argv) { %w[--secret JBSWY3DPEHPK3PXP --digest sha256] }
+
+    it 'prints the corresponding token' do
+      expect(output).to eq '544902'
+    end
+  end
+
   context 'generating a TOTP with no secret' do
     let(:argv) { %w[--time --secret] }
 
@@ -49,4 +57,12 @@
       expect(output).to eq '161024'
     end
   end
+
+  context 'generating a HOTP' do
+    let(:argv) { %W[--hmac --secret #{'a' * 32} --counter 1234 --digest sha256] }
+
+    it 'prints the corresponding token' do
+      expect(output).to eq '325941'
+    end
+  end
 end