Adding a quick Dock badge fix.

jdsimcoe · Aug 23, 2018 · 426cd9d · 426cd9d
1 parent 406b61e
commit 426cd9d
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/index.js b/index.js
@@ -29,10 +29,11 @@ if (isAlreadyRunning) {
 }
 
 function updateBadge(title) {
-	// ignore `Sindre messaged you` blinking
-	if (title.indexOf('Keyframes') === -1) {
-		return;
-	}
+
+  mainWindow.webContents.executeJavaScript(`
+    var count = document.getElementsByClassName('unread')["0"].dataset.unreadCount;
+    document.title = "Keyframes (" + count + ")";
+  `);
 
 	let messageCount = (/\(([0-9]+)\)/).exec(title);
 	messageCount = messageCount ? Number(messageCount[1]) : 0;