Add Cleanup binary for fixing broken DB/file relations

src/database/mod.rs

@@ -123,12 +123,15 @@ pub enum DbErrorKind {
     AlreadyExists,
     InsertError,
     QueryError,
+    UpdateError,
 }
 
 pub trait FavoriteDao: Sync + Send {
     fn add_favorite(&mut self, user_id: i32, favorite_path: &str) -> Result<usize, DbError>;
     fn remove_favorite(&mut self, user_id: i32, favorite_path: String);
     fn get_favorites(&mut self, user_id: i32) -> Result<Vec<Favorite>, DbError>;
+    fn update_path(&mut self, old_path: &str, new_path: &str) -> Result<(), DbError>;
+    fn get_all_paths(&mut self) -> Result<Vec<String>, DbError>;
 }
 
 pub struct SqliteFavoriteDao {
@@ -183,6 +186,26 @@ impl FavoriteDao for SqliteFavoriteDao {
             .load::<Favorite>(self.connection.lock().unwrap().deref_mut())
             .map_err(|_| DbError::new(DbErrorKind::QueryError))
     }
+
+    fn update_path(&mut self, old_path: &str, new_path: &str) -> Result<(), DbError> {
+        use schema::favorites::dsl::*;
+
+        diesel::update(favorites.filter(path.eq(old_path)))
+            .set(path.eq(new_path))
+            .execute(self.connection.lock().unwrap().deref_mut())
+            .map_err(|_| DbError::new(DbErrorKind::UpdateError))?;
+        Ok(())
+    }
+
+    fn get_all_paths(&mut self) -> Result<Vec<String>, DbError> {
+        use schema::favorites::dsl::*;
+
+        favorites
+            .select(path)
+            .distinct()
+            .load(self.connection.lock().unwrap().deref_mut())
+            .map_err(|_| DbError::new(DbErrorKind::QueryError))
+    }
 }
 
 pub trait ExifDao: Sync + Send {
@@ -208,6 +231,12 @@ pub trait ExifDao: Sync + Send {
 
     /// Get distinct camera makes with counts
     fn get_camera_makes(&mut self) -> Result<Vec<(String, i64)>, DbError>;
+
+    /// Update file path in EXIF database
+    fn update_file_path(&mut self, old_path: &str, new_path: &str) -> Result<(), DbError>;
+
+    /// Get all file paths from EXIF database
+    fn get_all_file_paths(&mut self) -> Result<Vec<String>, DbError>;
 }
 
 pub struct SqliteExifDao {
@@ -398,4 +427,27 @@ impl ExifDao for SqliteExifDao {
             })
             .map_err(|_| DbError::new(DbErrorKind::QueryError))
     }
+
+    fn update_file_path(&mut self, old_path: &str, new_path: &str) -> Result<(), DbError> {
+        use schema::image_exif::dsl::*;
+
+        let mut connection = self.connection.lock().expect("Unable to get ExifDao");
+
+        diesel::update(image_exif.filter(file_path.eq(old_path)))
+            .set(file_path.eq(new_path))
+            .execute(connection.deref_mut())
+            .map_err(|_| DbError::new(DbErrorKind::UpdateError))?;
+        Ok(())
+    }
+
+    fn get_all_file_paths(&mut self) -> Result<Vec<String>, DbError> {
+        use schema::image_exif::dsl::*;
+
+        let mut connection = self.connection.lock().expect("Unable to get ExifDao");
+
+        image_exif
+            .select(file_path)
+            .load(connection.deref_mut())
+            .map_err(|_| DbError::new(DbErrorKind::QueryError))
+    }
 }