diff --git a/app/controllers/notifications_controller.rb b/app/controllers/notifications_controller.rb index 8c94cc4c14a..dc0b5e4470a 100644 --- a/app/controllers/notifications_controller.rb +++ b/app/controllers/notifications_controller.rb @@ -3,7 +3,17 @@ class NotificationsController < ApplicationController before_filter :ensure_logged_in def index - notifications = current_user.notifications.recent.includes(:topic).all + notifications = current_user.notifications.recent.includes(:topic).all.to_a + + if notifications.present? + notifications += current_user.notifications + .order('created_at desc') + .where(read: false, notification_type: Notification.types[:private_message]) + .where('id < ?', notifications.last.id) + .limit(5) + .to_a + end + current_user.saw_notification_id(notifications.first.id) if notifications.present? current_user.reload current_user.publish_notifications_state