use chrono::DateTime;
use mongodb::bson::Bson;
use serde::{Deserialize, Serialize};
use sqlx::FromRow;
use utoipa::ToSchema;
use crate::domain::query::IntoBsonDocument;
use super::super::serde::{
deserialize_datetime, deserialize_object_id, deserialize_object_id_as_string,
serialize_object_id_as_string,
};
#[derive(Default, Debug, Clone, PartialEq, Eq, Deserialize, Serialize, FromRow, ToSchema)]
#[schema(example = json!({ "userId": String::default(), "roleId": String::default() }))]
pub struct UserRole {
#[serde(
rename = "_id",
default,
skip_serializing_if = "Option::is_none",
deserialize_with = "deserialize_object_id_as_string"
)]
pub id: Option<String>,
#[sqlx(rename = "userId")]
#[serde(
rename = "userId",
default,
serialize_with = "serialize_object_id_as_string",
deserialize_with = "deserialize_object_id"
)]
pub user_id: String,
#[sqlx(rename = "roleId")]
#[serde(
rename = "roleId",
default,
serialize_with = "serialize_object_id_as_string",
deserialize_with = "deserialize_object_id"
)]
pub role_id: String,
#[sqlx(rename = "issuedAt")]
#[serde(rename = "issuedAt", deserialize_with = "deserialize_datetime")]
pub issued_at: chrono::DateTime<chrono::Utc>,
}
impl UserRole {
pub fn new(user_id: &str, role_id: &str) -> Self {
Self {
id: None,
user_id: user_id.into(),
role_id: role_id.into(),
issued_at: chrono::Utc::now(),
}
}
}
impl IntoBsonDocument for UserRole {
fn into_bson_document(self) -> Result<mongodb::bson::Document, mongodb::bson::ser::Error> {
let mut doc = mongodb::bson::to_document(&self)?;
for key in &["issuedAt"] {
if let Some(Bson::String(s)) = doc.get(*key).cloned()
&& let Ok(dt) = DateTime::parse_from_rfc3339(&s)
{
doc.insert(
*key,
mongodb::bson::Bson::DateTime(mongodb::bson::DateTime::from_millis(
dt.timestamp_millis(),
)),
);
}
}
Ok(doc)
}
}