PiCloud/crates/manager-core/src/outbox_event_emitter.rs

//! `OutboxEventEmitter` — the real `ServiceEventEmitter` that replaces
//! v1.1.0's `NoopEventEmitter` once the triggers framework lands.
//!
//! On each `emit` (a KV mutation, future doc/file/pubsub event, etc.):
//!   1. Look up matching triggers for the event's (app_id, source, op,
//!      collection) tuple via `TriggerRepo::list_matching_*`.
//!   2. For each match, write one outbox row carrying the event payload
//!      serialized as a `TriggerEvent`.
//!
//! Defaults applied at write time so `OutboxRow.payload` carries
//! everything the dispatcher needs to reconstruct the executor
//! invocation without joining back to the trigger row.
//!
//! Non-KV `ServiceEvent` sources are silently dropped in v1.1.1 — the
//! dispatcher only knows how to fire KV triggers this release. Future
//! sources (docs/files/pubsub) add their own dispatch arm.

use std::sync::Arc;

use async_trait::async_trait;
use picloud_shared::{
    DocsEventOp, EmitError, KvEventOp, SdkCallCx, ServiceEvent, ServiceEventEmitter, TriggerEvent,
};

use crate::outbox_repo::{NewOutboxRow, OutboxRepo, OutboxSourceKind};
use crate::trigger_repo::TriggerRepo;

pub struct OutboxEventEmitter {
    triggers: Arc<dyn TriggerRepo>,
    outbox: Arc<dyn OutboxRepo>,
}

impl OutboxEventEmitter {
    #[must_use]
    pub fn new(triggers: Arc<dyn TriggerRepo>, outbox: Arc<dyn OutboxRepo>) -> Self {
        Self { triggers, outbox }
    }
}

#[async_trait]
impl ServiceEventEmitter for OutboxEventEmitter {
    async fn emit(&self, cx: &SdkCallCx, event: ServiceEvent) -> Result<(), EmitError> {
        match event.source {
            "kv" => self.emit_kv(cx, event).await,
            "docs" => self.emit_docs(cx, event).await,
            // Future sources land here. For now, silently drop — the
            // SDK calls `events.emit(...)` unconditionally for forward
            // compat, so swallowing without an error is correct.
            _ => Ok(()),
        }
    }
}

impl OutboxEventEmitter {
    async fn emit_kv(&self, cx: &SdkCallCx, event: ServiceEvent) -> Result<(), EmitError> {
        let Some(op) = KvEventOp::from_wire(event.op) else {
            return Ok(()); // unknown op — drop quietly
        };
        let Some(collection) = event.collection.clone() else {
            return Ok(()); // KV events always carry a collection — defensively skip
        };
        let key = event.key.clone().unwrap_or_default();

        let matches = self
            .triggers
            .list_matching_kv(cx.app_id, &collection, op)
            .await
            .map_err(|e| EmitError::Unavailable(format!("trigger lookup: {e}")))?;

        if matches.is_empty() {
            return Ok(());
        }

        // Serialize the originating event as a TriggerEvent so the
        // dispatcher can hand it to the script as `ctx.event` without
        // round-tripping back to the trigger row.
        let trigger_event = TriggerEvent::Kv {
            op,
            collection,
            key,
            value: event.payload.clone(),
        };
        let payload = serde_json::to_value(&trigger_event)
            .map_err(|e| EmitError::Rejected(format!("event serialize: {e}")))?;

        for m in matches {
            self.outbox
                .insert(NewOutboxRow {
                    app_id: cx.app_id,
                    source_kind: OutboxSourceKind::Kv,
                    trigger_id: Some(m.trigger_id),
                    script_id: Some(m.script_id),
                    reply_to: None,
                    payload: payload.clone(),
                    origin_principal: cx.principal.as_ref().map(|p| p.user_id),
                    trigger_depth: cx.trigger_depth.saturating_add(1),
                    root_execution_id: Some(cx.root_execution_id),
                })
                .await
                .map_err(|e| EmitError::Unavailable(format!("outbox insert: {e}")))?;
        }
        Ok(())
    }

    /// v1.1.2. Mirrors `emit_kv` — fan out a docs mutation across
    /// matching docs triggers + write one outbox row each. The
    /// `prev_data` change-data-capture surface is preserved from the
    /// `ServiceEvent.old_payload` field (set by `DocsServiceImpl` on
    /// update and delete; `None` for create).
    async fn emit_docs(&self, cx: &SdkCallCx, event: ServiceEvent) -> Result<(), EmitError> {
        let Some(op) = DocsEventOp::from_wire(event.op) else {
            return Ok(());
        };
        let Some(collection) = event.collection.clone() else {
            return Ok(());
        };
        let id = event.key.clone().unwrap_or_default();

        let matches = self
            .triggers
            .list_matching_docs(cx.app_id, &collection, op)
            .await
            .map_err(|e| EmitError::Unavailable(format!("trigger lookup: {e}")))?;

        if matches.is_empty() {
            return Ok(());
        }

        let trigger_event = TriggerEvent::Docs {
            op,
            collection,
            id,
            data: event.payload.clone(),
            prev_data: event.old_payload.clone(),
        };
        let payload = serde_json::to_value(&trigger_event)
            .map_err(|e| EmitError::Rejected(format!("event serialize: {e}")))?;

        for m in matches {
            self.outbox
                .insert(NewOutboxRow {
                    app_id: cx.app_id,
                    source_kind: OutboxSourceKind::Docs,
                    trigger_id: Some(m.trigger_id),
                    script_id: Some(m.script_id),
                    reply_to: None,
                    payload: payload.clone(),
                    origin_principal: cx.principal.as_ref().map(|p| p.user_id),
                    trigger_depth: cx.trigger_depth.saturating_add(1),
                    root_execution_id: Some(cx.root_execution_id),
                })
                .await
                .map_err(|e| EmitError::Unavailable(format!("outbox insert: {e}")))?;
        }
        Ok(())
    }
}