Refactor post-receive worker

This commit is contained in:
Dmitriy Zaporozhets 2013-04-29 09:43:18 +03:00
parent 348eb12598
commit a1704273ec
2 changed files with 32 additions and 18 deletions

View File

@ -1,5 +1,6 @@
class PostReceive
include Sidekiq::Worker
include Gitlab::Identifier
sidekiq_options queue: :post_receive
@ -8,7 +9,7 @@ class PostReceive
if repo_path.start_with?(Gitlab.config.gitlab_shell.repos_path.to_s)
repo_path.gsub!(Gitlab.config.gitlab_shell.repos_path.to_s, "")
else
Gitlab::GitLogger.error("POST-RECEIVE: Check gitlab.yml config for correct gitlab_shell.repos_path variable. \"#{Gitlab.config.gitlab_shell.repos_path}\" does not match \"#{repo_path}\"")
log("Check gitlab.yml config for correct gitlab_shell.repos_path variable. \"#{Gitlab.config.gitlab_shell.repos_path}\" does not match \"#{repo_path}\"")
end
repo_path.gsub!(/.git$/, "")
@ -17,31 +18,21 @@ class PostReceive
project = Project.find_with_namespace(repo_path)
if project.nil?
Gitlab::GitLogger.error("POST-RECEIVE: Triggered hook for non-existing project with full path \"#{repo_path} \"")
log("Triggered hook for non-existing project with full path \"#{repo_path} \"")
return false
end
user = if identifier.blank?
# Local push from gitlab
email = project.repository.commit(newrev).author_email rescue nil
User.find_by_email(email) if email
elsif identifier =~ /\Auser-\d+\Z/
# git push over http
user_id = identifier.gsub("user-", "")
User.find_by_id(user_id)
elsif identifier =~ /\Akey-\d+\Z/
# git push over ssh
key_id = identifier.gsub("key-", "")
Key.find_by_id(key_id).try(:user)
end
user = identify(identifier, project, newrev)
unless user
Gitlab::GitLogger.error("POST-RECEIVE: Triggered hook for non-existing user \"#{identifier} \"")
log("Triggered hook for non-existing user \"#{identifier} \"")
return false
end
GitPushService.new.execute(project, user, oldrev, newrev, ref)
end
def log(message)
Gitlab::GitLogger.error("POST-RECEIVE: #{message}")
end
end

23
lib/gitlab/identifier.rb Normal file
View File

@ -0,0 +1,23 @@
# Detect user based on identifier like
# key-13 or user-36 or last commit
module Gitlab
module Indentifier
def identify(identifier, project, newrev)
if identifier.blank?
# Local push from gitlab
email = project.repository.commit(newrev).author_email rescue nil
User.find_by_email(email) if email
elsif identifier =~ /\Auser-\d+\Z/
# git push over http
user_id = identifier.gsub("user-", "")
User.find_by_id(user_id)
elsif identifier =~ /\Akey-\d+\Z/
# git push over ssh
key_id = identifier.gsub("key-", "")
Key.find_by_id(key_id).try(:user)
end
end
end
end