module Sorcery
  module Model
    module Submodules
      # The Remember Me submodule takes care of setting the user's cookie so that he will 
      # be automatically logged in to the site on every visit,
      # until the cookie expires.
      module RememberMe
        def self.included(base)
          base.sorcery_config.class_eval do
            attr_accessor :remember_me_token_attribute_name,              # the attribute in the model class.
                          :remember_me_token_expires_at_attribute_name,   # the expires attribute in the model class.
                          :remember_me_for                                # how long in seconds to remember.

          end

          base.sorcery_config.instance_eval do
            @defaults.merge!(:@remember_me_token_attribute_name            => :remember_me_token,
                             :@remember_me_token_expires_at_attribute_name => :remember_me_token_expires_at,
                             :@remember_me_for                             => 7 * 60 * 60 * 24)

            reset!
          end

          base.send(:include, InstanceMethods)

          base.sorcery_config.after_config << :define_remember_me_mongoid_fields if defined?(Mongoid) and base.ancestors.include?(Mongoid::Document)
          if defined?(MongoMapper) and base.ancestors.include?(MongoMapper::Document)
            base.sorcery_config.after_config << :define_remember_me_mongo_mapper_fields
          end
          if defined?(DataMapper) and base.ancestors.include?(DataMapper::Resource)
            base.sorcery_config.after_config << :define_remember_me_datamapper_fields
          end

          base.extend(ClassMethods)
        end

        module ClassMethods
          protected

          def define_remember_me_mongoid_fields
            field sorcery_config.remember_me_token_attribute_name,            :type => String
            field sorcery_config.remember_me_token_expires_at_attribute_name, :type => Time
          end

          def define_remember_me_mongo_mapper_fields
            key sorcery_config.remember_me_token_attribute_name, String
            key sorcery_config.remember_me_token_expires_at_attribute_name, Time
          end

          def define_remember_me_datamapper_fields
            property sorcery_config.remember_me_token_attribute_name,            String
            property sorcery_config.remember_me_token_expires_at_attribute_name, Time
            [sorcery_config.remember_me_token_expires_at_attribute_name].each do |sym|
              alias_method "orig_#{sym}", sym
              define_method(sym) do
                t = send("orig_#{sym}")
                t && Time.new(t.year, t.month, t.day, t.hour, t.min, t.sec, 0)
              end
            end
          end
        end

        module InstanceMethods
          # You shouldn't really use this one yourself - it's called by the controller's 'remember_me!' method.
          def remember_me!
            config = sorcery_config
            self.update_many_attributes(config.remember_me_token_attribute_name => TemporaryToken.generate_random_token,
                                        config.remember_me_token_expires_at_attribute_name => Time.now.in_time_zone + config.remember_me_for)
          end
          
          # You shouldn't really use this one yourself - it's called by the controller's 'forget_me!' method.
          def forget_me!
            config = sorcery_config
            self.update_many_attributes(config.remember_me_token_attribute_name => nil,
                                        config.remember_me_token_expires_at_attribute_name => nil)
          end
        end
      end
    end
  end
end