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

package.json

@@ -30,7 +30,8 @@
         "electron-updater": "^6.3.9"
     },
     "optionalDependencies": {
-        "@vencord/venmic": "^6.1.0"
+        "@vencord/venmic": "^6.1.0",
+        "@homebridge/dbus-native": "0.6.0"
     },
     "devDependencies": {
         "@fal-works/esbuild-plugin-global-externals": "^2.1.2",

src/main/appBadge.ts

@@ -8,6 +8,8 @@ import { app, NativeImage, nativeImage } from "electron";
 import { join } from "path";
 import { BADGE_DIR } from "shared/paths";
 
+import { dbus, getSessionBus } from "./utils/dbus";
+
 const imgCache = new Map<number, NativeImage>();
 function loadBadge(index: number) {
     const cached = imgCache.get(index);
@@ -24,8 +26,28 @@ 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") {
+                throw new Error("count must be a number"); // sanitize
+            }
+
+            const sessionBus = getSessionBus();
+            sessionBus.connection.message({
+                type: dbus.messageType.signal,
+                serial: 1,
+                path: "/",
+                interface: "com.canonical.Unity.LauncherEntry",
+                member: "Update",
+                signature: "sa{sv}",
+                body: [
+                    process.env.container === "1"
+                        ? "application://dev.vencord.Vesktop.desktop" // flatpak handling
+                        : "application://vesktop.desktop",
+                    [
+                        ["count", ["x", count === -1 ? 0 : count]],
+                        ["count-visible", ["b", count !== 0]]
+                    ]
+                ]
+            });
             break;
         case "darwin":
             if (count === 0) {

src/main/mainWindow.ts

@@ -395,6 +395,7 @@ function initStaticTitle(win: BrowserWindow) {
             win.setTitle("Vesktop");
             win.on("page-title-updated", listener);
         } else {
+            win.setTitle(win.getTitle().replace(/^\(\d+\)\s*|•\s/, ""));
             win.off("page-title-updated", listener);
         }
     });

src/main/utils/dbus.ts

@@ -0,0 +1,16 @@
+/*
+ * SPDX-License-Identifier: GPL-3.0
+ * Vesktop, a desktop app aiming to give you a snappier Discord Experience
+ * Copyright (c) 2023 Vendicated and Vencord contributors
+ */
+
+import dbus from "@homebridge/dbus-native";
+
+let sessionBus: dbus.MessageBus | null;
+
+export function getSessionBus(): dbus.MessageBus {
+    if (!sessionBus) sessionBus = dbus.sessionBus();
+    return sessionBus;
+}
+
+export { dbus };

src/renderer/appBadge.ts

@@ -12,17 +12,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);
@@ -45,4 +47,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");