# frozen_string_literal: true
module Isomorfeus
module Transport
class RackMiddleware
WS_RESPONSE = [0, {}, []]
def initialize(app)
@app = app
end
def user_from_env(env)
cookies = env['HTTP_COOKIE']
if cookies
cookies = cookies.split('; ')
cookie = cookies.detect { |c| c.start_with?('session=') }
if cookie
session_id = cookie[8..-1]
if !session_id.nil? && !session_id.empty?
user = Isomorfeus.session_class&.get_user(session_id: session_id)
return [user, session_id] unless user.nil?
end
end
end
[Anonymous.new, nil]
end
def call(env)
case env['PATH_INFO']
when Isomorfeus.api_websocket_path
if env['rack.upgrade?'] == :websocket
user, session_id = user_from_env(env)
env['rack.upgrade'] = Isomorfeus::Transport::ServerSocketProcessor.new(user, session_id)
end
WS_RESPONSE
when Isomorfeus.cookie_eater_path
cookie_accessor, new_path = env['QUERY_STRING'].split('=')
cookie = Isomorfeus.session_class&.take_cookie(accessor: cookie_accessor)
if new_path.start_with?('/')
if cookie
[302, { 'Location' => new_path, 'Set-Cookie' => cookie }, ["Cookie eaten!"]]
else
[302, { 'Location' => new_path }, ["No Cookie!"]]
end
else
[404, {}, ["Must specify relative path!"]]
end
when Isomorfeus.api_logout_path
user, session_id = user_from_env(env)
if user
begin
Isomorfeus.session_class&.remove(session_id: session_id)
cookie = "session=#{session_id}; SameSite=Strict; HttpOnly; Path=/; expires=Thu, 01 Jan 1970 00:00:00 UTC#{'; Secure' if Isomorfeus.production?}"
return [302, { 'Location' => '/', 'Set-Cookie' => cookie }, ["Logged out!"]]
ensure
Thread.current[:isomorfeus_user] = nil
Thread.current[:isomorfeus_session_id] = nil
end
end
return [302, { 'Location' => '/', 'Set-Cookie' => cookie }, ["Tried to log out!"]]
else
user, session_id = user_from_env(env)
if user
Thread.current[:isomorfeus_user] = user
Thread.current[:isomorfeus_session_id] = session_id
end
begin
result = @app.call(env)
ensure
Thread.current[:isomorfeus_user] = nil
Thread.current[:isomorfeus_session_id] = nil
end
result
end
rescue Exception => e
e_text = "#{e.class}: #{e.message}\n #{e.backtrace.join("\n")}"
STDERR.puts e_text
message = if Isomorfeus.production?
"ErrorSorry, a error occured!"
else
"#{e.class}#{e_text}"
end
return [500, { Rack::CONTENT_TYPE => "text/html", Rack::CONTENT_LENGTH => message.length.to_s }, [message]]
end
end
end
end