mod.rs 84.3%

91 of 108 lines run, 18 of 32 functions · run #516 on main · run missed

//! Notifications (hub spec §10): who to tell about an event, the inbox
//! row, and email when the account asked for it and SMTP is configured.

use crate::db::{self, accounts::Account, notify as ndb};
use crate::Shared;
use std::collections::BTreeSet;

tokio::task_local! {
    /// The name of the agent-marked token this request came with, if any
    /// (oxygenhub#183): what an agent does with it is told to the token's
    /// owner too, though the owner is the actor.
    pub static AGENT: Option<String>;
}

/// The agent token acting now, when there is one.
pub fn agent_acting() -> Option<String> {
    AGENT.try_with(|a| a.clone()).ok().flatten()
}

/// An event worth telling people about.
pub struct Event<'a> {
    pub kind: &'a str,
    pub title: String,
    pub url: String,
    pub actor: &'a Account,
    pub repo_id: i64,
    /// `issue` or `proposal` and its id, for thread subscribers.
    pub thread: Option<(&'a str, i64)>,
    /// Bodies to scan for `@mentions`.
    pub mention_text: &'a str,
    /// Accounts to tell whatever their subscriptions (author, assignees).
    pub direct: Vec<i64>,
}

/// `@username` tokens in a text, unique.
pub fn mentions(text: &str) -> Vec<String> {
    let re = regex::Regex::new(r"(?:^|[^\w/])@([A-Za-z0-9][A-Za-z0-9._-]{0,63})").unwrap();
    let mut out = BTreeSet::new();
    for c in re.captures_iter(text) {
        out.insert(c[1].trim_end_matches('.').to_string());
    }
    out.into_iter().collect()
}

/// Deliver an event: subscribers of the thread, watchers of the
/// repository, mentioned accounts and direct targets, never the actor.
pub async fn deliver(app: &Shared, ev: Event<'_>) {
    let pool = &app.db;
    // Participants: the thread's subscribers, the mentioned, direct targets.
    // Watchers of the repository who are none of those are only watching.
    let mut participants: BTreeSet<i64> = BTreeSet::new();
    if let Some((kind, id)) = ev.thread {
        if let Ok(subs) = ndb::subscribers(pool, kind, id).await {
            participants.extend(subs);
        }
        // Acting on a thread subscribes the actor to it.
        ndb::subscribe(pool, ev.actor.id, kind, id).await.ok();
    }
    for name in mentions(ev.mention_text) {
        if let Ok(Some(a)) = db::accounts::account_by_name(pool, &name).await {
            participants.insert(a.id);
        }
    }
    participants.extend(ev.direct.iter().copied());
    let mut targets: Vec<(i64, ndb::Reason)> = participants.iter().map(|id| (*id, ndb::Reason::Participating)).collect();
    if let Ok(w) = ndb::watchers(pool, ev.repo_id).await {
        targets.extend(w.into_iter().filter(|id| !participants.contains(id)).map(|id| (id, ndb::Reason::Watching)));
    }
    // Never the actor, unless an agent acted with their token: then they
    // hear what their agent did.
    let agent = agent_acting();
    if agent.is_none() {
        targets.retain(|(id, _)| *id != ev.actor.id);
    }
    let actor = match &agent {
        Some(token) => format!("{} (agent token {token})", ev.actor.username),
        None => ev.actor.username.clone(),
    };
    tell(app, ev.kind, &ev.title, &ev.url, &actor, (ev.repo_id > 0).then_some(ev.repo_id), targets).await;
}

/// Tell each of `targets` about something: an inbox row, and an email
/// where the account asked for them and has proven its address.
pub async fn tell(app: &Shared, kind: &str, title: &str, url: &str, actor: &str, repo_id: Option<i64>, targets: Vec<(i64, ndb::Reason)>) {
    let pool = &app.db;
    for (id, reason) in targets {
        ndb::add(pool, ndb::New { account_id: id, repo_id, reason, kind, title, url, actor }).await.ok();
        app.bus.live(&crate::events::topic::user(id), "notification");
        // And to the account's phones, sealed (oxygenhub-android#16).
        let link = format!("{}{url}", app.config.base_url().trim_end_matches('/'));
        crate::push::send(app, id, crate::push::Alert { kind: kind.to_string(), title: subject(kind, title), url: link });
        if let Ok(Some(a)) = db::accounts::account_by_id(pool, id).await {
            // Only to addresses the account has proven it reads (hub spec §5.1.1).
            if a.email_notifications && a.email_verified && !a.email.is_empty() && crate::mail::enabled(app) {
                email(app, &a, kind, title, url, actor).await;
            }
        }
    }
}

/// What a kind of notification is called in an email's subject, with its
/// emoji, so a glance at the inbox says what happened (Andy, 2026-10-14).
pub fn label(kind: &str) -> Option<&'static str> {
    Some(match kind {
        "flow.failed" => "💥 Build Broken",
        "flow.fixed" => "🎉 Build Success",
        "flow.runner_drained" => "🛑 Runner Drained",
        "backup" => "⚠️ Backup Failed",
        "backup.recovered" => "✅ Backup Recovered",
        "backup.summary" => "📦 Daily Backup Summary",
        "proposal.opened" => "🔀 New Proposal",
        "proposal.review_requested" => "👀 Review Requested",
        "proposal.review" => "📝 New Review",
        "proposal.landed" => "🚀 Proposal Landed",
        "proposal.comment" | "issue.comment" | "drop.comment" => "💬 New Comment",
        "issue.opened" => "🐛 New Issue",
        "issue.assigned" => "📌 Issue Assigned",
        "issue.closed" => "✔️ Issue Closed",
        "account.registered" => "👋 New Sign-up",
        _ => return None,
    })
}

/// A notification's email subject: its label, then what happened.
pub fn subject(kind: &str, title: &str) -> String {
    match label(kind) {
        Some(l) => format!("{l}: {title}"),
        None => title.to_string(),
    }
}

async fn email(app: &Shared, to: &Account, kind: &str, title: &str, url: &str, actor: &str) {
    let unsubscribe = format!("{}/settings/notifications?unsubscribe={}", app.config.base_url(), to.unsubscribe_key);
    let link = format!("{}{url}", app.config.base_url());
    let body = format!("{actor}: {title}\n\n{link}\n\nStop these emails: {unsubscribe}\n");
    let action = crate::mail::Action { label: "Open it", url: &link };
    let subject = subject(kind, title);
    if let Err(e) = crate::mail::send_letter(app, to.name(), &to.email, &subject, &body, Some(action), Some(&unsubscribe)).await {
        tracing::warn!("email to {}: {e}", to.email);
        // Sent again from the work queue (#276), when mail is on at all.
        if crate::mail::enabled(app) {
            crate::queue::queue_mail(app, to.name(), &to.email, &subject, &body, Some(("Open it", &link)), Some(&unsubscribe)).await;
        }
    }
}

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

    #[test]
    fn subjects_say_what_happened() {
        assert_eq!(
            subject("flow.failed", "andy/demo: run #4 on main failed (test)"),
            "💥 Build Broken: andy/demo: run #4 on main failed (test)"
        );
        assert_eq!(
            subject("flow.fixed", "andy/demo: run #5 on main passed: fixed"),
            "🎉 Build Success: andy/demo: run #5 on main passed: fixed"
        );
        assert_eq!(subject("backup.summary", "x"), "📦 Daily Backup Summary: x");
        assert_eq!(subject("something.new", "x"), "x", "a kind without a label keeps its title");
    }

    #[test]
    fn finds_mentions() {
        assert_eq!(mentions("cc @andy and @bob. not an email [email protected], nor /@path"), vec!["andy", "bob"]);
        assert!(mentions("nothing here").is_empty());
    }
}

1		//! Notifications (hub spec §10): who to tell about an event, the inbox
2		//! row, and email when the account asked for it and SMTP is configured.
3
4		use crate::db::{self, accounts::Account, notify as ndb};
5		use crate::Shared;
6		use std::collections::BTreeSet;
7
8		tokio::task_local! {
9		/// The name of the agent-marked token this request came with, if any
10		/// (oxygenhub#183): what an agent does with it is told to the token's
11		/// owner too, though the owner is the actor.
12		pub static AGENT: Option<String>;
13		}
14
15		/// The agent token acting now, when there is one.
16	120×	pub fn agent_acting() -> Option<String> {
17	120×	AGENT.try_with(\|a\| a.clone()).ok().flatten()
18	120×	}
19
20		/// An event worth telling people about.
21		pub struct Event<'a> {
22		pub kind: &'a str,
23		pub title: String,
24		pub url: String,
25		pub actor: &'a Account,
26		pub repo_id: i64,
27		/// `issue` or `proposal` and its id, for thread subscribers.
28		pub thread: Option<(&'a str, i64)>,
29		/// Bodies to scan for `@mentions`.
30		pub mention_text: &'a str,
31		/// Accounts to tell whatever their subscriptions (author, assignees).
32		pub direct: Vec<i64>,
33		}
34
35		/// `@username` tokens in a text, unique.
36	122×	pub fn mentions(text: &str) -> Vec<String> {
37	122×	let re = regex::Regex::new(r"(?:^\|[^\w/])@([A-Za-z0-9][A-Za-z0-9._-]{0,63})").unwrap();
38	122×	let mut out = BTreeSet::new();
39	122×	for c in re.captures_iter(text) {
40	3×	out.insert(c[1].trim_end_matches('.').to_string());
41	3×	}
42	122×	out.into_iter().collect()
43	122×	}
44
45		/// Deliver an event: subscribers of the thread, watchers of the
46		/// repository, mentioned accounts and direct targets, never the actor.
47	120×	pub async fn deliver(app: &Shared, ev: Event<'_>) {
48	120×	let pool = &app.db;
49		// Participants: the thread's subscribers, the mentioned, direct targets.
50		// Watchers of the repository who are none of those are only watching.
51	120×	let mut participants: BTreeSet<i64> = BTreeSet::new();
52	120×	if let Some((kind, id)) = ev.thread {
53	98×	if let Ok(subs) = ndb::subscribers(pool, kind, id).await {
54	98×	participants.extend(subs);
55	98×	}
56		// Acting on a thread subscribes the actor to it.
57	98×	ndb::subscribe(pool, ev.actor.id, kind, id).await.ok();
58	22×	}
59	120×	for name in mentions(ev.mention_text) {
60	1×	if let Ok(Some(a)) = db::accounts::account_by_name(pool, &name).await {
61	1×	participants.insert(a.id);
62	1×	}
63		}
64	120×	participants.extend(ev.direct.iter().copied());
65	120×	let mut targets: Vec<(i64, ndb::Reason)> = participants.iter().map(\|id\| (*id, ndb::Reason::Participating)).collect();
66	120×	if let Ok(w) = ndb::watchers(pool, ev.repo_id).await {
67	126×	targets.extend(w.into_iter().filter(\|id\| !participants.contains(id)).map(\|id\| (id, ndb::Reason::Watching)));
68	0×	}
69		// Never the actor, unless an agent acted with their token: then they
70		// hear what their agent did.
71	120×	let agent = agent_acting();
72	120×	if agent.is_none() {
73	133×	targets.retain(\|(id, _)\| *id != ev.actor.id);
74	3×	}
75	120×	let actor = match &agent {
76	3×	Some(token) => format!("{} (agent token {token})", ev.actor.username),
77	117×	None => ev.actor.username.clone(),
78		};
79	120×	tell(app, ev.kind, &ev.title, &ev.url, &actor, (ev.repo_id > 0).then_some(ev.repo_id), targets).await;
80	120×	}
81
82		/// Tell each of `targets` about something: an inbox row, and an email
83		/// where the account asked for them and has proven its address.
84	168×	pub async fn tell(app: &Shared, kind: &str, title: &str, url: &str, actor: &str, repo_id: Option<i64>, targets: Vec<(i64, ndb::Reason)>) {
85	168×	let pool = &app.db;
86	168×	for (id, reason) in targets {
87	101×	ndb::add(pool, ndb::New { account_id: id, repo_id, reason, kind, title, url, actor }).await.ok();
88	101×	app.bus.live(&crate::events::topic::user(id), "notification");
89		// And to the account's phones, sealed (oxygenhub-android#16).
90	101×	let link = format!("{}{url}", app.config.base_url().trim_end_matches('/'));
91	101×	crate::push::send(app, id, crate::push::Alert { kind: kind.to_string(), title: subject(kind, title), url: link });
92	101×	if let Ok(Some(a)) = db::accounts::account_by_id(pool, id).await {
93		// Only to addresses the account has proven it reads (hub spec §5.1.1).
94	101×	if a.email_notifications && a.email_verified && !a.email.is_empty() && crate::mail::enabled(app) {
95	0×	email(app, &a, kind, title, url, actor).await;
96	101×	}
97	0×	}
98		}
99	168×	}
100
101		/// What a kind of notification is called in an email's subject, with its
102		/// emoji, so a glance at the inbox says what happened (Andy, 2026-10-14).
103	120×	pub fn label(kind: &str) -> Option<&'static str> {
104	120×	Some(match kind {
105	120×	"flow.failed" => "💥 Build Broken",
106	109×	"flow.fixed" => "🎉 Build Success",
107	107×	"flow.runner_drained" => "🛑 Runner Drained",
108	106×	"backup" => "⚠️ Backup Failed",
109	103×	"backup.recovered" => "✅ Backup Recovered",
110	102×	"backup.summary" => "📦 Daily Backup Summary",
111	100×	"proposal.opened" => "🔀 New Proposal",
112	84×	"proposal.review_requested" => "👀 Review Requested",
113	80×	"proposal.review" => "📝 New Review",
114	75×	"proposal.landed" => "🚀 Proposal Landed",
115	72×	"proposal.comment" \| "issue.comment" \| "drop.comment" => "💬 New Comment",
116	63×	"issue.opened" => "🐛 New Issue",
117	51×	"issue.assigned" => "📌 Issue Assigned",
118	48×	"issue.closed" => "✔️ Issue Closed",
119	47×	"account.registered" => "👋 New Sign-up",
120	11×	_ => return None,
121		})
122	120×	}
123
124		/// A notification's email subject: its label, then what happened.
125	105×	pub fn subject(kind: &str, title: &str) -> String {
126	105×	match label(kind) {
127	98×	Some(l) => format!("{l}: {title}"),
128	7×	None => title.to_string(),
129		}
130	105×	}
131
132	0×	async fn email(app: &Shared, to: &Account, kind: &str, title: &str, url: &str, actor: &str) {
133	0×	let unsubscribe = format!("{}/settings/notifications?unsubscribe={}", app.config.base_url(), to.unsubscribe_key);
134	0×	let link = format!("{}{url}", app.config.base_url());
135	0×	let body = format!("{actor}: {title}\n\n{link}\n\nStop these emails: {unsubscribe}\n");
136	0×	let action = crate::mail::Action { label: "Open it", url: &link };
137	0×	let subject = subject(kind, title);
138	0×	if let Err(e) = crate::mail::send_letter(app, to.name(), &to.email, &subject, &body, Some(action), Some(&unsubscribe)).await {
139	0×	tracing::warn!("email to {}: {e}", to.email);
140		// Sent again from the work queue (#276), when mail is on at all.
141	0×	if crate::mail::enabled(app) {
142	0×	crate::queue::queue_mail(app, to.name(), &to.email, &subject, &body, Some(("Open it", &link)), Some(&unsubscribe)).await;
143	0×	}
144	0×	}
145	0×	}
146
147		#[cfg(test)]
148		mod tests {
149		use super::*;
150
151		#[test]
152	1×	fn subjects_say_what_happened() {
153	1×	assert_eq!(
154	1×	subject("flow.failed", "andy/demo: run #4 on main failed (test)"),
155		"💥 Build Broken: andy/demo: run #4 on main failed (test)"
156		);
157	1×	assert_eq!(
158	1×	subject("flow.fixed", "andy/demo: run #5 on main passed: fixed"),
159		"🎉 Build Success: andy/demo: run #5 on main passed: fixed"
160		);
161	1×	assert_eq!(subject("backup.summary", "x"), "📦 Daily Backup Summary: x");
162	1×	assert_eq!(subject("something.new", "x"), "x", "a kind without a label keeps its title");
163	1×	}
164
165		#[test]
166	1×	fn finds_mentions() {
167	1×	assert_eq!(mentions("cc @andy and @bob. not an email [email protected], nor /@path"), vec!["andy", "bob"]);
168	1×	assert!(mentions("nothing here").is_empty());
169	1×	}
170		}

`/`	Search
`g` `c`	Code
`g` `h`	History
`g` `b`	Bookmarks
`g` `t`	Trust
`y`	Permalink to this state (files and trees)
`?`	This sheet

Coverage / crates/hub/src/notify/mod.rs 84.3%

Keyboard