require 'rack/utils'
require 'action_controller/session/abstract_store'
require 'action_controller/flash'
 
module ActionController
  module Session
    class RedisCacheStore < AbstractStore
      def initialize(app, options = {})
        require 'redis'
 
        # Support old :expires option
        options[:expire_after] ||= options[:expires]
 
        super
 
        @mutex = Mutex.new
 
        @default_options = {
          :namespace => 'rack:session',
          :host => 'localhost',
          :port => 6379
        }.merge(@default_options)
 
        @cache = options[:cache] || Redis.new(@default_options)
        
        @mutex = Mutex.new
 
        super
      end
 
      private
        def get_session(env, sid)
          sid ||= generate_sid
          begin
            session = Marshal.load(@cache[sid]) || {}
          rescue Exception, Errno::ECONNREFUSED
            session = {}
          end
          [sid, session]
        end
 
        def set_session(env, sid, session_data)
          options = env['rack.session.options']
          expiry  = options[:expire_after] || 0
          @cache.set(sid, Marshal.dump(session_data), expiry)
          return true
        rescue Exception, Errno::ECONNREFUSED
          return false
        end
    end
  end
end