merge_request.rb 25.7 KB
Newer Older
D
Dmitriy Zaporozhets 已提交
1
class MergeRequest < ActiveRecord::Base
2
  include InternalId
3
  include Issuable
4
  include Noteable
5
  include Referable
6
  include IgnorableColumn
7
  include TimeTrackable
8

9 10
  ignore_column :locked_at,
                :ref_fetched
11

12 13
  belongs_to :target_project, class_name: "Project"
  belongs_to :source_project, class_name: "Project"
Z
Zeger-Jan van de Weg 已提交
14
  belongs_to :merge_user, class_name: "User"
15

16
  has_many :merge_request_diffs
17
  has_one :merge_request_diff,
18
    -> { order('merge_request_diffs.id DESC') }, inverse_of: :merge_request
19

20 21
  belongs_to :head_pipeline, foreign_key: "head_pipeline_id", class_name: "Ci::Pipeline"

22
  has_many :events, as: :target, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
23

24 25 26
  has_many :merge_requests_closing_issues,
    class_name: 'MergeRequestsClosingIssues',
    dependent: :delete_all # rubocop:disable Cop/ActiveRecordDependent
27

28 29
  belongs_to :assignee, class_name: "User"

30
  serialize :merge_params, Hash # rubocop:disable Cop/ActiveRecordSerialize
Z
Zeger-Jan van de Weg 已提交
31

32 33
  after_create :ensure_merge_request_diff, unless: :importing?
  after_update :reload_diff_if_branch_changed
34
  after_commit :update_project_counter_caches, on: :destroy
35

D
Dmitriy Zaporozhets 已提交
36 37 38 39
  # When this attribute is true some MR validation is ignored
  # It allows us to close or modify broken merge requests
  attr_accessor :allow_broken

D
Dmitriy Zaporozhets 已提交
40 41
  # Temporary fields to store compare vars
  # when creating new merge request
42
  attr_accessor :can_be_created, :compare_commits, :diff_options, :compare
D
Dmitriy Zaporozhets 已提交
43

A
Andrew8xx8 已提交
44
  state_machine :state, initial: :opened do
A
Andrew8xx8 已提交
45
    event :close do
46
      transition [:opened] => :closed
A
Andrew8xx8 已提交
47 48
    end

49
    event :mark_as_merged do
50
      transition [:opened, :locked] => :merged
A
Andrew8xx8 已提交
51 52 53
    end

    event :reopen do
54
      transition closed: :opened
A
Andrew8xx8 已提交
55 56
    end

57
    event :lock_mr do
58
      transition [:opened] => :locked
D
Dmitriy Zaporozhets 已提交
59 60
    end

61
    event :unlock_mr do
62
      transition locked: :opened
D
Dmitriy Zaporozhets 已提交
63 64
    end

A
Andrew8xx8 已提交
65 66 67
    state :opened
    state :closed
    state :merged
D
Dmitriy Zaporozhets 已提交
68
    state :locked
A
Andrew8xx8 已提交
69 70
  end

71 72 73 74 75 76
  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
77
      transition [:unchecked, :cannot_be_merged] => :can_be_merged
78 79 80
    end

    event :mark_as_unmergeable do
81
      transition [:unchecked, :can_be_merged] => :cannot_be_merged
82 83
    end

84
    state :unchecked
85 86
    state :can_be_merged
    state :cannot_be_merged
87 88

    around_transition do |merge_request, transition, block|
89
      Gitlab::Timeless.timeless(merge_request, &block)
90
    end
91
  end
92

93
  validates :source_project, presence: true, unless: [:allow_broken, :importing?, :closed_without_fork?]
A
Andrey Kumanyaev 已提交
94
  validates :source_branch, presence: true
I
Izaak Alpert 已提交
95
  validates :target_project, presence: true
A
Andrey Kumanyaev 已提交
96
  validates :target_branch, presence: true
J
James Lopez 已提交
97
  validates :merge_user, presence: true, if: :merge_when_pipeline_succeeds?, unless: :importing?
98 99
  validate :validate_branches, unless: [:allow_broken, :importing?, :closed_without_fork?]
  validate :validate_fork, unless: :closed_without_fork?
100
  validate :validate_target_project, on: :create
D
Dmitriy Zaporozhets 已提交
101

102 103 104
  scope :by_source_or_target_branch, ->(branch_name) do
    where("source_branch = :branch OR target_branch = :branch", branch: branch_name)
  end
105
  scope :by_milestone, ->(milestone) { where(milestone_id: milestone) }
106
  scope :of_projects, ->(ids) { where(target_project_id: ids) }
107
  scope :from_project, ->(project) { where(source_project_id: project.id) }
108 109
  scope :merged, -> { with_state(:merged) }
  scope :closed_and_merged, -> { with_states(:closed, :merged) }
S
Scott Le 已提交
110
  scope :from_source_branches, ->(branches) { where(source_branch: branches) }
111

112 113
  scope :join_project, -> { joins(:target_project) }
  scope :references_project, -> { references(:target_project) }
114 115 116 117 118
  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
119

120 121
  after_save :keep_around_commit

122 123
  acts_as_paranoid

124 125 126 127
  def self.reference_prefix
    '!'
  end

128 129 130 131
  # Pattern used to extract `!123` merge request references from text
  #
  # This pattern supports cross-project references.
  def self.reference_pattern
132
    @reference_pattern ||= %r{
133
      (#{Project.reference_pattern})?
134 135 136 137
      #{Regexp.escape(reference_prefix)}(?<merge_request>\d+)
    }x
  end

138
  def self.link_reference_pattern
139
    @link_reference_pattern ||= super("merge_requests", /(?<merge_request>\d+)/)
140 141
  end

142 143 144 145
  def self.reference_valid?(reference)
    reference.to_i > 0 && reference.to_i <= Gitlab::Database::MAX_INT_VALUE
  end

146 147 148 149
  def self.project_foreign_key
    'target_project_id'
  end

150 151 152 153 154 155 156 157 158 159 160
  # 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)
M
mhasbini 已提交
161 162 163 164
    # 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)
165 166
    union  = Gitlab::SQL::Union.new([source, target])

167
    where("merge_requests.id IN (#{union.to_sql})") # rubocop:disable GitlabSecurity/SqlInjection
168 169
  end

T
Thomas Balthazar 已提交
170 171 172 173 174 175 176 177 178 179 180 181 182 183
  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

184
  def hook_attrs
185
    Gitlab::HookData::MergeRequestBuilder.new(self).build
186 187
  end

188 189 190 191 192 193 194 195
  # Returns a Hash of attributes to be used for Twitter card metadata
  def card_attributes
    {
      'Author'   => author.try(:name),
      'Assignee' => assignee.try(:name)
    }
  end

196
  # These method are needed for compatibility with issues to not mess view and other code
197 198 199 200
  def assignees
    Array(assignee)
  end

201 202 203 204 205 206 207 208
  def assignee_ids
    Array(assignee_id)
  end

  def assignee_ids=(ids)
    write_attribute(:assignee_id, ids.last)
  end

209 210 211 212
  def assignee_or_author?(user)
    author_id == user.id || assignee_id == user.id
  end

213
  # `from` argument can be a Namespace or Project.
214
  def to_reference(from = nil, full: false)
215 216
    reference = "#{self.class.reference_prefix}#{iid}"

217
    "#{project.to_reference(from, full: full)}#{reference}"
218 219
  end

220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249
  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

250 251 252
  # Calls `MergeWorker` to proceed with the merge process and
  # updates `merge_jid` with the MergeWorker#jid.
  # This helps tracking enqueued and ongoing merge jobs.
253
  def merge_async(user_id, params)
254 255 256 257
    jid = MergeWorker.perform_async(id, user_id, params)
    update_column(:merge_jid, jid)
  end

258 259
  def first_commit
    merge_request_diff ? merge_request_diff.first_commit : compare_commits.first
260
  end
261

262
  def raw_diffs(*args)
263
    merge_request_diff ? merge_request_diff.raw_diffs(*args) : compare.raw_diffs(*args)
S
Sean McGivern 已提交
264 265
  end

266
  def diffs(diff_options = {})
267
    if compare
268
      # When saving MR diffs, `expanded` is implicitly added (because we need
269 270
      # to save the entire contents to the DB), so add that here for
      # consistency.
271
      compare.diffs(diff_options.merge(expanded: true))
272
    else
273
      merge_request_diff.diffs(diff_options)
274
    end
S
Sean McGivern 已提交
275 276
  end

J
Jacob Vosmaer 已提交
277
  def diff_size
278 279
    # Calling `merge_request_diff.diffs.real_size` will also perform
    # highlighting, which we don't need here.
280
    merge_request_diff&.real_size || diffs.real_size
J
Jacob Vosmaer 已提交
281 282
  end

283
  def diff_base_commit
284
    if persisted?
285
      merge_request_diff.base_commit
286 287
    else
      branch_merge_base_commit
288 289 290 291 292 293 294 295
    end
  end

  def diff_start_commit
    if persisted?
      merge_request_diff.start_commit
    else
      target_branch_head
296 297 298
    end
  end

299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325
  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
326 327
    return unless source_project

328
    source_branch_ref = @source_branch_sha || source_branch
S
Sean McGivern 已提交
329
    source_project.repository.commit(source_branch_ref) if source_branch_ref
330 331 332 333
  end

  def target_branch_head
    target_branch_ref = @target_branch_sha || target_branch
S
Sean McGivern 已提交
334
    target_project.repository.commit(target_branch_ref) if target_branch_ref
335 336
  end

337 338 339 340 341 342 343 344 345
  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

346
  def target_branch_sha
347
    @target_branch_sha || target_branch_head.try(:sha)
348 349 350
  end

  def source_branch_sha
351
    @source_branch_sha || source_branch_head.try(:sha)
352 353
  end

354
  def diff_refs
355
    if persisted?
356
      merge_request_diff.diff_refs
357
    else
358 359 360 361 362
      Gitlab::Diff::DiffRefs.new(
        base_sha:  diff_base_sha,
        start_sha: diff_start_sha,
        head_sha:  diff_head_sha
      )
363
    end
364 365
  end

366 367 368 369
  def branch_merge_base_sha
    branch_merge_base_commit.try(:sha)
  end

370
  def validate_branches
371
    if target_project == source_project && target_branch == source_branch
I
Izaak Alpert 已提交
372
      errors.add :branch_conflict, "You can not use same project/branch for source and target"
373
    end
374

375
    if opened?
376
      similar_mrs = self.target_project.merge_requests.where(source_branch: source_branch, target_branch: target_branch, source_project_id: source_project.try(:id)).opened
377 378
      similar_mrs = similar_mrs.where('id not in (?)', self.id) if self.id
      if similar_mrs.any?
J
jubianchi 已提交
379
        errors.add :validate_branches,
G
Gabriel Mazetto 已提交
380
                   "Cannot Create: This merge request already exists: #{similar_mrs.pluck(:title)}"
381
      end
382
    end
383 384
  end

385 386 387 388 389 390
  def validate_target_project
    return true if target_project.merge_requests_enabled?

    errors.add :base, 'Target project has disabled merge requests'
  end

391
  def validate_fork
392
    return true unless target_project && source_project
393
    return true if target_project == source_project
394
    return true unless source_project_missing?
395

396
    errors.add :validate_fork,
397
               'Source project is not a fork of the target project'
398 399
  end

400
  def merge_ongoing?
401 402
    # While the MergeRequest is locked, it should present itself as 'merge ongoing'.
    # The unlocking process is handled by StuckMergeJobsWorker scheduled in Cron.
403 404 405
    return true if locked?

    !!merge_jid && !merged? && Gitlab::SidekiqStatus.running?(merge_jid)
406 407
  end

408
  def closed_without_fork?
409
    closed? && source_project_missing?
410 411
  end

412
  def source_project_missing?
413 414 415
    return false unless for_fork?
    return true unless source_project

416
    !source_project.in_fork_network_of?(target_project)
417 418
  end

419
  def reopenable?
420
    closed? && !source_project_missing? && source_branch_exists?
K
Katarzyna Kobierska 已提交
421 422
  end

423 424
  def ensure_merge_request_diff
    merge_request_diff || create_merge_request_diff
425 426
  end

427
  def create_merge_request_diff
428
    fetch_ref!
429

430 431 432 433 434
    # n+1: https://gitlab.com/gitlab-org/gitlab-ce/issues/37435
    Gitlab::GitalyClient.allow_n_plus_1_calls do
      merge_request_diffs.create
      reload_merge_request_diff
    end
435 436 437 438 439 440
  end

  def reload_merge_request_diff
    merge_request_diff(true)
  end

441 442 443 444 445 446 447 448 449
  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
D
Douwe Maan 已提交
450 451
    end

452
    @merge_request_diffs_by_diff_refs_or_sha[diff_refs_or_sha]
453 454
  end

455 456 457 458 459 460 461 462 463 464 465
  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

466
  def reload_diff_if_branch_changed
467 468
    if (source_branch_changed? || target_branch_changed?) &&
        (source_branch_head && target_branch_head)
469
      reload_diff
D
Dmitriy Zaporozhets 已提交
470 471 472
    end
  end

473
  def reload_diff(current_user = nil)
474 475
    return unless open?

476
    old_diff_refs = self.diff_refs
477

478
    create_merge_request_diff
479
    MergeRequests::MergeRequestDiffCacheService.new.execute(self)
480 481
    new_diff_refs = self.diff_refs

482
    update_diff_discussion_positions(
483
      old_diff_refs: old_diff_refs,
484 485
      new_diff_refs: new_diff_refs,
      current_user: current_user
486
    )
487 488
  end

489
  def check_if_can_be_merged
T
Toon Claes 已提交
490
    return unless unchecked? && Gitlab::Database.read_write?
491

492
    can_be_merged =
493
      !broken? && project.repository.can_be_merged?(diff_head_sha, target_branch)
494 495

    if can_be_merged
496 497 498 499
      mark_as_mergeable
    else
      mark_as_unmergeable
    end
500 501
  end

D
Dmitriy Zaporozhets 已提交
502
  def merge_event
503
    @merge_event ||= target_project.events.where(target_id: self.id, target_type: "MergeRequest", action: Event::MERGED).last
D
Dmitriy Zaporozhets 已提交
504 505
  end

506
  def closed_event
507
    @closed_event ||= target_project.events.where(target_id: self.id, target_type: "MergeRequest", action: Event::CLOSED).last
508 509
  end

510
  def work_in_progress?
T
Thomas Balthazar 已提交
511
    self.class.work_in_progress?(title)
512 513 514
  end

  def wipless_title
T
Thomas Balthazar 已提交
515 516 517 518 519
    self.class.wipless_title(self.title)
  end

  def wip_title
    self.class.wip_title(self.title)
520 521
  end

522 523
  def mergeable?(skip_ci_check: false)
    return false unless mergeable_state?(skip_ci_check: skip_ci_check)
524 525 526 527

    check_if_can_be_merged

    can_be_merged?
528 529
  end

530
  def mergeable_state?(skip_ci_check: false)
531 532 533
    return false unless open?
    return false if work_in_progress?
    return false if broken?
534
    return false unless skip_ci_check || mergeable_ci_state?
535
    return false unless mergeable_discussions_state?
536 537

    true
538 539
  end

540 541 542 543 544 545 546 547
  def ff_merge_possible?
    project.repository.ancestor?(target_branch_sha, diff_head_sha)
  end

  def should_be_rebased?
    project.ff_merge_must_be_possible? && !ff_merge_possible?
  end

J
James Lopez 已提交
548
  def can_cancel_merge_when_pipeline_succeeds?(current_user)
549
    can_be_merged_by?(current_user) || self.author == current_user
550 551
  end

552
  def can_remove_source_branch?(current_user)
553
    !ProtectedBranch.protected?(source_project, source_branch) &&
554
      !source_project.root_ref?(source_branch) &&
H
http://jneen.net/ 已提交
555
      Ability.allowed?(current_user, :push_code, source_project) &&
556
      diff_head_commit == source_branch_head
557 558
  end

559
  def should_remove_source_branch?
560
    Gitlab::Utils.to_boolean(merge_params['should_remove_source_branch'])
561 562 563
  end

  def force_remove_source_branch?
564
    Gitlab::Utils.to_boolean(merge_params['force_remove_source_branch'])
565 566 567 568 569 570
  end

  def remove_source_branch?
    should_remove_source_branch? || force_remove_source_branch?
  end

571
  def related_notes
572 573
    # Fetch comments only from last 100 commits
    commits_for_notes_limit = 100
574
    commit_ids = commit_shas.take(commits_for_notes_limit)
575

576 577 578
    commit_notes = Note
      .except(:order)
      .where(project_id: [source_project_id, target_project_id])
579
      .for_commit_id(commit_ids)
580 581 582 583 584 585 586 587 588 589

    # We're using a UNION ALL here since this results in better performance
    # compared to using OR statements. We're using UNION ALL since the queries
    # used won't produce any duplicates (e.g. a note for a commit can't also be
    # a note for an MR).
    union = Gitlab::SQL::Union
      .new([notes, commit_notes], remove_duplicates: false)
      .to_sql

    Note.from("(#{union}) #{Note.table_name}")
590
  end
591

592
  alias_method :discussion_notes, :related_notes
593

594 595 596
  def mergeable_discussions_state?
    return true unless project.only_allow_merge_if_all_discussions_are_resolved?

597
    !discussions_to_be_resolved?
598 599
  end

I
Izaak Alpert 已提交
600 601 602 603
  def for_fork?
    target_project != source_project
  end

604 605 606 607
  def project
    target_project
  end

608 609 610 611
  # 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.
612
  # This optimization does not apply to issues from external sources.
613
  def cache_merge_request_closes_issues!(current_user)
614
    return unless project.issues_enabled?
615

616
    transaction do
617
      self.merge_requests_closing_issues.delete_all
618

619
      closes_issues(current_user).each do |issue|
620 621
        next if issue.is_a?(ExternalIssue)

622
        self.merge_requests_closing_issues.create!(issue: issue)
623 624 625 626
      end
    end
  end

627
  # Return the set of issues that will be closed if this merge request is accepted.
628
  def closes_issues(current_user = self.author)
629
    if target_branch == project.default_branch
630
      messages = [title, description]
631
      messages.concat(commits.map(&:safe_message)) if merge_request_diff
632

633 634
      Gitlab::ClosingIssueExtractor.new(project, current_user)
        .closed_by_message(messages.join("\n"))
635 636 637 638 639
    else
      []
    end
  end

640
  def issues_mentioned_but_not_closing(current_user)
641
    return [] unless target_branch == project.default_branch
642

643
    ext = Gitlab::ReferenceExtractor.new(project, current_user)
644
    ext.analyze("#{title}\n#{description}")
645

646
    ext.issues - closes_issues(current_user)
647 648
  end

649 650
  def target_project_path
    if target_project
651
      target_project.full_path
652 653 654 655 656 657 658
    else
      "(removed)"
    end
  end

  def source_project_path
    if source_project
659
      source_project.full_path
660 661 662 663 664
    else
      "(removed)"
    end
  end

665 666
  def source_project_namespace
    if source_project && source_project.namespace
667
      source_project.namespace.full_path
668 669 670 671 672
    else
      "(removed)"
    end
  end

673 674
  def target_project_namespace
    if target_project && target_project.namespace
675
      target_project.namespace.full_path
676 677 678 679 680
    else
      "(removed)"
    end
  end

681 682 683
  def source_branch_exists?
    return false unless self.source_project

684
    self.source_project.repository.branch_exists?(self.source_branch)
685 686 687 688 689
  end

  def target_branch_exists?
    return false unless self.target_project

690
    self.target_project.repository.branch_exists?(self.target_branch)
691 692
  end

693
  def merge_commit_message(include_description: false)
694 695 696 697
    closes_issues_references = closes_issues.map do |issue|
      issue.to_reference(target_project)
    end

698 699 700 701
    message = [
      "Merge branch '#{source_branch}' into '#{target_branch}'",
      title
    ]
702

703
    if !include_description && closes_issues_references.present?
704
      message << "Closes #{closes_issues_references.to_sentence}"
705
    end
706
    message << "#{description}" if include_description && description.present?
707
    message << "See merge request #{to_reference(full: true)}"
708

709
    message.join("\n\n")
710
  end
711

J
James Lopez 已提交
712 713
  def reset_merge_when_pipeline_succeeds
    return unless merge_when_pipeline_succeeds?
714

J
James Lopez 已提交
715
    self.merge_when_pipeline_succeeds = false
Z
Zeger-Jan van de Weg 已提交
716
    self.merge_user = nil
717 718 719 720
    if merge_params
      merge_params.delete('should_remove_source_branch')
      merge_params.delete('commit_message')
    end
Z
Zeger-Jan van de Weg 已提交
721 722 723 724

    self.save
  end

725
  # Return array of possible target branches
S
Steven Burgart 已提交
726
  # depends on target project of MR
727 728 729 730 731 732 733 734 735
  def target_branches
    if target_project.nil?
      []
    else
      target_project.repository.branch_names
    end
  end

  # Return array of possible source branches
S
Steven Burgart 已提交
736
  # depends on source project of MR
737 738 739 740 741 742 743
  def source_branches
    if source_project.nil?
      []
    else
      source_project.repository.branch_names
    end
  end
744

745
  def has_ci?
746
    return false if has_no_commits?
747

748
    !!(head_pipeline_id || all_pipelines.any? || source_project&.ci_service)
749 750 751 752 753
  end

  def branch_missing?
    !source_branch_exists? || !target_branch_exists?
  end
754

755
  def broken?
756
    has_no_commits? || branch_missing? || cannot_be_merged?
757 758
  end

759
  def can_be_merged_by?(user)
760 761 762 763 764 765 766
    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)
767 768
  end

769
  def mergeable_ci_state?
J
James Lopez 已提交
770
    return true unless project.only_allow_merge_if_pipeline_succeeds?
771

772
    !head_pipeline || head_pipeline.success? || head_pipeline.skipped?
773 774
  end

D
Douwe Maan 已提交
775
  def environments_for(current_user)
776
    return [] unless diff_head_commit
777

D
Douwe Maan 已提交
778 779 780
    @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
781

D
Douwe Maan 已提交
782 783 784 785
      if source_project
        envs.concat EnvironmentsFinder.new(source_project, current_user,
          ref: source_branch, commit: diff_head_commit).execute
      end
786

D
Douwe Maan 已提交
787
      h[current_user] = envs.uniq
788
    end
D
Douwe Maan 已提交
789 790

    @environments[current_user]
791 792
  end

793 794 795 796 797 798 799 800 801
  def state_human_name
    if merged?
      "Merged"
    elsif closed?
      "Closed"
    else
      "Open"
    end
  end
802

803 804 805 806 807 808 809 810 811 812
  def state_icon_name
    if merged?
      "check"
    elsif closed?
      "times"
    else
      "circle-o"
    end
  end

813 814
  def fetch_ref!
    target_project.repository.fetch_source_branch!(source_project.repository, source_branch, ref_path)
815 816
  end

817
  def ref_path
818
    "refs/#{Repository::REF_MERGE_REQUEST}/#{iid}/head"
819 820
  end

821 822 823 824 825
  def in_locked_state
    begin
      lock_mr
      yield
    ensure
826
      unlock_mr
827 828
    end
  end
829

830 831 832
  def diverged_commits_count
    cache = Rails.cache.read(:"merge_request_#{id}_diverged_commits")

833
    if cache.blank? || cache[:source_sha] != source_branch_sha || cache[:target_sha] != target_branch_sha
834
      cache = {
835 836
        source_sha: source_branch_sha,
        target_sha: target_branch_sha,
837 838 839 840 841 842 843 844 845
        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
846
    return 0 unless source_branch_sha && target_branch_sha
847

848
    Gitlab::Git::Commit.between(target_project.repository.raw_repository, source_branch_sha, target_branch_sha).size
849
  end
850
  private :compute_diverged_commits_count
851 852 853 854 855

  def diverged_from_target_branch?
    diverged_commits_count > 0
  end

856
  def all_pipelines
857
    return Ci::Pipeline.none unless source_project
858

859
    @all_pipelines ||= source_project.pipelines
860
      .where(sha: all_commit_shas, ref: source_branch)
861
      .order(id: :desc)
862
  end
863

864
  # Note that this could also return SHA from now dangling commits
865
  #
866
  def all_commit_shas
867
    if persisted?
868 869 870 871 872 873 874 875 876 877 878 879 880
      # MySQL doesn't support LIMIT in a subquery.
      diffs_relation =
        if Gitlab::Database.postgresql?
          merge_request_diffs.order(id: :desc).limit(100)
        else
          merge_request_diffs
        end

      column_shas = MergeRequestDiffCommit
                      .where(merge_request_diff: diffs_relation)
                      .limit(10_000)
                      .pluck('sha')

881 882 883
      serialised_shas = merge_request_diffs.where.not(st_commits: nil).flat_map(&:commit_shas)

      (column_shas + serialised_shas).uniq
884 885
    elsif compare_commits
      compare_commits.to_a.reverse.map(&:id)
886
    else
887
      [diff_head_sha]
888
    end
889 890
  end

891 892 893 894
  def merge_commit
    @merge_commit ||= project.commit(merge_commit_sha) if merge_commit_sha
  end

895
  def can_be_reverted?(current_user)
896
    merge_commit && !merge_commit.has_been_reverted?(current_user, self)
897
  end
898 899

  def can_be_cherry_picked?
F
Fatih Acet 已提交
900
    merge_commit.present?
901
  end
902

903
  def has_complete_diff_refs?
904
    diff_refs && diff_refs.complete?
905 906
  end

907
  def update_diff_discussion_positions(old_diff_refs:, new_diff_refs:, current_user: nil)
908
    return unless has_complete_diff_refs?
909 910
    return if new_diff_refs == old_diff_refs

911 912
    active_diff_discussions = self.notes.new_diff_notes.discussions.select do |discussion|
      discussion.active?(old_diff_refs)
913
    end
914
    return if active_diff_discussions.empty?
915

916
    paths = active_diff_discussions.flat_map { |n| n.diff_file.paths }.uniq
917

918
    service = Discussions::UpdateDiffPositionService.new(
919
      self.project,
920
      current_user,
921 922 923 924 925
      old_diff_refs: old_diff_refs,
      new_diff_refs: new_diff_refs,
      paths: paths
    )

926 927
    active_diff_discussions.each do |discussion|
      service.execute(discussion)
928
    end
929 930 931 932 933 934

    if project.resolve_outdated_diff_discussions?
      MergeRequests::ResolvedDiscussionNotificationService
        .new(project, current_user)
        .execute(self)
    end
935 936
  end

937 938 939
  def keep_around_commit
    project.repository.keep_around(self.merge_commit_sha)
  end
940

941
  def has_commits?
942
    merge_request_diff && commits_count > 0
943 944 945 946 947
  end

  def has_no_commits?
    !has_commits?
  end
948

949
  def mergeable_with_quick_action?(current_user, autocomplete_precheck: false, last_diff_sha: nil)
950 951 952 953 954 955 956 957 958 959
    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
960

961 962 963 964
  def update_project_counter_caches
    Projects::OpenMergeRequestsCountService.new(target_project).refresh_cache
  end

M
micael.bergeron 已提交
965
  def first_contribution?
M
micael.bergeron 已提交
966
    return false if project.team.max_member_access(author_id) > Gitlab::Access::GUEST
M
micael.bergeron 已提交
967

M
micael.bergeron 已提交
968 969
    project.merge_requests.merged.where(author_id: author_id).empty?
  end
D
Dmitriy Zaporozhets 已提交
970
end