backbone-bucket 0.4.0

//! UserQuota Projector (Event Handler)
//!
//! Generated by backbone-schema. Do not edit manually.
//!
//! This projector listens to domain events and updates the projection.

use std::sync::Arc;

use anyhow::Result;
use async_trait::async_trait;
use chrono::Utc;

use super::UserQuotaProjection;
use crate::domain::entity::UserQuota;
use crate::domain::events::{
    UserQuotaCreatedEvent, UserQuotaUpdatedEvent, UserQuotaDeletedEvent,
};

// ============================================================================
// EVENT HANDLER TRAIT
// ============================================================================

/// Trait for handling UserQuota domain events
///
/// All handler methods receive the event sequence number for tracking
/// which events have been processed by the projector.
#[async_trait]
pub trait UserQuotaEventHandler: Send + Sync {
    /// Handle entity created event
    async fn on_created(&self, event: UserQuotaCreatedEvent, sequence: i64) -> Result<()>;
    /// Handle entity updated event
    async fn on_updated(&self, event: UserQuotaUpdatedEvent, sequence: i64) -> Result<()>;
    /// Handle entity deleted event
    async fn on_deleted(&self, event: UserQuotaDeletedEvent, sequence: i64) -> Result<()>;
}

// ============================================================================
// PROJECTION REPOSITORY
// ============================================================================

/// Repository for UserQuota projections
#[async_trait]
pub trait UserQuotaProjectionRepository: Send + Sync {
    async fn save(&self, projection: &UserQuotaProjection) -> Result<()>;
    async fn find_by_id(&self, id: uuid::Uuid) -> Result<Option<UserQuotaProjection>>;
    async fn delete(&self, id: uuid::Uuid) -> Result<()>;
    async fn rebuild_all(&self) -> Result<u64>;
}

// ============================================================================
// PROJECTOR
// ============================================================================

/// Projector that updates UserQuota projection from events
pub struct UserQuotaProjector<R: UserQuotaProjectionRepository> {
    repository: Arc<R>,
}

impl<R: UserQuotaProjectionRepository> UserQuotaProjector<R> {
    /// Create a new projector
    pub fn new(repository: Arc<R>) -> Self {
        Self { repository }
    }

    /// Rebuild all projections from event history
    pub async fn rebuild(&self) -> Result<u64> {
        self.repository.rebuild_all().await
    }

    /// Process a stored event from the event store
    ///
    /// This method dispatches the event to the appropriate handler based on event type.
    /// Returns Ok(true) if the event was processed, Ok(false) if it was skipped.
    pub async fn process_stored_event(&self, event: &crate::infrastructure::event_store::StoredEvent) -> Result<bool>
    where
        Self: UserQuotaEventHandler,
    {
        let sequence = event.sequence;

        match event.event_type.as_str() {
            "UserQuotaCreatedEvent" => {
                let created: UserQuotaCreatedEvent = serde_json::from_value(event.payload.clone())?;
                self.on_created(created, sequence).await?;
                Ok(true)
            }
            "UserQuotaUpdatedEvent" => {
                let updated: UserQuotaUpdatedEvent = serde_json::from_value(event.payload.clone())?;
                self.on_updated(updated, sequence).await?;
                Ok(true)
            }
            "UserQuotaDeletedEvent" => {
                let deleted: UserQuotaDeletedEvent = serde_json::from_value(event.payload.clone())?;
                self.on_deleted(deleted, sequence).await?;
                Ok(true)
            }
            _ => Ok(false), // Event type not handled by this projector
        }
    }
}

#[async_trait]
impl<R: UserQuotaProjectionRepository> UserQuotaEventHandler for UserQuotaProjector<R> {

    async fn on_created(&self, event: UserQuotaCreatedEvent, sequence: i64) -> Result<()> {
        let mut projection = UserQuotaProjection::new(
            event.id,
            event.user_id,
            event.limit_bytes,
            event.used_bytes,
            event.file_count,
            event.max_file_size,
            event.max_file_count,
            event.tier,
            event.quota_status,
            event.warning_threshold_percent,
            event.last_warning_sent_at,
            event.peak_usage_bytes,
            event.peak_usage_at,
            event.metadata
        );

        // Track the event sequence for this projection
        projection.apply_event(sequence);

        self.repository.save(&projection).await
    }

    async fn on_updated(&self, event: UserQuotaUpdatedEvent, sequence: i64) -> Result<()> {
        if let Some(mut projection) = self.repository.find_by_id(event.id).await? {
            // Apply updates from event
            projection.user_id = event.user_id;
            projection.limit_bytes = event.limit_bytes;
            projection.used_bytes = event.used_bytes;
            projection.file_count = event.file_count;
            projection.max_file_size = event.max_file_size;
            projection.max_file_count = event.max_file_count;
            projection.tier = event.tier;
            projection.quota_status = event.quota_status;
            projection.warning_threshold_percent = event.warning_threshold_percent;
            projection.last_warning_sent_at = event.last_warning_sent_at;
            projection.peak_usage_bytes = event.peak_usage_bytes;
            projection.peak_usage_at = event.peak_usage_at;
            projection.metadata = event.metadata;
            // Track the event sequence for this projection
            projection.apply_event(sequence);

            self.repository.save(&projection).await?;
        }

        Ok(())
    }

    async fn on_deleted(&self, event: UserQuotaDeletedEvent, _sequence: i64) -> Result<()> {
        self.repository.delete(event.id).await
    }
}

// ============================================================================
// CUSTOM EVENT HANDLERS
// ============================================================================
// <<< CUSTOM HANDLERS START >>>

// <<< CUSTOM HANDLERS START >>>
// Add custom event handlers here
// <<< CUSTOM HANDLERS END >>>