Add Favorite GET, and POST endpoints

src/database/mod.rs

@@ -3,7 +3,7 @@ use diesel::prelude::*;
 use diesel::sqlite::SqliteConnection;
 use dotenv::dotenv;
 
-use crate::database::models::{InsertUser, User};
+use crate::database::models::{Favorite, InsertFavorite, InsertUser, User};
 
 mod models;
 mod schema;
@@ -68,3 +68,25 @@ pub fn user_exists(name: &str) -> bool {
         .first()
         .is_some()
 }
+
+pub fn add_favorite(user_id: i32, favorite_path: String) {
+    use schema::favorites::dsl::*;
+
+    let connection = connect();
+    diesel::insert_into(favorites)
+        .values(InsertFavorite {
+            userid: &user_id,
+            path: &favorite_path,
+        })
+        .execute(&connection)
+        .unwrap();
+}
+
+pub fn get_favorites(user_id: i32) -> Vec<Favorite> {
+    use schema::favorites::dsl::*;
+
+    favorites
+        .filter(userid.eq(user_id))
+        .load::<Favorite>(&connect())
+        .unwrap_or_default()
+}