diff options
4 files changed, 21 insertions, 20 deletions
diff --git a/app/controllers/application_controller.rb b/app/controllers/application_controller.rb index 693bd86..be7aa1f 100644 --- a/app/controllers/application_controller.rb +++ b/app/controllers/application_controller.rb @@ -1,10 +1,5 @@  class ApplicationController < ActionController::Base    protect_from_forgery -  helper_method :current_user - -  private -  def current_user -    @current_user ||= User.find(session[:user_id]) if session[:user_id] -  end +  ActiveSupport.run_load_hooks(:application_controller, self)  end diff --git a/users/app/controllers/application_controller.rb b/users/app/controllers/application_controller.rb deleted file mode 100644 index 64e1a55..0000000 --- a/users/app/controllers/application_controller.rb +++ /dev/null @@ -1,14 +0,0 @@ -class ApplicationController < ActionController::Base -  protect_from_forgery - -  private - -  def current_user -    @current_user ||= User.find(session[:user_id]) if session[:user_id] -  end -  helper_method :current_user - -  def authorize -    redirect_to login_url, alert: "Not authorized" if current_user.nil? -  end -end diff --git a/users/app/controllers/controller_extension/authentication.rb b/users/app/controllers/controller_extension/authentication.rb new file mode 100644 index 0000000..507b62f --- /dev/null +++ b/users/app/controllers/controller_extension/authentication.rb @@ -0,0 +1,17 @@ +module ControllerExtension::Authentication +  extend ActiveSupport::Concern + +  private + +  included do +    helper_method :current_user +  end + +  def current_user +    @current_user ||= User.find(session[:user_id]) if session[:user_id] +  end + +  def authorize +    redirect_to login_url, :alert => "Not authorized" if current_user.nil? +  end +end diff --git a/users/config/initializers/add_controller_methods.rb b/users/config/initializers/add_controller_methods.rb new file mode 100644 index 0000000..2579176 --- /dev/null +++ b/users/config/initializers/add_controller_methods.rb @@ -0,0 +1,3 @@ +ActiveSupport.on_load(:application_controller) do +  include ControllerExtension::Authentication +end  | 
