Class: SoldierOfCode::DejaVu::Middleware

Inherits:
Object
  • Object
show all
Defined in:
lib/deja-vu.rb

Instance Method Summary collapse

Constructor Details

#initialize(app, opt = {:enable_record=>true, :cookie_name=>'deja-vu'}) ⇒ Middleware

Possible Options:

:cookie_name => optional - name of the cookie to watch for and use as identifier - defaults to deja-vu :enable_record => optional - (true|false) :session_items => optional - [array of session keys] if present will capture



22
23
24
25
26
 
# File 'lib/deja-vu.rb', line 22

def initialize(app, opt={:enable_record=>true, :cookie_name=>'deja-vu'})
  @app = app
  @opt = opt
  @identifier = opt[:cookie_name]
end

Instance Method Details

#call(env) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/deja-vu.rb', line 28

def call(env)
  if @opt[:enable_record] then
    user_identifier_before = get_user_identifier(env)
    env_before = env
    t_start = Time.new
#          puts "#{__FILE__}:#{__LINE__} #{Thread.current} HERE BEFORE APP CALL"
    resp = @app.call(env) # would be nice to capture all log output from the downstream as well.... TODO add log object as logger
#          puts "#{__FILE__}:#{__LINE__} #{Thread.current} HERE AFTER APP CALL"
    t_stop = Time.new
    user_identifier_after = get_user_identifier(env)

    current_request = Rack::Request.new(env)

    dejavu_recorder = Recorder.new(@opt)
    if user_identifier_before != user_identifier_after then
#            puts "#{__FILE__}:#{__LINE__} #{Thread.current} HERE SOMETHING CHANGED"
      dejavu_recorder.identifier_change(user_identifier_after)
    end

#          puts "#{__FILE__}:#{__LINE__} #{Thread.current} ABOUT TO DO THE RECORD"
    dejavu_recorder.record(env, resp, current_request, t_start, t_stop, user_identifier_after)
#          puts "#{__FILE__}:#{__LINE__} #{Thread.current} FINISHED THE RECORD OPERATION"
    resp
  else
#          puts "#{__FILE__}:#{__LINE__} #{Thread.current} HERE SKIPPED COMPLETLEY"
    @app.call(env)
  end
end

#get_user_identifier(env) ⇒ Object

will use the identified cookie if available otherwise will create a hash based on the env data available



62
63
64
65
66
67
68
69
 
# File 'lib/deja-vu.rb', line 62

def get_user_identifier(env) # => a string representing a specific browser client user
#        puts "#{__FILE__}:#{__LINE__} #{Thread.current} HERE DOING IDENTIFIER WORK"
  http_accept = env['HTTP_ACCEPT']
  http_agent = env['HTTP_USER_AGENT']
  user_ip = env['REMOTE_ADDR']

  Rack::Request.new(env).cookies[@identifier] || Digest::SHA1.hexdigest("#{user_ip}#{http_agent}#{http_accept}")
end