refactor

ddboline · Sep 15, 2023 · 5efa20c · 5efa20c
1 parent 1d3b73f
commit 5efa20c
Showing 1 changed file with 16 additions and 6 deletions.
diff --git a/movie_collection_lib/src/movie_collection.rs b/movie_collection_lib/src/movie_collection.rs
@@ -359,12 +359,22 @@ impl MovieCollection {
     /// # Errors
     /// Returns error if db queries fail
     pub async fn get_collection_index(&self, path: &str) -> Result<Option<Uuid>, Error> {
-        let query = query!(
-            r#"SELECT idx FROM movie_collection WHERE path = $path"#,
-            path = path
-        );
-        let conn = self.pool.get().await?;
-        let id = query.fetch_opt(&conn).await?;
+        let id = if path.starts_with("/") {
+            let query = query!(
+                r#"SELECT idx FROM movie_collection WHERE path = $path LIMIT 1"#,
+                path = path
+            );
+            let conn = self.pool.get().await?;
+            query.fetch_opt(&conn).await?
+        } else {
+            let path = format_sstr!("%{path}");
+            let query = query!(
+                r#"SELECT idx FROM movie_collection WHERE path like $path LIMIT 1"#,
+                path = path
+            );
+            let conn = self.pool.get().await?;
+            query.fetch_opt(&conn).await?
+        };
         Ok(id.map(|(x,)| x))
     }