82 lines
2.0 KiB
Rust
82 lines
2.0 KiB
Rust
use sqlx::SqlitePool;
|
|
|
|
use super::planned_event::PlannedEvent;
|
|
|
|
pub struct UserTrip {}
|
|
|
|
impl UserTrip {
|
|
pub async fn create(
|
|
db: &SqlitePool,
|
|
user_id: i64,
|
|
trip_details_id: i64,
|
|
) -> Result<(), UserTripError> {
|
|
if PlannedEvent::is_full(db, trip_details_id).await {
|
|
return Err(UserTripError::EventAlreadyFull);
|
|
}
|
|
|
|
//check if cox if own event
|
|
let is_cox = sqlx::query!(
|
|
"SELECT count(*) as amount
|
|
FROM trip
|
|
WHERE trip_details_id = ?
|
|
AND cox_id = ?",
|
|
trip_details_id,
|
|
user_id
|
|
)
|
|
.fetch_one(db)
|
|
.await
|
|
.unwrap();
|
|
if is_cox.amount > 0 {
|
|
return Err(UserTripError::AlreadyRegisteredAsCox);
|
|
}
|
|
|
|
//check if cox if planned_event
|
|
let is_cox = sqlx::query!(
|
|
"SELECT count(*) as amount
|
|
FROM trip
|
|
WHERE planned_event_id = (
|
|
SELECT id FROM planned_event WHERE trip_details_id = ?
|
|
)
|
|
AND cox_id = ?",
|
|
trip_details_id,
|
|
user_id
|
|
)
|
|
.fetch_one(db)
|
|
.await
|
|
.unwrap();
|
|
if is_cox.amount > 0 {
|
|
return Err(UserTripError::AlreadyRegisteredAsCox);
|
|
}
|
|
|
|
match sqlx::query!(
|
|
"INSERT INTO user_trip (user_id, trip_details_id) VALUES(?, ?)",
|
|
user_id,
|
|
trip_details_id
|
|
)
|
|
.execute(db)
|
|
.await
|
|
{
|
|
Ok(_) => Ok(()),
|
|
Err(_) => Err(UserTripError::AlreadyRegistered),
|
|
}
|
|
}
|
|
|
|
pub async fn delete(db: &SqlitePool, user_id: i64, trip_details_id: i64) {
|
|
//TODO: Check if > 2 hrs to event
|
|
let _ = sqlx::query!(
|
|
"DELETE FROM user_trip WHERE user_id = ? AND trip_details_id = ?",
|
|
user_id,
|
|
trip_details_id
|
|
)
|
|
.execute(db)
|
|
.await
|
|
.is_ok();
|
|
}
|
|
}
|
|
|
|
pub enum UserTripError {
|
|
AlreadyRegistered,
|
|
AlreadyRegisteredAsCox,
|
|
EventAlreadyFull,
|
|
}
|