From 97337657da56d65f53ec7a46f6164c7391130a71 Mon Sep 17 00:00:00 2001 From: Pine Wong Date: Tue, 19 Feb 2019 19:06:23 +0800 Subject: [PATCH] Add failure reason argument for `after_failed_login` callback --- lib/sorcery/controller.rb | 6 +++--- lib/sorcery/controller/submodules/brute_force_protection.rb | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/lib/sorcery/controller.rb b/lib/sorcery/controller.rb index 7b40adce..759511e3 100644 --- a/lib/sorcery/controller.rb +++ b/lib/sorcery/controller.rb @@ -36,7 +36,7 @@ def login(*credentials) user_class.authenticate(*credentials) do |user, failure_reason| if failure_reason - after_failed_login!(credentials) + after_failed_login!(credentials, failure_reason) yield(user, failure_reason) if block_given? @@ -143,8 +143,8 @@ def after_login!(user, credentials = []) Config.after_login.each { |c| send(c, user, credentials) } end - def after_failed_login!(credentials) - Config.after_failed_login.each { |c| send(c, credentials) } + def after_failed_login!(credentials, failure_reason) + Config.after_failed_login.each { |c| send(c, credentials, failure_reason) } end def before_logout! diff --git a/lib/sorcery/controller/submodules/brute_force_protection.rb b/lib/sorcery/controller/submodules/brute_force_protection.rb index e4756b7b..66b168f8 100644 --- a/lib/sorcery/controller/submodules/brute_force_protection.rb +++ b/lib/sorcery/controller/submodules/brute_force_protection.rb @@ -20,7 +20,7 @@ module InstanceMethods # Increments the failed logins counter on every failed login. # Runs as a hook after a failed login. - def update_failed_logins_count!(credentials) + def update_failed_logins_count!(credentials, _failure_reason) user = user_class.sorcery_adapter.find_by_credentials(credentials) user.register_failed_login! if user end