implement is_locked for trip_details

src/model/trip.rs

@@ -22,6 +22,7 @@ pub struct Trip {
     pub allow_guests: bool,
     trip_type_id: Option<i64>,
     always_show: bool,
+    is_locked: bool,
 }
 
 #[derive(Serialize)]
@@ -48,7 +49,7 @@ impl Trip {
         sqlx::query_as!(
             Self,
             "
-SELECT trip.id, cox_id, user.name as cox_name, trip_details_id, planned_starting_time, max_people, day, notes, allow_guests, trip_type_id, always_show
+SELECT trip.id, cox_id, user.name as cox_name, trip_details_id, planned_starting_time, max_people, day, notes, allow_guests, trip_type_id, always_show, is_locked
 FROM trip 
 INNER JOIN trip_details ON trip.trip_details_id = trip_details.id
 INNER JOIN user ON trip.cox_id = user.id
@@ -71,6 +72,14 @@ WHERE trip.id=?
             return Err(CoxHelpError::AlreadyRegisteredAsRower);
         }
 
+        if planned_event.trip_details(db).await.is_locked {
+            return Err(CoxHelpError::DetailsLocked);
+        }
+
+        if planned_event.is_rower_registered(db, cox).await {
+            return Err(CoxHelpError::AlreadyRegisteredAsRower);
+        }
+
         match sqlx::query!(
             "INSERT INTO trip (cox_id, planned_event_id) VALUES(?, ?)",
             cox.id,
@@ -89,7 +98,7 @@ WHERE trip.id=?
         let trips = sqlx::query_as!(
             Trip,
             "
-SELECT trip.id, cox_id, user.name as cox_name, trip_details_id, planned_starting_time, max_people, day, notes, allow_guests, trip_type_id, always_show
+SELECT trip.id, cox_id, user.name as cox_name, trip_details_id, planned_starting_time, max_people, day, notes, allow_guests, trip_type_id, always_show, is_locked
 FROM trip 
 INNER JOIN trip_details ON trip.trip_details_id = trip_details.id
 INNER JOIN user ON trip.cox_id = user.id
@@ -141,6 +150,7 @@ FROM user_trip WHERE trip_details_id = (SELECT trip_details_id FROM trip WHERE i
         notes: Option<&str>,
         trip_type: Option<i64>, //TODO: Move to `TripType`
         always_show: bool,
+        is_locked: bool,
     ) -> Result<(), TripUpdateError> {
         if !trip.is_trip_from_user(cox.id) {
             return Err(TripUpdateError::NotYourTrip);
@@ -158,11 +168,12 @@ FROM user_trip WHERE trip_details_id = (SELECT trip_details_id FROM trip WHERE i
         };
 
         sqlx::query!(
-            "UPDATE trip_details SET max_people = ?, notes = ?, trip_type_id = ?, always_show = ? WHERE id = ?",
+            "UPDATE trip_details SET max_people = ?, notes = ?, trip_type_id = ?, always_show = ?, is_locked = ? WHERE id = ?",
             max_people,
             notes,
             trip_type,
             always_show,
+            is_locked,
             trip_details_id
         )
         .execute(db)
@@ -172,12 +183,23 @@ FROM user_trip WHERE trip_details_id = (SELECT trip_details_id FROM trip WHERE i
         Ok(())
     }
 
+    pub async fn trip_details(&self, db: &SqlitePool) -> Option<TripDetails> {
+        if let Some(trip_details_id) = self.trip_type_id {
+            return TripDetails::find_by_id(db, trip_details_id).await;
+        }
+        None
+    }
+
     pub async fn delete_by_planned_event(
         db: &SqlitePool,
         cox: &CoxUser,
         planned_event: &PlannedEvent,
-    ) -> bool {
-        sqlx::query!(
+    ) -> Result<(), TripHelpDeleteError> {
+        if planned_event.trip_details(db).await.is_locked {
+            return Err(TripHelpDeleteError::DetailsLocked);
+        }
+
+        let affected_rows = sqlx::query!(
             "DELETE FROM trip WHERE cox_id = ? AND planned_event_id = ?",
             cox.id,
             planned_event.id
@@ -185,8 +207,13 @@ FROM user_trip WHERE trip_details_id = (SELECT trip_details_id FROM trip WHERE i
         .execute(db)
         .await
         .unwrap()
-        .rows_affected()
-            > 0
+        .rows_affected();
+
+        if affected_rows == 0 {
+            return Err(TripHelpDeleteError::CoxNotHelping);
+        }
+
+        Ok(())
     }
 
     pub(crate) async fn delete(
@@ -233,6 +260,13 @@ FROM user_trip WHERE trip_details_id = (SELECT trip_details_id FROM trip WHERE i
 pub enum CoxHelpError {
     AlreadyRegisteredAsRower,
     AlreadyRegisteredAsCox,
+    DetailsLocked,
+}
+
+#[derive(Debug, PartialEq)]
+pub enum TripHelpDeleteError {
+    DetailsLocked,
+    CoxNotHelping,
 }
 
 #[derive(Debug, PartialEq)]
@@ -333,9 +367,11 @@ mod test {
 
         let trip = Trip::find_by_id(&pool, 1).await.unwrap();
 
-        assert!(Trip::update_own(&pool, &cox, &trip, 10, None, None, false)
-            .await
-            .is_ok());
+        assert!(
+            Trip::update_own(&pool, &cox, &trip, 10, None, None, false, false)
+                .await
+                .is_ok()
+        );
 
         let trip = Trip::find_by_id(&pool, 1).await.unwrap();
         assert_eq!(trip.max_people, 10);
@@ -354,7 +390,7 @@ mod test {
         let trip = Trip::find_by_id(&pool, 1).await.unwrap();
 
         assert!(
-            Trip::update_own(&pool, &cox, &trip, 10, None, Some(1), false)
+            Trip::update_own(&pool, &cox, &trip, 10, None, Some(1), false, false)
                 .await
                 .is_ok()
         );
@@ -376,9 +412,11 @@ mod test {
 
         let trip = Trip::find_by_id(&pool, 1).await.unwrap();
 
-        assert!(Trip::update_own(&pool, &cox, &trip, 10, None, None, false)
-            .await
-            .is_err());
+        assert!(
+            Trip::update_own(&pool, &cox, &trip, 10, None, None, false, false)
+                .await
+                .is_err()
+        );
         assert_eq!(trip.max_people, 1);
     }
 
@@ -398,7 +436,9 @@ mod test {
 
         //TODO: check why following assert fails
         //assert!(Trip::find_by_id(&pool, 2).await.is_some());
-        Trip::delete_by_planned_event(&pool, &cox, &planned_event).await;
+        Trip::delete_by_planned_event(&pool, &cox, &planned_event)
+            .await
+            .unwrap();
         assert!(Trip::find_by_id(&pool, 2).await.is_none());
     }