File: connection.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (35 lines) | stat: -rw-r--r-- 917 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# frozen_string_literal: true

module ApplicationCable
  class Connection < ActionCable::Connection::Base
    include Logging
    include Gitlab::Auth::AuthFinders

    identified_by :current_user

    public :request

    def connect
      self.current_user = find_user_from_bearer_token || find_user_from_session_store
    rescue Gitlab::Auth::AuthenticationError
      reject_unauthorized_connection
    end

    private

    def find_user_from_session_store
      session = ActiveSession.sessions_from_ids(Array.wrap(session_id)).first
      Warden::SessionSerializer.new('rack.session' => session).fetch(:user)
    end

    def session_id
      session_cookie = cookies[Gitlab::Application.config.session_options[:key]]

      Rack::Session::SessionId.new(session_cookie).private_id if session_cookie.present?
    end

    def notification_payload(_)
      super.merge!(params: request.params)
    end
  end
end