1 files changed, 20 insertions, 5 deletions
diff --git a/ajax/loadfeed.php b/ajax/loadfeed.php
index 644f28714..03f143fe1 100644
--- a/ajax/loadfeed.php
+++ b/ajax/loadfeed.php
@@ -25,12 +25,27 @@ $itemsTpl = new OCP\Template("news", "part.items");
 $itemsTpl->assign('feedid', $feedId);
 $feedItems = $itemsTpl->fetchPage();
 
-$feedMapper = new OCA\News\FeedMapper();
-$feed = $feedMapper->findById($feedId);
-$feedTitle = $feed->getTitle();
-
 $itemMapper = new OCA\News\ItemMapper();
-$unreadItemCount = $itemMapper->countAllStatus($feedId, OCA\News\StatusFlag::UNREAD);
+
+
+switch ($feedId) {
+    case -1:
+        $feedTitle = $l->t('Starred');
+        $unreadItemCount = $itemMapper->countAllStatus($feedId, OCA\News\StatusFlag::IMPORTANT);
+        break;
+
+    case -2:
+        $feedTitle = $l->t('New articles');
+        $unreadItemCount = $itemMapper->countEveryItemByStatus(OCA\News\StatusFlag::UNREAD);
+        break;
+    
+    default:
+        $feedMapper = new OCA\News\FeedMapper();
+        $feed = $feedMapper->findById($feedId);
+        $feedTitle = $feed->getTitle();
+        $unreadItemCount = $itemMapper->countAllStatus($feedId, OCA\News\StatusFlag::UNREAD);
+        break;
+}
 
 OCP\JSON::success(array('data' => array( 'message' => $l->t('Feed loaded!'),
                                         'feedTitle' => $feedTitle,