vote feedback

frontend/src/app/lobby/page.tsx

@@ -17,6 +17,8 @@ const MAX_PARTY_SIZE = 8;
 
 export default function Lobby() {
   const [userList, setUserList] = useState<any[]>([]);
+  const [noList, setNoList] = useState<any[]>([]);
+  const [yesList, setYesList] = useState<any[]>([]);
   const [sessionSocket, setSessionSocket] = useState<WebSocket | null>(null);
   const { sessionId, setSessionId } = useSession();
   const [sessionState, setSessionState] = useState("lobby");
@@ -34,6 +36,8 @@ export default function Lobby() {
         addUser(data[key]);
       } else if (key == "track") {
         getTrack();
+      } else if (key == "voted") {
+        voteUser(data[key]);
       }
     }
     console.log(data);
@@ -78,6 +82,7 @@ export default function Lobby() {
       id: user["id"],
       name: user["username"],
       avatar: `https://i.pravatar.cc/150?img=${user["id"]}`,
+      hidden: false,
     };
 
     setUserList((prevUserList) => {
@@ -89,8 +94,67 @@ export default function Lobby() {
     });
   }
 
+  // Visually processes the incoming vote for a user.
+  function voteUser(vote: object) {
+    const userId = vote["id"];
+    const userName = vote["username"];
+    const points = vote["points"];
+    if (points === 0) {
+      setYesList((prevYesList) => {
+        //if (prevYesList.find((u) => u.id === userId)) return prevYesList;
+        return prevYesList.filter((u) => u.id !== userId);
+      });
+      setNoList((prevNoList) => {
+        if (prevNoList.find((u) => u.id === userId)) {
+          return prevNoList;
+        }
+        return [
+          ...prevNoList,
+          {
+            id: userId,
+            name: userName,
+            avatar: `https://i.pravatar.cc/150?img=${userId}`,
+          },
+        ];
+      });
+    } else {
+      setNoList((prevNoList) => {
+        return prevNoList.filter((u) => u.id !== userId);
+      });
+      setYesList((prevYesList) => {
+        if (prevYesList.find((u) => u.id === userId)) {
+          return prevYesList;
+        }
+        return [
+          ...prevYesList,
+          {
+            id: userId,
+            name: userName,
+            avatar: `https://i.pravatar.cc/150?img=${userId}`,
+          },
+        ];
+      });
+    }
+    setUserList((prevUserList) => {
+      // Set hidden true on the user that voted:
+      return prevUserList.map((user) => {
+        if (user.id === userId) {
+          return { ...user, hidden: true };
+        }
+        return user;
+      });
+    });
+  }
+
   async function getTrack() {
     setSongLoading(true);
+    setUserList((prevUserList) => {
+      return prevUserList.map((user) => {
+        return { ...user, hidden: false };
+      });
+    });
+    setYesList([]);
+    setNoList([]);
     const { data: session } = await api.sessions.get({ session_id: sessionId });
     const trackId = session?.current_track;
     if (trackId) {
@@ -207,7 +271,7 @@ export default function Lobby() {
         <div className="flex flex-col items-center justify-center">
           <div className="flex flex-row items-center justify-center gap-x-32">
             <div className="grid min-w-52 grid-flow-row grid-cols-3 items-start justify-start gap-4 rounded bg-blue-950/80 p-4">
-              {userList.map((user) => (
+              {noList.map((user) => (
                 <div
                   key={user.id}
                   className="flex h-[120px] w-[120px] flex-col items-center justify-center p-4"
@@ -244,7 +308,7 @@ export default function Lobby() {
               </div>
             )}
             <div className="grid min-w-52 grid-flow-row grid-cols-3 items-start justify-start gap-4 rounded bg-blue-950/80 p-4">
-              {userList.map((user) => (
+              {yesList.map((user) => (
                 <div
                   key={user.id}
                   className="flex h-[120px] w-[120px] flex-col items-center justify-center p-4"
@@ -282,7 +346,7 @@ export default function Lobby() {
           {userList.map((user) => (
             <div
               key={user.id}
-              className="flex h-[100px] w-[100px] flex-col items-center justify-center p-4"
+              className={`flex h-[100px] w-[100px] flex-col items-center justify-center p-4 ${user.hidden ? "hidden" : ""}`}
             >
               <img src={user.avatar} alt={user.name} width={75} height={75} />
               <div className="mt-1 min-w-[90px] rounded-3xl px-3 text-center text-3xl font-semibold text-white">