add follow function

manupacheco · May 7, 2018 · a798c7e · a798c7e
1 parent c872ad3
commit a798c7e
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 0 deletions.
diff --git a/models/company.js b/models/company.js
@@ -20,6 +20,7 @@ const companySchema = new Schema({
   messages: [Msg.schema],
   send: [Msg.schema],
   influencersFavs: [Schema.Types.ObjectId],
+  followers: [Schema.Types.ObjectId],
 }, {
   timestamps: { createdAt: 'created_at', updatedAt: 'updated_at' },
 });

diff --git a/models/influencer.js b/models/influencer.js
@@ -35,6 +35,8 @@ const influencerSchema = new Schema({
   messages: [Msg.schema],
   send: [Msg.schema],
   campaignsFavs: [Schema.Types.ObjectId],
+  companiesFavs: [Schema.Types.ObjectId],
+  followers: [Schema.Types.ObjectId],
   instagram: {
     username: String,
     biography: String,

diff --git a/routes/api.js b/routes/api.js
@@ -598,4 +598,76 @@ router.put('/messages/read/:idMessage', (req, res, next) => {
   }
 });
 
+// Follow & Unfollow function
+
+router.put('/follow/:userId', (req, res, next) => {
+  if (!req.session.currentUser) {
+    return res.status(401).json({ error: 'unauthorized' });
+  }
+  const options = {
+    new: true,
+  };
+
+  const { _id, role } = req.session.currentUser;
+  const { userId } = req.params;
+  const id = mongoose.Types.ObjectId(userId);
+
+  if (role === 'company') {
+    Company.findByIdAndUpdate({ _id }, { $push: { influencersFavs: userId } }, options)
+      .then(() => {
+        Influencer.findByIdAndUpdate({ _id: id }, { $push: { followers: _id } }, options)
+          .then((result) => {
+            res.status(200).json(result);
+          })
+          .catch(next);
+      })
+      .catch(next);
+  } else if (role === 'influencer') {
+    Influencer.findByIdAndUpdate({ _id }, { $push: { companiesFavs: userId } }, options)
+      .then(() => {
+        Company.findByIdAndUpdate({ _id: id }, { $push: { followers: _id } }, options)
+          .then((result) => {
+            res.status(200).json(result);
+          })
+          .catch(next);
+      })
+      .catch(next);
+  }
+});
+
+router.put('/unfollow/:userId', (req, res, next) => {
+  if (!req.session.currentUser) {
+    return res.status(401).json({ error: 'unauthorized' });
+  }
+  const options = {
+    new: true,
+  };
+
+  const { _id, role } = req.session.currentUser;
+  const { userId } = req.params;
+  const id = mongoose.Types.ObjectId(userId);
+
+  if (role === 'company') {
+    Company.findByIdAndUpdate({ _id }, { $pull: { influencersFavs: userId } }, options)
+      .then(() => {
+        Influencer.findByIdAndUpdate({ _id: id }, { $pull: { followers: _id } }, options)
+          .then((result) => {
+            res.status(200).json(result);
+          })
+          .catch(next);
+      })
+      .catch(next);
+  } else if (role === 'influencer') {
+    Influencer.findByIdAndUpdate({ _id }, { $pull: { companiesFavs: userId } }, options)
+      .then(() => {
+        Company.findByIdAndUpdate({ _id: id }, { $pull: { followers: _id } }, options)
+          .then((result) => {
+            res.status(200).json(result);
+          })
+          .catch(next);
+      })
+      .catch(next);
+  }
+});
+
 module.exports = router;