Add after_login_lock callback

CHANGELOG.md

@@ -1,6 +1,8 @@
 # Changelog
 ## HEAD
 
+* Add `after_login_lock` callback [#236](https://github.com/Sorcery/sorcery/pull/236)
+
 ## 0.16.0
 
 * Add BattleNet Provider [#260](https://github.com/Sorcery/sorcery/pull/260)

lib/sorcery/controller.rb

@@ -162,6 +162,10 @@ def after_remember_me!(user)
         Config.after_remember_me.each { |c| send(c, user) }
       end
 
+      def after_login_lock!(credentials)
+        Config.after_login_lock.each { |c| send(c, credentials) }
+      end
+
       def user_class
         @user_class ||= Config.user_class.to_s.constantize
       rescue NameError

lib/sorcery/controller/config.rb

@@ -19,6 +19,7 @@ class << self
         attr_accessor :before_logout
         attr_accessor :after_logout
         attr_accessor :after_remember_me
+        attr_accessor :after_login_lock
 
         def init!
           @defaults = {
@@ -31,6 +32,7 @@ def init!
             :@before_logout                        => Set.new,
             :@after_logout                         => Set.new,
             :@after_remember_me                    => Set.new,
+            :@after_login_lock                     => Set.new,
             :@save_return_to_url                   => true,
             :@cookie_domain                        => nil
           }

lib/sorcery/controller/submodules/brute_force_protection.rb

@@ -22,7 +22,10 @@ module InstanceMethods
           # Runs as a hook after a failed login.
           def update_failed_logins_count!(credentials)
             user = user_class.sorcery_adapter.find_by_credentials(credentials)
-            user.register_failed_login! if user
+            if user && !user.login_locked?
+              user.register_failed_login!
+              after_login_lock!(credentials) if user.login_locked?
+            end
           end
 
           # Resets the failed logins counter.

spec/controllers/controller_brute_force_protection_spec.rb

@@ -22,6 +22,7 @@ def request_test_login
     it 'counts login retries' do
       allow(User).to receive(:authenticate) { |&block| block.call(nil, :other) }
       allow(User.sorcery_adapter).to receive(:find_by_credentials).with(['[email protected]', 'blabla']).and_return(user)
+      allow(user).to receive(:login_locked?).and_return(false)
 
       expect(user).to receive(:register_failed_login!).exactly(3).times
 
@@ -37,5 +38,16 @@ def request_test_login
 
       get :test_login, params: { email: '[email protected]', password: 'secret' }
     end
+
+    it 'calls after_login_lock when user locked' do
+      allow(User).to receive(:authenticate) { |&block| block.call(nil, :other) }
+      allow(User.sorcery_adapter).to receive(:find_by_credentials).with(['[email protected]', 'blabla']).and_return(user)
+      allow(user).to receive(:register_failed_login!)
+      allow(user).to receive(:login_locked?).and_return(false, true)
+
+      expect(@controller).to receive(:after_login_lock!).exactly(1).times
+
+      request_test_login
+    end
   end
 end