debian-mirror-gitlab/app/models/note.rb

467 lines
13 KiB
Ruby
Raw Normal View History

2017-08-17 22:00:37 +05:30
# A note on the root of an issue, merge request, commit, or snippet.
#
# A note of this type is never resolvable.
2014-09-02 18:07:02 +05:30
class Note < ActiveRecord::Base
2016-06-02 11:05:42 +05:30
extend ActiveModel::Naming
2015-09-11 14:41:01 +05:30
include Participable
2015-10-24 18:46:33 +05:30
include Mentionable
include Awardable
2016-06-22 15:30:34 +05:30
include Importable
2016-09-13 17:45:13 +05:30
include FasterCacheKeys
2016-11-03 12:29:30 +05:30
include CacheMarkdownField
2017-08-17 22:00:37 +05:30
include AfterCommitQueue
include ResolvableNote
include IgnorableColumn
include Editable
2018-03-17 18:26:18 +05:30
include Gitlab::SQL::Pattern
include ThrottledTouch
module SpecialRole
FIRST_TIME_CONTRIBUTOR = :first_time_contributor
class << self
def values
constants.map {|const| self.const_get(const)}
end
end
end
2016-11-03 12:29:30 +05:30
2017-08-17 22:00:37 +05:30
ignore_column :original_discussion_id
cache_markdown_field :note, pipeline: :note, issuable_state_filter_enabled: true
# Aliases to make application_helper#edited_time_ago_with_tooltip helper work properly with notes.
# See https://gitlab.com/gitlab-org/gitlab-ce/merge_requests/10392/diffs#note_28719102
alias_attribute :last_edited_at, :updated_at
alias_attribute :last_edited_by, :updated_by
2014-09-02 18:07:02 +05:30
2016-08-24 12:49:21 +05:30
# Attribute containing rendered and redacted Markdown as generated by
# Banzai::ObjectRenderer.
2016-11-03 12:29:30 +05:30
attr_accessor :redacted_note_html
2016-08-24 12:49:21 +05:30
# An Array containing the number of visible references as generated by
# Banzai::ObjectRenderer
attr_accessor :user_visible_reference_count
2018-03-17 18:26:18 +05:30
# Attribute used to store the attributes that have been changed by quick actions.
2017-08-17 22:00:37 +05:30
attr_accessor :commands_changes
2018-03-17 18:26:18 +05:30
# A special role that may be displayed on issuable's discussions
attr_accessor :special_role
2014-09-02 18:07:02 +05:30
default_value_for :system, false
attr_mentionable :note, pipeline: :note
2015-10-24 18:46:33 +05:30
participant :author
2014-09-02 18:07:02 +05:30
belongs_to :project
2018-03-17 18:26:18 +05:30
belongs_to :noteable, polymorphic: true # rubocop:disable Cop/PolymorphicAssociations
2014-09-02 18:07:02 +05:30
belongs_to :author, class_name: "User"
2015-09-11 14:41:01 +05:30
belongs_to :updated_by, class_name: "User"
2017-08-17 22:00:37 +05:30
belongs_to :last_edited_by, class_name: 'User'
2016-09-13 17:45:13 +05:30
2018-03-17 18:26:18 +05:30
has_many :todos
2017-09-10 17:25:29 +05:30
has_many :events, as: :target, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
2017-08-17 22:00:37 +05:30
has_one :system_note_metadata
2016-04-02 18:10:28 +05:30
2016-06-02 11:05:42 +05:30
delegate :gfm_reference, :local_reference, to: :noteable
2014-09-02 18:07:02 +05:30
delegate :name, to: :project, prefix: true
delegate :name, :email, to: :author, prefix: true
2016-06-02 11:05:42 +05:30
delegate :title, to: :noteable, allow_nil: true
2014-09-02 18:07:02 +05:30
2017-08-17 22:00:37 +05:30
validates :note, presence: true
2018-03-17 18:26:18 +05:30
validates :project, presence: true, if: :for_project_noteable?
2015-04-26 12:48:37 +05:30
# Attachments are deprecated and are handled by Markdown uploader
validates :attachment, file_size: { maximum: :max_attachment_size }
2014-09-02 18:07:02 +05:30
validates :noteable_type, presence: true
2016-06-22 15:30:34 +05:30
validates :noteable_id, presence: true, unless: [:for_commit?, :importing?]
validates :commit_id, presence: true, if: :for_commit?
2015-11-26 14:37:03 +05:30
validates :author, presence: true
2017-08-17 22:00:37 +05:30
validates :discussion_id, presence: true, format: { with: /\A\h{40}\z/ }
2014-09-02 18:07:02 +05:30
2018-03-27 19:54:05 +05:30
validate unless: [:for_commit?, :importing?, :skip_project_check?] do |note|
unless note.noteable.try(:project) == note.project
2017-08-17 22:00:37 +05:30
errors.add(:project, 'does not match noteable project')
end
end
2018-03-17 18:26:18 +05:30
# @deprecated attachments are handler by the MarkdownUploader
2014-09-02 18:07:02 +05:30
mount_uploader :attachment, AttachmentUploader
# Scopes
scope :for_commit_id, ->(commit_id) { where(noteable_type: "Commit", commit_id: commit_id) }
2018-03-17 18:26:18 +05:30
scope :system, -> { where(system: true) }
scope :user, -> { where(system: false) }
scope :common, -> { where(noteable_type: ["", nil]) }
scope :fresh, -> { order(created_at: :asc, id: :asc) }
scope :updated_after, ->(time) { where('updated_at > ?', time) }
scope :inc_author_project, -> { includes(:project, :author) }
scope :inc_author, -> { includes(:author) }
2017-08-17 22:00:37 +05:30
scope :inc_relations_for_view, -> do
includes(:project, :author, :updated_by, :resolved_by, :award_emoji, :system_note_metadata)
end
2014-09-02 18:07:02 +05:30
2018-03-17 18:26:18 +05:30
scope :diff_notes, -> { where(type: %w(LegacyDiffNote DiffNote)) }
scope :new_diff_notes, -> { where(type: 'DiffNote') }
scope :non_diff_notes, -> { where(type: ['Note', 'DiscussionNote', nil]) }
2016-06-02 11:05:42 +05:30
2015-10-24 18:46:33 +05:30
scope :with_associations, -> do
2016-08-24 12:49:21 +05:30
# FYI noteable cannot be loaded for LegacyDiffNote for commits
2015-10-24 18:46:33 +05:30
includes(:author, :noteable, :updated_by,
project: [:project_members, { group: [:group_members] }])
end
2018-03-17 18:26:18 +05:30
scope :with_metadata, -> { includes(:system_note_metadata) }
2015-10-24 18:46:33 +05:30
2016-09-13 17:45:13 +05:30
after_initialize :ensure_discussion_id
2016-08-24 12:49:21 +05:30
before_validation :nullify_blank_type, :nullify_blank_line_code
2017-08-17 22:00:37 +05:30
before_validation :set_discussion_id, on: :create
2018-03-17 18:26:18 +05:30
after_save :keep_around_commit, if: :for_project_noteable?
2017-08-17 22:00:37 +05:30
after_save :expire_etag_cache
2018-03-17 18:26:18 +05:30
after_save :touch_noteable
2017-08-17 22:00:37 +05:30
after_destroy :expire_etag_cache
2014-09-02 18:07:02 +05:30
class << self
2016-06-02 11:05:42 +05:30
def model_name
ActiveModel::Name.new(self, nil, 'note')
end
2017-08-17 22:00:37 +05:30
def discussions(context_noteable = nil)
2017-09-10 17:25:29 +05:30
Discussion.build_collection(all.includes(:noteable).fresh, context_noteable)
2016-06-02 11:05:42 +05:30
end
2014-09-02 18:07:02 +05:30
2017-08-17 22:00:37 +05:30
def find_discussion(discussion_id)
notes = where(discussion_id: discussion_id).fresh.to_a
2018-03-27 19:54:05 +05:30
2017-08-17 22:00:37 +05:30
return if notes.empty?
Discussion.build(notes)
2016-09-13 17:45:13 +05:30
end
2018-03-17 18:26:18 +05:30
# Group diff discussions by line code or file path.
# It is not needed to group by line code when comment is
# on an image.
2017-08-17 22:00:37 +05:30
def grouped_diff_discussions(diff_refs = nil)
groups = {}
diff_notes.fresh.discussions.each do |discussion|
2018-03-17 18:26:18 +05:30
group_key =
if discussion.on_image?
discussion.file_new_path
else
discussion.line_code_in_diffs(diff_refs)
end
if group_key
discussions = groups[group_key] ||= []
2017-09-10 17:25:29 +05:30
discussions << discussion
end
2017-08-17 22:00:37 +05:30
end
groups
2014-09-02 18:07:02 +05:30
end
2017-08-17 22:00:37 +05:30
def count_for_collection(ids, type)
2017-09-10 17:25:29 +05:30
user.select('noteable_id', 'COUNT(*) as count')
.group(:noteable_id)
.where(noteable_type: type, noteable_id: ids)
2014-09-02 18:07:02 +05:30
end
2018-03-17 18:26:18 +05:30
def has_special_role?(role, note)
note.special_role == role
end
def search(query)
fuzzy_search(query, [:note])
end
2015-09-11 14:41:01 +05:30
end
2015-04-26 12:48:37 +05:30
2015-09-11 14:41:01 +05:30
def cross_reference?
2018-03-17 18:26:18 +05:30
return unless system?
if force_cross_reference_regex_check?
matches_cross_reference_regex?
else
SystemNoteService.cross_reference?(note)
end
2015-04-26 12:48:37 +05:30
end
2016-06-02 11:05:42 +05:30
def diff_note?
2014-09-02 18:07:02 +05:30
false
end
2016-06-02 11:05:42 +05:30
def active?
true
2015-04-26 12:48:37 +05:30
end
2016-06-02 11:05:42 +05:30
def max_attachment_size
2018-03-17 18:26:18 +05:30
Gitlab::CurrentSettings.max_attachment_size.megabytes.to_i
2015-04-26 12:48:37 +05:30
end
2016-06-02 11:05:42 +05:30
def hook_attrs
attributes
2014-09-02 18:07:02 +05:30
end
def for_commit?
noteable_type == "Commit"
end
def for_issue?
noteable_type == "Issue"
end
def for_merge_request?
noteable_type == "MergeRequest"
end
2016-06-02 11:05:42 +05:30
def for_snippet?
2015-04-26 12:48:37 +05:30
noteable_type == "Snippet"
end
2017-08-17 22:00:37 +05:30
def for_personal_snippet?
noteable.is_a?(PersonalSnippet)
end
2018-03-17 18:26:18 +05:30
def for_project_noteable?
!for_personal_snippet?
end
2017-08-17 22:00:37 +05:30
def skip_project_check?
2018-03-27 19:54:05 +05:30
!for_project_noteable?
2017-08-17 22:00:37 +05:30
end
2018-03-17 18:26:18 +05:30
def commit
@commit ||= project.commit(commit_id) if commit_id.present?
end
2014-09-02 18:07:02 +05:30
# override to return commits, which are not active record
def noteable
2018-03-17 18:26:18 +05:30
return commit if for_commit?
super
2014-09-02 18:07:02 +05:30
rescue
2018-03-17 18:26:18 +05:30
# Temp fix to prevent app crash
# if note commit id doesn't exist
2014-09-02 18:07:02 +05:30
nil
end
# FIXME: Hack for polymorphic associations with STI
2015-04-26 12:48:37 +05:30
# For more information visit http://api.rubyonrails.org/classes/ActiveRecord/Associations/ClassMethods.html#label-Polymorphic+Associations
2015-10-24 18:46:33 +05:30
def noteable_type=(noteable_type)
super(noteable_type.to_s.classify.constantize.base_class.to_s)
2014-09-02 18:07:02 +05:30
end
2018-03-17 18:26:18 +05:30
def special_role=(role)
raise "Role is undefined, #{role} not found in #{SpecialRole.values}" unless SpecialRole.values.include?(role)
@special_role = role
end
def has_special_role?(role)
self.class.has_special_role?(role, self)
end
def specialize_for_first_contribution!(noteable)
return unless noteable.author_id == self.author_id
self.special_role = Note::SpecialRole::FIRST_TIME_CONTRIBUTOR
end
2018-04-05 14:03:07 +05:30
def confidential?
noteable.try(:confidential?)
end
2015-04-26 12:48:37 +05:30
def editable?
!system?
2015-12-23 02:04:40 +05:30
end
def cross_reference_not_visible_for?(user)
2016-08-24 12:49:21 +05:30
cross_reference? && !has_referenced_mentionables?(user)
end
def has_referenced_mentionables?(user)
if user_visible_reference_count.present?
user_visible_reference_count > 0
else
referenced_mentionables(user).any?
end
end
def award_emoji?
2016-08-24 12:49:21 +05:30
can_be_award_emoji? && contains_emoji_only?
2015-12-23 02:04:40 +05:30
end
def emoji_awardable?
!system?
end
2015-12-23 02:04:40 +05:30
2016-08-24 12:49:21 +05:30
def can_be_award_emoji?
2017-08-17 22:00:37 +05:30
noteable.is_a?(Awardable) && !part_of_discussion?
2015-12-23 02:04:40 +05:30
end
def contains_emoji_only?
note =~ /\A#{Banzai::Filter::EmojiFilter.emoji_pattern}\s?\Z/
end
2017-08-17 22:00:37 +05:30
def to_ability_name
for_personal_snippet? ? 'personal_snippet' : noteable_type.underscore
end
def can_be_discussion_note?
self.noteable.supports_discussions? && !part_of_discussion?
end
2018-03-27 19:54:05 +05:30
def can_create_todo?
# Skip system notes, and notes on project snippet
!system? && !for_snippet?
end
2017-08-17 22:00:37 +05:30
def discussion_class(noteable = nil)
# When commit notes are rendered on an MR's Discussion page, they are
# displayed in one discussion instead of individually.
# See also `#discussion_id` and `Discussion.override_discussion_id`.
if noteable && noteable != self.noteable
OutOfContextDiscussion
else
IndividualNoteDiscussion
end
end
# See `Discussion.override_discussion_id` for details.
def discussion_id(noteable = nil)
discussion_class(noteable).override_discussion_id(self) || super()
end
# Returns a discussion containing just this note.
# This method exists as an alternative to `#discussion` to use when the methods
# we intend to call on the Discussion object don't require it to have all of its notes,
# and just depend on the first note or the type of discussion. This saves us a DB query.
def to_discussion(noteable = nil)
Discussion.build([self], noteable)
end
# Returns the entire discussion this note is part of.
# Consider using `#to_discussion` if we do not need to render the discussion
# and all its notes and if we don't care about the discussion's resolvability status.
def discussion
full_discussion = self.noteable.notes.find_discussion(self.discussion_id) if part_of_discussion?
full_discussion || to_discussion
end
def part_of_discussion?
!to_discussion.individual_note?
end
def in_reply_to?(other)
case other
when Note
if part_of_discussion?
in_reply_to?(other.noteable) && in_reply_to?(other.to_discussion)
else
in_reply_to?(other.noteable)
end
when Discussion
self.discussion_id == other.id
when Noteable
self.noteable == other
else
false
end
2016-08-24 12:49:21 +05:30
end
2018-03-17 18:26:18 +05:30
def references
refs = [noteable]
if part_of_discussion?
refs += discussion.notes.take_while { |n| n.id < id }
end
refs
end
def expire_etag_cache
return unless noteable&.discussions_rendered_on_frontend?
key = Gitlab::Routing.url_helpers.project_noteable_notes_path(
project,
target_type: noteable_type.underscore,
target_id: noteable_id
)
Gitlab::EtagCaching::Store.new.touch(key)
end
def touch(*args)
# We're not using an explicit transaction here because this would in all
# cases result in all future queries going to the primary, even if no writes
# are performed.
#
# We touch the noteable first so its SELECT query can run before our writes,
# ensuring it runs on a secondary (if no prior write took place).
touch_noteable
super
end
# By default Rails will issue an "SELECT *" for the relation, which is
# overkill for just updating the timestamps. To work around this we manually
# touch the data so we can SELECT only the columns we need.
def touch_noteable
# Commits are not stored in the DB so we can't touch them.
return if for_commit?
assoc = association(:noteable)
noteable_object =
if assoc.loaded?
noteable
else
# If the object is not loaded (e.g. when notes are loaded async) we
# _only_ want the data we actually need.
assoc.scope.select(:id, :updated_at).take
end
noteable_object&.touch
# We return the noteable object so we can re-use it in EE for ElasticSearch.
noteable_object
end
def banzai_render_context(field)
super.merge(noteable: noteable)
end
2016-08-24 12:49:21 +05:30
private
def keep_around_commit
project.repository.keep_around(self.commit_id)
end
def nullify_blank_type
self.type = nil if self.type.blank?
end
def nullify_blank_line_code
self.line_code = nil if self.line_code.blank?
2015-04-26 12:48:37 +05:30
end
2016-09-13 17:45:13 +05:30
def ensure_discussion_id
return unless self.persisted?
# Needed in case the SELECT statement doesn't ask for `discussion_id`
return unless self.has_attribute?(:discussion_id)
return if self.discussion_id
set_discussion_id
update_column(:discussion_id, self.discussion_id)
end
def set_discussion_id
2017-08-17 22:00:37 +05:30
self.discussion_id ||= discussion_class.discussion_id(self)
2016-09-13 17:45:13 +05:30
end
2018-03-17 18:26:18 +05:30
def force_cross_reference_regex_check?
return unless system?
2017-08-17 22:00:37 +05:30
2018-03-17 18:26:18 +05:30
SystemNoteMetadata::TYPES_WITH_CROSS_REFERENCES.include?(system_note_metadata&.action)
2016-09-13 17:45:13 +05:30
end
2014-09-02 18:07:02 +05:30
end