openswarm/frontend/src/shared/notifications.ts

// Native notifications for agent completion; fires only when document is hidden (user switched away).
import { store } from '@/shared/state/store';

function notificationsEnabled(kind: 'agent' | 'workflow'): boolean {
  const d = store.getState().settings.data as { notify_agent_completion?: boolean; notify_workflow_runs?: boolean };
  return kind === 'agent' ? d.notify_agent_completion !== false : d.notify_workflow_runs !== false;
}

const FIRED_RECENTLY = new Set<string>();
const COOLDOWN_MS = 30_000;

let permissionRequested = false;

function ensurePermission(): NotificationPermission {
  if (typeof Notification === 'undefined') return 'denied';
  if (Notification.permission === 'granted') return 'granted';
  if (Notification.permission === 'denied') return 'denied';
  if (!permissionRequested) {
    permissionRequested = true;
    Notification.requestPermission().catch(() => {});
  }
  return 'default';
}

export interface AgentCompletionPayload {
  sessionId: string;
  sessionName: string;
  dashboardId?: string;
  status: 'completed' | 'error';
  bodyExcerpt?: string;
}

export function notifyAgentCompletion(p: AgentCompletionPayload): void {
  if (typeof document === 'undefined') return;
  if (!notificationsEnabled('agent')) return;
  // Same-window: skip noise (hidden = tab-switched, minimized, or another BrowserWindow in front).
  if (!document.hidden) return;
  if (typeof Notification === 'undefined') return;
  const perm = ensurePermission();
  if (perm !== 'granted') return;

  // Per-session debounce: collapse rapid completed/error/completed flips.
  const key = `${p.sessionId}:${p.status}`;
  if (FIRED_RECENTLY.has(key)) return;
  FIRED_RECENTLY.add(key);
  setTimeout(() => FIRED_RECENTLY.delete(key), COOLDOWN_MS);

  const title = p.status === 'error'
    ? `${p.sessionName} hit an error`
    : `${p.sessionName} finished`;
  const body = (p.bodyExcerpt || '').slice(0, 140);

  try {
    const n = new Notification(title, {
      body,
      tag: p.sessionId,
      silent: false,
    });
    n.onclick = () => {
      try { window.focus(); } catch {}
      window.dispatchEvent(new CustomEvent('openswarm:notification-click', {
        detail: { sessionId: p.sessionId, dashboardId: p.dashboardId },
      }));
      n.close();
    };
  } catch {
    // Notification API can throw if sandboxed or headless; fail silently.
  }
}

export type WorkflowNotificationOutcome = 'open' | 'ack' | 'rerun' | 'edit';

export interface WorkflowRunNotification {
  workflowId: string;
  workflowTitle: string;
  runId?: string;
  sessionId?: string;
  status: string;
  tierKind?: string;
  fallback?: boolean;
}

const SUCCESS_TITLES = ['{name} is done', '{name} just wrapped up', 'Heads up: {name} finished', '{name} is ready'];
const FAILURE_TITLES = ['{name} hit a snag', "{name} couldn't finish", 'Something went sideways on {name}'];
const LATE_TITLES = ['{name} caught up late', '{name} ran late but made it'];

function workflowTitleFor(p: WorkflowRunNotification): string {
  const name = p.workflowTitle || 'Workflow';
  const pool = p.status === 'success' ? SUCCESS_TITLES
    : p.status === 'failure' ? FAILURE_TITLES
      : p.status === 'ran_late' ? LATE_TITLES
        : null;
  if (!pool) return `${name}: ${p.status}`;
  // Seed by workflow id + current minute so two workflows pick different copy while one workflow stays stable across a few minutes.
  const seed = Math.abs((p.workflowId.length + Math.floor(Date.now() / 60_000)) | 0);
  return pool[seed % pool.length].replace('{name}', name);
}

function workflowBodyFor(p: WorkflowRunNotification): string {
  if (p.tierKind && p.fallback) {
    return `Would have ${p.tierKind === 'call' ? 'called' : 'texted'} you. (Cloud SMS not wired yet.)`;
  }
  const verb = typeof navigator !== 'undefined' && /Mac/i.test(navigator.platform) ? 'Tap' : 'Click';
  if (p.status === 'success') return `${verb} to see what it did.`;
  if (p.status === 'failure') return `${verb} to see what went wrong.`;
  return `${verb} to open the run.`;
}

/** Native OS notification for a finished workflow run. Prefers the Electron main process, which reaches Notification Center even with the window hidden or the renderer backgrounded; the renderer's own Notification API is the browser-only fallback and it only fires when the tab is hidden. */
export function notifyWorkflowRun(p: WorkflowRunNotification): void {
  if (!notificationsEnabled('workflow')) return;
  const bridge = typeof window !== 'undefined' ? window.openswarm : undefined;
  if (!bridge?.notify) {
    notifyAgentCompletion({
      sessionId: p.sessionId || p.workflowId,
      sessionName: p.workflowTitle || 'Workflow',
      status: p.status === 'success' ? 'completed' : 'error',
    });
    return;
  }
  bridge.notify({
    title: workflowTitleFor(p),
    body: workflowBodyFor(p),
    deepLink: `openswarm://workflow/${p.workflowId}/run/${p.runId || ''}`,
    runId: p.runId,
    workflowId: p.workflowId,
    actions: [
      { text: 'Looks good', outcome: 'ack' },
      { text: 'Re-run', outcome: 'rerun' },
      { text: 'Adjust', outcome: 'edit' },
    ],
  }).catch(() => { /* the OS refused it; the run is still on the canvas */ });
}