81 lines
2.2 KiB
Ruby
81 lines
2.2 KiB
Ruby
require 'gitlab/o_auth/user'
|
|
|
|
# LDAP extension for User model
|
|
#
|
|
# * Find or create user from omniauth.auth data
|
|
# * Links LDAP account with existing user
|
|
# * Auth LDAP user with login and password
|
|
#
|
|
module Gitlab
|
|
module LDAP
|
|
class User < Gitlab::OAuth::User
|
|
class << self
|
|
def find_by_uid_and_provider(uid, provider)
|
|
# LDAP distinguished name is case-insensitive
|
|
identity = ::Identity.
|
|
where(provider: provider).
|
|
where('lower(extern_uid) = ?', uid.downcase).last
|
|
identity && identity.user
|
|
end
|
|
end
|
|
|
|
def initialize(auth_hash)
|
|
super
|
|
update_user_attributes
|
|
end
|
|
|
|
# instance methods
|
|
def gl_user
|
|
@gl_user ||= find_by_uid_and_provider || find_by_email || build_new_user
|
|
end
|
|
|
|
def find_by_uid_and_provider
|
|
self.class.find_by_uid_and_provider(
|
|
auth_hash.uid.downcase, auth_hash.provider)
|
|
end
|
|
|
|
def find_by_email
|
|
::User.find_by(email: auth_hash.email)
|
|
end
|
|
|
|
def update_user_attributes
|
|
return unless persisted?
|
|
|
|
gl_user.skip_reconfirmation!
|
|
gl_user.email = auth_hash.email
|
|
|
|
# find_or_initialize_by doesn't update `gl_user.identities`, and isn't autosaved.
|
|
identity = gl_user.identities.find { |identity| identity.provider == auth_hash.provider }
|
|
identity ||= gl_user.identities.build(provider: auth_hash.provider)
|
|
|
|
# For a new user set extern_uid to the LDAP DN
|
|
# For an existing user with matching email but changed DN, update the DN.
|
|
# For an existing user with no change in DN, this line changes nothing.
|
|
identity.extern_uid = auth_hash.uid
|
|
|
|
gl_user
|
|
end
|
|
|
|
def changed?
|
|
gl_user.changed? || gl_user.identities.any?(&:changed?)
|
|
end
|
|
|
|
def block_after_signup?
|
|
ldap_config.block_auto_created_users
|
|
end
|
|
|
|
def allowed?
|
|
Gitlab::LDAP::Access.allowed?(gl_user)
|
|
end
|
|
|
|
def ldap_config
|
|
Gitlab::LDAP::Config.new(auth_hash.provider)
|
|
end
|
|
|
|
def auth_hash=(auth_hash)
|
|
@auth_hash = Gitlab::LDAP::AuthHash.new(auth_hash)
|
|
end
|
|
end
|
|
end
|
|
end
|