128 lines
2.7 KiB
Ruby
128 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
class DraftNote < ApplicationRecord
|
|
include DiffPositionableNote
|
|
include Gitlab::Utils::StrongMemoize
|
|
include Sortable
|
|
include ShaAttribute
|
|
|
|
PUBLISH_ATTRS = %i(noteable_id noteable_type type note).freeze
|
|
DIFF_ATTRS = %i(position original_position change_position commit_id).freeze
|
|
|
|
sha_attribute :commit_id
|
|
|
|
# Attribute used to store quick actions changes and users referenced.
|
|
attr_accessor :commands_changes
|
|
attr_accessor :users_referenced
|
|
|
|
# Text with quick actions filtered out
|
|
attr_accessor :rendered_note
|
|
|
|
attr_accessor :review
|
|
|
|
belongs_to :author, class_name: 'User'
|
|
belongs_to :merge_request
|
|
|
|
validates :merge_request_id, presence: true
|
|
validates :author_id, presence: true, uniqueness: { scope: [:merge_request_id, :discussion_id] }, if: :discussion_id?
|
|
validates :discussion_id, allow_nil: true, format: { with: /\A\h{40}\z/ }
|
|
validates :line_code, length: { maximum: 255 }, allow_nil: true
|
|
|
|
scope :authored_by, ->(u) { where(author_id: u.id) }
|
|
|
|
delegate :file_path, :file_hash, :file_identifier_hash, to: :diff_file, allow_nil: true
|
|
|
|
def self.positions
|
|
where.not(position: nil)
|
|
.select(:position)
|
|
.map(&:position)
|
|
end
|
|
|
|
def project
|
|
merge_request.target_project
|
|
end
|
|
|
|
# noteable_id and noteable_type methods
|
|
# are used to generate discussion_id on Discussion.discussion_id
|
|
def noteable_id
|
|
merge_request_id
|
|
end
|
|
|
|
def noteable
|
|
merge_request
|
|
end
|
|
|
|
def noteable_type
|
|
"MergeRequest"
|
|
end
|
|
|
|
def for_commit?
|
|
commit_id.present?
|
|
end
|
|
|
|
def importing?
|
|
false
|
|
end
|
|
|
|
def resolvable?
|
|
false
|
|
end
|
|
|
|
def emoji_awardable?
|
|
false
|
|
end
|
|
|
|
def on_diff?
|
|
position&.complete?
|
|
end
|
|
|
|
def type
|
|
return 'DiffNote' if on_diff?
|
|
return 'DiscussionNote' if discussion_id.present?
|
|
|
|
'Note'
|
|
end
|
|
|
|
def references
|
|
{
|
|
users: users_referenced,
|
|
commands: commands_changes
|
|
}
|
|
end
|
|
|
|
def line_code
|
|
super.presence || find_line_code
|
|
end
|
|
|
|
def find_line_code
|
|
write_attribute(:line_code, diff_file&.line_code_for_position(original_position))
|
|
end
|
|
|
|
def publish_params
|
|
attrs = PUBLISH_ATTRS.dup
|
|
attrs.concat(DIFF_ATTRS) if on_diff?
|
|
params = slice(*attrs)
|
|
params[:in_reply_to_discussion_id] = discussion_id if discussion_id.present?
|
|
params[:review_id] = review.id if review.present?
|
|
|
|
params
|
|
end
|
|
|
|
def self.preload_author(draft_notes)
|
|
ActiveRecord::Associations::Preloader.new(records: draft_notes, associations: { author: :status }).call
|
|
end
|
|
|
|
def diff_file
|
|
strong_memoize(:diff_file) do
|
|
file = original_position&.diff_file(project.repository)
|
|
|
|
file&.unfold_diff_lines(original_position)
|
|
|
|
file
|
|
end
|
|
end
|
|
|
|
def commit
|
|
@commit ||= project.commit(commit_id) if commit_id.present?
|
|
end
|
|
end
|