summaryrefslogtreecommitdiff
path: root/lib/gitlab/ldap/auth_hash.rb
blob: 4fbc5fa526218148ace326d28bc736c0ab905dda (plain)
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
36
# Class to parse and transform the info provided by omniauth
#
module Gitlab
  module LDAP
    class AuthHash < Gitlab::OAuth::AuthHash
      private

      def get_info(key)
        attributes = ldap_config.attributes[key.to_s]
        return super unless attributes

        attributes = Array(attributes)

        value = nil
        attributes.each do |attribute|
          value = get_raw(attribute)
          value = value.first if value
          break if value.present?
        end

        return super unless value

        Gitlab::Utils.force_utf8(value)
        value
      end

      def get_raw(key)
        auth_hash.extra[:raw_info][key] if auth_hash.extra
      end

      def ldap_config
        @ldap_config ||= Gitlab::LDAP::Config.new(self.provider)
      end
    end
  end
end