Fix issue with header too large

Author: tungleduyxyz

app/controllers/kaui/engine_controller.rb

@@ -12,11 +12,12 @@ class Kaui::EngineController < ApplicationController
   # Common options for the Kill Bill client
   def options_for_klient(options = {})
     user_tenant_options = Kaui.current_tenant_user_options(current_user, session)
-    user_tenant_options.merge(options)
+    user_tenant_options.merge!(options)
     # Pass the X-Request-Id seen by Rails to Kill Bill
     # Note that this means that subsequent requests issued by a single action will share the same X-Request-Id in Kill Bill
     user_tenant_options[:request_id] ||= request.request_id
-    user_tenant_options[:jwt_token] ||= session[:jwt_token]
+    # Use session storage for JWT tokens instead of cookies to avoid header size limits
+    user_tenant_options[:jwt_token] = session[:aviate_jwt_token]
     user_tenant_options
   end
 

app/controllers/kaui/sessions_controller.rb

@@ -28,6 +28,7 @@ def after_sign_in_path_for(_resource)
     end
 
     def after_sign_out_path_for(_resource)
+      cookies.delete(:jwt_token)
       kaui_path
     end
 

lib/kaui.rb

@@ -297,7 +297,6 @@ def self.current_tenant_user_options(user, session)
       password: user.password,
       session_id: user.kb_session_id
     }
-    result[:jwt_token] = session[:jwt_token] if session[:jwt_token]
     if user_tenant
       result[:api_key] = user_tenant.api_key
       result[:api_secret] = user_tenant.api_secret