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

127 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.preload(draft_notes, { author: :status })
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