controller.rb

module Sorcery
  module Controller
    def self.included(klass)
      klass.class_eval do
        extend ClassMethods
        include InstanceMethods
        Config.submodules.each do |mod|
          begin
            include Submodules.const_get(mod.to_s.split("_").map {|p| p.capitalize}.join("")) 
          rescue NameError
            # don't stop on a missing submodule.
          end
        end
        Config.update!
      end
    end
    
    module ClassMethods
      def activate_sorcery!(&block)
        yield Config if block_given?
        after_config!
      end
      
      def after_config!
        Config.after_config.each {|c| send(c)}       
      end
    end
    
    module InstanceMethods
      # To be used as before_filter.
      # Will trigger auto-login attempts via the call to logged_in?
      # If all attempts to auto-login fail, the failure callback will be called.
      def require_login
        if !logged_in?
          session[:user_wanted_url] = request.url if Config.save_user_wanted_url
          self.send(Config.not_authenticated_action) 
        end
      end
      
      def login(*credentials)
        user = Config.user_class.authenticate(*credentials)
        if user
          reset_session # protect from session fixation attacks
          login_user(user)
          after_login!(user, credentials)
          logged_in_user
        else
          after_failed_login!(credentials)
          nil
        end
      end
      
      def logout
        if logged_in?
          before_logout!(logged_in_user)
          reset_session
          after_logout!
        end
      end
      
      def logged_in?
        !!logged_in_user
      end
      
      # attempts to auto-login from the sources defined (session, basic_auth, cookie, etc.)
      # returns the logged in user if found, false if not (using old restful-authentication trick, nil != false).
      def logged_in_user
        @logged_in_user ||= login_from_session || login_from_other_sources unless @logged_in_user == false
      end
      
      def login_from_other_sources
        result = nil
        Config.login_sources.find do |source|
          result = send(source)
        end
        result || false
      end
      
      def not_authenticated
        redirect_to root_path
      end
      
      protected
      
      def login_user(user)
        session[:user_id] = user.id
      end
      
      def login_from_session
        @logged_in_user = (Config.user_class.find_by_id(session[:user_id]) if session[:user_id]) || false
      end
      
      def after_login!(user, credentials)
        Config.after_login.each {|c| self.send(c, user, credentials)}
      end
      
      def after_failed_login!(credentials)
        Config.after_failed_login.each {|c| self.send(c, credentials)}
      end
      
      def before_logout!(user)
        Config.before_logout.each {|c| self.send(c, user)}
      end
      
      def after_logout!
        Config.after_logout.each {|c| self.send(c)}
      end
      
    end
    
    module Config
      class << self
        attr_accessor :submodules,
        
                      :user_class,                    # what class to use as the user class.
                      
                      :not_authenticated_action,      # what controller action to call for non-authenticated users.
                      
                      :save_user_wanted_url,          # when a non logged in user tries to enter a page that requires login, save the URL he wanted to reach, 
                                                      # and send him there after login.
                      
                      :login_sources,
                      :after_login,
                      :after_failed_login,
                      :before_logout,
                      :after_logout,
                      :after_config
                      
                      
        def init!
          @defaults = {
            :@user_class                           => nil,
            :@submodules                           => [],
            :@not_authenticated_action             => :not_authenticated,
            :@login_sources                        => [],
            :@after_login                          => [],
            :@after_failed_login                   => [],
            :@before_logout                        => [],
            :@after_logout                         => [],
            :@after_config                         => [],
            :@save_user_wanted_url                 => true
          }
        end
        
        # Resets all configuration options to their default values.
        def reset!
          @defaults.each do |k,v|
            instance_variable_set(k,v)
          end       
        end
        
        def update!
          @defaults.each do |k,v|
            instance_variable_set(k,v) if !instance_variable_defined?(k)
          end
        end
      end
      init!
      reset!
    end
  end
end