DRY'ed up the user model
This commit is contained in:
parent
1b0198f1d3
commit
0dd94cd86e
1 changed files with 32 additions and 37 deletions
|
@ -86,36 +86,42 @@ class User < ActiveRecord::Base
|
||||||
where('id NOT IN (SELECT DISTINCT(user_id) FROM users_projects)')
|
where('id NOT IN (SELECT DISTINCT(user_id) FROM users_projects)')
|
||||||
end
|
end
|
||||||
|
|
||||||
def self.find_or_new_for_omniauth(oauth)
|
def self.create_from_omniauth(auth, ldap = false)
|
||||||
provider, uid = oauth['provider'], oauth['uid']
|
provider, uid = auth.provider, auth.uid
|
||||||
|
name = auth.info.name.force_encoding("utf-8")
|
||||||
|
email = auth.info.email.downcase unless auth.info.email.nil?
|
||||||
|
|
||||||
|
ldap_prefix = ldap ? '(LDAP) ' : ''
|
||||||
|
raise OmniAuth::Error, "#{ldap_prefix}#{provider} does not provide an email"\
|
||||||
|
" address" if auth.info.email.blank?
|
||||||
|
|
||||||
|
logger.info "#{ldap_prefix}Creating user from #{provider} login"\
|
||||||
|
" {uid => #{uid}, name => #{name}, email => #{email}}"
|
||||||
|
password = Devise.friendly_token[0, 8].downcase
|
||||||
|
@user = User.new(
|
||||||
|
extern_uid: uid,
|
||||||
|
provider: provider,
|
||||||
|
name: name,
|
||||||
|
email: email,
|
||||||
|
password: password,
|
||||||
|
password_confirmation: password,
|
||||||
|
projects_limit: Gitlab.config.default_projects_limit,
|
||||||
|
)
|
||||||
|
if Gitlab.config.omniauth.block_auto_created_users && !ldap
|
||||||
|
@user.blocked = true
|
||||||
|
end
|
||||||
|
@user.save!
|
||||||
|
@user
|
||||||
|
end
|
||||||
|
|
||||||
|
def self.find_or_new_for_omniauth(auth)
|
||||||
|
provider, uid = auth.provider, auth.uid
|
||||||
|
|
||||||
if @user = User.find_by_provider_and_extern_uid(provider, uid)
|
if @user = User.find_by_provider_and_extern_uid(provider, uid)
|
||||||
@user
|
@user
|
||||||
else
|
else
|
||||||
if Gitlab.config.omniauth.allow_single_sign_on
|
if Gitlab.config.omniauth.allow_single_sign_on
|
||||||
# Ensure here that all required attributes were passed along with the
|
@user = User.create_from_omniauth(auth)
|
||||||
# oauth request:
|
|
||||||
%w(first_name last_name email).each do |attr|
|
|
||||||
unless oauth[:info][attr].present?
|
|
||||||
raise OmniAuth::Error,
|
|
||||||
"#{provider} does not provide the required field #{attr}"
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
password = Devise.friendly_token[0, 8].downcase
|
|
||||||
@user = User.new(
|
|
||||||
extern_uid: uid,
|
|
||||||
provider: provider,
|
|
||||||
name: "#{oauth[:info][:first_name]} #{oauth[:info][:last_name]}",
|
|
||||||
email: oauth[:info][:email],
|
|
||||||
password: password,
|
|
||||||
password_confirmation: password,
|
|
||||||
projects_limit: Gitlab.config.default_projects_limit,
|
|
||||||
)
|
|
||||||
|
|
||||||
@user.blocked = true if Gitlab.config.omniauth.block_auto_created_users
|
|
||||||
@user.save!
|
|
||||||
|
|
||||||
@user
|
@user
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
@ -124,7 +130,6 @@ class User < ActiveRecord::Base
|
||||||
def self.find_for_ldap_auth(auth, signed_in_resource=nil)
|
def self.find_for_ldap_auth(auth, signed_in_resource=nil)
|
||||||
uid = auth.info.uid
|
uid = auth.info.uid
|
||||||
provider = auth.provider
|
provider = auth.provider
|
||||||
name = auth.info.name.force_encoding("utf-8")
|
|
||||||
email = auth.info.email.downcase unless auth.info.email.nil?
|
email = auth.info.email.downcase unless auth.info.email.nil?
|
||||||
raise OmniAuth::Error, "LDAP accounts must provide an uid and email address" if uid.nil? or email.nil?
|
raise OmniAuth::Error, "LDAP accounts must provide an uid and email address" if uid.nil? or email.nil?
|
||||||
|
|
||||||
|
@ -136,17 +141,7 @@ class User < ActiveRecord::Base
|
||||||
@user.update_attributes(:extern_uid => uid, :provider => provider)
|
@user.update_attributes(:extern_uid => uid, :provider => provider)
|
||||||
@user
|
@user
|
||||||
else
|
else
|
||||||
logger.info "Creating user from LDAP login {uid => #{uid}, name => #{name}, email => #{email}}"
|
create_from_omniauth(auth)
|
||||||
password = Devise.friendly_token[0, 8].downcase
|
|
||||||
@user = User.create(
|
|
||||||
:extern_uid => uid,
|
|
||||||
:provider => provider,
|
|
||||||
:name => name,
|
|
||||||
:email => email,
|
|
||||||
:password => password,
|
|
||||||
:password_confirmation => password,
|
|
||||||
:projects_limit => Gitlab.config.default_projects_limit
|
|
||||||
)
|
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue