discourse/app/models/topic_notifier.rb
Louis Rose 1574485443 Perform the where(...).first to find_by(...) refactoring.
This refactoring was automated using the command: bundle exec "ruby refactorings/where_dot_first_to_find_by/app.rb"
2014-05-06 14:41:59 +01:00

43 lines
1.0 KiB
Ruby

class TopicNotifier
def initialize(topic)
@topic = topic
end
{ :watch! => :watching,
:tracking! => :tracking,
:regular! => :regular,
:muted! => :muted }.each_pair do |method_name, level|
define_method method_name do |user_id|
change_level user_id, level
end
end
def watch_topic!(user_id, reason = :created_topic)
change_level user_id, :watching, reason
end
# Enable/disable the mute on the topic
def toggle_mute(user_id)
change_level user_id, (muted?(user_id) ? levels[:regular] : levels[:muted])
end
def muted?(user_id)
tu = @topic.topic_users.find_by(user_id: user_id)
tu && tu.notification_level == levels[:muted]
end
private
def levels
@notification_levels ||= TopicUser.notification_levels
end
def change_level(user_id, level, reason=nil)
attrs = {notification_level: levels[level]}
attrs.merge!(notifications_reason_id: TopicUser.notification_reasons[reason]) if reason
TopicUser.change(user_id, @topic.id, attrs)
end
end