iris_chat_core/core/

notifications.rs

//! Single source of truth for "should this chat raise a desktop / shell
//! notification?". All non-iOS-APNS callers (foreground macOS, Linux GTK,
//! Windows WPF, Android in-app) compare two consecutive AppState snapshots
//! and call [`decide_notifications`] to get the resulting candidate list.
//!
//! iOS APNS resolves encrypted events against persisted state in its service
//! extension. Its notification-filtering entitlement permits suppression; the
//! background resolver applies the same mention exception after decryption.

use std::collections::HashMap;

use crate::state::{ChatThreadSnapshot, PreferencesSnapshot, Router, Screen};

#[derive(uniffi::Record, Clone, Debug, PartialEq, Eq)]
pub struct NotificationCandidate {
    pub chat_id: String,
    #[uniffi(default = None)]
    pub message_id: Option<String>,
    pub title: String,
    pub body: String,
}

/// Decide which chats should raise a notification given the previous and
/// next snapshots of the chat list. Suppression rules — all must pass:
///
/// 1. `preferences.desktop_notifications_enabled` is true.
/// 2. The chat is not muted, or it is a group mentioning this account and the
///    user permits mention notifications while muted.
/// 3. The ordinary candidate is known-incoming, or a muted group has a new
///    unread incoming mention ID. A later ordinary/outgoing message in the
///    same coalesced snapshot cannot hide that mention.
/// 4. The chat is not currently open with the app foregrounded.
/// 5. For ordinary chats, `unread_count` strictly increased. Muted group
///    mentions compare exact unread IDs, so concurrent reads cannot hide them.
pub fn decide_notifications(
    previous_chats: &[ChatThreadSnapshot],
    next_chats: &[ChatThreadSnapshot],
    preferences: &PreferencesSnapshot,
    app_foreground: bool,
    open_chat_id: Option<&str>,
) -> Vec<NotificationCandidate> {
    if !preferences.desktop_notifications_enabled {
        return Vec::new();
    }

    let previous_by_id: HashMap<&str, &ChatThreadSnapshot> = previous_chats
        .iter()
        .map(|c| (c.chat_id.as_str(), c))
        .collect();

    let suppressing_open_chat = if app_foreground { open_chat_id } else { None };

    let mut out = Vec::new();
    for chat in next_chats {
        if suppressing_open_chat == Some(chat.chat_id.as_str()) {
            continue;
        }
        let previous = previous_by_id.get(chat.chat_id.as_str()).copied();
        let mention = if chat.is_muted {
            if chat.kind != crate::state::ChatKind::Group
                || preferences.silent_mention_chat_ids.contains(&chat.chat_id)
            {
                continue;
            }
            let mention = chat.unread_mention_messages.iter().rev().find(|mention| {
                !previous.is_some_and(|old| {
                    old.unread_mention_messages
                        .iter()
                        .any(|item| item.message_id == mention.message_id)
                })
            });
            let Some(mention) = mention else {
                continue;
            };
            Some(mention)
        } else {
            if chat.unread_count <= previous.map_or(0, |chat| chat.unread_count) {
                continue;
            }
            None
        };
        if mention.is_none() && chat.last_message_is_outgoing != Some(false) {
            continue;
        }
        let body = mention
            .map(|mention| mention.preview.as_str())
            .or(chat.last_message_preview.as_deref())
            .map(str::trim)
            .filter(|s| !s.is_empty())
            .map(str::to_owned)
            .unwrap_or_else(|| "New message".to_string());
        out.push(NotificationCandidate {
            chat_id: chat.chat_id.clone(),
            message_id: mention
                .map(|mention| mention.message_id.clone())
                .or_else(|| chat.last_message_id.clone()),
            title: chat.display_name.clone(),
            body,
        });
    }
    out
}

/// Pull the topmost chat id out of a router stack, falling back to the
/// default screen. Mirrors what every shell does inline so they don't
/// each have to reach into `Screen::Chat` themselves.
pub fn active_chat_id(router: &Router) -> Option<String> {
    if let Some(id) = router.screen_stack.iter().rev().find_map(screen_chat_id) {
        return Some(id);
    }
    screen_chat_id(&router.default_screen)
}

fn screen_chat_id(screen: &Screen) -> Option<String> {
    match screen {
        Screen::Chat { chat_id } => Some(chat_id.clone()),
        _ => None,
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::state::ChatKind;

    fn chat(
        id: &str,
        unread: u64,
        last_is_outgoing: Option<bool>,
        muted: bool,
        preview: Option<&str>,
    ) -> ChatThreadSnapshot {
        ChatThreadSnapshot {
            social_connection: None,
            chat_id: id.to_string(),
            kind: ChatKind::Direct,
            display_name: format!("name-{id}"),
            nickname: None,
            contact_note: None,
            profile_name: None,
            subtitle: None,
            picture_url: None,
            about: None,
            member_count: 0,
            last_message_preview: preview.map(str::to_string),
            last_message_at_secs: Some(100),
            last_message_is_outgoing: last_is_outgoing,
            last_message_id: None,
            unread_mention_messages: Vec::new(),
            last_message_delivery: None,
            unread_count: unread,
            is_typing: false,
            is_muted: muted,
            is_pinned: false,
            draft: String::new(),
            is_request: false,
        }
    }

    fn prefs(enabled: bool) -> PreferencesSnapshot {
        PreferencesSnapshot {
            desktop_notifications_enabled: enabled,
            ..PreferencesSnapshot::default()
        }
    }

    #[test]
    fn fires_when_unread_strictly_increases() {
        let prev = vec![chat("a", 0, Some(false), false, Some("hi"))];
        let next = vec![chat("a", 1, Some(false), false, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(true), false, None);
        assert_eq!(out.len(), 1);
        assert_eq!(out[0].chat_id, "a");
        assert_eq!(out[0].body, "hi");
    }

    #[test]
    fn falls_back_to_new_message_when_preview_empty() {
        let prev = vec![chat("a", 0, Some(false), false, None)];
        let next = vec![chat("a", 1, Some(false), false, Some("   "))];
        let out = decide_notifications(&prev, &next, &prefs(true), false, None);
        assert_eq!(out[0].body, "New message");
    }

    #[test]
    fn suppresses_muted_chat() {
        let prev = vec![chat("a", 0, Some(false), true, Some("hi"))];
        let next = vec![chat("a", 1, Some(false), true, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(true), false, None);
        assert!(out.is_empty());
    }

    #[test]
    fn mentions_only_override_muted_groups_with_notifications_enabled() {
        let mut next = chat("group:friends", 1, Some(false), true, Some("Hi @Alex"));
        next.unread_mention_messages
            .push(crate::state::MentionNotificationSnapshot {
                message_id: "mention".into(),
                preview: "Hi @Alex".into(),
            });
        assert!(
            decide_notifications(&[], &[next.clone()], &prefs(true), false, None).is_empty(),
            "direct chats stay muted"
        );
        next.kind = crate::state::ChatKind::Group;
        assert_eq!(
            decide_notifications(&[], &[next.clone()], &prefs(true), false, None).len(),
            1
        );
        let mut opted_out = prefs(true);
        opted_out.silent_mention_chat_ids.push(next.chat_id.clone());
        assert!(decide_notifications(&[], &[next.clone()], &opted_out, false, None).is_empty());
        assert!(decide_notifications(&[], &[next.clone()], &prefs(false), false, None).is_empty());
        assert!(decide_notifications(
            &[],
            &[next.clone()],
            &prefs(true),
            true,
            Some(&next.chat_id)
        )
        .is_empty());
        next.last_message_is_outgoing = Some(true);
        assert_eq!(
            decide_notifications(&[], &[next.clone()], &prefs(true), false, None).len(),
            1,
            "a later outgoing message cannot hide a new incoming mention"
        );
        next.unread_mention_messages.clear();
        assert!(decide_notifications(&[], &[next], &prefs(true), false, None).is_empty());
    }

    #[test]
    fn suppresses_outgoing_message() {
        let prev = vec![chat("a", 0, Some(true), false, Some("hi"))];
        let next = vec![chat("a", 1, Some(true), false, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(true), false, None);
        assert!(out.is_empty());
    }

    #[test]
    fn suppresses_unknown_direction_message() {
        let prev = vec![chat("a", 0, None, false, Some("hi"))];
        let next = vec![chat("a", 1, None, false, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(true), false, None);
        assert!(out.is_empty());
    }

    #[test]
    fn suppresses_when_chat_open_and_app_foreground() {
        let prev = vec![chat("a", 0, Some(false), false, Some("hi"))];
        let next = vec![chat("a", 1, Some(false), false, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(true), true, Some("a"));
        assert!(out.is_empty());
    }

    #[test]
    fn fires_when_chat_open_but_app_backgrounded() {
        let prev = vec![chat("a", 0, Some(false), false, Some("hi"))];
        let next = vec![chat("a", 1, Some(false), false, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(true), false, Some("a"));
        assert_eq!(out.len(), 1);
    }

    #[test]
    fn fires_when_app_foreground_but_a_different_chat_open() {
        let prev = vec![chat("a", 0, Some(false), false, Some("hi"))];
        let next = vec![chat("a", 1, Some(false), false, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(true), true, Some("b"));
        assert_eq!(out.len(), 1);
    }

    #[test]
    fn suppresses_when_unread_unchanged_or_decreased() {
        let prev = vec![chat("a", 2, Some(false), false, Some("hi"))];
        let next = vec![chat("a", 1, Some(false), false, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(true), false, None);
        assert!(out.is_empty());
    }

    #[test]
    fn returns_empty_when_preference_disabled() {
        let prev = vec![chat("a", 0, Some(false), false, Some("hi"))];
        let next = vec![chat("a", 1, Some(false), false, Some("hi"))];
        let out = decide_notifications(&prev, &next, &prefs(false), false, None);
        assert!(out.is_empty());
    }

    #[test]
    fn treats_new_chat_as_previous_unread_zero() {
        let prev: Vec<ChatThreadSnapshot> = Vec::new();
        let next = vec![chat("a", 1, Some(false), false, Some("hello"))];
        let out = decide_notifications(&prev, &next, &prefs(true), false, None);
        assert_eq!(out.len(), 1);
    }

    #[test]
    fn active_chat_id_returns_topmost_chat_screen() {
        let router = Router {
            default_screen: Screen::ChatList,
            screen_stack: vec![Screen::Chat {
                chat_id: "x".to_string(),
            }],
        };
        assert_eq!(active_chat_id(&router), Some("x".to_string()));
    }

    #[test]
    fn active_chat_id_falls_back_to_default_screen() {
        let router = Router {
            default_screen: Screen::Chat {
                chat_id: "default-x".to_string(),
            },
            screen_stack: vec![],
        };
        assert_eq!(active_chat_id(&router), Some("default-x".to_string()));
    }

    #[test]
    fn active_chat_id_none_when_not_on_a_chat_screen() {
        let router = Router {
            default_screen: Screen::ChatList,
            screen_stack: vec![Screen::Settings],
        };
        assert_eq!(active_chat_id(&router), None);
    }
}