feat: init

service/notify/subscriptions.rs

@@ -0,0 +1,183 @@
+use crate::error::AppError;
+use crate::models::common::{EventType, SubscriptionLevel, TargetType};
+use crate::models::notifications::NotificationSubscription;
+use crate::service::NotificationService;
+use crate::session::Session;
+use chrono::{DateTime, Utc};
+use serde::{Deserialize, Serialize};
+use uuid::Uuid;
+
+use super::util::clamp_limit_offset;
+
+#[derive(Debug, Deserialize, Serialize)]
+pub struct CreateSubscriptionParams {
+    pub workspace_id: Option<Uuid>,
+    pub repo_id: Option<Uuid>,
+    pub target_type: TargetType,
+    pub target_id: Option<Uuid>,
+    pub event_types: Vec<EventType>,
+    pub channels: Vec<String>,
+    pub level: SubscriptionLevel,
+}
+
+#[derive(Debug, Deserialize, Serialize)]
+pub struct UpdateSubscriptionParams {
+    pub event_types: Option<Vec<EventType>>,
+    pub channels: Option<Vec<String>>,
+    pub level: Option<SubscriptionLevel>,
+    pub muted: Option<bool>,
+    pub muted_until: Option<DateTime<Utc>>,
+}
+
+impl NotificationSubscription {
+    pub async fn find_by_id(
+        pool: &sqlx::PgPool,
+        id: Uuid,
+        user_id: Uuid,
+    ) -> Result<Option<Self>, AppError> {
+        sqlx::query_as::<_, NotificationSubscription>(
+            "SELECT id, user_id, workspace_id, repo_id, target_type, target_id, event_types, \
+             channels, level, muted, muted_until, created_at, updated_at \
+             FROM notification_subscription WHERE id = $1 AND user_id = $2",
+        )
+        .bind(id)
+        .bind(user_id)
+        .fetch_optional(pool)
+        .await
+        .map_err(AppError::Database)
+    }
+
+    pub async fn list_for_user(
+        pool: &sqlx::PgPool,
+        user_id: Uuid,
+        limit: i64,
+        offset: i64,
+    ) -> Result<Vec<Self>, AppError> {
+        sqlx::query_as::<_, NotificationSubscription>(
+            "SELECT id, user_id, workspace_id, repo_id, target_type, target_id, event_types, \
+             channels, level, muted, muted_until, created_at, updated_at \
+             FROM notification_subscription WHERE user_id = $1 \
+             ORDER BY created_at DESC LIMIT $2 OFFSET $3",
+        )
+        .bind(user_id)
+        .bind(limit)
+        .bind(offset)
+        .fetch_all(pool)
+        .await
+        .map_err(AppError::Database)
+    }
+}
+
+impl NotificationService {
+    pub async fn list_subscriptions(
+        &self,
+        session: &Session,
+        limit: i64,
+        offset: i64,
+    ) -> Result<Vec<NotificationSubscription>, AppError> {
+        let user_id = session.user().ok_or(AppError::Unauthorized)?;
+        let (limit, offset) = clamp_limit_offset(limit, offset);
+        NotificationSubscription::list_for_user(self.ctx.db.reader(), user_id, limit, offset).await
+    }
+
+    pub async fn create_subscription(
+        &self,
+        session: &Session,
+        params: CreateSubscriptionParams,
+    ) -> Result<NotificationSubscription, AppError> {
+        let user_id = session.user().ok_or(AppError::Unauthorized)?;
+        let id = Uuid::now_v7();
+        let now = Utc::now();
+
+        sqlx::query(
+            "INSERT INTO notification_subscription \
+             (id, user_id, workspace_id, repo_id, target_type, target_id, event_types, channels, level, muted, muted_until, created_at, updated_at) \
+             VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, false, NULL, $10, $10)",
+        )
+        .bind(id)
+        .bind(user_id)
+        .bind(params.workspace_id)
+        .bind(params.repo_id)
+        .bind(params.target_type.as_str())
+        .bind(params.target_id)
+        .bind(&params.event_types)
+        .bind(&params.channels)
+        .bind(params.level.as_str())
+        .bind(now)
+        .execute(self.ctx.db.writer())
+        .await
+        .map_err(AppError::Database)?;
+
+        NotificationSubscription::find_by_id(self.ctx.db.reader(), id, user_id)
+            .await?
+            .ok_or(AppError::InternalServerError(
+                "failed to fetch created subscription".into(),
+            ))
+    }
+
+    pub async fn update_subscription(
+        &self,
+        session: &Session,
+        subscription_id: Uuid,
+        params: UpdateSubscriptionParams,
+    ) -> Result<NotificationSubscription, AppError> {
+        let user_id = session.user().ok_or(AppError::Unauthorized)?;
+        let now = Utc::now();
+
+        let existing =
+            NotificationSubscription::find_by_id(self.ctx.db.reader(), subscription_id, user_id)
+                .await?
+                .ok_or(AppError::NotFound("subscription not found".into()))?;
+
+        let event_types = params.event_types.unwrap_or(existing.event_types);
+        let channels = params.channels.unwrap_or(existing.channels);
+        let level = params.level.unwrap_or(existing.level);
+        let muted = params.muted.unwrap_or(existing.muted);
+        let muted_until = params.muted_until.or(existing.muted_until);
+
+        sqlx::query(
+            "UPDATE notification_subscription \
+             SET event_types = $1, channels = $2, level = $3, muted = $4, muted_until = $5, updated_at = $6 \
+             WHERE id = $7 AND user_id = $8",
+        )
+        .bind(&event_types)
+        .bind(&channels)
+        .bind(level.as_str())
+        .bind(muted)
+        .bind(muted_until)
+        .bind(now)
+        .bind(subscription_id)
+        .bind(user_id)
+        .execute(self.ctx.db.writer())
+        .await
+        .map_err(AppError::Database)?;
+
+        NotificationSubscription::find_by_id(self.ctx.db.reader(), subscription_id, user_id)
+            .await?
+            .ok_or(AppError::InternalServerError(
+                "failed to fetch updated subscription".into(),
+            ))
+    }
+
+    pub async fn delete_subscription(
+        &self,
+        session: &Session,
+        subscription_id: Uuid,
+    ) -> Result<(), AppError> {
+        let user_id = session.user().ok_or(AppError::Unauthorized)?;
+
+        let result =
+            sqlx::query("DELETE FROM notification_subscription WHERE id = $1 AND user_id = $2")
+                .bind(subscription_id)
+                .bind(user_id)
+                .execute(self.ctx.db.writer())
+                .await
+                .map_err(AppError::Database)?;
+
+        if result.rows_affected() == 0 {
+            return Err(AppError::NotFound("subscription not found".into()));
+        }
+
+        Ok(())
+    }
+}