use serde::{Deserialize, Serialize}; use sqlx::{FromRow, SqlitePool}; use super::{ notification::Notification, trip::{Trip, TripWithUserAndType}, tripdetails::TripDetails, user::{CoxUser, User}, }; use crate::model::tripdetails::{Action, CoxAtTrip::Yes}; #[derive(FromRow, Debug, Serialize, Deserialize, Clone)] pub struct UserTrip { pub user_id: Option, pub user_note: Option, pub trip_details_id: i64, pub created_at: String, // TODO: switch to NaiveDateTime } impl UserTrip { pub async fn create( db: &SqlitePool, user: &User, trip_details: &TripDetails, user_note: Option, ) -> Result { if trip_details.is_full(db).await { return Err(UserTripError::EventAlreadyFull); } if trip_details.is_locked { return Err(UserTripError::DetailsLocked); } if user.has_role(db, "scheckbuch").await && !trip_details.allow_guests { return Err(UserTripError::GuestNotAllowedForThisEvent); } if !trip_details.user_sees_trip(db, user).await { return Err(UserTripError::NotVisibleToUser); } //TODO: Check if user sees the event (otherwise she could forge trip_details_id) let is_cox = trip_details.user_is_cox(db, user).await; let name_newly_registered_person = if user_note.is_none() { if let Yes(action) = is_cox { match action { Action::Helping => return Err(UserTripError::AlreadyRegisteredAsCox), Action::Own => return Err(UserTripError::CantRegisterAtOwnEvent), }; } if trip_details.user_is_rower(db, user).await { return Err(UserTripError::AlreadyRegistered); } sqlx::query!( "INSERT INTO user_trip (user_id, trip_details_id) VALUES(?, ?)", user.id, trip_details.id, ) .execute(db) .await .unwrap(); user.name.clone() } else { if !trip_details.user_allowed_to_change(db, user).await { return Err(UserTripError::NotAllowedToAddGuest); } sqlx::query!( "INSERT INTO user_trip (user_note, trip_details_id) VALUES(?, ?)", user_note, trip_details.id, ) .execute(db) .await .unwrap(); user_note.clone().unwrap() }; if let Some(trip) = Trip::find_by_trip_details(db, trip_details.id).await { if user_note.is_none() { // Don't show notification if we add guest (as only we are // allowed to do so) let cox = User::find_by_id(db, trip.cox_id as i32).await.unwrap(); Notification::create( db, &cox, &format!( "{} hat sich für deine Ausfahrt am {} registriert", name_newly_registered_person, trip.day ), "Registrierung bei deiner Ausfahrt", None, None, ) .await; } trip_details.check_free_spaces(db).await; } Ok(name_newly_registered_person) } pub async fn tripdetails(&self, db: &SqlitePool) -> TripDetails { TripDetails::find_by_id(db, self.trip_details_id) .await .unwrap() } pub async fn find_by_userid_and_trip_detail_id( db: &SqlitePool, user_id: i64, trip_detail_id: i64, ) -> Option { sqlx::query_as!(Self, "SELECT user_id, user_note, trip_details_id, created_at FROM user_trip WHERE user_id= ? AND trip_details_id = ?", user_id, trip_detail_id) .fetch_one(db) .await .ok() } pub async fn self_delete(&self, db: &SqlitePool) -> Result<(), UserTripDeleteError> { let trip_details = self.tripdetails(db).await; if let Some(id) = self.user_id { let user = User::find_by_id(db, id as i32).await.unwrap(); return Self::delete(db, &user, &trip_details, self.user_note.clone()).await; } Ok(()) // TODO: fixme } //TODO: cleaner code pub async fn delete( db: &SqlitePool, user: &User, trip_details: &TripDetails, name: Option, ) -> Result<(), UserTripDeleteError> { if trip_details.is_locked { return Err(UserTripDeleteError::DetailsLocked); } if !trip_details.user_sees_trip(db, user).await { return Err(UserTripDeleteError::NotVisibleToUser); } let mut trip_to_delete = None; let mut some_trip = None; if let Some(trip) = Trip::find_by_trip_details(db, trip_details.id).await { some_trip = Some(trip.clone()); // If trip is cancelled, and lost rower just unregistered, delete the trip if TripDetails::find_by_id(db, trip_details.id) .await .unwrap() .cancelled() { let trip = TripWithUserAndType::from(db, trip.clone()).await; if trip.rower.len() == 1 { trip_to_delete = Some(trip.trip); } } } if let Some(name) = name.clone() { if !trip_details.user_allowed_to_change(db, user).await { return Err(UserTripDeleteError::NotAllowedToDeleteGuest); } if sqlx::query!( "DELETE FROM user_trip WHERE user_note = ? AND trip_details_id = ?", name, trip_details.id ) .execute(db) .await .unwrap() .rows_affected() == 0 { return Err(UserTripDeleteError::GuestNotParticipating); } } else { let _ = sqlx::query!( "DELETE FROM user_trip WHERE user_id = ? AND trip_details_id = ?", user.id, trip_details.id ) .execute(db) .await .unwrap(); } let mut add_info = ""; if let Some(trip) = &trip_to_delete { let cox = User::find_by_id(db, trip.cox_id as i32).await.unwrap(); trip.delete(db, &CoxUser::new(db, cox).await.unwrap()) .await .unwrap(); add_info = " Das war die letzte angemeldete Person. Nachdem nun alle Bescheid wissen, wird die Ausfahrt ab sofort nicht mehr angezeigt."; } if let Some(trip) = some_trip { let opt_cancelled = if trip_to_delete.is_some() { "abgesagten " } else { "" }; if let Some(name) = name { if !add_info.is_empty() { let cox = User::find_by_id(db, trip.cox_id as i32).await.unwrap(); Notification::create( db, &cox, &format!( "Du hast {} von deiner {}Ausfahrt am {} abgemeldet.{}", name, opt_cancelled, trip.day, add_info ), "Abmeldung von deiner Ausfahrt", None, None, ) .await; } } else { let cox = User::find_by_id(db, trip.cox_id as i32).await.unwrap(); Notification::create( db, &cox, &format!( "{} hat sich von deiner {}Ausfahrt am {} abgemeldet.{}", user.name, opt_cancelled, trip.day, add_info ), "Abmeldung von deiner Ausfahrt", None, None, ) .await; } } Ok(()) } } #[derive(Debug, PartialEq)] pub enum UserTripError { AlreadyRegistered, AlreadyRegisteredAsCox, EventAlreadyFull, DetailsLocked, CantRegisterAtOwnEvent, GuestNotAllowedForThisEvent, NotAllowedToAddGuest, NotVisibleToUser, } #[derive(Debug, PartialEq)] pub enum UserTripDeleteError { DetailsLocked, GuestNotParticipating, NotAllowedToDeleteGuest, NotVisibleToUser, } #[cfg(test)] mod test { use crate::{ model::{ event::Event, trip::Trip, tripdetails::TripDetails, user::CoxUser, usertrip::UserTripError, }, testdb, }; use super::{User, UserTrip}; use sqlx::SqlitePool; #[sqlx::test] fn test_succ_create() { let pool = testdb!(); let user = User::find_by_name(&pool, "rower".into()).await.unwrap(); let trip_details = TripDetails::find_by_id(&pool, 1).await.unwrap(); UserTrip::create(&pool, &user, &trip_details, None) .await .unwrap(); } #[sqlx::test] fn test_fail_create_full() { let pool = testdb!(); let user = User::find_by_name(&pool, "rower".into()).await.unwrap(); let user2 = User::find_by_name(&pool, "cox".into()).await.unwrap(); let user3 = User::find_by_name(&pool, "rower2".into()).await.unwrap(); let trip_details = TripDetails::find_by_id(&pool, 1).await.unwrap(); UserTrip::create(&pool, &user, &trip_details, None) .await .unwrap(); UserTrip::create(&pool, &user2, &trip_details, None) .await .unwrap(); let result = UserTrip::create(&pool, &user3, &trip_details, None) .await .expect_err("Expect registration to fail because trip is already full"); assert_eq!(result, UserTripError::EventAlreadyFull); } #[sqlx::test] fn test_fail_create_already_registered() { let pool = testdb!(); let user = User::find_by_name(&pool, "cox".into()).await.unwrap(); let trip_details = TripDetails::find_by_id(&pool, 1).await.unwrap(); UserTrip::create(&pool, &user, &trip_details, None) .await .unwrap(); let result = UserTrip::create(&pool, &user, &trip_details, None) .await .expect_err("Expect registration to fail because user is same as responsible cox"); assert_eq!(result, UserTripError::AlreadyRegistered); } #[sqlx::test] fn test_fail_create_is_cox_own_trip() { let pool = testdb!(); let user = User::find_by_name(&pool, "cox".into()).await.unwrap(); let trip_details = TripDetails::find_by_id(&pool, 2).await.unwrap(); let result = UserTrip::create(&pool, &user, &trip_details, None) .await .expect_err("Expect registration to fail because user is same as responsible cox"); assert_eq!(result, UserTripError::CantRegisterAtOwnEvent); } #[sqlx::test] fn test_fail_create_is_cox_planned_event() { let pool = testdb!(); let cox = CoxUser::new( &pool, User::find_by_name(&pool, "cox".into()).await.unwrap(), ) .await .unwrap(); let event = Event::find_by_id(&pool, 1).await.unwrap(); Trip::new_join(&pool, &cox, &event).await.unwrap(); let trip_details = TripDetails::find_by_id(&pool, 1).await.unwrap(); let result = UserTrip::create(&pool, &cox, &trip_details, None) .await .expect_err("Expect registration to fail because user is already registered as cox"); assert_eq!(result, UserTripError::AlreadyRegisteredAsCox); } #[sqlx::test] fn test_fail_create_guest() { let pool = testdb!(); let user = User::find_by_name(&pool, "guest".into()).await.unwrap(); let trip_details = TripDetails::find_by_id(&pool, 1).await.unwrap(); let result = UserTrip::create(&pool, &user, &trip_details, None) .await .expect_err("Not allowed for guests"); assert_eq!(result, UserTripError::GuestNotAllowedForThisEvent); } }