//! 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.events.live.publish(&crate::events::topic::user(id), "notification");

        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, title, url, actor).await;

            }

        }

    }

}

async fn email(app: &Shared, to: &Account, title: &str, url: &str, actor: &str) {

    let unsubscribe = format!("{}/settings/notifications?unsubscribe={}", app.config.base_url(), to.unsubscribe_key);

    let body = format!("{actor}: {title}\n\n{}{url}\n\nStop these emails: {unsubscribe}\n", app.config.base_url());

    if let Err(e) = crate::mail::send(app, to.name(), &to.email, title, &body).await {

        tracing::warn!("email to {}: {e}", to.email);

    }

}

#[cfg(test)]

mod tests {

    use super::*;

    #[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());

    }

}