gitlab-org--gitlab-foss/app/services/git_push_service.rb

class GitPushService
  attr_accessor :project, :user, :push_data, :push_commits
  include Gitlab::CurrentSettings
  include Gitlab::Access

  def initialize(project, user, oldrev, newrev, ref)
    # TODO: Consider passing an options hash instead https://github.com/bbatsov/ruby-style-guide#too-many-params
    @project = project
    @user = user
    @oldrev = oldrev
    @newrev = newrev
    @ref = ref
  end

  # This method will be called after each git update
  # and only if the provided user and project is present in GitLab.
  #
  # All callbacks for post receive action should be placed here.
  #
  # Next, this method:
  #  1. Creates the push event
  #  2. Updates merge requests
  #  3. Recognizes cross-references from commit messages
  #  4. Executes the project's web hooks
  #  5. Executes the project's services
  #
  def execute
    @project.repository.expire_cache(branch_name)

    if push_remove_branch?
      @project.repository.expire_has_visible_content_cache
      @push_commits = []
    elsif push_to_new_branch?
      @project.repository.expire_has_visible_content_cache

      # Re-find the pushed commits.
      if is_default_branch?
        # Initial push to the default branch. Take the full history of that branch as "newly pushed".
        process_default_branch
      else
        # Use the pushed commits that aren't reachable by the default branch
        # as a heuristic. This may include more commits than are actually pushed, but
        # that shouldn't matter because we check for existing cross-references later.
        @push_commits = @project.repository.commits_between(@project.default_branch, @newrev)

        # don't process commits for the initial push to the default branch
        process_commit_messages
      end
    elsif push_to_existing_branch?
      # Collect data for this git push
      @push_commits = @project.repository.commits_between(@oldrev, @newrev)
      process_commit_messages
    end
    # Update merge requests that may be affected by this push. A new branch
    # could cause the last commit of a merge request to change.
    update_merge_requests
  end

  protected

  def update_merge_requests
    @project.update_merge_requests(@oldrev, @newrev, @ref, @user)

    EventCreateService.new.push(@project, @user, build_push_data)
    @project.execute_hooks(build_push_data.dup, :push_hooks)
    @project.execute_services(build_push_data.dup, :push_hooks)
    CreateCommitBuildsService.new.execute(@project, @user, build_push_data)
    ProjectCacheWorker.perform_async(@project.id)
  end

  def process_default_branch
    @push_commits = project.repository.commits(@newrev)

    # Ensure HEAD points to the default branch in case it is not master
    project.change_head(branch_name)

    # Set protection on the default branch if configured
    if (current_application_settings.default_branch_protection != PROTECTION_NONE)
      developers_can_push = current_application_settings.default_branch_protection == PROTECTION_DEV_CAN_PUSH ? true : false
      @project.protected_branches.create({ name: @project.default_branch, developers_can_push: developers_can_push })
    end
  end

  # Extract any GFM references from the pushed commit messages. If the configured issue-closing regex is matched,
  # close the referenced Issue. Create cross-reference Notes corresponding to any other referenced Mentionables.
  def process_commit_messages
    is_default_branch = is_default_branch?

    authors = Hash.new do |hash, commit|
      email = commit.author_email
      next hash[email] if hash.has_key?(email)

      hash[email] = commit_user(commit)
    end

    @push_commits.each do |commit|
      # Keep track of the issues that will be actually closed because they are on a default branch.
      # Hence, when creating cross-reference notes, the not-closed issues (on non-default branches)
      # will also have cross-reference.
      closed_issues = []

      if is_default_branch
        # Close issues if these commits were pushed to the project's default branch and the commit message matches the
        # closing regex. Exclude any mentioned Issues from cross-referencing even if the commits are being pushed to
        # a different branch.
        closed_issues = commit.closes_issues(user)
        closed_issues.each do |issue|
          Issues::CloseService.new(project, authors[commit], {}).execute(issue, commit)
        end
      end

      commit.create_cross_references!(authors[commit], closed_issues)
    end
  end

  def build_push_data
    @push_data ||= Gitlab::PushDataBuilder.
        build(@project, @user, @oldrev, @newrev, @ref, push_commits)
  end

  def push_to_existing_branch?
    # Return if this is not a push to a branch (e.g. new commits)
    Gitlab::Git.branch_ref?(@ref) && !Gitlab::Git.blank_ref?(@oldrev)
  end

  def push_to_new_branch?
    Gitlab::Git.branch_ref?(@ref) && Gitlab::Git.blank_ref?(@oldrev)
  end

  def push_remove_branch?
    Gitlab::Git.branch_ref?(@ref) && Gitlab::Git.blank_ref?(@newrev)
  end

  def push_to_branch?
    Gitlab::Git.branch_ref?(@ref)
  end

  def is_default_branch?
    Gitlab::Git.branch_ref?(@ref) &&
        (Gitlab::Git.ref_name(@ref) == project.default_branch || project.default_branch.nil?)
  end

  def commit_user(commit)
    commit.author || user
  end

  def branch_name
    @_branch_name ||= Gitlab::Git.ref_name(@ref)
  end
end