Pramith/backend routes

backend/controllers/userController.js

@@ -1,4 +1,5 @@
 const user = require("../models/User");
+const course = require("../models/Course");
 
 exports.addUser = async (req, res) => {
     // Check that required data is given
@@ -48,4 +49,68 @@ exports.updateUser = async (req, res) => {
     // Object of fields and values to update in the user object
     const updateParams = req.body;
     res.status(200).send("Updated user.");
+};
+
+// DEFAULTS TO ADDING USER AS A STUDENT
+exports.addUserToCourse = async (req, res) => {
+    const courseUUID = req.params.courseId;
+    let userObj = req.user;
+    if (!courseUUID || !userObj) {
+        res.status(422).json({
+            status: 422,
+            error: "Missing parameter: courseUUID or userUUID"
+        });
+        return;
+    };
+
+
+    // Adds the user to the course as a student. If fails, responds with an error. Only works with users we've manually made
+    try {
+        let courseObj = await course.getCourseById(courseUUID);
+        await courseObj.addStudent(userObj.getUUID());
+        await userObj.addStudentCourse(courseObj.getUUID());
+        res.status(200).send("Added user to course.");
+    } catch (e) {
+        res.status(410).json({
+            status: 410,
+            error: e
+        });
+    };
+};
+
+exports.getUserType = async (req, res) => {
+    const courseUUID = req.params.courseId;
+    const userObj = req.user;
+
+    if (!courseUUID || !userObj) {
+        res.status(422).json({
+            status: 422,
+            error: "Missing parameter: courseUUID or user"
+        });
+        return;
+    };
+
+    // Grabs the user's type based on the courseUUID. If fails, responds with an error.
+    try {
+        const courseObj = await course.getCourseById(courseUUID);
+
+        if (courseObj.getInstructorList().indexOf(userObj.getUUID()) != -1) {
+            res.status(200).json({
+                type: "Instructor"
+            });
+        } else if (courseObj.getStudentList().indexOf(userObj.getUUID()) != -1) {
+            res.status(200).json({
+                type: "Student"
+            });
+        } else {
+            res.status(200).json({
+                error: "User not in this class"
+            });
+        }
+    } catch (e) {
+        res.status(410).json({
+            status: 410,
+            error: e
+        });
+    };
 };

backend/models/Comment.js

@@ -62,6 +62,11 @@ class Comment {
         await this.push();
     }
 
+    decrementScore = async () => {
+        this.props.score--;
+        await this.push();
+    }
+
     setEndorsed = async (newValue) => {
         this.props.isEndorsed = newValue;
         await this.push();

backend/models/Course.js

@@ -17,7 +17,7 @@ class Course {
     }
 
     getInstructorList() {
-        return this.props.instructorList;
+        return this.props.instructorList.slice(1, this.props.instructorList.length);
     }
 
     getUUID() {

backend/models/Post.js

@@ -78,6 +78,11 @@ class Post {
         await this.push();
     }
 
+    decrementScore = async () => {
+        this.props.score--;
+        await this.push();
+    }
+
     setAnnouncement = async (newValue) => {
         this.props.isAnnouncement = newValue;
         await this.push();

backend/models/User.js

@@ -35,6 +35,34 @@ class User {
         await this.push();
     }
 
+    addLikedPost = async (postId) => {
+        this.props.likedPostList.push(postId);
+        let post = await getPostById(postId);
+        post.incrementScore();
+        await this.push();
+    }
+
+    removeLikedPost = async (postId) => {
+        this.props.likedPostList.splice(likedPostList.indexOf(postId), 1);
+        let post = await getPostById(postId);
+        post.decrementScore();
+        await this.push();
+    }
+
+    addLikedComment = async (commentId) => {
+        this.props.likedCommentList.push(commentId);
+        let comment = await getCommentById(commentId);
+        comment.incrementScore();
+        await this.push();
+    }
+
+    removeLikedComment = async (commentId) => {
+        this.props.likedCommentList.splice(likedCommentList.indexOf(commentId), 1);
+        let comment = await getCommentById(commentId);
+        comment.decrementScore();
+        await this.push();
+    }
+
     getName() {
         return this.props.name;
     }
@@ -67,6 +95,14 @@ class User {
         return this.props.followingList.slice(1, this.props.followingList.length);
     }
 
+    getLikedPostList() {
+        return this.props.likedPostList.slice(1, this.props.likedPostList.length);
+    }
+
+    getLikedCommentList() {
+        return this.props.likedCommentList.slice(1, this.props.likedCommentList.length);
+    }
+
     getIcon() {
         return this.props.icon;
     }
@@ -88,6 +124,9 @@ class User {
             postList: this.props.postList,
             commentList: this.props.commentList,
             followingList: this.props.followingList,
+            // Add this ternary for new fields to prevent old data from crashing
+            likedPostList: this.props.likedPostList ? this.props.likedPostList : [],
+            likedCommentList: this.props.likedCommentList ? this.props.likedCommentList : [],
             icon: this.props.icon
         });
     } 
@@ -109,6 +148,8 @@ module.exports.pushUserToFirebase = (updateParams) => {
                 postList: ["dummy_post_id"],
                 commentList: ["dummy_comment_id"],
                 followingList: ["dummy_post_id"],
+                likedPostList: ["dummy_post_id"],
+                likedCommentList: ["dummy_comment_id"],
                 icon: "anonymous.jpg"
             });
             resolve("Everything worked");
@@ -128,7 +169,6 @@ getUserById = async (uuid) => {
     return new Promise((resolve, reject) => {
         ref.once("value", function(snapshot) {
             const r = new User(snapshot.val());
-            //console.log(r.props.name);
             resolve(r);
         }, function (errorObject) {
             reject(errorObject);

backend/routes/api/user.js

@@ -8,4 +8,8 @@ router.route("/")
     .get(userController.getUser) // Read
     .put(userController.updateUser) // Update
 
+// Matches with "/api/user/courseId"
+router.route("/:courseId")   // regex? how do we format the courseID
+    .post(userController.addUserToCourse)
+    .get(userController.getUserType)
 module.exports = router;

backend/test/courseTest.js

@@ -23,8 +23,8 @@ describe('course', () => {
             name: "cse110", 
             term: "sp20",
             uuid: "course1",
-            instructorList: "user2", 
-            studentList: "user1",
+            instructorList: ["user2"], 
+            studentList: ["user1"],
             tagList: ["dummy_tag"],
             postList: ["dummy_post"],
         }