Add Update Methods for Shopping Tours

TrueDoctor · May 29, 2024 · 261606f · 261606f
1 parent fb93e25
commit 261606f
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 2 deletions.
diff --git a/foodlib/src/events.rs b/foodlib/src/events.rs
@@ -320,15 +320,41 @@ impl FoodBase {
         tour_id: i32,
         date: NaiveDateTime,
     ) -> eyre::Result<ShoppingTour> {
-        todo!()
+        let result = sqlx::query_as!(
+            ShoppingTour,
+            r#"
+                UPDATE shopping_tours
+                SET tour_date = $2 
+                WHERE tour_id = $1
+                RETURNING *
+            "#,
+            tour_id,
+            date,
+        )
+        .fetch_one(&*self.pg_pool)
+        .await?;
+        Ok(result)
     }
 
     pub async fn update_event_shopping_tour_store(
         &self,
         tour_id: i32,
         store_id: i32,
     ) -> eyre::Result<ShoppingTour> {
-        todo!()
+        let result = sqlx::query_as!(
+            ShoppingTour,
+            r#"
+                UPDATE shopping_tours
+                SET store_id = $2 
+                WHERE tour_id = $1
+                RETURNING *
+            "#,
+            tour_id,
+            store_id
+        )
+        .fetch_one(&*self.pg_pool)
+        .await?;
+        Ok(result)
     }
 
     pub async fn get_event_food_prep(&self, event_id: i32) -> eyre::Result<Vec<FoodPrep>> {

diff --git a/foodlib/src/ingredients.rs b/foodlib/src/ingredients.rs
@@ -596,6 +596,34 @@ impl FoodBase {
         .await
         .map_err(Into::into)
     }
+
+    pub async fn get_store_by_ref(&self, store_ref: String) -> eyre::Result<Store> {
+        if let Ok(store_id) = store_ref.parse::<i32>() {
+            let result = sqlx::query_as!(
+                Store,
+                r#"
+                    SELECT * FROM stores
+                    WHERE store_id = $1
+                "#,
+                store_id
+            )
+            .fetch_one(&*self.pg_pool)
+            .await;
+            Ok(result?)
+        } else {
+            let result = sqlx::query_as!(
+                Store,
+                r#"
+                    SELECT * FROM stores
+                    WHERE name = $1
+                "#,
+                store_ref
+            )
+            .fetch_one(&*self.pg_pool)
+            .await;
+            Ok(result?)
+        }
+    }
 }
 #[derive(Clone, Debug, Default, PartialEq, Eq, Serialize, Deserialize)]
 pub struct Store {