debian-mirror-gitlab/app/models/concerns/subscribable.rb

83 lines
2 KiB
Ruby
Raw Normal View History

2016-06-02 11:05:42 +05:30
# == Subscribable concern
#
# Users can subscribe to these models.
#
# Used by Issue, MergeRequest, Label
#
module Subscribable
extend ActiveSupport::Concern
included do
2017-09-10 17:25:29 +05:30
has_many :subscriptions, dependent: :destroy, as: :subscribable # rubocop:disable Cop/ActiveRecordDependent
2016-06-02 11:05:42 +05:30
end
2017-08-17 22:00:37 +05:30
def subscribed?(user, project = nil)
if subscription = subscriptions.find_by(user: user, project: project)
2016-06-02 11:05:42 +05:30
subscription.subscribed
else
2017-08-17 22:00:37 +05:30
subscribed_without_subscriptions?(user, project)
2016-06-02 11:05:42 +05:30
end
end
# Override this method to define custom logic to consider a subscribable as
# subscribed without an explicit subscription record.
2017-08-17 22:00:37 +05:30
def subscribed_without_subscriptions?(user, project)
2016-06-02 11:05:42 +05:30
false
end
2017-08-17 22:00:37 +05:30
def subscribers(project)
2017-09-10 17:25:29 +05:30
subscriptions_available(project)
.where(subscribed: true)
.map(&:user)
2016-06-02 11:05:42 +05:30
end
2017-08-17 22:00:37 +05:30
def toggle_subscription(user, project = nil)
unsubscribe_from_other_levels(user, project)
2017-09-10 17:25:29 +05:30
find_or_initialize_subscription(user, project)
.update(subscribed: !subscribed?(user, project))
2017-08-17 22:00:37 +05:30
end
def subscribe(user, project = nil)
unsubscribe_from_other_levels(user, project)
find_or_initialize_subscription(user, project)
.update(subscribed: true)
end
def unsubscribe(user, project = nil)
unsubscribe_from_other_levels(user, project)
find_or_initialize_subscription(user, project)
.update(subscribed: false)
2016-06-02 11:05:42 +05:30
end
2017-08-17 22:00:37 +05:30
private
def unsubscribe_from_other_levels(user, project)
other_subscriptions = subscriptions.where(user: user)
other_subscriptions =
if project.blank?
other_subscriptions.where.not(project: nil)
else
other_subscriptions.where(project: nil)
end
other_subscriptions.update_all(subscribed: false)
end
def find_or_initialize_subscription(user, project)
2017-09-10 17:25:29 +05:30
subscriptions
.find_or_initialize_by(user_id: user.id, project_id: project.try(:id))
2016-06-02 11:05:42 +05:30
end
2017-08-17 22:00:37 +05:30
def subscriptions_available(project)
t = Subscription.arel_table
2017-09-10 17:25:29 +05:30
subscriptions
.where(t[:project_id].eq(nil).or(t[:project_id].eq(project.try(:id))))
2016-06-02 11:05:42 +05:30
end
end