Merge pull request #91 from shuyi320/main

favorite

backend/database/models/userInfo.js

@@ -5,7 +5,7 @@ const userModel = new Schema({
     userName: String,
     imageUrl: String,
     petPreference: Object,
-    favoritePets: [String],
+    favorites: { type: [String], default: [] },
     friends: { type: [String], default: [] }
 });
 
@@ -47,6 +47,29 @@ export async function addFriend(user, friend) {
 
 }
 
+export async function addPetToFavorites(userId, petId) {
+    const user = await users.findOne({ userId });
+    if (!user) throw new Error("User not found");
+
+    if (user.favorites.includes(petId)) {
+        throw new Error("Pet is already in favorites");
+    }
+
+    user.favorites.push(petId);
+    await user.save();
+    return user.favorites; // Return updated favorites
+}
+
+export async function removePetFromFavorites(userId, petId) {
+    const user = await users.findOne({ userId });
+    if (!user) throw new Error("User not found");
+
+    user.favorites = user.favorites.filter((id) => id !== petId);
+    await user.save();
+    return user.favorites; // Return updated favorites
+}
+
+
 // export async function addPetToUser(userID, petID) {
 //     const user = await getUserByID(userID);
 //     if (!user)

backend/routes/api.users.js

@@ -3,6 +3,8 @@ import {
   getUserByName,
   getFriends,
   addFriend,
+  addPetToFavorites,
+  removePetFromFavorites,
   create,
 } from "../database/models/userInfo.js";
 import { createClerkClient } from "@clerk/backend";
@@ -82,4 +84,50 @@ router.post("/addFriend", async (req, res) => {
   }
 });
 
+router.post("/addFavorite", async (req, res) => {
+    try {
+        const { userId, petId } = req.body;
+
+        if (!userId || !petId) {
+            return res.status(400).json({ error: "Missing userId or petId" });
+        }
+
+        const updatedFavorites = await addPetToFavorites(userId, petId);
+        res.status(200).json({ message: "Pet added to favorites", favorites: updatedFavorites });
+    } catch (error) {
+        res.status(400).json({ error: error.message });
+    }
+});
+
+router.post("/removePetFromFavorites", async (req, res) => {
+    try {
+        const { userId, petId } = req.body;
+
+        if (!userId || !petId) {
+            return res.status(400).json({ error: "Missing userId or petId" });
+        }
+
+        const updatedFavorites = await removePetFromFavorites(userId, petId);
+        res.status(200).json({ message: "Pet removed from favorites", favorites: updatedFavorites });
+    } catch (error) {
+        res.status(400).json({ error: error.message });
+    }
+});
+
+router.get("/getFavorites/:userId", async (req, res) => {
+    try {
+        const { userId } = req.params;
+        const user = await getUserById(userId);
+
+        if (!user) {
+            return res.status(404).json({ error: "User not found" });
+        }
+
+        res.status(200).json({ favorites: user.favorites });
+    } catch (error) {
+        res.status(400).json({ error: error.message });
+    }
+});
+
+
 export default router;