use rustauth_core::db::{DbFieldType, DbValue, IdGeneration, SqlParam};
use rustauth_core::error::RustAuthError;
use tokio_postgres::types::ToSql;
pub fn postgres_params(
params: &[SqlParam],
) -> Result<Vec<Box<dyn ToSql + Sync + Send>>, RustAuthError> {
params.iter().map(postgres_param).collect()
}
pub fn param_refs(values: &[Box<dyn ToSql + Sync + Send>]) -> Vec<&(dyn ToSql + Sync)> {
values
.iter()
.map(|value| &**value as &(dyn ToSql + Sync))
.collect()
}
fn postgres_param(param: &SqlParam) -> Result<Box<dyn ToSql + Sync + Send>, RustAuthError> {
match ¶m.value {
DbValue::String(value) if param.generated_id == Some(IdGeneration::Uuid) => {
let value = uuid::Uuid::parse_str(value).map_err(|error| {
RustAuthError::Adapter(format!("invalid PostgreSQL UUID value `{value}`: {error}"))
})?;
Ok(Box::new(value))
}
DbValue::String(value) => Ok(Box::new(value.clone())),
DbValue::Number(value) => Ok(Box::new(*value)),
DbValue::Boolean(value) => Ok(Box::new(*value)),
DbValue::Timestamp(value) => Ok(Box::new(*value)),
DbValue::Json(value) => Ok(Box::new(value.clone())),
DbValue::StringArray(value) => Ok(Box::new(value.clone())),
DbValue::NumberArray(value) => Ok(Box::new(value.clone())),
DbValue::Record(_) | DbValue::RecordArray(_) => Err(RustAuthError::Adapter(
"joined records cannot be bound as SQL values".to_owned(),
)),
DbValue::Null => match param.field_type {
DbFieldType::String if param.generated_id == Some(IdGeneration::Uuid) => {
Ok(Box::new(Option::<uuid::Uuid>::None))
}
DbFieldType::String => Ok(Box::new(Option::<String>::None)),
DbFieldType::Number => Ok(Box::new(Option::<i64>::None)),
DbFieldType::Boolean => Ok(Box::new(Option::<bool>::None)),
DbFieldType::Timestamp => Ok(Box::new(Option::<time::OffsetDateTime>::None)),
DbFieldType::Json => Ok(Box::new(Option::<serde_json::Value>::None)),
DbFieldType::StringArray => Ok(Box::new(Option::<Vec<String>>::None)),
DbFieldType::NumberArray => Ok(Box::new(Option::<Vec<i64>>::None)),
},
}
}