diffs_controller.rb 2.6 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

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

    @commit ||= @project.commit(commit_id)
  end

44
  def find_merge_request_diff_compare
D
Douwe Maan 已提交
45
    @merge_request_diff =
46 47
      if diff_id = params[:diff_id].presence
        @merge_request.merge_request_diffs.viewable.find_by(id: diff_id)
D
Douwe Maan 已提交
48 49 50 51
      else
        @merge_request.merge_request_diff
      end

52 53
    return unless @merge_request_diff

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

56
    if @start_sha = params[:start_sha].presence
D
Douwe Maan 已提交
57 58 59 60 61 62 63 64
      @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

65 66 67 68 69
    if @start_sha
      @merge_request_diff.compare_with(@start_sha)
    else
      @merge_request_diff
    end
D
Douwe Maan 已提交
70 71
  end

F
Felipe Artur 已提交
72 73 74 75 76 77 78 79 80 81 82 83 84
  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 已提交
85 86 87
  def define_diff_comment_vars
    @new_diff_note_attrs = {
      noteable_type: 'MergeRequest',
88 89
      noteable_id: @merge_request.id,
      commit_id: @commit&.id
D
Douwe Maan 已提交
90 91 92 93 94 95 96
    }

    @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)
97
    @notes = prepare_notes_for_rendering(@grouped_diff_discussions.values.flatten.flat_map(&:notes), @merge_request)
D
Douwe Maan 已提交
98 99
  end
end