diff --git a/crates/adapters/postgres/src/boost.rs b/crates/adapters/postgres/src/boost.rs index d72d75d..a1f0431 100644 --- a/crates/adapters/postgres/src/boost.rs +++ b/crates/adapters/postgres/src/boost.rs @@ -1,2 +1,80 @@ -pub struct PgBoostRepository { _pool: sqlx::PgPool } -impl PgBoostRepository { pub fn new(pool: sqlx::PgPool) -> Self { Self { _pool: pool } } } +use async_trait::async_trait; +use chrono::{DateTime, Utc}; +use sqlx::PgPool; +use domain::{errors::DomainError, models::social::Boost, ports::BoostRepository, value_objects::{BoostId, ThoughtId, UserId}}; + +pub struct PgBoostRepository { pool: PgPool } +impl PgBoostRepository { pub fn new(pool: PgPool) -> Self { Self { pool } } } + +#[async_trait] +impl BoostRepository for PgBoostRepository { + async fn save(&self, b: &Boost) -> Result<(), DomainError> { + sqlx::query( + "INSERT INTO boosts(id,user_id,thought_id,ap_id,created_at) VALUES($1,$2,$3,$4,$5) ON CONFLICT(user_id,thought_id) DO NOTHING" + ) + .bind(b.id.as_uuid()).bind(b.user_id.as_uuid()).bind(b.thought_id.as_uuid()).bind(&b.ap_id).bind(b.created_at) + .execute(&self.pool).await.map_err(|e| DomainError::Internal(e.to_string())).map(|_| ()) + } + + async fn delete(&self, user_id: &UserId, thought_id: &ThoughtId) -> Result<(), DomainError> { + let r = sqlx::query("DELETE FROM boosts WHERE user_id=$1 AND thought_id=$2") + .bind(user_id.as_uuid()).bind(thought_id.as_uuid()) + .execute(&self.pool).await.map_err(|e| DomainError::Internal(e.to_string()))?; + if r.rows_affected() == 0 { return Err(DomainError::NotFound); } + Ok(()) + } + + async fn find(&self, user_id: &UserId, thought_id: &ThoughtId) -> Result, DomainError> { + #[derive(sqlx::FromRow)] + struct Row { id: uuid::Uuid, user_id: uuid::Uuid, thought_id: uuid::Uuid, ap_id: Option, created_at: DateTime } + sqlx::query_as::<_, Row>("SELECT id,user_id,thought_id,ap_id,created_at FROM boosts WHERE user_id=$1 AND thought_id=$2") + .bind(user_id.as_uuid()).bind(thought_id.as_uuid()) + .fetch_optional(&self.pool).await + .map_err(|e| DomainError::Internal(e.to_string())) + .map(|o| o.map(|r| Boost { id: BoostId::from_uuid(r.id), user_id: UserId::from_uuid(r.user_id), thought_id: ThoughtId::from_uuid(r.thought_id), ap_id: r.ap_id, created_at: r.created_at })) + } + + async fn count_for_thought(&self, thought_id: &ThoughtId) -> Result { + sqlx::query_scalar("SELECT COUNT(*) FROM boosts WHERE thought_id=$1") + .bind(thought_id.as_uuid()).fetch_one(&self.pool).await + .map_err(|e| DomainError::Internal(e.to_string())) + } +} + +#[cfg(test)] +mod tests { + use super::*; + use chrono::Utc; + use domain::{models::{thought::{Thought, Visibility}, user::User}, value_objects::*}; + use crate::{thought::PgThoughtRepository, user::PgUserRepository}; + use domain::ports::{ThoughtRepository, UserRepository}; + + async fn seed(pool: &sqlx::PgPool) -> (User, Thought) { + let urepo = PgUserRepository::new(pool.clone()); + let trepo = PgThoughtRepository::new(pool.clone()); + let u = User::new_local(UserId::new(), Username::new("alice").unwrap(), Email::new("alice@ex.com").unwrap(), PasswordHash("h".into())); + urepo.save(&u).await.unwrap(); + let t = Thought::new_local(ThoughtId::new(), u.id.clone(), Content::new_local("hi").unwrap(), None, Visibility::Public, None, false); + trepo.save(&t).await.unwrap(); + (u, t) + } + + #[sqlx::test(migrations = "./migrations")] + async fn boost_and_count(pool: sqlx::PgPool) { + let (user, thought) = seed(&pool).await; + let repo = PgBoostRepository::new(pool); + let boost = Boost { id: BoostId::new(), user_id: user.id.clone(), thought_id: thought.id.clone(), ap_id: None, created_at: Utc::now() }; + repo.save(&boost).await.unwrap(); + assert_eq!(repo.count_for_thought(&thought.id).await.unwrap(), 1); + } + + #[sqlx::test(migrations = "./migrations")] + async fn unboost(pool: sqlx::PgPool) { + let (user, thought) = seed(&pool).await; + let repo = PgBoostRepository::new(pool); + let boost = Boost { id: BoostId::new(), user_id: user.id.clone(), thought_id: thought.id.clone(), ap_id: None, created_at: Utc::now() }; + repo.save(&boost).await.unwrap(); + repo.delete(&user.id, &thought.id).await.unwrap(); + assert_eq!(repo.count_for_thought(&thought.id).await.unwrap(), 0); + } +} diff --git a/crates/adapters/postgres/src/like.rs b/crates/adapters/postgres/src/like.rs index 6970da9..556d486 100644 --- a/crates/adapters/postgres/src/like.rs +++ b/crates/adapters/postgres/src/like.rs @@ -1,2 +1,80 @@ -pub struct PgLikeRepository { _pool: sqlx::PgPool } -impl PgLikeRepository { pub fn new(pool: sqlx::PgPool) -> Self { Self { _pool: pool } } } +use async_trait::async_trait; +use chrono::{DateTime, Utc}; +use sqlx::PgPool; +use domain::{errors::DomainError, models::social::Like, ports::LikeRepository, value_objects::{LikeId, ThoughtId, UserId}}; + +pub struct PgLikeRepository { pool: PgPool } +impl PgLikeRepository { pub fn new(pool: PgPool) -> Self { Self { pool } } } + +#[async_trait] +impl LikeRepository for PgLikeRepository { + async fn save(&self, l: &Like) -> Result<(), DomainError> { + sqlx::query( + "INSERT INTO likes(id,user_id,thought_id,ap_id,created_at) VALUES($1,$2,$3,$4,$5) ON CONFLICT(user_id,thought_id) DO NOTHING" + ) + .bind(l.id.as_uuid()).bind(l.user_id.as_uuid()).bind(l.thought_id.as_uuid()).bind(&l.ap_id).bind(l.created_at) + .execute(&self.pool).await.map_err(|e| DomainError::Internal(e.to_string())).map(|_| ()) + } + + async fn delete(&self, user_id: &UserId, thought_id: &ThoughtId) -> Result<(), DomainError> { + let r = sqlx::query("DELETE FROM likes WHERE user_id=$1 AND thought_id=$2") + .bind(user_id.as_uuid()).bind(thought_id.as_uuid()) + .execute(&self.pool).await.map_err(|e| DomainError::Internal(e.to_string()))?; + if r.rows_affected() == 0 { return Err(DomainError::NotFound); } + Ok(()) + } + + async fn find(&self, user_id: &UserId, thought_id: &ThoughtId) -> Result, DomainError> { + #[derive(sqlx::FromRow)] + struct Row { id: uuid::Uuid, user_id: uuid::Uuid, thought_id: uuid::Uuid, ap_id: Option, created_at: DateTime } + sqlx::query_as::<_, Row>("SELECT id,user_id,thought_id,ap_id,created_at FROM likes WHERE user_id=$1 AND thought_id=$2") + .bind(user_id.as_uuid()).bind(thought_id.as_uuid()) + .fetch_optional(&self.pool).await + .map_err(|e| DomainError::Internal(e.to_string())) + .map(|o| o.map(|r| Like { id: LikeId::from_uuid(r.id), user_id: UserId::from_uuid(r.user_id), thought_id: ThoughtId::from_uuid(r.thought_id), ap_id: r.ap_id, created_at: r.created_at })) + } + + async fn count_for_thought(&self, thought_id: &ThoughtId) -> Result { + sqlx::query_scalar("SELECT COUNT(*) FROM likes WHERE thought_id=$1") + .bind(thought_id.as_uuid()).fetch_one(&self.pool).await + .map_err(|e| DomainError::Internal(e.to_string())) + } +} + +#[cfg(test)] +mod tests { + use super::*; + use chrono::Utc; + use domain::{models::{thought::{Thought, Visibility}, user::User}, value_objects::*}; + use crate::{thought::PgThoughtRepository, user::PgUserRepository}; + use domain::ports::{ThoughtRepository, UserRepository}; + + async fn seed(pool: &sqlx::PgPool) -> (User, Thought) { + let urepo = PgUserRepository::new(pool.clone()); + let trepo = PgThoughtRepository::new(pool.clone()); + let u = User::new_local(UserId::new(), Username::new("alice").unwrap(), Email::new("alice@ex.com").unwrap(), PasswordHash("h".into())); + urepo.save(&u).await.unwrap(); + let t = Thought::new_local(ThoughtId::new(), u.id.clone(), Content::new_local("hi").unwrap(), None, Visibility::Public, None, false); + trepo.save(&t).await.unwrap(); + (u, t) + } + + #[sqlx::test(migrations = "./migrations")] + async fn like_and_count(pool: sqlx::PgPool) { + let (user, thought) = seed(&pool).await; + let repo = PgLikeRepository::new(pool); + let like = Like { id: LikeId::new(), user_id: user.id.clone(), thought_id: thought.id.clone(), ap_id: None, created_at: Utc::now() }; + repo.save(&like).await.unwrap(); + assert_eq!(repo.count_for_thought(&thought.id).await.unwrap(), 1); + } + + #[sqlx::test(migrations = "./migrations")] + async fn unlike(pool: sqlx::PgPool) { + let (user, thought) = seed(&pool).await; + let repo = PgLikeRepository::new(pool); + let like = Like { id: LikeId::new(), user_id: user.id.clone(), thought_id: thought.id.clone(), ap_id: None, created_at: Utc::now() }; + repo.save(&like).await.unwrap(); + repo.delete(&user.id, &thought.id).await.unwrap(); + assert_eq!(repo.count_for_thought(&thought.id).await.unwrap(), 0); + } +}