From 09c82c76a3b7c7190d1e44531715a5a1ceebdfa3 Mon Sep 17 00:00:00 2001
From: Juan Tejada <juans.tejada@gmail.com>
Date: Thu, 28 Jan 2016 11:10:05 -0800
Subject: [PATCH] fix(count): Fix unread count for badge and tray icon in mac

---
 src/flux/stores/unread-badge-store.coffee | 24 +++++++++--------------
 src/mailbox-perspective.coffee            |  2 +-
 2 files changed, 10 insertions(+), 16 deletions(-)

diff --git a/src/flux/stores/unread-badge-store.coffee b/src/flux/stores/unread-badge-store.coffee
index c2dc55cde..4b48f403e 100644
--- a/src/flux/stores/unread-badge-store.coffee
+++ b/src/flux/stores/unread-badge-store.coffee
@@ -2,15 +2,14 @@ Reflux = require 'reflux'
 _ = require 'underscore'
 NylasStore = require 'nylas-store'
 FocusedPerspectiveStore = require './focused-perspective-store'
-CategoryStore = require './category-store'
 ThreadCountsStore = require './thread-counts-store'
 
 class UnreadBadgeStore extends NylasStore
 
   constructor: ->
-    @_count = 0
+    @_count = FocusedPerspectiveStore.current().threadUnreadCount()
 
-    @listenTo CategoryStore, @_updateCount
+    @listenTo FocusedPerspectiveStore, @_updateCount
     @listenTo ThreadCountsStore, @_updateCount
     NylasEnv.config.observe 'core.notifications.unreadBadge', (val) =>
       if val is true
@@ -25,18 +24,13 @@ class UnreadBadgeStore extends NylasStore
     @_count
 
   _updateCount: =>
-    account = FocusedPerspectiveStore.current().account
-    category = CategoryStore.getStandardCategory(account, 'inbox')
-    if category
-      count = ThreadCountsStore.unreadCountForCategoryId(category.id) ? 0
-    else
-      count = 0
-
-    return if @_count is count
-
-    @_count = count
-    @_setBadgeForCount()
-    @trigger()
+    current = FocusedPerspectiveStore.current()
+    if current.isInbox()
+      count = current.threadUnreadCount()
+      return if @_count is count
+      @_count = count
+      @_setBadgeForCount()
+      @trigger()
 
   _setBadgeForCount: =>
     return unless NylasEnv.config.get('core.notifications.unreadBadge')
diff --git a/src/mailbox-perspective.coffee b/src/mailbox-perspective.coffee
index 34b4a9c3b..a56384477 100644
--- a/src/mailbox-perspective.coffee
+++ b/src/mailbox-perspective.coffee
@@ -64,7 +64,7 @@ class MailboxPerspective
     []
 
   category: =>
-    return null unless @categories().length isnt 0
+    return null unless @categories().length is 1
     return @categories()[0]
 
   threads: =>