Enable notification badges

Closes #1695

progress.js

@@ -27,6 +27,7 @@ const Pango = imports.gi.Pango;
 const St = imports.gi.St;
 const Signals = imports.signals;
 const Utils = Me.imports.utils;
+const Main = imports.ui.main;
 
 
 var ProgressManager = class {
@@ -285,6 +286,7 @@ var ProgressIndicator = class {
     constructor(source, progressManager) {
         this._source = source;
         this._progressManager = progressManager;
+        this._messageTray = Main.messageTray;
         this._signalsHandler = new Utils.GlobalSignalsHandler();
 
         this._sourceDestroyId = this._source.connect('destroy', () => {
@@ -317,6 +319,18 @@ var ProgressIndicator = class {
             this._progressManager,
             'progress-entry-removed',
             this._onEntryRemoved.bind(this)
+        ], [
+            this._messageTray,
+            'source-added',
+            this._updateBadge.bind(this)
+        ], [
+            this._messageTray,
+            'source-removed',
+            this._updateBadge.bind(this)
+        ], [
+            this._messageTray,
+            'queue-changed',
+            this._updateBadge.bind(this)
         ]);
     }
 
@@ -368,6 +382,27 @@ var ProgressIndicator = class {
         }
     }
 
+    _updateBadge() {
+        // FIXME - Probably not the best way to get the show counter badge setting
+        // It is already present in AppProgress class, not sure how to get it here
+        const showBadge = Me.settings.get_boolean('progress-show-count');
+        if (showBadge) {
+            let items = this._messageTray.getSources();
+            let unreadCount = 0;
+            for (let i = 0; i < items.length; i++) {
+                let item = items[i];
+                // FIXME - Check if these 12 symbols are enough
+                if (this._source.app.get_name().includes(item.title.substring(0, 12))) {
+                    unreadCount += item.notifications.length;
+                }
+            }
+            if (unreadCount != this._notificationBadgeCount) {
+                this.setNotificationBadge(unreadCount);
+                this.toggleNotificationBadge(showBadge);
+            }
+        }
+    }
+
     setNotificationBadge(count) {
         this._notificationBadgeCount = count;
         let text = this._notificationBadgeCountToText(count);