128 lines
3.1 KiB
Ruby
128 lines
3.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module DiffPositionableNote
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
before_validation :set_original_position, on: :create
|
|
before_validation :update_position, on: :create, if: :should_update_position?, unless: :importing?
|
|
|
|
serialize :original_position, type: Gitlab::Diff::Position # rubocop:disable Cop/ActiveRecordSerialize
|
|
serialize :position, type: Gitlab::Diff::Position # rubocop:disable Cop/ActiveRecordSerialize
|
|
serialize :change_position, type: Gitlab::Diff::Position # rubocop:disable Cop/ActiveRecordSerialize
|
|
|
|
validate :diff_refs_match_commit, if: :for_commit?
|
|
validates :position, json_schema: { filename: "position", hash_conversion: true }
|
|
end
|
|
|
|
%i[original_position position change_position].each do |meth|
|
|
define_method "#{meth}=" do |new_position|
|
|
if new_position.is_a?(String)
|
|
new_position = begin
|
|
Gitlab::Json.parse(new_position)
|
|
rescue StandardError
|
|
nil
|
|
end
|
|
end
|
|
|
|
if new_position.is_a?(Hash)
|
|
new_position = new_position.with_indifferent_access
|
|
new_position = Gitlab::Diff::Position.new(new_position)
|
|
elsif !new_position.is_a?(Gitlab::Diff::Position)
|
|
new_position = nil
|
|
end
|
|
|
|
return if new_position == read_attribute(meth)
|
|
|
|
super(new_position)
|
|
end
|
|
end
|
|
|
|
def should_update_position?
|
|
on_text? || on_file?
|
|
end
|
|
|
|
def on_text?
|
|
!!position&.on_text?
|
|
end
|
|
|
|
def on_file?
|
|
!!position&.on_file?
|
|
end
|
|
|
|
def on_image?
|
|
!!position&.on_image?
|
|
end
|
|
|
|
def supported?
|
|
for_commit? || self.noteable.has_complete_diff_refs?
|
|
end
|
|
|
|
def active?(diff_refs = nil)
|
|
return false unless supported?
|
|
return true if for_commit?
|
|
|
|
diff_refs ||= noteable.diff_refs
|
|
|
|
self.position.diff_refs == diff_refs
|
|
end
|
|
|
|
def set_original_position
|
|
return unless position
|
|
|
|
self.original_position = self.position.dup unless self.original_position&.complete?
|
|
end
|
|
|
|
def update_position
|
|
return unless supported?
|
|
return if for_commit?
|
|
|
|
return if active?
|
|
return unless position
|
|
|
|
tracer = Gitlab::Diff::PositionTracer.new(
|
|
project: self.project,
|
|
old_diff_refs: self.position.diff_refs,
|
|
new_diff_refs: self.noteable.diff_refs,
|
|
paths: self.position.paths
|
|
)
|
|
|
|
result = tracer.trace(self.position)
|
|
return unless result
|
|
|
|
if result[:outdated]
|
|
self.change_position = result[:position]
|
|
else
|
|
self.position = result[:position]
|
|
end
|
|
end
|
|
|
|
def diff_refs_match_commit
|
|
return if self.original_position.diff_refs == commit&.diff_refs
|
|
|
|
errors.add(:commit_id, 'does not match the diff refs')
|
|
end
|
|
|
|
def keep_around_commits
|
|
repository.keep_around(*shas, source: "#{noteable_type}/#{self.class.name}")
|
|
end
|
|
|
|
def repository
|
|
noteable.respond_to?(:repository) ? noteable.repository : project.repository
|
|
end
|
|
|
|
def shas
|
|
[
|
|
original_position.base_sha,
|
|
original_position.start_sha,
|
|
original_position.head_sha
|
|
].tap do |a|
|
|
if position != original_position
|
|
a << position.base_sha
|
|
a << position.start_sha
|
|
a << position.head_sha
|
|
end
|
|
end
|
|
end
|
|
end
|