diffs_controller.rb 2.8 KB
Newer Older
D
Douwe Maan 已提交
1 2 3 4 5 6
class Projects::MergeRequests::DiffsController < Projects::MergeRequests::ApplicationController
  include DiffForPath
  include DiffHelper
  include RendersNotes

  before_action :apply_diff_view_cookie!
M
micael.bergeron 已提交
7
  before_action :commit
D
Douwe Maan 已提交
8 9 10 11
  before_action :define_diff_vars
  before_action :define_diff_comment_vars

  def show
F
Felipe Artur 已提交
12
    render_diffs
D
Douwe Maan 已提交
13 14 15
  end

  def diff_for_path
F
Felipe Artur 已提交
16
    render_diffs
D
Douwe Maan 已提交
17 18 19 20
  end

  private

F
Felipe Artur 已提交
21 22 23
  def render_diffs
    @environment = @merge_request.environments_for(current_user).last

24 25
    @diffs.write_cache

F
Felipe Artur 已提交
26 27 28
    render json: DiffsSerializer.new(current_user: current_user).represent(@diffs, additional_attributes)
  end

D
Douwe Maan 已提交
29
  def define_diff_vars
30
    @merge_request_diffs = @merge_request.merge_request_diffs.viewable.order_id_desc
31
    @compare = commit || find_merge_request_diff_compare
32 33 34 35 36
    return render_404 unless @compare

    @diffs = @compare.diffs(diff_options)
  end

37
  # rubocop: disable CodeReuse/ActiveRecord
M
micael.bergeron 已提交
38 39
  def commit
    return nil unless commit_id = params[:commit_id].presence
40
    return nil unless @merge_request.all_commits.exists?(sha: commit_id)
M
micael.bergeron 已提交
41 42 43

    @commit ||= @project.commit(commit_id)
  end
44
  # rubocop: enable CodeReuse/ActiveRecord
M
micael.bergeron 已提交
45

46
  # rubocop: disable CodeReuse/ActiveRecord
47
  def find_merge_request_diff_compare
D
Douwe Maan 已提交
48
    @merge_request_diff =
49 50
      if diff_id = params[:diff_id].presence
        @merge_request.merge_request_diffs.viewable.find_by(id: diff_id)
D
Douwe Maan 已提交
51 52 53 54
      else
        @merge_request.merge_request_diff
      end

55 56
    return unless @merge_request_diff

D
Douwe Maan 已提交
57 58
    @comparable_diffs = @merge_request_diffs.select { |diff| diff.id < @merge_request_diff.id }

59
    if @start_sha = params[:start_sha].presence
D
Douwe Maan 已提交
60 61 62 63 64 65 66 67
      @start_version = @comparable_diffs.find { |diff| diff.head_commit_sha == @start_sha }

      unless @start_version
        @start_sha = @merge_request_diff.head_commit_sha
        @start_version = @merge_request_diff
      end
    end

68 69 70 71 72
    if @start_sha
      @merge_request_diff.compare_with(@start_sha)
    else
      @merge_request_diff
    end
D
Douwe Maan 已提交
73
  end
74
  # rubocop: enable CodeReuse/ActiveRecord
D
Douwe Maan 已提交
75

F
Felipe Artur 已提交
76 77 78 79 80 81 82 83 84 85 86 87 88
  def additional_attributes
    {
      environment: @environment,
      merge_request: @merge_request,
      merge_request_diff: @merge_request_diff,
      merge_request_diffs: @merge_request_diffs,
      start_version: @start_version,
      start_sha: @start_sha,
      commit: @commit,
      latest_diff: @merge_request_diff&.latest?
    }
  end

D
Douwe Maan 已提交
89 90 91
  def define_diff_comment_vars
    @new_diff_note_attrs = {
      noteable_type: 'MergeRequest',
92 93
      noteable_id: @merge_request.id,
      commit_id: @commit&.id
D
Douwe Maan 已提交
94 95 96 97 98 99 100
    }

    @diff_notes_disabled = false

    @use_legacy_diff_notes = !@merge_request.has_complete_diff_refs?

    @grouped_diff_discussions = @merge_request.grouped_diff_discussions(@compare.diff_refs)
101
    @notes = prepare_notes_for_rendering(@grouped_diff_discussions.values.flatten.flat_map(&:notes), @merge_request)
D
Douwe Maan 已提交
102 103
  end
end