summaryrefslogtreecommitdiff
path: root/lib/gitlab/saml/user.rb
blob: b1e30110ef55b9d7c3044d5193c766dbf9d61066 (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
37
38
39
40
41
42
43
44
45
46
47
# SAML extension for User model
#
# * Find GitLab user based on SAML uid and provider
# * Create new user from SAML data
#
module Gitlab
  module Saml
    class User < Gitlab::OAuth::User

      def save
        super('SAML')
      end

      def gl_user
        @user ||= find_by_uid_and_provider

        if auto_link_ldap_user?
          @user ||= find_or_create_ldap_user
        end

        if auto_link_saml_enabled?
          @user ||= find_by_email
        end

        if signup_enabled?
          @user ||= build_new_user
        end

        @user
      end

      def find_by_email
        if auth_hash.has_email?
          user = ::User.find_by(email: auth_hash.email.downcase)
          user.identities.new(extern_uid: auth_hash.uid, provider: auth_hash.provider) if user
          user
        end
      end

      protected

      def auto_link_saml_enabled?
        Gitlab.config.omniauth.auto_link_saml_user
      end
    end
  end
end