refactor: move dock-badge service to trpc (#274)

Author: jonathanlab

apps/array/src/main/di/container.ts

@@ -1,6 +1,7 @@
 import "reflect-metadata";
 import { Container } from "inversify";
 import { ContextMenuService } from "../services/context-menu/service.js";
+import { DockBadgeService } from "../services/dock-badge/service.js";
 import { ExternalAppsService } from "../services/external-apps/service.js";
 import { GitService } from "../services/git/service.js";
 import { MAIN_TOKENS } from "./tokens.js";
@@ -9,6 +10,7 @@ export const container = new Container({
   defaultScope: "Singleton",
 });
 
+container.bind(MAIN_TOKENS.DockBadgeService).to(DockBadgeService);
 container.bind(MAIN_TOKENS.ExternalAppsService).to(ExternalAppsService);
 container.bind(MAIN_TOKENS.GitService).to(GitService);
 container.bind(MAIN_TOKENS.ContextMenuService).to(ContextMenuService);

apps/array/src/main/di/tokens.ts

@@ -7,6 +7,7 @@
 export const MAIN_TOKENS = Object.freeze({
   // Services
   ContextMenuService: Symbol.for("Main.ContextMenuService"),
+  DockBadgeService: Symbol.for("Main.DockBadgeService"),
   ExternalAppsService: Symbol.for("Main.ExternalAppsService"),
   GitService: Symbol.for("Main.GitService"),
 });

apps/array/src/main/index.ts

@@ -20,7 +20,9 @@ import {
 import { createIPCHandler } from "trpc-electron/main";
 import "./lib/logger";
 import { ANALYTICS_EVENTS } from "../types/analytics.js";
-import { dockBadgeService } from "./services/dockBadge.js";
+import { container } from "./di/container.js";
+import { MAIN_TOKENS } from "./di/tokens.js";
+import type { DockBadgeService } from "./services/dock-badge/service.js";
 import {
   cleanupAgentSessions,
   registerAgentIpc,
@@ -254,7 +256,7 @@ app.whenReady().then(() => {
   ensureClaudeConfigDir();
 
   // Initialize dock badge service for notification badges
-  dockBadgeService.initialize(() => mainWindow);
+  container.get<DockBadgeService>(MAIN_TOKENS.DockBadgeService);
 
   // Initialize PostHog analytics
   initializePostHog();

apps/array/src/main/preload.ts

@@ -337,10 +337,6 @@ contextBridge.exposeInMainWorld("electronAPI", {
       }>,
     ): (() => void) => createIpcListener("workspace:warning", listener),
   },
-  // Dock Badge API
-  dockBadge: {
-    show: (): Promise<void> => ipcRenderer.invoke("dock-badge:show"),
-  },
   shellExecute: (
     cwd: string,
     command: string,

apps/array/src/main/services/dock-badge/service.ts

@@ -0,0 +1,36 @@
+import { app } from "electron";
+import { injectable, postConstruct } from "inversify";
+import { logger } from "../../lib/logger";
+
+const log = logger.scope("dock-badge");
+
+@injectable()
+export class DockBadgeService {
+  private hasBadge = false;
+
+  @postConstruct()
+  init(): void {
+    app.on("browser-window-focus", () => this.clear());
+    log.info("Dock badge service initialized");
+  }
+
+  show(): void {
+    if (this.hasBadge) return;
+
+    this.hasBadge = true;
+    if (process.platform === "darwin" || process.platform === "linux") {
+      app.setBadgeCount(1);
+    }
+    log.info("Dock badge shown");
+  }
+
+  private clear(): void {
+    if (!this.hasBadge) return;
+
+    this.hasBadge = false;
+    if (process.platform === "darwin" || process.platform === "linux") {
+      app.setBadgeCount(0);
+    }
+    log.info("Dock badge cleared");
+  }
+}

apps/array/src/main/services/dockBadge.ts

@@ -3,7 +3,6 @@
  * This file is auto-generated by vite-plugin-auto-services.ts
  */
 
-import "./dockBadge.js";
 import "./fileWatcher.js";
 import "./folders.js";
 import "./fs.js";

apps/array/src/main/services/index.ts

@@ -1,4 +1,5 @@
 import { contextMenuRouter } from "./routers/context-menu.js";
+import { dockBadgeRouter } from "./routers/dock-badge.js";
 import { encryptionRouter } from "./routers/encryption.js";
 import { externalAppsRouter } from "./routers/external-apps.js";
 import { gitRouter } from "./routers/git.js";
@@ -8,13 +9,14 @@ import { secureStoreRouter } from "./routers/secure-store.js";
 import { router } from "./trpc.js";
 
 export const trpcRouter = router({
-  os: osRouter,
-  logs: logsRouter,
-  secureStore: secureStoreRouter,
-  encryption: encryptionRouter,
-  git: gitRouter,
   contextMenu: contextMenuRouter,
+  dockBadge: dockBadgeRouter,
+  encryption: encryptionRouter,
   externalApps: externalAppsRouter,
+  git: gitRouter,
+  logs: logsRouter,
+  os: osRouter,
+  secureStore: secureStoreRouter,
 });
 
 export type TrpcRouter = typeof trpcRouter;

apps/array/src/main/trpc/router.ts

@@ -0,0 +1,11 @@
+import { container } from "../../di/container.js";
+import { MAIN_TOKENS } from "../../di/tokens.js";
+import type { DockBadgeService } from "../../services/dock-badge/service.js";
+import { publicProcedure, router } from "../trpc.js";
+
+const getService = () =>
+  container.get<DockBadgeService>(MAIN_TOKENS.DockBadgeService);
+
+export const dockBadgeRouter = router({
+  show: publicProcedure.mutation(() => getService().show()),
+});

apps/array/src/main/trpc/routers/dock-badge.ts

@@ -13,6 +13,7 @@ import {
 import { Box } from "@radix-ui/themes";
 import { logger } from "@renderer/lib/logger";
 import { useNavigationStore } from "@renderer/stores/navigationStore";
+import { trpcVanilla } from "@renderer/trpc/client";
 import type { Task } from "@shared/types";
 import { useCallback, useEffect, useRef } from "react";
 
@@ -83,7 +84,7 @@ export function TaskLogsPanel({ taskId, task }: TaskLogsPanelProps) {
 
         const isWindowFocused = document.hasFocus();
         if (!isWindowFocused) {
-          window.electronAPI.dockBadge.show();
+          trpcVanilla.dockBadge.show.mutate();
         }
       } catch (error) {
         log.error("Failed to send prompt", error);