tfa 0.0.10 → 0.0.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 757123af55e697b21d9d564d78f59c27708a861b
-  data.tar.gz: e89cd5ca7e10c9883fb8fa27574d28d1c7611e59
+  metadata.gz: 1fb7180ac2ccb7c877e05e1927763d9bdb0d4e6c
+  data.tar.gz: f7d18a024879662a22404c42b1a3ef01975b027e
 SHA512:
-  metadata.gz: 9f402f377b28ad237caec3c01d8e8811fe187b604467498063ee5dfb809f5fa6f307e4a82ca36a1acb1cc3159faf97faa376ce1bd7fa30e48982bc9ae29c6d40
-  data.tar.gz: b5e0240fbc9a499f224df6d1a09ee753e69897c7983bddfed103c27da880e8e37fec63281e5a345c4378c1c9f6ec0a425bddab4beeb3802285cb549e84853a6d
+  metadata.gz: 79e854463b20f67ec678ac3c0e0e08e3c00df4ce9fc27df0a0b7b320647d65de78d938fa38f9a3074f3050dd6789fa4f2166c6e24b173c7e540eb33d40fd6efd
+  data.tar.gz: 0741cd7c80b47730893ad23fc962d0448a2e8ae3be72a4c98c5ddefab31fe423d8bd6ec4fbefeaa3a2c2b9b2d17c902a37642f73e360a2dfc3cbb725cc79ca23

data/lib/tfa/cli.rb

@@ -7,6 +7,7 @@ module TFA
 
     desc "add NAME SECRET", "add a new secret to the database"
     def add(name, secret)
+      secret = clean(secret)
       storage.save(name, secret)
       "Added #{name}"
     end
@@ -18,7 +19,7 @@ module TFA
 
     desc "totp NAME", "generate a Time based One Time Password"
     def totp(name = nil)
-      TotpCommand.new(storage).run([name])
+      TotpCommand.new(storage).run(name)
     end
 
     private
@@ -26,5 +27,13 @@ module TFA
     def storage
       @storage ||= Storage.new(filename: options[:filename] || 'tfa')
     end
+
+    def clean(secret)
+      if secret.include?("=")
+        /secret=([^&]*)/.match(secret).captures.first
+      else
+        secret
+      end
+    end
   end
 end

data/lib/tfa/version.rb

@@ -1,3 +1,3 @@
 module TFA
-  VERSION = "0.0.10"
+  VERSION = "0.0.11"
 end

data/spec/lib/cli_spec.rb

@@ -0,0 +1,38 @@
+module TFA
+  describe CLI do
+    subject { CLI.new }
+
+    def code_for(secret)
+      ::ROTP::TOTP.new(secret).now
+    end
+
+    let(:secret) { ::ROTP::Base32.random_base32 }
+
+    describe "#add" do
+      context "when a secret is added" do
+        it "adds the secret" do
+          subject.add("development", secret)
+          expect(subject.show("development")).to eql(secret)
+        end
+      end
+
+      context "when a full otpauth string is added" do
+        it "strips out the url for just the secret" do
+          url = "otpauth://totp/email@email.com?secret=#{secret}&issuer="
+
+          subject.add("development", url)
+          expect(subject.show("development")).to eql(secret)
+        end
+      end
+    end
+
+    describe "#totp" do
+      context "when a single key is given" do
+        it "returns a time based one time password" do
+          subject.add("development", secret)
+          expect(subject.totp("development")).to eql(code_for(secret))
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tfa
 version: !ruby/object:Gem::Version
-  version: 0.0.10
+  version: 0.0.11
 platform: ruby
 authors:
 - mo khan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-05-26 00:00:00.000000000 Z
+date: 2015-08-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rotp
@@ -101,6 +101,7 @@ files:
 - lib/tfa/storage.rb
 - lib/tfa/totp_command.rb
 - lib/tfa/version.rb
+- spec/lib/cli_spec.rb
 - spec/lib/console_spec.rb
 - spec/lib/totp_command_spec.rb
 - spec/spec_helper.rb
@@ -130,6 +131,7 @@ signing_key:
 specification_version: 4
 summary: A CLI to manage your one time passwords.
 test_files:
+- spec/lib/cli_spec.rb
 - spec/lib/console_spec.rb
 - spec/lib/totp_command_spec.rb
 - spec/spec_helper.rb