emails_on_push_worker.rb 2.81 KB
Newer Older
1 2 3
class EmailsOnPushWorker
  include Sidekiq::Worker

4
  sidekiq_options queue: :mailers
5 6
  attr_reader :email, :skip_premailer

Douwe Maan's avatar
Douwe Maan committed
7 8 9
  def perform(project_id, recipients, push_data, options = {})
    options.symbolize_keys!
    options.reverse_merge!(
10
      send_from_committer_email:  false,
Douwe Maan's avatar
Douwe Maan committed
11 12 13 14 15
      disable_diffs:              false
    )
    send_from_committer_email = options[:send_from_committer_email]
    disable_diffs = options[:disable_diffs]

16 17 18
    project = Project.find(project_id)
    before_sha = push_data["before"]
    after_sha = push_data["after"]
19
    ref = push_data["ref"]
20 21
    author_id = push_data["user_id"]

22
    action =
23
      if Gitlab::Git.blank_ref?(before_sha)
24
        :create
25 26 27 28 29
      elsif Gitlab::Git.blank_ref?(after_sha)
        :delete
      else
        :push
      end
30

31
    diff_refs = nil
32 33
    compare = nil
    reverse_compare = false
34

35
    if action == :push
36
      merge_base_sha = project.merge_base_commit(before_sha, after_sha).try(:sha)
37
      compare = Gitlab::Git::Compare.new(project.repository.raw_repository, before_sha, after_sha)
38 39 40 41 42 43

      diff_refs = Gitlab::Diff::DiffRefs.new(
        base_sha: merge_base_sha,
        start_sha: before_sha,
        head_sha: after_sha
      )
44

45
      return false if compare.same
46

47 48
      if compare.commits.empty?
        compare = Gitlab::Git::Compare.new(project.repository.raw_repository, after_sha, before_sha)
49 50 51 52 53 54

        diff_refs = Gitlab::Diff::DiffRefs.new(
          base_sha: merge_base_sha,
          start_sha: after_sha,
          head_sha: before_sha
        )
55

56
        reverse_compare = true
57

58 59
        return false if compare.commits.empty?
      end
60
    end
61

62
    recipients.split.each do |recipient|
63
      begin
64
        send_email(
65
          recipient,
66
          project_id,
67 68 69 70 71 72 73 74
          author_id:                 author_id,
          ref:                       ref,
          action:                    action,
          compare:                   compare,
          reverse_compare:           reverse_compare,
          diff_refs:                 diff_refs,
          send_from_committer_email: send_from_committer_email,
          disable_diffs:             disable_diffs
75 76
        )

77 78 79 80
      # These are input errors and won't be corrected even if Sidekiq retries
      rescue Net::SMTPFatalError, Net::SMTPSyntaxError => e
        logger.info("Failed to send e-mail for project '#{project.name_with_namespace}' to #{recipient}: #{e}")
      end
81
    end
82
  ensure
83
    @email = nil
84 85
    compare = nil
    GC.start
86
  end
87 88 89 90 91 92 93 94 95 96 97 98 99

  private

  def send_email(recipient, project_id, options)
    # Generating the body of this email can be expensive, so only do it once
    @skip_premailer ||= email.present?
    @email ||= Notify.repository_push_email(project_id, options)

    email.to = recipient
    email.add_message_id
    email.header[:skip_premailer] = true if skip_premailer
    email.deliver_now
  end
100
end