git-next/crates/webhook-actor/src/router.rs

//
use std::collections::BTreeMap;

use actix::prelude::*;
use derive_more::Constructor;
use git_next_config::{ForgeAlias, RepoAlias};
use git_next_repo_actor::messages::WebhookNotification;
use tracing::info;

pub struct WebhookRouter {
    span: tracing::Span,
    recipients: BTreeMap<ForgeAlias, BTreeMap<RepoAlias, Recipient<WebhookNotification>>>,
}
impl Default for WebhookRouter {
    fn default() -> Self {
        Self::new()
    }
}
impl WebhookRouter {
    pub fn new() -> Self {
        let span = tracing::info_span!("WebhookRouter");
        Self {
            span,
            recipients: Default::default(),
        }
    }
}
impl Actor for WebhookRouter {
    type Context = Context<Self>;
}

impl Handler<WebhookNotification> for WebhookRouter {
    type Result = ();

    fn handle(&mut self, msg: WebhookNotification, _ctx: &mut Self::Context) -> Self::Result {
        let _gaurd = self.span.enter();
        let forge_alias = msg.forge_alias();
        let repo_alias = msg.repo_alias();
        tracing::debug!(forge = %forge_alias, repo = %repo_alias, "Router...");
        let Some(forge_repos) = self.recipients.get(forge_alias) else {
            tracing::warn!(forge = %forge_alias, "No forge repos found");
            return;
        };
        let Some(recipient) = forge_repos.get(repo_alias) else {
            tracing::debug!(forge = %forge_alias, repo = %repo_alias, "No recipient found");
            return;
        };
        recipient.do_send(msg);
    }
}

#[derive(Message, Constructor)]
#[rtype(result = "()")]
pub struct AddWebhookRecipient {
    pub forge_alias: ForgeAlias,
    pub repo_alias: RepoAlias,
    pub recipient: Recipient<WebhookNotification>,
}
impl Handler<AddWebhookRecipient> for WebhookRouter {
    type Result = ();

    fn handle(&mut self, msg: AddWebhookRecipient, _ctx: &mut Self::Context) -> Self::Result {
        let _gaurd = self.span.enter();
        info!(forge = %msg.forge_alias, repo = %msg.repo_alias, "Register Recipient");
        if !self.recipients.contains_key(&msg.forge_alias) {
            self.recipients
                .insert(msg.forge_alias.clone(), BTreeMap::new());
        }
        self.recipients
            .get_mut(&msg.forge_alias)
            .map(|repos| repos.insert(msg.repo_alias, msg.recipient));
    }
}