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

72 lines
1.9 KiB
Ruby
Raw Normal View History

2016-06-02 11:05:42 +05:30
class NotificationSetting < ActiveRecord::Base
2017-09-10 17:25:29 +05:30
include IgnorableColumn
ignore_column :events
2016-06-22 15:30:34 +05:30
enum level: { global: 3, watch: 2, mention: 4, participating: 1, disabled: 0, custom: 5 }
2016-06-02 11:05:42 +05:30
default_value_for :level, NotificationSetting.levels[:global]
belongs_to :user
2017-09-10 17:25:29 +05:30
belongs_to :source, polymorphic: true # rubocop:disable Cop/PolymorphicAssociations
2016-08-24 12:49:21 +05:30
belongs_to :project, foreign_key: 'source_id'
2016-06-02 11:05:42 +05:30
validates :user, presence: true
validates :level, presence: true
validates :user_id, uniqueness: { scope: [:source_type, :source_id],
message: "already exists in source",
allow_nil: true }
scope :for_groups, -> { where(source_type: 'Namespace') }
2016-08-24 12:49:21 +05:30
# Exclude projects not included by the Project model's default scope (those that are
# pending delete).
#
scope :for_projects, -> do
2017-09-10 17:25:29 +05:30
includes(:project).references(:projects).where(source_type: 'Project').where.not(projects: { id: nil, pending_delete: true })
2016-08-24 12:49:21 +05:30
end
2016-06-02 11:05:42 +05:30
2016-06-22 15:30:34 +05:30
EMAIL_EVENTS = [
:new_note,
:new_issue,
:reopen_issue,
:close_issue,
:reassign_issue,
:new_merge_request,
:reopen_merge_request,
:close_merge_request,
:reassign_merge_request,
2017-08-17 22:00:37 +05:30
:merge_merge_request,
:failed_pipeline,
:success_pipeline
].freeze
EXCLUDED_WATCHER_EVENTS = [
:success_pipeline
].freeze
2016-06-22 15:30:34 +05:30
2016-06-02 11:05:42 +05:30
def self.find_or_create_for(source)
setting = find_or_initialize_by(source: source)
unless setting.persisted?
setting.save
end
setting
end
2016-06-22 15:30:34 +05:30
2017-08-17 22:00:37 +05:30
# Allow people to receive failed pipeline notifications if they already have
# custom notifications enabled, as these are more like mentions than the other
# custom settings.
def failed_pipeline
bool = super
bool.nil? || bool
end
2017-09-10 17:25:29 +05:30
alias_method :failed_pipeline?, :failed_pipeline
def event_enabled?(event)
2018-03-17 18:26:18 +05:30
respond_to?(event) && !!public_send(event) # rubocop:disable GitlabSecurity/PublicSend
2017-09-10 17:25:29 +05:30
end
2016-06-02 11:05:42 +05:30
end