970 lines
		
	
	
		
			25 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			970 lines
		
	
	
		
			25 KiB
		
	
	
	
		
			Ruby
		
	
	
	
class MergeRequest < ActiveRecord::Base
 | 
						|
  include InternalId
 | 
						|
  include Issuable
 | 
						|
  include Noteable
 | 
						|
  include Referable
 | 
						|
  include Sortable
 | 
						|
  include IgnorableColumn
 | 
						|
  include CreatedAtFilterable
 | 
						|
 | 
						|
  ignore_column :locked_at
 | 
						|
 | 
						|
  belongs_to :target_project, class_name: "Project"
 | 
						|
  belongs_to :source_project, class_name: "Project"
 | 
						|
  belongs_to :merge_user, class_name: "User"
 | 
						|
 | 
						|
  has_many :merge_request_diffs
 | 
						|
  has_one :merge_request_diff,
 | 
						|
    -> { order('merge_request_diffs.id DESC') }, inverse_of: :merge_request
 | 
						|
 | 
						|
  belongs_to :head_pipeline, foreign_key: "head_pipeline_id", class_name: "Ci::Pipeline"
 | 
						|
 | 
						|
  has_many :events, as: :target, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
 | 
						|
 | 
						|
  has_many :merge_requests_closing_issues,
 | 
						|
    class_name: 'MergeRequestsClosingIssues',
 | 
						|
    dependent: :delete_all # rubocop:disable Cop/ActiveRecordDependent
 | 
						|
 | 
						|
  belongs_to :assignee, class_name: "User"
 | 
						|
 | 
						|
  serialize :merge_params, Hash # rubocop:disable Cop/ActiveRecordSerialize
 | 
						|
 | 
						|
  after_create :ensure_merge_request_diff, unless: :importing?
 | 
						|
  after_update :reload_diff_if_branch_changed
 | 
						|
  after_commit :update_project_counter_caches, on: :destroy
 | 
						|
 | 
						|
  # When this attribute is true some MR validation is ignored
 | 
						|
  # It allows us to close or modify broken merge requests
 | 
						|
  attr_accessor :allow_broken
 | 
						|
 | 
						|
  # Temporary fields to store compare vars
 | 
						|
  # when creating new merge request
 | 
						|
  attr_accessor :can_be_created, :compare_commits, :diff_options, :compare
 | 
						|
 | 
						|
  state_machine :state, initial: :opened do
 | 
						|
    event :close do
 | 
						|
      transition [:opened] => :closed
 | 
						|
    end
 | 
						|
 | 
						|
    event :mark_as_merged do
 | 
						|
      transition [:opened, :locked] => :merged
 | 
						|
    end
 | 
						|
 | 
						|
    event :reopen do
 | 
						|
      transition closed: :opened
 | 
						|
    end
 | 
						|
 | 
						|
    event :lock_mr do
 | 
						|
      transition [:opened] => :locked
 | 
						|
    end
 | 
						|
 | 
						|
    event :unlock_mr do
 | 
						|
      transition locked: :opened
 | 
						|
    end
 | 
						|
 | 
						|
    state :opened
 | 
						|
    state :closed
 | 
						|
    state :merged
 | 
						|
    state :locked
 | 
						|
  end
 | 
						|
 | 
						|
  state_machine :merge_status, initial: :unchecked do
 | 
						|
    event :mark_as_unchecked do
 | 
						|
      transition [:can_be_merged, :cannot_be_merged] => :unchecked
 | 
						|
    end
 | 
						|
 | 
						|
    event :mark_as_mergeable do
 | 
						|
      transition [:unchecked, :cannot_be_merged] => :can_be_merged
 | 
						|
    end
 | 
						|
 | 
						|
    event :mark_as_unmergeable do
 | 
						|
      transition [:unchecked, :can_be_merged] => :cannot_be_merged
 | 
						|
    end
 | 
						|
 | 
						|
    state :unchecked
 | 
						|
    state :can_be_merged
 | 
						|
    state :cannot_be_merged
 | 
						|
 | 
						|
    around_transition do |merge_request, transition, block|
 | 
						|
      Gitlab::Timeless.timeless(merge_request, &block)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  validates :source_project, presence: true, unless: [:allow_broken, :importing?, :closed_without_fork?]
 | 
						|
  validates :source_branch, presence: true
 | 
						|
  validates :target_project, presence: true
 | 
						|
  validates :target_branch, presence: true
 | 
						|
  validates :merge_user, presence: true, if: :merge_when_pipeline_succeeds?, unless: :importing?
 | 
						|
  validate :validate_branches, unless: [:allow_broken, :importing?, :closed_without_fork?]
 | 
						|
  validate :validate_fork, unless: :closed_without_fork?
 | 
						|
  validate :validate_target_project, on: :create
 | 
						|
 | 
						|
  scope :by_source_or_target_branch, ->(branch_name) do
 | 
						|
    where("source_branch = :branch OR target_branch = :branch", branch: branch_name)
 | 
						|
  end
 | 
						|
  scope :by_milestone, ->(milestone) { where(milestone_id: milestone) }
 | 
						|
  scope :of_projects, ->(ids) { where(target_project_id: ids) }
 | 
						|
  scope :from_project, ->(project) { where(source_project_id: project.id) }
 | 
						|
  scope :merged, -> { with_state(:merged) }
 | 
						|
  scope :closed_and_merged, -> { with_states(:closed, :merged) }
 | 
						|
  scope :from_source_branches, ->(branches) { where(source_branch: branches) }
 | 
						|
 | 
						|
  scope :join_project, -> { joins(:target_project) }
 | 
						|
  scope :references_project, -> { references(:target_project) }
 | 
						|
  scope :assigned, -> { where("assignee_id IS NOT NULL") }
 | 
						|
  scope :unassigned, -> { where("assignee_id IS NULL") }
 | 
						|
  scope :assigned_to, ->(u) { where(assignee_id: u.id)}
 | 
						|
 | 
						|
  participant :assignee
 | 
						|
 | 
						|
  after_save :keep_around_commit
 | 
						|
 | 
						|
  def self.reference_prefix
 | 
						|
    '!'
 | 
						|
  end
 | 
						|
 | 
						|
  # Pattern used to extract `!123` merge request references from text
 | 
						|
  #
 | 
						|
  # This pattern supports cross-project references.
 | 
						|
  def self.reference_pattern
 | 
						|
    @reference_pattern ||= %r{
 | 
						|
      (#{Project.reference_pattern})?
 | 
						|
      #{Regexp.escape(reference_prefix)}(?<merge_request>\d+)
 | 
						|
    }x
 | 
						|
  end
 | 
						|
 | 
						|
  def self.link_reference_pattern
 | 
						|
    @link_reference_pattern ||= super("merge_requests", /(?<merge_request>\d+)/)
 | 
						|
  end
 | 
						|
 | 
						|
  def self.reference_valid?(reference)
 | 
						|
    reference.to_i > 0 && reference.to_i <= Gitlab::Database::MAX_INT_VALUE
 | 
						|
  end
 | 
						|
 | 
						|
  def self.project_foreign_key
 | 
						|
    'target_project_id'
 | 
						|
  end
 | 
						|
 | 
						|
  # Returns all the merge requests from an ActiveRecord:Relation.
 | 
						|
  #
 | 
						|
  # This method uses a UNION as it usually operates on the result of
 | 
						|
  # ProjectsFinder#execute. PostgreSQL in particular doesn't always like queries
 | 
						|
  # using multiple sub-queries especially when combined with an OR statement.
 | 
						|
  # UNIONs on the other hand perform much better in these cases.
 | 
						|
  #
 | 
						|
  # relation - An ActiveRecord::Relation that returns a list of Projects.
 | 
						|
  #
 | 
						|
  # Returns an ActiveRecord::Relation.
 | 
						|
  def self.in_projects(relation)
 | 
						|
    # unscoping unnecessary conditions that'll be applied
 | 
						|
    # when executing `where("merge_requests.id IN (#{union.to_sql})")`
 | 
						|
    source = unscoped.where(source_project_id: relation).select(:id)
 | 
						|
    target = unscoped.where(target_project_id: relation).select(:id)
 | 
						|
    union  = Gitlab::SQL::Union.new([source, target])
 | 
						|
 | 
						|
    where("merge_requests.id IN (#{union.to_sql})") # rubocop:disable GitlabSecurity/SqlInjection
 | 
						|
  end
 | 
						|
 | 
						|
  WIP_REGEX = /\A\s*(\[WIP\]\s*|WIP:\s*|WIP\s+)+\s*/i.freeze
 | 
						|
 | 
						|
  def self.work_in_progress?(title)
 | 
						|
    !!(title =~ WIP_REGEX)
 | 
						|
  end
 | 
						|
 | 
						|
  def self.wipless_title(title)
 | 
						|
    title.sub(WIP_REGEX, "")
 | 
						|
  end
 | 
						|
 | 
						|
  def self.wip_title(title)
 | 
						|
    work_in_progress?(title) ? title : "WIP: #{title}"
 | 
						|
  end
 | 
						|
 | 
						|
  # Returns a Hash of attributes to be used for Twitter card metadata
 | 
						|
  def card_attributes
 | 
						|
    {
 | 
						|
      'Author'   => author.try(:name),
 | 
						|
      'Assignee' => assignee.try(:name)
 | 
						|
    }
 | 
						|
  end
 | 
						|
 | 
						|
  # These method are needed for compatibility with issues to not mess view and other code
 | 
						|
  def assignees
 | 
						|
    Array(assignee)
 | 
						|
  end
 | 
						|
 | 
						|
  def assignee_ids
 | 
						|
    Array(assignee_id)
 | 
						|
  end
 | 
						|
 | 
						|
  def assignee_ids=(ids)
 | 
						|
    write_attribute(:assignee_id, ids.last)
 | 
						|
  end
 | 
						|
 | 
						|
  def assignee_or_author?(user)
 | 
						|
    author_id == user.id || assignee_id == user.id
 | 
						|
  end
 | 
						|
 | 
						|
  # `from` argument can be a Namespace or Project.
 | 
						|
  def to_reference(from = nil, full: false)
 | 
						|
    reference = "#{self.class.reference_prefix}#{iid}"
 | 
						|
 | 
						|
    "#{project.to_reference(from, full: full)}#{reference}"
 | 
						|
  end
 | 
						|
 | 
						|
  def commits
 | 
						|
    if persisted?
 | 
						|
      merge_request_diff.commits
 | 
						|
    elsif compare_commits
 | 
						|
      compare_commits.reverse
 | 
						|
    else
 | 
						|
      []
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def commits_count
 | 
						|
    if persisted?
 | 
						|
      merge_request_diff.commits_count
 | 
						|
    elsif compare_commits
 | 
						|
      compare_commits.size
 | 
						|
    else
 | 
						|
      0
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def commit_shas
 | 
						|
    if persisted?
 | 
						|
      merge_request_diff.commit_shas
 | 
						|
    elsif compare_commits
 | 
						|
      compare_commits.reverse.map(&:sha)
 | 
						|
    else
 | 
						|
      []
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  # Calls `MergeWorker` to proceed with the merge process and
 | 
						|
  # updates `merge_jid` with the MergeWorker#jid.
 | 
						|
  # This helps tracking enqueued and ongoing merge jobs.
 | 
						|
  def merge_async(user_id, params)
 | 
						|
    jid = MergeWorker.perform_async(id, user_id, params)
 | 
						|
    update_column(:merge_jid, jid)
 | 
						|
  end
 | 
						|
 | 
						|
  def first_commit
 | 
						|
    merge_request_diff ? merge_request_diff.first_commit : compare_commits.first
 | 
						|
  end
 | 
						|
 | 
						|
  def raw_diffs(*args)
 | 
						|
    merge_request_diff ? merge_request_diff.raw_diffs(*args) : compare.raw_diffs(*args)
 | 
						|
  end
 | 
						|
 | 
						|
  def diffs(diff_options = {})
 | 
						|
    if compare
 | 
						|
      # When saving MR diffs, `expanded` is implicitly added (because we need
 | 
						|
      # to save the entire contents to the DB), so add that here for
 | 
						|
      # consistency.
 | 
						|
      compare.diffs(diff_options.merge(expanded: true))
 | 
						|
    else
 | 
						|
      merge_request_diff.diffs(diff_options)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_size
 | 
						|
    # Calling `merge_request_diff.diffs.real_size` will also perform
 | 
						|
    # highlighting, which we don't need here.
 | 
						|
    merge_request_diff&.real_size || diffs.real_size
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_base_commit
 | 
						|
    if persisted?
 | 
						|
      merge_request_diff.base_commit
 | 
						|
    else
 | 
						|
      branch_merge_base_commit
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_start_commit
 | 
						|
    if persisted?
 | 
						|
      merge_request_diff.start_commit
 | 
						|
    else
 | 
						|
      target_branch_head
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_head_commit
 | 
						|
    if persisted?
 | 
						|
      merge_request_diff.head_commit
 | 
						|
    else
 | 
						|
      source_branch_head
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_start_sha
 | 
						|
    diff_start_commit.try(:sha)
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_base_sha
 | 
						|
    diff_base_commit.try(:sha)
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_head_sha
 | 
						|
    diff_head_commit.try(:sha)
 | 
						|
  end
 | 
						|
 | 
						|
  # When importing a pull request from GitHub, the old and new branches may no
 | 
						|
  # longer actually exist by those names, but we need to recreate the merge
 | 
						|
  # request diff with the right source and target shas.
 | 
						|
  # We use these attributes to force these to the intended values.
 | 
						|
  attr_writer :target_branch_sha, :source_branch_sha
 | 
						|
 | 
						|
  def source_branch_head
 | 
						|
    return unless source_project
 | 
						|
 | 
						|
    source_branch_ref = @source_branch_sha || source_branch
 | 
						|
    source_project.repository.commit(source_branch_ref) if source_branch_ref
 | 
						|
  end
 | 
						|
 | 
						|
  def target_branch_head
 | 
						|
    target_branch_ref = @target_branch_sha || target_branch
 | 
						|
    target_project.repository.commit(target_branch_ref) if target_branch_ref
 | 
						|
  end
 | 
						|
 | 
						|
  def branch_merge_base_commit
 | 
						|
    start_sha = target_branch_sha
 | 
						|
    head_sha  = source_branch_sha
 | 
						|
 | 
						|
    if start_sha && head_sha
 | 
						|
      target_project.merge_base_commit(start_sha, head_sha)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def target_branch_sha
 | 
						|
    @target_branch_sha || target_branch_head.try(:sha)
 | 
						|
  end
 | 
						|
 | 
						|
  def source_branch_sha
 | 
						|
    @source_branch_sha || source_branch_head.try(:sha)
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_refs
 | 
						|
    if persisted?
 | 
						|
      merge_request_diff.diff_refs
 | 
						|
    else
 | 
						|
      Gitlab::Diff::DiffRefs.new(
 | 
						|
        base_sha:  diff_base_sha,
 | 
						|
        start_sha: diff_start_sha,
 | 
						|
        head_sha:  diff_head_sha
 | 
						|
      )
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def branch_merge_base_sha
 | 
						|
    branch_merge_base_commit.try(:sha)
 | 
						|
  end
 | 
						|
 | 
						|
  def validate_branches
 | 
						|
    if target_project == source_project && target_branch == source_branch
 | 
						|
      errors.add :branch_conflict, "You can not use same project/branch for source and target"
 | 
						|
    end
 | 
						|
 | 
						|
    if opened?
 | 
						|
      similar_mrs = self.target_project.merge_requests.where(source_branch: source_branch, target_branch: target_branch, source_project_id: source_project.try(:id)).opened
 | 
						|
      similar_mrs = similar_mrs.where('id not in (?)', self.id) if self.id
 | 
						|
      if similar_mrs.any?
 | 
						|
        errors.add :validate_branches,
 | 
						|
                   "Cannot Create: This merge request already exists: #{similar_mrs.pluck(:title)}"
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate_target_project
 | 
						|
    return true if target_project.merge_requests_enabled?
 | 
						|
 | 
						|
    errors.add :base, 'Target project has disabled merge requests'
 | 
						|
  end
 | 
						|
 | 
						|
  def validate_fork
 | 
						|
    return true unless target_project && source_project
 | 
						|
    return true if target_project == source_project
 | 
						|
    return true unless source_project_missing?
 | 
						|
 | 
						|
    errors.add :validate_fork,
 | 
						|
               'Source project is not a fork of the target project'
 | 
						|
  end
 | 
						|
 | 
						|
  def merge_ongoing?
 | 
						|
    !!merge_jid && !merged?
 | 
						|
  end
 | 
						|
 | 
						|
  def closed_without_fork?
 | 
						|
    closed? && source_project_missing?
 | 
						|
  end
 | 
						|
 | 
						|
  def source_project_missing?
 | 
						|
    return false unless for_fork?
 | 
						|
    return true unless source_project
 | 
						|
 | 
						|
    !source_project.forked_from?(target_project)
 | 
						|
  end
 | 
						|
 | 
						|
  def reopenable?
 | 
						|
    closed? && !source_project_missing? && source_branch_exists?
 | 
						|
  end
 | 
						|
 | 
						|
  def ensure_merge_request_diff
 | 
						|
    merge_request_diff || create_merge_request_diff
 | 
						|
  end
 | 
						|
 | 
						|
  def create_merge_request_diff
 | 
						|
    merge_request_diffs.create
 | 
						|
    reload_merge_request_diff
 | 
						|
  end
 | 
						|
 | 
						|
  def reload_merge_request_diff
 | 
						|
    merge_request_diff(true)
 | 
						|
  end
 | 
						|
 | 
						|
  def merge_request_diff_for(diff_refs_or_sha)
 | 
						|
    @merge_request_diffs_by_diff_refs_or_sha ||= Hash.new do |h, diff_refs_or_sha|
 | 
						|
      diffs = merge_request_diffs.viewable.select_without_diff
 | 
						|
      h[diff_refs_or_sha] =
 | 
						|
        if diff_refs_or_sha.is_a?(Gitlab::Diff::DiffRefs)
 | 
						|
          diffs.find_by_diff_refs(diff_refs_or_sha)
 | 
						|
        else
 | 
						|
          diffs.find_by(head_commit_sha: diff_refs_or_sha)
 | 
						|
        end
 | 
						|
    end
 | 
						|
 | 
						|
    @merge_request_diffs_by_diff_refs_or_sha[diff_refs_or_sha]
 | 
						|
  end
 | 
						|
 | 
						|
  def version_params_for(diff_refs)
 | 
						|
    if diff = merge_request_diff_for(diff_refs)
 | 
						|
      { diff_id: diff.id }
 | 
						|
    elsif diff = merge_request_diff_for(diff_refs.head_sha)
 | 
						|
      {
 | 
						|
        diff_id: diff.id,
 | 
						|
        start_sha: diff_refs.start_sha
 | 
						|
      }
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def reload_diff_if_branch_changed
 | 
						|
    if (source_branch_changed? || target_branch_changed?) &&
 | 
						|
        (source_branch_head && target_branch_head)
 | 
						|
      reload_diff
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def reload_diff(current_user = nil)
 | 
						|
    return unless open?
 | 
						|
 | 
						|
    old_diff_refs = self.diff_refs
 | 
						|
    create_merge_request_diff
 | 
						|
    MergeRequests::MergeRequestDiffCacheService.new.execute(self)
 | 
						|
    new_diff_refs = self.diff_refs
 | 
						|
 | 
						|
    update_diff_discussion_positions(
 | 
						|
      old_diff_refs: old_diff_refs,
 | 
						|
      new_diff_refs: new_diff_refs,
 | 
						|
      current_user: current_user
 | 
						|
    )
 | 
						|
  end
 | 
						|
 | 
						|
  def check_if_can_be_merged
 | 
						|
    return unless unchecked?
 | 
						|
 | 
						|
    can_be_merged =
 | 
						|
      !broken? && project.repository.can_be_merged?(diff_head_sha, target_branch)
 | 
						|
 | 
						|
    if can_be_merged
 | 
						|
      mark_as_mergeable
 | 
						|
    else
 | 
						|
      mark_as_unmergeable
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def merge_event
 | 
						|
    @merge_event ||= target_project.events.where(target_id: self.id, target_type: "MergeRequest", action: Event::MERGED).last
 | 
						|
  end
 | 
						|
 | 
						|
  def closed_event
 | 
						|
    @closed_event ||= target_project.events.where(target_id: self.id, target_type: "MergeRequest", action: Event::CLOSED).last
 | 
						|
  end
 | 
						|
 | 
						|
  def work_in_progress?
 | 
						|
    self.class.work_in_progress?(title)
 | 
						|
  end
 | 
						|
 | 
						|
  def wipless_title
 | 
						|
    self.class.wipless_title(self.title)
 | 
						|
  end
 | 
						|
 | 
						|
  def wip_title
 | 
						|
    self.class.wip_title(self.title)
 | 
						|
  end
 | 
						|
 | 
						|
  def mergeable?(skip_ci_check: false)
 | 
						|
    return false unless mergeable_state?(skip_ci_check: skip_ci_check)
 | 
						|
 | 
						|
    check_if_can_be_merged
 | 
						|
 | 
						|
    can_be_merged?
 | 
						|
  end
 | 
						|
 | 
						|
  def mergeable_state?(skip_ci_check: false)
 | 
						|
    return false unless open?
 | 
						|
    return false if work_in_progress?
 | 
						|
    return false if broken?
 | 
						|
    return false unless skip_ci_check || mergeable_ci_state?
 | 
						|
    return false unless mergeable_discussions_state?
 | 
						|
 | 
						|
    true
 | 
						|
  end
 | 
						|
 | 
						|
  def can_cancel_merge_when_pipeline_succeeds?(current_user)
 | 
						|
    can_be_merged_by?(current_user) || self.author == current_user
 | 
						|
  end
 | 
						|
 | 
						|
  def can_remove_source_branch?(current_user)
 | 
						|
    !ProtectedBranch.protected?(source_project, source_branch) &&
 | 
						|
      !source_project.root_ref?(source_branch) &&
 | 
						|
      Ability.allowed?(current_user, :push_code, source_project) &&
 | 
						|
      diff_head_commit == source_branch_head
 | 
						|
  end
 | 
						|
 | 
						|
  def should_remove_source_branch?
 | 
						|
    Gitlab::Utils.to_boolean(merge_params['should_remove_source_branch'])
 | 
						|
  end
 | 
						|
 | 
						|
  def force_remove_source_branch?
 | 
						|
    Gitlab::Utils.to_boolean(merge_params['force_remove_source_branch'])
 | 
						|
  end
 | 
						|
 | 
						|
  def remove_source_branch?
 | 
						|
    should_remove_source_branch? || force_remove_source_branch?
 | 
						|
  end
 | 
						|
 | 
						|
  def related_notes
 | 
						|
    # Fetch comments only from last 100 commits
 | 
						|
    commits_for_notes_limit = 100
 | 
						|
    commit_ids = commit_shas.take(commits_for_notes_limit)
 | 
						|
 | 
						|
    Note.where(
 | 
						|
      "(project_id = :target_project_id AND noteable_type = 'MergeRequest' AND noteable_id = :mr_id) OR" +
 | 
						|
      "((project_id = :source_project_id OR project_id = :target_project_id) AND noteable_type = 'Commit' AND commit_id IN (:commit_ids))",
 | 
						|
      mr_id: id,
 | 
						|
      commit_ids: commit_ids,
 | 
						|
      target_project_id: target_project_id,
 | 
						|
      source_project_id: source_project_id
 | 
						|
    )
 | 
						|
  end
 | 
						|
 | 
						|
  alias_method :discussion_notes, :related_notes
 | 
						|
 | 
						|
  def mergeable_discussions_state?
 | 
						|
    return true unless project.only_allow_merge_if_all_discussions_are_resolved?
 | 
						|
 | 
						|
    !discussions_to_be_resolved?
 | 
						|
  end
 | 
						|
 | 
						|
  def hook_attrs
 | 
						|
    attrs = {
 | 
						|
      source: source_project.try(:hook_attrs),
 | 
						|
      target: target_project.hook_attrs,
 | 
						|
      last_commit: nil,
 | 
						|
      work_in_progress: work_in_progress?,
 | 
						|
      total_time_spent: total_time_spent,
 | 
						|
      human_total_time_spent: human_total_time_spent,
 | 
						|
      human_time_estimate: human_time_estimate
 | 
						|
    }
 | 
						|
 | 
						|
    if diff_head_commit
 | 
						|
      attrs[:last_commit] = diff_head_commit.hook_attrs
 | 
						|
    end
 | 
						|
 | 
						|
    attributes.merge!(attrs)
 | 
						|
  end
 | 
						|
 | 
						|
  def for_fork?
 | 
						|
    target_project != source_project
 | 
						|
  end
 | 
						|
 | 
						|
  def project
 | 
						|
    target_project
 | 
						|
  end
 | 
						|
 | 
						|
  # If the merge request closes any issues, save this information in the
 | 
						|
  # `MergeRequestsClosingIssues` model. This is a performance optimization.
 | 
						|
  # Calculating this information for a number of merge requests requires
 | 
						|
  # running `ReferenceExtractor` on each of them separately.
 | 
						|
  # This optimization does not apply to issues from external sources.
 | 
						|
  def cache_merge_request_closes_issues!(current_user)
 | 
						|
    return unless project.issues_enabled?
 | 
						|
 | 
						|
    transaction do
 | 
						|
      self.merge_requests_closing_issues.delete_all
 | 
						|
 | 
						|
      closes_issues(current_user).each do |issue|
 | 
						|
        next if issue.is_a?(ExternalIssue)
 | 
						|
 | 
						|
        self.merge_requests_closing_issues.create!(issue: issue)
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  # Return the set of issues that will be closed if this merge request is accepted.
 | 
						|
  def closes_issues(current_user = self.author)
 | 
						|
    if target_branch == project.default_branch
 | 
						|
      messages = [title, description]
 | 
						|
      messages.concat(commits.map(&:safe_message)) if merge_request_diff
 | 
						|
 | 
						|
      Gitlab::ClosingIssueExtractor.new(project, current_user)
 | 
						|
        .closed_by_message(messages.join("\n"))
 | 
						|
    else
 | 
						|
      []
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def issues_mentioned_but_not_closing(current_user)
 | 
						|
    return [] unless target_branch == project.default_branch
 | 
						|
 | 
						|
    ext = Gitlab::ReferenceExtractor.new(project, current_user)
 | 
						|
    ext.analyze("#{title}\n#{description}")
 | 
						|
 | 
						|
    ext.issues - closes_issues(current_user)
 | 
						|
  end
 | 
						|
 | 
						|
  def target_project_path
 | 
						|
    if target_project
 | 
						|
      target_project.full_path
 | 
						|
    else
 | 
						|
      "(removed)"
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def source_project_path
 | 
						|
    if source_project
 | 
						|
      source_project.full_path
 | 
						|
    else
 | 
						|
      "(removed)"
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def source_project_namespace
 | 
						|
    if source_project && source_project.namespace
 | 
						|
      source_project.namespace.full_path
 | 
						|
    else
 | 
						|
      "(removed)"
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def target_project_namespace
 | 
						|
    if target_project && target_project.namespace
 | 
						|
      target_project.namespace.full_path
 | 
						|
    else
 | 
						|
      "(removed)"
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def source_branch_exists?
 | 
						|
    return false unless self.source_project
 | 
						|
 | 
						|
    self.source_project.repository.branch_names.include?(self.source_branch)
 | 
						|
  end
 | 
						|
 | 
						|
  def target_branch_exists?
 | 
						|
    return false unless self.target_project
 | 
						|
 | 
						|
    self.target_project.repository.branch_names.include?(self.target_branch)
 | 
						|
  end
 | 
						|
 | 
						|
  def merge_commit_message(include_description: false)
 | 
						|
    closes_issues_references = closes_issues.map do |issue|
 | 
						|
      issue.to_reference(target_project)
 | 
						|
    end
 | 
						|
 | 
						|
    message = [
 | 
						|
      "Merge branch '#{source_branch}' into '#{target_branch}'",
 | 
						|
      title
 | 
						|
    ]
 | 
						|
 | 
						|
    if !include_description && closes_issues_references.present?
 | 
						|
      message << "Closes #{closes_issues_references.to_sentence}"
 | 
						|
    end
 | 
						|
    message << "#{description}" if include_description && description.present?
 | 
						|
    message << "See merge request #{to_reference(full: true)}"
 | 
						|
 | 
						|
    message.join("\n\n")
 | 
						|
  end
 | 
						|
 | 
						|
  def reset_merge_when_pipeline_succeeds
 | 
						|
    return unless merge_when_pipeline_succeeds?
 | 
						|
 | 
						|
    self.merge_when_pipeline_succeeds = false
 | 
						|
    self.merge_user = nil
 | 
						|
    if merge_params
 | 
						|
      merge_params.delete('should_remove_source_branch')
 | 
						|
      merge_params.delete('commit_message')
 | 
						|
    end
 | 
						|
 | 
						|
    self.save
 | 
						|
  end
 | 
						|
 | 
						|
  # Return array of possible target branches
 | 
						|
  # depends on target project of MR
 | 
						|
  def target_branches
 | 
						|
    if target_project.nil?
 | 
						|
      []
 | 
						|
    else
 | 
						|
      target_project.repository.branch_names
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  # Return array of possible source branches
 | 
						|
  # depends on source project of MR
 | 
						|
  def source_branches
 | 
						|
    if source_project.nil?
 | 
						|
      []
 | 
						|
    else
 | 
						|
      source_project.repository.branch_names
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def has_ci?
 | 
						|
    has_ci_integration = source_project.try(:ci_service)
 | 
						|
    uses_gitlab_ci = all_pipelines.any?
 | 
						|
 | 
						|
    (has_ci_integration || uses_gitlab_ci) && commits.any?
 | 
						|
  end
 | 
						|
 | 
						|
  def branch_missing?
 | 
						|
    !source_branch_exists? || !target_branch_exists?
 | 
						|
  end
 | 
						|
 | 
						|
  def broken?
 | 
						|
    has_no_commits? || branch_missing? || cannot_be_merged?
 | 
						|
  end
 | 
						|
 | 
						|
  def can_be_merged_by?(user)
 | 
						|
    access = ::Gitlab::UserAccess.new(user, project: project)
 | 
						|
    access.can_push_to_branch?(target_branch) || access.can_merge_to_branch?(target_branch)
 | 
						|
  end
 | 
						|
 | 
						|
  def can_be_merged_via_command_line_by?(user)
 | 
						|
    access = ::Gitlab::UserAccess.new(user, project: project)
 | 
						|
    access.can_push_to_branch?(target_branch)
 | 
						|
  end
 | 
						|
 | 
						|
  def mergeable_ci_state?
 | 
						|
    return true unless project.only_allow_merge_if_pipeline_succeeds?
 | 
						|
 | 
						|
    !head_pipeline || head_pipeline.success? || head_pipeline.skipped?
 | 
						|
  end
 | 
						|
 | 
						|
  def environments_for(current_user)
 | 
						|
    return [] unless diff_head_commit
 | 
						|
 | 
						|
    @environments ||= Hash.new do |h, current_user|
 | 
						|
      envs = EnvironmentsFinder.new(target_project, current_user,
 | 
						|
        ref: target_branch, commit: diff_head_commit, with_tags: true).execute
 | 
						|
 | 
						|
      if source_project
 | 
						|
        envs.concat EnvironmentsFinder.new(source_project, current_user,
 | 
						|
          ref: source_branch, commit: diff_head_commit).execute
 | 
						|
      end
 | 
						|
 | 
						|
      h[current_user] = envs.uniq
 | 
						|
    end
 | 
						|
 | 
						|
    @environments[current_user]
 | 
						|
  end
 | 
						|
 | 
						|
  def state_human_name
 | 
						|
    if merged?
 | 
						|
      "Merged"
 | 
						|
    elsif closed?
 | 
						|
      "Closed"
 | 
						|
    else
 | 
						|
      "Open"
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def state_icon_name
 | 
						|
    if merged?
 | 
						|
      "check"
 | 
						|
    elsif closed?
 | 
						|
      "times"
 | 
						|
    else
 | 
						|
      "circle-o"
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def fetch_ref
 | 
						|
    write_ref
 | 
						|
    update_column(:ref_fetched, true)
 | 
						|
  end
 | 
						|
 | 
						|
  def ref_path
 | 
						|
    "refs/#{Repository::REF_MERGE_REQUEST}/#{iid}/head"
 | 
						|
  end
 | 
						|
 | 
						|
  def ref_fetched?
 | 
						|
    super ||
 | 
						|
      begin
 | 
						|
        computed_value = project.repository.ref_exists?(ref_path)
 | 
						|
        update_column(:ref_fetched, true) if computed_value
 | 
						|
 | 
						|
        computed_value
 | 
						|
      end
 | 
						|
  end
 | 
						|
 | 
						|
  def ensure_ref_fetched
 | 
						|
    fetch_ref unless ref_fetched?
 | 
						|
  end
 | 
						|
 | 
						|
  def in_locked_state
 | 
						|
    begin
 | 
						|
      lock_mr
 | 
						|
      yield
 | 
						|
    ensure
 | 
						|
      unlock_mr
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def diverged_commits_count
 | 
						|
    cache = Rails.cache.read(:"merge_request_#{id}_diverged_commits")
 | 
						|
 | 
						|
    if cache.blank? || cache[:source_sha] != source_branch_sha || cache[:target_sha] != target_branch_sha
 | 
						|
      cache = {
 | 
						|
        source_sha: source_branch_sha,
 | 
						|
        target_sha: target_branch_sha,
 | 
						|
        diverged_commits_count: compute_diverged_commits_count
 | 
						|
      }
 | 
						|
      Rails.cache.write(:"merge_request_#{id}_diverged_commits", cache)
 | 
						|
    end
 | 
						|
 | 
						|
    cache[:diverged_commits_count]
 | 
						|
  end
 | 
						|
 | 
						|
  def compute_diverged_commits_count
 | 
						|
    return 0 unless source_branch_sha && target_branch_sha
 | 
						|
 | 
						|
    Gitlab::Git::Commit.between(target_project.repository.raw_repository, source_branch_sha, target_branch_sha).size
 | 
						|
  end
 | 
						|
  private :compute_diverged_commits_count
 | 
						|
 | 
						|
  def diverged_from_target_branch?
 | 
						|
    diverged_commits_count > 0
 | 
						|
  end
 | 
						|
 | 
						|
  def all_pipelines
 | 
						|
    return Ci::Pipeline.none unless source_project
 | 
						|
 | 
						|
    @all_pipelines ||= source_project.pipelines
 | 
						|
      .where(sha: all_commit_shas, ref: source_branch)
 | 
						|
      .order(id: :desc)
 | 
						|
  end
 | 
						|
 | 
						|
  # Note that this could also return SHA from now dangling commits
 | 
						|
  #
 | 
						|
  def all_commit_shas
 | 
						|
    if persisted?
 | 
						|
      column_shas = MergeRequestDiffCommit.where(merge_request_diff: merge_request_diffs).pluck('DISTINCT(sha)')
 | 
						|
      serialised_shas = merge_request_diffs.where.not(st_commits: nil).flat_map(&:commit_shas)
 | 
						|
 | 
						|
      (column_shas + serialised_shas).uniq
 | 
						|
    elsif compare_commits
 | 
						|
      compare_commits.to_a.reverse.map(&:id)
 | 
						|
    else
 | 
						|
      [diff_head_sha]
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def merge_commit
 | 
						|
    @merge_commit ||= project.commit(merge_commit_sha) if merge_commit_sha
 | 
						|
  end
 | 
						|
 | 
						|
  def can_be_reverted?(current_user)
 | 
						|
    merge_commit && !merge_commit.has_been_reverted?(current_user, self)
 | 
						|
  end
 | 
						|
 | 
						|
  def can_be_cherry_picked?
 | 
						|
    merge_commit.present?
 | 
						|
  end
 | 
						|
 | 
						|
  def has_complete_diff_refs?
 | 
						|
    diff_refs && diff_refs.complete?
 | 
						|
  end
 | 
						|
 | 
						|
  def update_diff_discussion_positions(old_diff_refs:, new_diff_refs:, current_user: nil)
 | 
						|
    return unless has_complete_diff_refs?
 | 
						|
    return if new_diff_refs == old_diff_refs
 | 
						|
 | 
						|
    active_diff_discussions = self.notes.new_diff_notes.discussions.select do |discussion|
 | 
						|
      discussion.active?(old_diff_refs)
 | 
						|
    end
 | 
						|
    return if active_diff_discussions.empty?
 | 
						|
 | 
						|
    paths = active_diff_discussions.flat_map { |n| n.diff_file.paths }.uniq
 | 
						|
 | 
						|
    service = Discussions::UpdateDiffPositionService.new(
 | 
						|
      self.project,
 | 
						|
      current_user,
 | 
						|
      old_diff_refs: old_diff_refs,
 | 
						|
      new_diff_refs: new_diff_refs,
 | 
						|
      paths: paths
 | 
						|
    )
 | 
						|
 | 
						|
    active_diff_discussions.each do |discussion|
 | 
						|
      service.execute(discussion)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def keep_around_commit
 | 
						|
    project.repository.keep_around(self.merge_commit_sha)
 | 
						|
  end
 | 
						|
 | 
						|
  def has_commits?
 | 
						|
    merge_request_diff && commits_count > 0
 | 
						|
  end
 | 
						|
 | 
						|
  def has_no_commits?
 | 
						|
    !has_commits?
 | 
						|
  end
 | 
						|
 | 
						|
  def mergeable_with_quick_action?(current_user, autocomplete_precheck: false, last_diff_sha: nil)
 | 
						|
    return false unless can_be_merged_by?(current_user)
 | 
						|
 | 
						|
    return true if autocomplete_precheck
 | 
						|
 | 
						|
    return false unless mergeable?(skip_ci_check: true)
 | 
						|
    return false if head_pipeline && !(head_pipeline.success? || head_pipeline.active?)
 | 
						|
    return false if last_diff_sha != diff_head_sha
 | 
						|
 | 
						|
    true
 | 
						|
  end
 | 
						|
 | 
						|
  def update_project_counter_caches?
 | 
						|
    state_changed?
 | 
						|
  end
 | 
						|
 | 
						|
  def update_project_counter_caches
 | 
						|
    return unless update_project_counter_caches?
 | 
						|
 | 
						|
    Projects::OpenMergeRequestsCountService.new(target_project).refresh_cache
 | 
						|
  end
 | 
						|
 | 
						|
  private
 | 
						|
 | 
						|
  def write_ref
 | 
						|
    target_project.repository.with_repo_branch_commit(
 | 
						|
      source_project.repository, source_branch) do |commit|
 | 
						|
        if commit
 | 
						|
          target_project.repository.write_ref(ref_path, commit.sha)
 | 
						|
        else
 | 
						|
          raise Rugged::ReferenceError, 'source repository is empty'
 | 
						|
        end
 | 
						|
      end
 | 
						|
  end
 | 
						|
end
 |