add freinds feature and display on home page

Author: harsh1206

assets/js/toggle_freindship.js

@@ -0,0 +1,70 @@
+console.log('freindship script loaded');
+
+class ToggleFreindship{
+
+
+    constructor(toggleElement){
+        this.toggler=toggleElement;
+        this.toggleFreindship();
+    }
+
+    toggleFreindship(){
+
+
+        $(this.toggler).click(function(e){
+
+
+            e.preventDefault();
+            let self=this;
+             
+            $.ajax({
+                type:'POST',
+                url:$(self).attr('href')
+            })
+            .done(function(data){
+                
+                console.log(data);
+
+                if(data.data.added==true){
+                    
+                    $(self).html('ADD');
+                     
+                    new Noty({
+                        theme:'relax',
+                         text: 'Removed Freind',
+                         type:'success',
+                        layout:'topRight',
+                        timeout:1500
+       
+                     }).show();
+
+
+                }else{
+
+                    $(self).html('REMOVE');
+                    new Noty({
+                        theme:'relax',
+                         text: 'Added Freind',
+                         type:'success',
+                        layout:'topRight',
+                        timeout:1500
+       
+                     }).show();
+                }
+
+            })
+            .fail(function(errData){
+
+                console.log('error in completing the request');
+
+            });
+
+        });
+
+
+
+    }
+
+
+
+}

controllers/freindship_controller.js

@@ -0,0 +1,59 @@
+const Freindship=require('../models/freindship');
+const User=require('../models/user');
+const Friendship = require('../models/freindship');
+
+module.exports.toggleFriend=async function(req,res){
+    
+    try{
+        
+     let added = false;
+
+     let from=await User.findById(req.user._id);
+     let to=await User.findById(req.query.id);
+
+     let exsitingFreindship=await Freindship.findOne({
+
+        from_user:from._id,
+        to_user:to._id
+
+     });
+    
+     if(exsitingFreindship){
+       
+        from.freindship.pull(exsitingFreindship._id);
+        to.freindship.pull(exsitingFreindship._id);
+        from.save();
+        to.save();
+        exsitingFreindship.remove();
+        added=true; 
+
+     }else{
+        
+        let newFreindship = await Friendship.create({
+            from_user: from._id,
+            to_user:to._id
+        })
+
+        from.freindship.push(newFreindship._id);
+        to.freindship.push(newFreindship._id);
+        from.save();
+        to.save();
+        
+     }
+
+     return res.json(200, {
+        message: "Request successful!",
+        data: {
+            added: added
+        }
+    })
+
+   
+    }catch(err){
+         
+        console.log(err);
+
+
+    } 
+
+}

controllers/home_controller.js

@@ -1,6 +1,8 @@
 const Posts=require('../models/posts');
 const Post = require('../models/posts');
 const User=require('../models/user');
+const Friendship = require('../models/freindship');
+
 
 
 
@@ -24,17 +26,34 @@ module.exports.home =async function(req, res){
         }
       }).populate('likes');
 
-    
-    let users=await User.find({});
+      // console.log(req.user);   
+        
+      const users=await User.find({});
+
+      if(req.user){
+
+      
+     let freindships = await Friendship.find({from_user:req.user._id}).populate('to_user');     
 
 
       return res.render('home',{
 
           all_users:users,
           title:"Home",
-          posts:posts
-
+          posts:posts,
+          freindships:freindships
       }); 
+
+     }else{
+        
+      return res.render('home',{
+         
+        all_users:users,
+        title:"Home",
+        posts:posts,
+        }); 
+       
+     }
 
       }catch(err){
         console.log('Error',err);

controllers/users_controller.js

@@ -2,17 +2,38 @@ const User=require('../models/user');
 const fs=require('fs');
 const path=require('path');
 const Noty=require('noty');
+const Friendship = require('../models/freindship');
 
 
 
 module.exports.profile = function(req, res){
-    User.findById(req.params.id,function(err,user){
-    return res.render('user_profile', {
-        title: 'User Profile',
-        profile_user:user
-    })
+       
+      
+
+    User.findById(req.params.id,function(err,user){   
+           
+      
+         
+
+         console.log(user);
+        
+
+         return res.render('user_profile', {
+            title: 'User Profile',
+             profile_user:user,
+         }) 
 
    });
+
+    //   let user=User.findById(req.params.id);
+
+    //   console.log(user);
+
+    //   return res.render('user_profile',{
+    //       title:'User Profile',
+    //       profile_user:user
+    //   });
+
 };
 
 

models/freindship.js

@@ -0,0 +1,27 @@
+const mongoose=require('mongoose');
+
+
+const friendshipSchema=new mongoose.Schema({
+
+    // user who sent the request
+    from_user:{
+
+        type:mongoose.Schema.Types.ObjectId,
+        ref:'User'
+    },
+    // the user who acccepted the request,the naming is just to understand,otherwise the user's wont see a difference
+    to_user:{
+         
+       type:mongoose.Schema.Types.ObjectId,
+       ref:'User' 
+    }  
+
+
+},{
+     
+    timestamps:true
+    
+});
+
+const Friendship=mongoose.model('Friendship',friendshipSchema);
+module.exports=Friendship;

models/user.js

@@ -24,7 +24,16 @@ const userSchema=new mongoose.Schema({
     },
     avatar:{
           type:String
-    }
+    },
+    freindship: [
+
+         {
+             type:mongoose.Schema.Types.ObjectId,
+             ref:'Freindship'
+         }
+
+    ]
+
 
 }, {
 

routes/freindship.js

@@ -0,0 +1,10 @@
+const express = require('express');
+
+const router = express.Router();
+
+const freindshipController=require('../controllers/freindship_controller');
+
+
+router.post('/toggle',freindshipController.toggleFriend);
+
+module.exports=router;

routes/index.js

@@ -12,6 +12,7 @@ router.use('/posts',require('./posts'));
 router.use('/comments',require('./comments'))
 router.use('/reset-password',require('./reset_password'));
 router.use('/likes',require('./likes'));
+router.use('/freindship',require('./freindship'));
 
 router.use('/api',require('./api'));
 

views/home.ejs

@@ -35,15 +35,28 @@
 
  <section id="user-freinds">
 
-    <h4>Freinds</h4>
+    <h4>ALL-USERS</h4>
    <% for(u of all_users){%>
 
         <p>
           <a href="/users/profile/<%= u.id %>"><%= u.name %></a>
         </p>
-       
 
    <%}%>   
+   
+    <h4>FREINDS</h4>
+   
+   <% if(locals.user) {%> 
+
+     <% for(u of freindships){%>
+           
+        <p>
+           <a href="/users/profile/<%= u.to_user._id %>"><%= u.to_user.name %></a>
+        </p> 
+    
+    <%}%>
+
+    <%}%>
 
 
  </section>

views/user_profile.ejs

@@ -22,4 +22,20 @@
             <p><%= profile_user.name %></p>
             <p><%= profile_user.email %></p>
 
-        <%}%>
+        <%}%>
+
+ 
+        <a href="/freindship/toggle/?id=<%= profile_user.id %>" class="toggle-freindship-button">ADD/REMOVE</a> 
+        
+          
+
+        <script src="/js/toggle_freindship.js"></script>
+        <script>
+
+            $('.toggle-freindship-button').each(function(){
+             let self = this;
+             let toggleFreindship = new ToggleFreindship(self);
+           });
+
+        </script>  
+