user.rb 2.4 KB
Newer Older
1 2
require 'gitlab/oauth/user'

3 4 5 6
# LDAP extension for User model
#
# * Find or create user from omniauth.auth data
# * Links LDAP account with existing user
7
# * Auth LDAP user with login and password
8 9 10
#
module Gitlab
  module LDAP
11
    class User < Gitlab::OAuth::User
12
      class << self
13
        def authenticate(login, password)
14 15 16 17
          # Check user against LDAP backend if user is not authenticated
          # Only check with valid login and password to prevent anonymous bind results
          return nil unless ldap_conf.enabled && login.present? && password.present?

18 19
          ldap_user = adapter.bind_as(
            filter: user_filter(login),
20 21 22 23 24 25 26
            size: 1,
            password: password
          )

          find_by_uid(ldap_user.dn) if ldap_user
        end

27 28 29 30
        def adapter
          @adapter ||= OmniAuth::LDAP::Adaptor.new(ldap_conf)
        end

31 32 33 34 35 36 37 38 39
        def user_filter(login)
          filter = Net::LDAP::Filter.eq(adapter.uid, login)
          # Apply LDAP user filter if present
          if ldap_conf['user_filter'].present?
            user_filter = Net::LDAP::Filter.construct(ldap_conf['user_filter'])
            filter = Net::LDAP::Filter.join(filter, user_filter)
          end
          filter
        end
40

41 42
        def ldap_conf
          Gitlab.config.ldap
43 44
        end

45
        def find_by_uid(uid)
46 47
          # LDAP distinguished name is case-insensitive
          model.where("provider = ? and lower(extern_uid) = ?", provider, uid.downcase).last
48 49
        end

50 51
        def provider
          'ldap'
52
        end
53
      end
54

55 56 57 58
      def initialize(auth_hash)
        super
        update_user_attributes
      end
59

60 61 62 63
      # instance methods
      def gl_user
        @gl_user ||= find_by_uid_and_provider || find_by_email || build_new_user
      end
64

65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
      def find_by_uid_and_provider
        # LDAP distinguished name is case-insensitive
        model.
          where(provider: auth_hash.provider).
          where('lower(extern_uid) = ?', auth_hash.uid.downcase).last
      end

      def find_by_email
        model.find_by(email: auth_hash.email)
      end

      def update_user_attributes
        gl_user.attributes = {
          extern_uid: auth_hash.uid,
          provider: auth_hash.provider,
          email: auth_hash.email
        }
      end

      def changed?
        gl_user.changed?
86
      end
87 88 89 90

      def needs_blocking?
        false
      end
91 92 93
    end
  end
end