65 lines
1.7 KiB
Ruby
65 lines
1.7 KiB
Ruby
# Detect user based on identifier like
|
|
# key-13 or user-36 or last commit
|
|
module Gitlab
|
|
module Identifier
|
|
def identify(identifier, project, newrev)
|
|
if identifier.blank?
|
|
# Local push from gitlab
|
|
identify_using_commit(project, newrev)
|
|
elsif identifier =~ /\Auser-\d+\Z/
|
|
# git push over http
|
|
identify_using_user(identifier)
|
|
elsif identifier =~ /\Akey-\d+\Z/
|
|
# git push over ssh
|
|
identify_using_ssh_key(identifier)
|
|
end
|
|
end
|
|
|
|
# Tries to identify a user based on a commit SHA.
|
|
def identify_using_commit(project, ref)
|
|
commit = project.commit(ref)
|
|
|
|
return if !commit || !commit.author_email
|
|
|
|
email = commit.author_email
|
|
|
|
identify_with_cache(:email, email) do
|
|
User.find_by(email: email)
|
|
end
|
|
end
|
|
|
|
# Tries to identify a user based on a user identifier (e.g. "user-123").
|
|
def identify_using_user(identifier)
|
|
user_id = identifier.gsub("user-", "")
|
|
|
|
identify_with_cache(:user, user_id) do
|
|
User.find_by(id: user_id)
|
|
end
|
|
end
|
|
|
|
# Tries to identify a user based on an SSH key identifier (e.g. "key-123").
|
|
def identify_using_ssh_key(identifier)
|
|
key_id = identifier.gsub("key-", "")
|
|
|
|
identify_with_cache(:ssh_key, key_id) do
|
|
User.find_by_ssh_key_id(key_id)
|
|
end
|
|
end
|
|
|
|
def identify_with_cache(category, key)
|
|
if identification_cache[category].key?(key)
|
|
identification_cache[category][key]
|
|
else
|
|
identification_cache[category][key] = yield
|
|
end
|
|
end
|
|
|
|
def identification_cache
|
|
@identification_cache ||= {
|
|
email: {},
|
|
user: {},
|
|
ssh_key: {}
|
|
}
|
|
end
|
|
end
|
|
end
|