Add update and get-by-tag for schedules

src/db/errors.rs

@@ -10,6 +10,7 @@ pub enum DatabaseError {
     InsertGetError,
     NotFound,
     Protected,
+    UpdateError(diesel::result::Error),
 }
 
 impl DatabaseError {
@@ -45,6 +46,7 @@ impl From<&DatabaseError> for String {
             DatabaseError::NotFound => String::from("model was not found in database"),
             DatabaseError::DeleteError => String::from("error on deleting from database"),
             DatabaseError::Protected => String::from("model is protected"),
+            DatabaseError::UpdateError(_) => String::from("error on updating the model"),
         }
     }
 }

src/db/schedules.rs

@@ -1,3 +1,4 @@
+use std::borrow::Borrow;
 use diesel::dsl::sql;
 use diesel::prelude::*;
 
@@ -37,6 +38,15 @@ pub fn get_schedule_by_uid(filter_uid: EmgauwaUid) -> Result<Schedule, DatabaseE
     Ok(result)
 }
 
+pub fn get_schedules_by_tag(tag: &Tag) -> Vec<Schedule> {
+    let connection = get_connection();
+    JunctionTag::belonging_to(tag)
+        .inner_join(schedules)
+        .select(schema::schedules::all_columns)
+        .load::<Schedule>(&connection)
+        .expect("Error loading tags")
+}
+
 pub fn delete_schedule_by_uid(filter_uid: EmgauwaUid) -> Result<(), DatabaseError> {
     let filter_uid = match filter_uid {
         EmgauwaUid::Off => Err(DatabaseError::Protected),
@@ -79,6 +89,25 @@ pub fn create_schedule(new_name: &str, new_periods: &Periods) -> Result<Schedule
     Ok(result)
 }
 
+pub fn update_schedule(schedule: &Schedule, new_name: &str, new_periods: &Periods) -> Result<Schedule, DatabaseError> {
+    let connection = get_connection();
+
+    let new_periods = match schedule.uid {
+        EmgauwaUid::Off | EmgauwaUid::On => schedule.periods.borrow(),
+        EmgauwaUid::Any(_) => new_periods,
+    };
+
+    diesel::update(schedule)
+        .set((
+            schema::schedules::name.eq(new_name),
+            schema::schedules::periods.eq(new_periods),
+        ))
+        .execute(&connection)
+        .map_err(DatabaseError::UpdateError)?;
+
+    get_schedule_by_uid(schedule.uid.clone())
+}
+
 pub fn set_schedule_tags(schedule: &Schedule, new_tags: &[String]) -> Result<(), DatabaseError> {
     let connection = get_connection();
     diesel::delete(junction_tag.filter(schema::junction_tag::schedule_id.eq(schedule.id)))

src/db/tag.rs

@@ -3,7 +3,7 @@ use diesel::prelude::*;
 
 
 use crate::db::errors::DatabaseError;
-use crate::db::{get_connection};
+use crate::db::{get_connection, schema};
 use crate::db::models::*;
 use crate::db::schema::tags::dsl::tags;
 use crate::db::schema::junction_tag::dsl::junction_tag;
@@ -29,6 +29,17 @@ pub fn create_tag(new_tag: &str) -> Result<Tag, DatabaseError> {
     Ok(result)
 }
 
+pub fn get_tag(target_tag: &str) -> Result<Tag, DatabaseError> {
+    let connection = get_connection();
+
+    let result = tags
+        .filter(schema::tags::tag.eq(target_tag))
+        .first::<Tag>(&connection)
+        .or(Err(DatabaseError::NotFound))?;
+
+    Ok(result)
+}
+
 pub fn create_junction_tag(target_tag: Tag, target_relay: Option<&Relay>, target_schedule: Option<&Schedule>) -> Result<JunctionTag, DatabaseError> {
     let connection = get_connection();