notifications.ts

import { useSettingsStore } from "@features/settings/stores/settingsStore";
import { logger } from "@renderer/lib/logger";
import { playCompletionSound } from "@renderer/lib/sounds";
import { trpcVanilla } from "@renderer/trpc/client";

const log = logger.scope("notifications");

const MAX_TITLE_LENGTH = 50;

function truncateTitle(title: string): string {
  if (title.length <= MAX_TITLE_LENGTH) return title;
  return `${title.slice(0, MAX_TITLE_LENGTH)}...`;
}

function sendDesktopNotification(
  title: string,
  body: string,
  silent: boolean,
): void {
  trpcVanilla.notification.send.mutate({ title, body, silent }).catch((err) => {
    log.error("Failed to send notification", err);
  });
}

function showDockBadge(): void {
  trpcVanilla.notification.showDockBadge.mutate().catch((err) => {
    log.error("Failed to show dock badge", err);
  });
}

function bounceDock(): void {
  trpcVanilla.notification.bounceDock.mutate().catch((err) => {
    log.error("Failed to bounce dock", err);
  });
}

export function notifyPromptComplete(
  taskTitle: string,
  stopReason: string,
): void {
  if (stopReason !== "end_turn") return;

  const {
    completionSound,
    completionVolume,
    desktopNotifications,
    dockBadgeNotifications,
    dockBounceNotifications,
  } = useSettingsStore.getState();

  const isWindowFocused = document.hasFocus();
  if (isWindowFocused) return;

  const willPlayCustomSound = completionSound !== "none";
  playCompletionSound(completionSound, completionVolume);

  if (desktopNotifications) {
    sendDesktopNotification(
      "Twig",
      `"${truncateTitle(taskTitle)}" finished`,
      willPlayCustomSound,
    );
  }
  if (dockBadgeNotifications) {
    showDockBadge();
  }
  if (dockBounceNotifications) {
    bounceDock();
  }
}

export function notifyPermissionRequest(taskTitle: string): void {
  const {
    completionSound,
    completionVolume,
    desktopNotifications,
    dockBadgeNotifications,
    dockBounceNotifications,
  } = useSettingsStore.getState();
  const isWindowFocused = document.hasFocus();

  if (!isWindowFocused) {
    const willPlayCustomSound = completionSound !== "none";
    playCompletionSound(completionSound, completionVolume);

    if (desktopNotifications) {
      sendDesktopNotification(
        "Twig",
        `"${truncateTitle(taskTitle)}" needs your input`,
        willPlayCustomSound,
      );
    }
    if (dockBadgeNotifications) {
      showDockBadge();
    }
    if (dockBounceNotifications) {
      bounceDock();
    }
  }
}