docbrown/app/labor/notification_counter.rb
rhymes e588fa7ece Code cleanups (#659)
* Initial automatic cleanup with rubocop

* Fix syntax error introduced by rubocop

* Cleanup seeds file

* Cleanup lib folder

* Exclude bin folder because it contains auto generated files

* Make Rubocop a little bit more chatty

* Block length should not include comments in the count

* Cleanup config folder

* Cleanup specs

* Updated Rubocop version and generated a todo file

* Fix broken ArticlesApi spec

* Fix tests

* Restored rubocop pre-commit hook
2018-08-07 11:00:13 -04:00

14 lines
329 B
Ruby

class NotificationCounter
def initialize(user)
@user = user
end
def unread_notification_count
return 0 if Rails.env.test?
StreamRails.feed_manager.get_notification_feed(@user.id).get["unseen"]
end
def set_to_zero
StreamRails.feed_manager.get_notification_feed(@user.id).get(mark_seen: true)
end
end