feat(linux): Emit D-Bus signal directly for dock notification badges

src/main/appBadge.ts

@@ -7,6 +7,7 @@
 import { app, NativeImage, nativeImage } from "electron";
 import { join } from "path";
 import { BADGE_DIR } from "shared/paths";
+import { execFile } from "child_process";
 
 const imgCache = new Map<number, NativeImage>();
 function loadBadge(index: number) {
@@ -24,8 +25,24 @@ let lastIndex: null | number = -1;
 export function setBadgeCount(count: number) {
     switch (process.platform) {
         case "linux":
-            if (count === -1) count = 0;
-            app.setBadgeCount(count);
+            if (typeof count !== "number") { //sanitize
+                console.log("what the hel- *kaboom*") 
+                break;
+            }
+
+            function emitDBusBadge(count: number, visible: boolean) {
+                execFile ("gdbus", [
+                    "emit",
+                    "--session",
+                    "--object-path",
+                    "/",
+                    "--signal",
+                    "com.canonical.Unity.LauncherEntry.Update",
+                    "application://vesktop.desktop",
+                    `{\'count\': <int64 ${count}>, \'count-visible\': <${visible}>}`
+                ]);
+            }
+            emitDBusBadge(count === -1 ? 0 : count, count !== 0);
             break;
         case "darwin":
             if (count === 0) {

src/main/mainWindow.ts

@@ -445,6 +445,10 @@ function createMainWindow() {
     });
 
     if (Settings.store.staticTitle) win.on("page-title-updated", e => e.preventDefault());
+    else if (Settings.store.appBadge) mainWin.webContents.on('page-title-updated', (_, title) => {
+        let cleanedTitle = title.replace(/^\(\d+\)\s*|•\s/, '');
+        mainWin.setTitle(cleanedTitle);    
+    });
 
     initWindowBoundsListeners(win);
     if (!isDeckGameMode && (Settings.store.tray ?? true) && process.platform !== "darwin") initTray(win);

src/renderer/appBadge.ts

@@ -11,17 +11,19 @@ import { Settings } from "./settings";
 
 let GuildReadStateStore: any;
 let NotificationSettingsStore: any;
+let MessageRequestStore: any;
 
 export function setBadge() {
     if (Settings.store.appBadge === false) return;
 
     try {
         const mentionCount = GuildReadStateStore.getTotalMentionCount();
         const pendingRequests = RelationshipStore.getPendingCount();
+        const messageRequests = MessageRequestStore.getMessageRequestsCount();
         const hasUnread = GuildReadStateStore.hasAnyUnread();
         const disableUnreadBadge = NotificationSettingsStore.getDisableUnreadBadge();
 
-        let totalCount = mentionCount + pendingRequests;
+        let totalCount = mentionCount + pendingRequests + messageRequests;
         if (!totalCount && hasUnread && !disableUnreadBadge) totalCount = -1;
 
         VesktopNative.app.setBadgeCount(totalCount);
@@ -44,4 +46,5 @@ function waitForAndSubscribeToStore(name: string, cb?: (m: any) => void) {
 
 waitForAndSubscribeToStore("GuildReadStateStore", store => (GuildReadStateStore = store));
 waitForAndSubscribeToStore("NotificationSettingsStore", store => (NotificationSettingsStore = store));
+waitForAndSubscribeToStore("MessageRequestStore", store => (MessageRequestStore = store));
 waitForAndSubscribeToStore("RelationshipStore");