gitlab-org--gitlab-foss/app/models/diff_note.rb

156 lines
4.2 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
2017-03-30 23:06:09 -04:00
# A note on merge request or commit diffs
#
# A note of this type can be resolvable.
2016-06-20 13:20:39 -04:00
class DiffNote < Note
include NoteOnDiff
include DiffPositionableNote
include Gitlab::Utils::StrongMemoize
2016-06-20 13:20:39 -04:00
def self.noteable_types
%w(MergeRequest Commit)
end
2017-03-17 15:25:52 -04:00
2016-06-20 13:20:39 -04:00
validates :original_position, presence: true
validates :position, presence: true
2017-10-07 00:25:17 -04:00
validates :line_code, presence: true, line_code: true, if: :on_text?
validates :noteable_type, inclusion: { in: noteable_types }
2016-06-20 13:20:39 -04:00
validate :positions_complete
validate :verify_supported
validate :diff_refs_match_commit, if: :for_commit?
2016-06-20 13:20:39 -04:00
before_validation :set_line_code, if: :on_text?
2016-07-04 12:26:28 -04:00
after_save :keep_around_commits
after_commit :create_diff_file, on: :create
2016-06-20 13:20:39 -04:00
def discussion_class(*)
DiffDiscussion
end
def create_diff_file
return unless should_create_diff_file?
diff_file = fetch_diff_file
diff_line = diff_file.line_for_position(self.original_position)
creation_params = diff_file.diff.to_hash
.except(:too_large)
.merge(diff: diff_file.diff_hunk(diff_line))
create_note_diff_file(creation_params)
end
2016-06-20 13:20:39 -04:00
def diff_file
strong_memoize(:diff_file) do
enqueue_diff_file_creation_job if should_create_diff_file?
fetch_diff_file
end
2016-06-20 13:20:39 -04:00
end
def diff_line
@diff_line ||= diff_file&.line_for_position(self.original_position)
2016-06-20 13:20:39 -04:00
end
def original_line_code
2017-10-07 00:25:17 -04:00
return unless on_text?
self.diff_file.line_code(self.diff_line)
end
def created_at_diff?(diff_refs)
return false unless supported?
return true if for_commit?
self.original_position.diff_refs == diff_refs
end
def discussion_first_note?
self == discussion.first_note
end
private
def enqueue_diff_file_creation_job
# Avoid enqueuing multiple file creation jobs at once for a note (i.e.
# parallel calls to `DiffNote#diff_file`).
lease = Gitlab::ExclusiveLease.new("note_diff_file_creation:#{id}", timeout: 1.hour.to_i)
return unless lease.try_obtain
CreateNoteDiffFileWorker.perform_async(id)
end
def should_create_diff_file?
on_text? && note_diff_file.nil? && discussion_first_note?
end
def fetch_diff_file
file =
if note_diff_file
diff = Gitlab::Git::Diff.new(note_diff_file.to_hash)
Gitlab::Diff::File.new(diff,
repository: project.repository,
diff_refs: original_position.diff_refs)
elsif created_at_diff?(noteable.diff_refs)
# We're able to use the already persisted diffs (Postgres) if we're
# presenting a "current version" of the MR discussion diff.
# So no need to make an extra Gitaly diff request for it.
# As an extra benefit, the returned `diff_file` already
# has `highlighted_diff_lines` data set from Redis on
# `Diff::FileCollection::MergeRequestDiff`.
noteable.diffs(original_position.diff_options).diff_files.first
else
original_position.diff_file(self.project.repository)
end
# Since persisted diff files already have its content "unfolded"
# there's no need to make it pass through the unfolding process.
file&.unfold_diff_lines(position) unless note_diff_file
file
end
def supported?
2016-08-17 14:29:05 -04:00
for_commit? || self.noteable.has_complete_diff_refs?
end
def set_line_code
self.line_code = self.position.line_code(self.project.repository)
end
2016-06-20 13:20:39 -04:00
def verify_supported
return if supported?
errors.add(:noteable, "doesn't support new-style diff notes")
end
def positions_complete
return if self.original_position.complete? && self.position.complete?
errors.add(:position, "is invalid")
end
2016-07-04 12:26:28 -04:00
def diff_refs_match_commit
return if self.original_position.diff_refs == self.commit.diff_refs
errors.add(:commit_id, 'does not match the diff refs')
end
2016-07-04 12:26:28 -04:00
def keep_around_commits
shas = [
self.original_position.base_sha,
self.original_position.start_sha,
self.original_position.head_sha
]
2016-07-04 12:26:28 -04:00
if self.position != self.original_position
shas << self.position.base_sha
shas << self.position.start_sha
shas << self.position.head_sha
2016-07-04 12:26:28 -04:00
end
project.repository.keep_around(*shas)
2016-07-04 12:26:28 -04:00
end
2016-06-20 13:20:39 -04:00
end