debian-mirror-gitlab/lib/gitlab/background_migration/user_mentions/models/note.rb
2020-10-24 23:57:45 +05:30

72 lines
1.7 KiB
Ruby

# frozen_string_literal: true
# rubocop:disable Style/Documentation
module Gitlab
module BackgroundMigration
module UserMentions
module Models
class Note < ActiveRecord::Base
include EachBatch
include Concerns::IsolatedMentionable
include CacheMarkdownField
self.table_name = 'notes'
self.inheritance_column = :_type_disabled
attr_mentionable :note, pipeline: :note
cache_markdown_field :note, pipeline: :note, issuable_state_filter_enabled: true
belongs_to :author, class_name: "User"
belongs_to :noteable, polymorphic: true
belongs_to :project
def for_personal_snippet?
noteable && noteable.class.name == 'PersonalSnippet'
end
def for_project_noteable?
!for_personal_snippet? && !for_epic?
end
def skip_project_check?
!for_project_noteable?
end
def for_epic?
noteable && noteable_type == 'Epic'
end
def user_mention_resource_id
noteable_id || commit_id
end
def user_mention_note_id
id
end
def noteable
super unless for_commit?
end
def for_commit?
noteable_type == "Commit"
end
private
def mentionable_params
return super unless for_epic?
super.merge(banzai_context_params)
end
def banzai_context_params
return {} unless noteable
{ group: noteable.group, label_url_method: :group_epics_url }
end
end
end
end
end
end