Sorcery/sorcery

module Sorcery
  module Model
    module Submodules
      # This submodule keeps track of events such as login, logout, and last activity time, per user.
      # It helps in estimating which users are active now in the site.
      # This cannot be determined absolutely because a user might be reading a page without clicking anything
      # for a while.
      # This is the model part of the submodule, which provides configuration options.
      module ActivityLogging
        def self.included(base)
          base.extend(ClassMethods)
          base.send(:include, InstanceMethods)

          base.sorcery_config.class_eval do
            # last login attribute name.
            attr_accessor :last_login_at_attribute_name
            # last logout attribute name.
            attr_accessor :last_logout_at_attribute_name
            # last activity attribute name.
            attr_accessor :last_activity_at_attribute_name
            # last activity login source
            attr_accessor :last_login_from_ip_address_name
            # how long since last activity is the user defined offline
            attr_accessor :activity_timeout
          end

          base.sorcery_config.instance_eval do
            @defaults.merge!(:@last_login_at_attribute_name                => :last_login_at,
                             :@last_logout_at_attribute_name               => :last_logout_at,
                             :@last_activity_at_attribute_name             => :last_activity_at,
                             :@last_login_from_ip_address_name             => :last_login_from_ip_address,
                             :@activity_timeout                            => 10 * 60)
            reset!
          end

          base.sorcery_config.after_config << :define_activity_logging_fields
        end

        module InstanceMethods
          def set_last_login_at(time)
            sorcery_adapter.update_attribute(sorcery_config.last_login_at_attribute_name, time)
          end

          def set_last_logout_at(time)
            sorcery_adapter.update_attribute(sorcery_config.last_logout_at_attribute_name, time)
          end

          def set_last_activity_at(time)
            sorcery_adapter.update_attribute(sorcery_config.last_activity_at_attribute_name, time)
          end

          def set_last_ip_address(ip_address)
            sorcery_adapter.update_attribute(sorcery_config.last_login_from_ip_address_name, ip_address)
          end

          # online method shows if user is active (logout action makes user inactive too)
          def online?
            return false if send(sorcery_config.last_activity_at_attribute_name).nil?

            logged_in? && send(sorcery_config.last_activity_at_attribute_name) > sorcery_config.activity_timeout.seconds.ago
          end

          #  shows if user is logged in, but it not show if user is online - see online?
          def logged_in?
            return false if send(sorcery_config.last_login_at_attribute_name).nil?
            return true if send(sorcery_config.last_login_at_attribute_name).present? && send(sorcery_config.last_logout_at_attribute_name).nil?

            send(sorcery_config.last_login_at_attribute_name) > send(sorcery_config.last_logout_at_attribute_name)
          end

          def logged_out?
            !logged_in?
          end
        end

        module ClassMethods
          protected

          def define_activity_logging_fields
            sorcery_adapter.define_field sorcery_config.last_login_at_attribute_name,    Time
            sorcery_adapter.define_field sorcery_config.last_logout_at_attribute_name,   Time
            sorcery_adapter.define_field sorcery_config.last_activity_at_attribute_name, Time
            sorcery_adapter.define_field sorcery_config.last_login_from_ip_address_name, String
          end
        end
      end
    end
  end
end