Support sinatra extraction

Author: Inbal Tako

lib/context/hanami_context.rb

@@ -7,7 +7,11 @@ def self.get_client_token(request)
     begin
       request.env[SECURENATIVE_COOKIE]
     rescue StandardError
-      nil
+      begin
+        request.cookies[SECURENATIVE_COOKIE]
+      rescue StandardError
+        nil
+      end
     end
   end
 

lib/context/rails_context.rb

@@ -35,7 +35,7 @@ def self.get_method(request)
 
   def self.get_headers(request)
     begin
-      # Note: At the moment we're filtering out everything but user-agent since rails payload is way too big
+      # Note: At the moment we're filtering out everything but user-agent since ruby's payload is way too big
       {"user-agent" => request.headers['HTTP_USER_AGENT']}
     rescue StandardError
       nil

lib/context/sinatra_context.rb

@@ -7,29 +7,34 @@ def self.get_client_token(request)
     begin
       request.env[SECURENATIVE_COOKIE]
     rescue StandardError
-      nil
+      begin
+        request.cookies[SECURENATIVE_COOKIE]
+      rescue StandardError
+        nil
+      end
     end
   end
 
   def self.get_url(request)
     begin
-      request.url
+      request.env['REQUEST_URI']
     rescue StandardError
       nil
     end
   end
 
   def self.get_method(request)
     begin
-      request.method
+      request.env['REQUEST_METHOD']
     rescue StandardError
       nil
     end
   end
 
   def self.get_headers(request)
     begin
-      request.headers.to_h
+      # Note: At the moment we're filtering out everything but user-agent since ruby's payload is way too big
+      {"user-agent" => request.env['HTTP_USER_AGENT']}
     rescue StandardError
       nil
     end

lib/models/verify_result.rb

@@ -9,4 +9,8 @@ def initialize(risk_level: nil, score: nil, triggers: nil)
     @score = score
     @triggers = triggers
   end
+
+  def to_s
+    "risk_level: #{@risk_level}, score: #{@score}, triggers: #{@triggers}"
+  end
 end