Add notification auto-stop countdown

Show the 30-second auto-stop countdown in the compact notification, animate the Stop button progress, and stop listening when the notification times out.

Author: ComputelessComputer

apps/desktop/src/services/event-listeners.tsx

@@ -69,15 +69,15 @@ function shouldAutoStartNotificationSession(
 }
 
 function handleAutoStopEndedNotification(
-  type: "notification_confirm" | "notification_accept",
+  type: "notification_confirm" | "notification_accept" | "notification_timeout",
   key: string,
 ): boolean {
   const sessionId = parseAutoStopEndedNotificationKey(key);
   if (!sessionId) {
     return false;
   }
 
-  if (type !== "notification_accept") {
+  if (type === "notification_confirm") {
     return true;
   }
 
@@ -362,12 +362,17 @@ function useNotificationEvents() {
       .listen(({ payload }) => {
         if (
           payload.type === "notification_confirm" ||
-          payload.type === "notification_accept"
+          payload.type === "notification_accept" ||
+          payload.type === "notification_timeout"
         ) {
           if (handleAutoStopEndedNotification(payload.type, payload.key)) {
             return;
           }
 
+          if (payload.type === "notification_timeout") {
+            return;
+          }
+
           const eventId =
             payload.source?.type === "calendar_event"
               ? payload.source.event_id

apps/desktop/src/stt/auto-stop-notification.ts

@@ -1,5 +1,6 @@
 export const AUTO_STOP_ENDED_NOTIFICATION_KEY_PREFIX =
   "auto-stop-ended:" as const;
+export const AUTO_STOP_CONFIRM_TIMEOUT_SECONDS = 30;
 
 const AUTO_STOP_ENDED_NOTIFICATION_KEY_NONCE_SEPARATOR = ":prompt:";
 

apps/desktop/src/stt/contexts.test.tsx

@@ -887,8 +887,8 @@ describe("ListenerProvider detect events", () => {
       expect(notification).toEqual({
         key: expect.stringContaining("auto-stop-ended:session-1"),
         title: "Did your meeting end?",
-        message: `${browser.name} stopped using the microphone before the scheduled end time.`,
-        timeout: { secs: 60, nanos: 0 },
+        message: "Anarlog will stop listening in 30 seconds.",
+        timeout: { secs: 30, nanos: 0 },
         source: null,
         start_time: null,
         participants: null,

apps/desktop/src/stt/contexts.tsx

@@ -11,7 +11,10 @@ import {
   type NotificationIcon,
 } from "@hypr/plugin-notification";
 
-import { createAutoStopEndedNotificationKey } from "./auto-stop-notification";
+import {
+  AUTO_STOP_CONFIRM_TIMEOUT_SECONDS,
+  createAutoStopEndedNotificationKey,
+} from "./auto-stop-notification";
 
 import { getSessionEventById } from "~/session/utils";
 import * as main from "~/store/tinybase/store/main";
@@ -261,11 +264,6 @@ function getAutoStopActiveCheckAppIds(
   return [...new Set([...candidateAppIds, ...unreliableTriggerAppIds])];
 }
 
-function getStoppedAppLabel(app: MicApp | null) {
-  const name = app ? getNotificationAppName(app).trim() : "";
-  return name || "The meeting app";
-}
-
 function showMeetingEndedPrompt({
   sessionId,
   stoppedTriggerAppIds,
@@ -280,8 +278,8 @@ function showMeetingEndedPrompt({
   void notificationCommands.showNotification({
     key: createAutoStopEndedNotificationKey(sessionId),
     title: "Did your meeting end?",
-    message: `${getStoppedAppLabel(app)} stopped using the microphone before the scheduled end time.`,
-    timeout: { secs: 60, nanos: 0 },
+    message: `Anarlog will stop listening in ${AUTO_STOP_CONFIRM_TIMEOUT_SECONDS} seconds.`,
+    timeout: { secs: AUTO_STOP_CONFIRM_TIMEOUT_SECONDS, nanos: 0 },
     source: null,
     start_time: null,
     participants: null,

crates/notification-macos/swift-lib/src/Models.swift

@@ -230,6 +230,10 @@ struct NotificationPayload: Codable {
     return actionVariant == .destructive
   }
 
+  var showsStopCountdown: Bool {
+    return isDestructiveAction && actionLabel == "Stop" && timeoutSeconds > 0
+  }
+
   var hasOptions: Bool {
     guard let options = options else { return false }
     return !options.isEmpty

crates/notification-macos/swift-lib/src/NotificationButtons.swift

@@ -215,6 +215,8 @@ class CompactActionButton: ActionButton {
     }
   }
   var onProgressComplete: (() -> Void)?
+  private var progressBaseBackgroundColor = Colors.compactActionButtonElapsedBg
+  private var progressFillColor = Colors.compactActionButtonRemainingBg
 
   override init(frame frameRect: NSRect) {
     super.init(frame: frameRect)
@@ -237,7 +239,7 @@ class CompactActionButton: ActionButton {
       layer?.cornerCurve = .continuous
     }
 
-    progressLayer.backgroundColor = Colors.compactActionButtonRemainingBg
+    progressLayer.backgroundColor = progressFillColor
     progressLayer.anchorPoint = CGPoint(x: 0, y: 0.5)
     progressLayer.isHidden = true
     layer?.addSublayer(progressLayer)
@@ -258,6 +260,12 @@ class CompactActionButton: ActionButton {
     }
   }
 
+  func configureDestructiveCountdownStyle() {
+    progressBaseBackgroundColor = Colors.actionButtonDestructivePressedBg
+    progressFillColor = Colors.actionButtonDestructiveBg
+    progressLayer.backgroundColor = progressFillColor
+  }
+
   private func syncProgressLayerFrame() {
     let width = isPaused ? progressLayerFullWidth * progressRatio : progressLayerFullWidth
     progressLayer.frame = CGRect(x: 0, y: 0, width: width, height: bounds.height)
@@ -317,7 +325,8 @@ class CompactActionButton: ActionButton {
     isCountdownActive = true
     progressRatio = 1.0
 
-    layer?.backgroundColor = Colors.compactActionButtonElapsedBg
+    layer?.backgroundColor = progressBaseBackgroundColor
+    progressLayer.backgroundColor = progressFillColor
     progressLayer.isHidden = false
     progressLayer.removeAllAnimations()
     syncProgressLayerFrame()
@@ -363,7 +372,7 @@ class CompactActionButton: ActionButton {
     clearProgressState()
 
     if showsProgress {
-      layer?.backgroundColor = Colors.buttonNormalBg
+      layer?.backgroundColor = progressBaseBackgroundColor
     }
 
     CATransaction.begin()

crates/notification-macos/swift-lib/src/NotificationInstance.swift

@@ -23,6 +23,8 @@ class NotificationInstance {
   var meetingStartTime: Date?
   weak var timerLabel: NSTextField?
   weak var compactMessageLabel: NSTextField?
+  weak var stopCountdownLabel: NSTextField?
+  var stopCountdownTimer: Timer?
 
   init(
     payload: NotificationPayload, panel: NSPanel, clickableView: ClickableView, creationIndex: Int
@@ -60,6 +62,11 @@ class NotificationInstance {
     timerLabel = nil
   }
 
+  func bindStopCountdownLabel(_ label: NSTextField) {
+    stopCountdownLabel = label
+    updateStopCountdownLabel(remainingSeconds: payload.timeoutSeconds)
+  }
+
   func startScheduleUpdates() {
     guard let meetingStartTime else { return }
     updateScheduleLabels()
@@ -75,8 +82,11 @@ class NotificationInstance {
   func stopScheduleUpdates() {
     countdownTimer?.invalidate()
     countdownTimer = nil
+    stopCountdownTimer?.invalidate()
+    stopCountdownTimer = nil
     timerLabel = nil
     compactMessageLabel = nil
+    stopCountdownLabel = nil
   }
 
   private func updateScheduleLabels() {
@@ -110,6 +120,7 @@ class NotificationInstance {
     remainingDismissSeconds = timeoutSeconds
     dismissStartTime = Date()
     scheduleDismissTimer(after: timeoutSeconds)
+    startStopCountdownUpdates()
 
     if let compactActionButton {
       compactActionButton.startProgress(duration: timeoutSeconds)
@@ -125,6 +136,9 @@ class NotificationInstance {
     }
     dismissTimer?.invalidate()
     dismissTimer = nil
+    stopCountdownTimer?.invalidate()
+    stopCountdownTimer = nil
+    updateStopCountdownLabel(remainingSeconds: remainingDismissSeconds)
 
     if let compactActionButton {
       compactActionButton.pauseProgress()
@@ -135,6 +149,7 @@ class NotificationInstance {
     guard timeoutSeconds > 0, remainingDismissSeconds > 0 else { return }
     dismissStartTime = Date()
     scheduleDismissTimer(after: remainingDismissSeconds)
+    startStopCountdownUpdates()
 
     if let compactActionButton {
       compactActionButton.resumeProgress()
@@ -148,6 +163,7 @@ class NotificationInstance {
     remainingDismissSeconds = timeoutSeconds
     dismissStartTime = Date()
     scheduleDismissTimer(after: timeoutSeconds)
+    startStopCountdownUpdates()
 
     if let compactActionButton {
       compactActionButton.startProgress(duration: timeoutSeconds)
@@ -157,6 +173,8 @@ class NotificationInstance {
   func dismiss() {
     dismissTimer?.invalidate()
     dismissTimer = nil
+    stopCountdownTimer?.invalidate()
+    stopCountdownTimer = nil
     dismissStartTime = nil
     remainingDismissSeconds = 0
     compactActionButton?.resetProgress()
@@ -191,8 +209,40 @@ class NotificationInstance {
     }
   }
 
+  func stopCountdownText(_ remainingSeconds: Double) -> String {
+    let seconds = max(0, Int(ceil(remainingSeconds)))
+    return "Anarlog will stop listening in \(seconds) seconds."
+  }
+
+  private func startStopCountdownUpdates() {
+    guard stopCountdownLabel != nil else { return }
+    updateStopCountdownLabel()
+
+    stopCountdownTimer?.invalidate()
+    stopCountdownTimer = Timer.scheduledTimer(withTimeInterval: 1.0, repeats: true) {
+      [weak self] _ in
+      self?.updateStopCountdownLabel()
+    }
+  }
+
+  private func updateStopCountdownLabel(remainingSeconds: Double? = nil) {
+    guard stopCountdownLabel != nil else { return }
+
+    let remaining: Double
+    if let remainingSeconds {
+      remaining = remainingSeconds
+    } else if let dismissStartTime {
+      remaining = max(0, remainingDismissSeconds - Date().timeIntervalSince(dismissStartTime))
+    } else {
+      remaining = remainingDismissSeconds
+    }
+
+    stopCountdownLabel?.stringValue = stopCountdownText(remaining)
+  }
+
   deinit {
     countdownTimer?.invalidate()
     dismissTimer?.invalidate()
+    stopCountdownTimer?.invalidate()
   }
 }

crates/notification-macos/swift-lib/src/NotificationManager+CompactView.swift

@@ -14,7 +14,10 @@ extension NotificationManager {
     let actionLabel = notification.payload.actionLabel ?? "Open Anarlog"
     if notification.payload.isDestructiveAction {
       actionButton.configureDestructiveAction(label: actionLabel)
-      actionButton.showsProgress = false
+      actionButton.showsProgress = notification.payload.showsStopCountdown
+      if notification.payload.showsStopCountdown {
+        actionButton.configureDestructiveCountdownStyle()
+      }
     } else {
       actionButton.title = actionLabel
     }
@@ -64,10 +67,14 @@ extension NotificationManager {
 
     textStack.addArrangedSubview(titleLabel)
 
-    let compactMessage =
-      notification.meetingStartTime != nil
-      ? "Starting soon"
-      : notification.payload.message
+    let compactMessage: String
+    if notification.meetingStartTime != nil {
+      compactMessage = "Starting soon"
+    } else if notification.payload.showsStopCountdown {
+      compactMessage = notification.stopCountdownText(notification.payload.timeoutSeconds)
+    } else {
+      compactMessage = notification.payload.message
+    }
     if !compactMessage.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty {
       let bodyLabel = NSTextField(labelWithString: compactMessage)
       bodyLabel.font = NSFont.systemFont(ofSize: Fonts.bodySize, weight: Fonts.bodyWeight)
@@ -82,6 +89,8 @@ extension NotificationManager {
 
       if notification.meetingStartTime != nil {
         notification.bindCompactMessageLabel(bodyLabel)
+      } else if notification.payload.showsStopCountdown {
+        notification.bindStopCountdownLabel(bodyLabel)
       }
     }