use async_trait::async_trait;
use mas_data_model::User;
use mas_storage::{user::UserTermsRepository, Clock};
use rand::RngCore;
use sqlx::PgConnection;
use ulid::Ulid;
use url::Url;
use uuid::Uuid;
use crate::{tracing::ExecuteExt, DatabaseError};
pub struct PgUserTermsRepository<'c> {
conn: &'c mut PgConnection,
}
impl<'c> PgUserTermsRepository<'c> {
pub fn new(conn: &'c mut PgConnection) -> Self {
Self { conn }
}
}
#[async_trait]
impl<'c> UserTermsRepository for PgUserTermsRepository<'c> {
type Error = DatabaseError;
#[tracing::instrument(
name = "db.user_terms.accept_terms",
skip_all,
fields(
db.query.text,
%user.id,
user_terms.id,
%user_terms.url = terms_url.as_str(),
),
err,
)]
async fn accept_terms(
&mut self,
rng: &mut (dyn RngCore + Send),
clock: &dyn Clock,
user: &User,
terms_url: Url,
) -> Result<(), Self::Error> {
let created_at = clock.now();
let id = Ulid::from_datetime_with_source(created_at.into(), rng);
tracing::Span::current().record("user_terms.id", tracing::field::display(id));
sqlx::query!(
r#"
INSERT INTO user_terms (user_terms_id, user_id, terms_url, created_at)
VALUES ($1, $2, $3, $4)
ON CONFLICT (user_id, terms_url) DO NOTHING
"#,
Uuid::from(id),
Uuid::from(user.id),
terms_url.as_str(),
created_at,
)
.traced()
.execute(&mut *self.conn)
.await?;
Ok(())
}
}