Following and followers

public/js/followersAndFollowing.js

@@ -36,6 +36,18 @@ function outputUsers(results, container) {
 function createUserHtml(userData, showFollowButton) {
 
     const name = userData.firstName + " " + userData.lastName;
+    const isFollowing = (userLoggedIn.following && userLoggedIn.following.includes(userData._id))
+    let followButton = ""
+
+    const text = isFollowing ? "Подписан" : "Подписаться"
+    const buttonClass = isFollowing ? "followButton following" : "followButton"
+
+
+    if (showFollowButton && userLoggedIn._id != userData._id) {
+        followButton = `<div class='followButtonContainer'>
+                            <button class='${buttonClass}' data-user='${userData._id}'>${text}</button>
+                    </div>`
+    }
 
     return `<div class='user'>
                 <div class='userImageContainer'>
@@ -47,5 +59,6 @@ function createUserHtml(userData, showFollowButton) {
                         <span class='username'>@${userData.username}</span>
                     </div>
                 </div>
+                ${followButton}
             </div>`;
 }

public/js/home.js

@@ -1,5 +1,5 @@
 $(document).ready(() => {
-    $.get("/api/posts", results => {
+    $.get("/api/posts", { followingOnly: true }, results => {
         outputPosts(results, $(".postsContainer"))       
     })
 })

routes/api/posts.js

@@ -11,12 +11,33 @@ router.get("/", async (req, res, next) => {
 
     const searchObj = req.query;
 
-    if (searchObj.isReply !== undefined) {
+    if(searchObj.isReply !== undefined) {
         const isReply = searchObj.isReply == "true";
         searchObj.replyTo = { $exists: isReply };
         delete searchObj.isReply;
     }
 
+    if(searchObj.followingOnly !== undefined) {
+        const followingOnly = searchObj.followingOnly == "true";
+
+        if(followingOnly) {
+            let objectIds = [];
+
+            if(!req.session.user.following) {
+                req.session.user.following = [];
+            }
+
+            req.session.user.following.forEach(user => {
+                objectIds.push(user);
+            })
+
+            objectIds.push(req.session.user._id);
+            searchObj.postedBy = { $in: objectIds };
+        }
+
+        delete searchObj.followingOnly;
+    }
+
     const results = await getPosts(searchObj);
 
     res.status(200).send(results)