oneaws 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2eafc4ba671ad4dc11df4fade663322d6b4ed6d7f7c89f8b2490c2c9eab63b08
-  data.tar.gz: f1fec019a30a413da02a4d05cb7f4f25aeea959999f46027b2969209dc2319d3
+  metadata.gz: e224b3f6d58994ed4f8f52cfe00fd22a04980d9f5ff1ff959a04b4bc51672463
+  data.tar.gz: 65c32e08d72e7e3d740727d69bfa4b69b7bb4aafe2b9b7a8dd2b91f423ff5eb2
 SHA512:
-  metadata.gz: f0cc8de11530ae39439165ddeb7efab54ab1bf0a14542e2724464e8073ae5f5bf75caf650b39a89f06663f0fe36a9d4cac0396a04309dd293aa745fb7800d3c3
-  data.tar.gz: 833d79033134bf4bfe4728aca62b7b10c3eeeeb2fe7b17c7573f87bd7c4a8b6d9a9de468a51457a6c14bde1aedf9a6803e10ec26c64a877abc4063432a203a05
+  metadata.gz: 185a66fbe4c028298bd0bf41f2563e8ec0477af3c1bd17103c01815dadbcb25dbceb8f06e3c70ca45f129868a9bb0f55458984875958bbec266e50c37972b8a0
+  data.tar.gz: 47e3f4f1a5b89a9f01f4ba9929a467425205673149962c56f6758c6addafafacbb6488f0934b7f7229e4c1df6e4f0e763872d9058cc45d6d3ef90399f8e72d6f

data/lib/oneaws/client.rb

@@ -4,6 +4,7 @@ require 'aws-sdk-core'
 module Oneaws
   class Client
     class SamlRequestError < StandardError; end
+    class MfaDeviceNotFoundError < StandardError; end;
 
     def initialize
       @onelogin = OneLogin::Api::Client.new({
@@ -27,18 +28,24 @@ module Oneaws
       if response.nil?
         raise SamlRequestError.new("#{@onelogin.error} #{@onelogin.error_description}")
       end
-      
+
       mfa = response.mfa
 
       # sent push notification to OneLogin Protect
-      response = @onelogin.get_saml_assertion_verifying(app_id, mfa.devices[0].id, mfa.state_token, nil, nil, false)
+      mfa_device = mfa.devices.select{|device| device.type == "OneLogin Protect"}&.first
+
+      if mfa_device.nil?
+        raise MfaDeviceNotFoundError.new("OneLogin Protect device not found.")
+      end
+
+      response = @onelogin.get_saml_assertion_verifying(app_id, mfa_device.id, mfa.state_token, nil, nil, false)
       if response.nil?
         raise SamlRequestError.new("#{@onelogin.error} #{@onelogin.error_description}")
       end
 
       while response.type != "success" do
         sleep 1
-        response = @onelogin.get_saml_assertion_verifying(app_id, mfa.devices[0].id, mfa.state_token, nil, nil, true)
+        response = @onelogin.get_saml_assertion_verifying(app_id, mfa_device.id, mfa.state_token, nil, nil, true)
         if response.nil?
           raise SamlRequestError.new("#{@onelogin.error} #{@onelogin.error_description}")
         end

data/lib/oneaws/version.rb

@@ -1,3 +1,3 @@
 module Oneaws
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: oneaws
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Yuki Koya
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-04-28 00:00:00.000000000 Z
+date: 2020-05-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor