vendor/rhosync/lib/x_domain_session_wrapper.rb

x_domain_session_wrapper.rb

require "cgi"

class XDomainSessionWrapper
  def initialize(app, opts={})
    @app = app
    @session_cookie = opts[:session_cookie] || 'rhosync_session'
    yield self if block_given?
  end

  def is_sync_protocol(env)
    # if it is rhosync protocol URI
    /\A\/application/.match(env['REQUEST_PATH'])
  end

  def call(env)
    if is_sync_protocol(env)
      #env.each{|f| puts '------->> ' +f.inspect}
      # extract cookie from url if empty
      env['HTTP_COOKIE'] = env['HTTP_COOKIE'] || CGI.unescape(get_session_from_url(env))
      puts "FIXED HTTP_COOKIE: #{env['HTTP_COOKIE']}"
    end

    status, headers, body = @app.call(env)

    if is_sync_protocol(env)
      cookies = headers['Set-Cookie'].to_s
      #puts "<----- Cookies: #{cookies}"
      # put cookies to body as JSON on login success
      if /\/clientlogin/.match(env['REQUEST_PATH']) && status == 200
        body = session_json_from(cookies)
        headers['Content-Length'] = body.length.to_s
      end
    end

    [status, headers, body]
  end

  def session_json_from(cookies)
    rexp = Regexp.new(@session_cookie +'=[^\s]*')
    sc = cookies.to_s.slice rexp
    "{\"" +@session_cookie +"\": \"#{CGI.escape sc.to_s}\"}"
  end

  def get_session_from_url(env)
    rexp = Regexp.new(@session_cookie +'=.*\Z')
    qs = env['QUERY_STRING'].to_s.slice rexp
    qs = qs.to_s.split(/&/)[0]
    nv = qs.to_s.split(/=/)
    return nv[1] if nv.length > 1
    ''
  end
end