merge_request.rb 25.6 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 Sortable
7
  include IgnorableColumn
J
James Lopez 已提交
8
  include CreatedAtFilterable
9
  include TimeTrackable
10

11 12
  ignore_column :locked_at,
                :ref_fetched
13

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

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

22 23
  belongs_to :head_pipeline, foreign_key: "head_pipeline_id", class_name: "Ci::Pipeline"

24
  has_many :events, as: :target, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
25

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

30 31
  belongs_to :assignee, class_name: "User"

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

34 35
  after_create :ensure_merge_request_diff, unless: :importing?
  after_update :reload_diff_if_branch_changed
36
  after_commit :update_project_counter_caches, on: :destroy
37

D
Dmitriy Zaporozhets 已提交
38 39 40 41
  # 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 已提交
42 43
  # Temporary fields to store compare vars
  # when creating new merge request
44
  attr_accessor :can_be_created, :compare_commits, :diff_options, :compare
D
Dmitriy Zaporozhets 已提交
45

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

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

    event :reopen do
56
      transition closed: :opened
A
Andrew8xx8 已提交
57 58
    end

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

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

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

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

    event :mark_as_unmergeable do
83
      transition [:unchecked, :can_be_merged] => :cannot_be_merged
84 85
    end

86
    state :unchecked
87 88
    state :can_be_merged
    state :cannot_be_merged
89 90

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

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

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

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

122 123
  after_save :keep_around_commit

124 125
  acts_as_paranoid

126 127 128 129
  def self.reference_prefix
    '!'
  end

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

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

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

148 149 150 151
  def self.project_foreign_key
    'target_project_id'
  end

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

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

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

186
  def hook_attrs
187
    Gitlab::HookData::MergeRequestBuilder.new(self).build
188 189
  end

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

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

203 204 205 206 207 208 209 210
  def assignee_ids
    Array(assignee_id)
  end

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

211 212 213 214
  def assignee_or_author?(user)
    author_id == user.id || assignee_id == user.id
  end

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

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

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 250 251
  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

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

260 261
  def first_commit
    merge_request_diff ? merge_request_diff.first_commit : compare_commits.first
262
  end
263

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

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

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

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

  def diff_start_commit
    if persisted?
      merge_request_diff.start_commit
    else
      target_branch_head
298 299 300
    end
  end

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 326 327
  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
328 329
    return unless source_project

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

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

339 340 341 342 343 344 345 346 347
  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

348
  def target_branch_sha
349
    @target_branch_sha || target_branch_head.try(:sha)
350 351 352
  end

  def source_branch_sha
353
    @source_branch_sha || source_branch_head.try(:sha)
354 355
  end

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

368 369 370 371
  def branch_merge_base_sha
    branch_merge_base_commit.try(:sha)
  end

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

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

387 388 389 390 391 392
  def validate_target_project
    return true if target_project.merge_requests_enabled?

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

393
  def validate_fork
394
    return true unless target_project && source_project
395
    return true if target_project == source_project
396
    return true unless source_project_missing?
397

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

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

    !!merge_jid && !merged? && Gitlab::SidekiqStatus.running?(merge_jid)
408 409
  end

410
  def closed_without_fork?
411
    closed? && source_project_missing?
412 413
  end

414
  def source_project_missing?
415 416 417
    return false unless for_fork?
    return true unless source_project

418
    !source_project.in_fork_network_of?(target_project)
419 420
  end

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

425 426
  def ensure_merge_request_diff
    merge_request_diff || create_merge_request_diff
427 428
  end

429
  def create_merge_request_diff
430
    fetch_ref!
431

432 433 434 435 436
    # 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
437 438 439 440 441 442
  end

  def reload_merge_request_diff
    merge_request_diff(true)
  end

443 444 445 446 447 448 449 450 451
  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 已提交
452 453
    end

454
    @merge_request_diffs_by_diff_refs_or_sha[diff_refs_or_sha]
455 456
  end

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

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

475
  def reload_diff(current_user = nil)
476 477
    return unless open?

478
    old_diff_refs = self.diff_refs
479

480
    create_merge_request_diff
481
    MergeRequests::MergeRequestDiffCacheService.new.execute(self)
482 483
    new_diff_refs = self.diff_refs

484
    update_diff_discussion_positions(
485
      old_diff_refs: old_diff_refs,
486 487
      new_diff_refs: new_diff_refs,
      current_user: current_user
488
    )
489 490
  end

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

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

    if can_be_merged
498 499 500 501
      mark_as_mergeable
    else
      mark_as_unmergeable
    end
502 503
  end

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

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

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

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

  def wip_title
    self.class.wip_title(self.title)
522 523
  end

524 525
  def mergeable?(skip_ci_check: false)
    return false unless mergeable_state?(skip_ci_check: skip_ci_check)
526 527 528 529

    check_if_can_be_merged

    can_be_merged?
530 531
  end

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

    true
540 541
  end

542 543 544 545 546 547 548 549
  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 已提交
550
  def can_cancel_merge_when_pipeline_succeeds?(current_user)
551
    can_be_merged_by?(current_user) || self.author == current_user
552 553
  end

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

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

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

  def remove_source_branch?
    should_remove_source_branch? || force_remove_source_branch?
  end

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

578 579 580 581 582 583 584 585 586 587 588 589 590 591
    commit_notes = Note
      .except(:order)
      .where(project_id: [source_project_id, target_project_id])
      .where(noteable_type: 'Commit', commit_id: commit_ids)

    # 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}")
592
  end
593

594
  alias_method :discussion_notes, :related_notes
595

596 597 598
  def mergeable_discussions_state?
    return true unless project.only_allow_merge_if_all_discussions_are_resolved?

599
    !discussions_to_be_resolved?
600 601
  end

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

606 607 608 609
  def project
    target_project
  end

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

618
    transaction do
619
      self.merge_requests_closing_issues.delete_all
620

621
      closes_issues(current_user).each do |issue|
622 623
        next if issue.is_a?(ExternalIssue)

624
        self.merge_requests_closing_issues.create!(issue: issue)
625 626 627 628
      end
    end
  end

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

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

642
  def issues_mentioned_but_not_closing(current_user)
643
    return [] unless target_branch == project.default_branch
644

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

648
    ext.issues - closes_issues(current_user)
649 650
  end

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

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

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

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

683 684 685
  def source_branch_exists?
    return false unless self.source_project

686
    self.source_project.repository.branch_exists?(self.source_branch)
687 688 689 690 691
  end

  def target_branch_exists?
    return false unless self.target_project

692
    self.target_project.repository.branch_exists?(self.target_branch)
693 694
  end

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

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

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

711
    message.join("\n\n")
712
  end
713

J
James Lopez 已提交
714 715
  def reset_merge_when_pipeline_succeeds
    return unless merge_when_pipeline_succeeds?
716

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

    self.save
  end

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

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

747
  def has_ci?
748
    return false if has_no_commits?
749

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

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

757
  def broken?
758
    has_no_commits? || branch_missing? || cannot_be_merged?
759 760
  end

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

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

774
    !head_pipeline || head_pipeline.success? || head_pipeline.skipped?
775 776
  end

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

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

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

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

    @environments[current_user]
793 794
  end

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

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

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

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

823 824 825 826 827
  def in_locked_state
    begin
      lock_mr
      yield
    ensure
828
      unlock_mr
829 830
    end
  end
831

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

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

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

  def diverged_from_target_branch?
    diverged_commits_count > 0
  end

858
  def all_pipelines
859
    return Ci::Pipeline.none unless source_project
860

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

866
  # Note that this could also return SHA from now dangling commits
867
  #
868
  def all_commit_shas
869
    if persisted?
870
      column_shas = MergeRequestDiffCommit.where(merge_request_diff: merge_request_diffs).limit(10_000).pluck('sha')
871 872 873
      serialised_shas = merge_request_diffs.where.not(st_commits: nil).flat_map(&:commit_shas)

      (column_shas + serialised_shas).uniq
874 875
    elsif compare_commits
      compare_commits.to_a.reverse.map(&:id)
876
    else
877
      [diff_head_sha]
878
    end
879 880
  end

881 882 883 884
  def merge_commit
    @merge_commit ||= project.commit(merge_commit_sha) if merge_commit_sha
  end

885
  def can_be_reverted?(current_user)
886
    merge_commit && !merge_commit.has_been_reverted?(current_user, self)
887
  end
888 889

  def can_be_cherry_picked?
F
Fatih Acet 已提交
890
    merge_commit.present?
891
  end
892

893
  def has_complete_diff_refs?
894
    diff_refs && diff_refs.complete?
895 896
  end

897
  def update_diff_discussion_positions(old_diff_refs:, new_diff_refs:, current_user: nil)
898
    return unless has_complete_diff_refs?
899 900
    return if new_diff_refs == old_diff_refs

901 902
    active_diff_discussions = self.notes.new_diff_notes.discussions.select do |discussion|
      discussion.active?(old_diff_refs)
903
    end
904
    return if active_diff_discussions.empty?
905

906
    paths = active_diff_discussions.flat_map { |n| n.diff_file.paths }.uniq
907

908
    service = Discussions::UpdateDiffPositionService.new(
909
      self.project,
910
      current_user,
911 912 913 914 915
      old_diff_refs: old_diff_refs,
      new_diff_refs: new_diff_refs,
      paths: paths
    )

916 917
    active_diff_discussions.each do |discussion|
      service.execute(discussion)
918
    end
919 920 921 922 923 924

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

927 928 929
  def keep_around_commit
    project.repository.keep_around(self.merge_commit_sha)
  end
930

931
  def has_commits?
932
    merge_request_diff && commits_count > 0
933 934 935 936 937
  end

  def has_no_commits?
    !has_commits?
  end
938

939
  def mergeable_with_quick_action?(current_user, autocomplete_precheck: false, last_diff_sha: nil)
940 941 942 943 944 945 946 947 948 949
    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
950

951 952 953 954
  def update_project_counter_caches?
    state_changed?
  end

955 956 957 958
  def update_project_counter_caches
    Projects::OpenMergeRequestsCountService.new(target_project).refresh_cache
  end

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

M
micael.bergeron 已提交
962 963
    project.merge_requests.merged.where(author_id: author_id).empty?
  end
D
Dmitriy Zaporozhets 已提交
964
end