AgentCompany/src/main/notification-service.ts

import { Notification, app } from "electron";
import type { BrowserWindow } from "electron";
import type { DesktopEvent, SectionId } from "@shared/types";

export interface NotificationOptions {
  title: string;
  body: string;
  urgency: "critical" | "informational";
  navigation?: {
    section: SectionId;
    entityId?: string;
  };
  /** Batch key -- notifications with same key within window are batched */
  batchKey?: string;
}

interface NotificationServiceDependencies {
  getMainWindow: () => BrowserWindow | null;
  emitEvent: (event: DesktopEvent) => void;
  logger: {
    info: (...args: unknown[]) => void;
    warn: (...args: unknown[]) => void;
    error: (...args: unknown[]) => void;
  };
  /** Returns the active UI locale ("en", "zh", ...). Batch summary copy is
   * localized; fallbacks to "en" when unset. */
  getLocale?: () => string | null | undefined;
}

interface BatchState {
  count: number;
  timer: ReturnType<typeof setTimeout>;
  /** Latest section observed during the window. Summary navigates here on click. */
  section: SectionId;
}

const BATCH_WINDOW_MS = 10_000;

/**
 * Batch keys that should NEVER be batched -- always fire individually.
 * These represent events the user must see each instance of.
 */
const NEVER_BATCH: ReadonlySet<string> = new Set([
  "pending_approval",
  "crash_recovery",
  "update_available",
  "manual_intervention",
  "budget_exhaustion",
]);

interface LocalizedBatchCopy {
  title: (count: number) => string;
  body: string;
}

/**
 * Batch summary copy per batch key, per-locale. Used when the batch timer
 * expires and accumulated events need a summary notification.
 */
const BATCH_SUMMARY_COPY: Record<string, Record<"en" | "zh", LocalizedBatchCopy>> = {
  failed_runs: {
    en: { title: (count) => `${count} tasks failed`, body: "Click to view failed runs" },
    zh: { title: (count) => `${count} 个任务失败`, body: "点击查看失败记录" },
  },
  agent_hired: {
    en: { title: (count) => `${count} agents hired`, body: "Click to view organization" },
    zh: { title: (count) => `${count} 位新员工入职`, body: "点击查看组织架构" },
  },
  budget_alerts: {
    en: { title: (count) => `${count} budget alerts`, body: "Click to view cost details" },
    zh: { title: (count) => `${count} 个预算提醒`, body: "点击查看成本详情" },
  },
  task_completed: {
    en: { title: (count) => `${count} tasks completed`, body: "Click to view task list" },
    zh: { title: (count) => `${count} 个任务已完成`, body: "点击查看任务列表" },
  },
  agent_paused: {
    en: { title: (count) => `${count} agents paused`, body: "Click to view paused agents" },
    zh: { title: (count) => `${count} 位员工已暂停`, body: "点击查看已暂停员工" },
  },
};

function pickLocale(locale: string | null | undefined): "en" | "zh" {
  return locale === "zh" ? "zh" : "en";
}

export function createNotificationService(deps: NotificationServiceDependencies) {
  /**
   * Prevent GC on Windows -- Notification objects must be retained
   * while visible so click handlers fire. Removed on "close" event.
   */
  const activeNotifications: Set<Notification> = new Set();

  /** Active batch state per batch key */
  const activeBatches: Map<string, BatchState> = new Map();

  function notify(options: NotificationOptions) {
    if (!Notification.isSupported()) return;

    const key = options.batchKey;

    // If no batch key, or the key is in the NEVER_BATCH list, fire immediately
    if (!key || NEVER_BATCH.has(key)) {
      fireNotification(options);
      return;
    }

    const existing = activeBatches.get(key);
    if (!existing) {
      // First event in batch -- fire immediately, start batch window
      fireNotification(options);
      activeBatches.set(key, {
        count: 0,
        timer: setTimeout(() => flushBatch(key), BATCH_WINDOW_MS),
        section: options.navigation?.section ?? "overview",
      });
      return;
    }

    // Accumulate -- do not fire individual notification.
    // Track the most recent section so the summary links to where the latest event happened.
    existing.count++;
    const nextSection = options.navigation?.section;
    if (nextSection) {
      existing.section = nextSection;
    }
  }

  function flushBatch(key: string) {
    const batch = activeBatches.get(key);
    activeBatches.delete(key);

    if (batch && batch.count > 0) {
      const locale = pickLocale(deps.getLocale?.());
      const copySet = BATCH_SUMMARY_COPY[key];
      const copy = copySet?.[locale];
      const title = copy ? copy.title(batch.count) : `${batch.count} more events`;
      const body = copy ? copy.body : "Click to view";

      fireNotification({
        title,
        body,
        urgency: "informational",
        navigation: { section: batch.section },
        // No entityId -- batched summary navigates to section overview
      });
    }
  }

  function fireNotification(options: NotificationOptions) {
    if (!Notification.isSupported()) return;

    const notificationOptions: Electron.NotificationConstructorOptions = {
      title: options.title,
      body: options.body,
      silent: options.urgency === "informational",
    };

    // Platform-specific urgency handling
    if (process.platform === "win32" && options.urgency === "critical") {
      notificationOptions.timeoutType = "never";
    }
    if (process.platform === "linux") {
      notificationOptions.urgency = options.urgency === "critical" ? "critical" : "normal";
    }

    const notification = new Notification(notificationOptions);

    // Click handler: restore window and navigate (D-02 window restoration sequence)
    notification.on("click", () => {
      app.focus({ steal: true });
      const win = deps.getMainWindow();
      if (win && !win.isDestroyed()) {
        if (win.isMinimized()) win.restore();
        win.show();
        win.focus();
      }
      if (options.navigation) {
        deps.emitEvent({
          type: "notification-click",
          section: options.navigation.section,
          entityId: options.navigation.entityId,
        });
      }
    });

    // GC protection: hold reference until notification closes
    notification.on("close", () => {
      activeNotifications.delete(notification);
    });

    notification.show();
    activeNotifications.add(notification);
  }

  function destroy() {
    // Clear all batch timers
    for (const batch of activeBatches.values()) {
      clearTimeout(batch.timer);
    }
    activeBatches.clear();
    // Dismiss any still-visible notifications so Windows toast host does not
    // leave ghost toasts after quit.
    for (const notification of activeNotifications) {
      try {
        notification.close();
      } catch (error) {
        deps.logger.warn(
          `[notification] notification.close() during destroy failed: ${error instanceof Error ? error.message : String(error)}`,
        );
      }
    }
    activeNotifications.clear();
  }

  return {
    notify,
    destroy,
  };
}