Refactor utils and organize routes/schema

This commit is contained in:
Johnathon Slightham
2021-05-19 20:08:04 -04:00
parent ac404a0ac5
commit cafb324ddb
15 changed files with 128 additions and 115 deletions

77
routes/category.route.js Normal file
View File

@@ -0,0 +1,77 @@
const utils = require('../utils/utils');
const express = require('express');
const categoryRoutes = express.Router();
let Post = require('../schema/post.model');
let Category = require('../schema/category.model');
let Session = require('../schema/session.model');
let User = require('../schema/user.model');
/*
POST - /category/create
Create a category
Response: 201 - Created
401 - Unauthorized
*/
categoryRoutes.route('/create').post((req, res) => {
if (!req.body) {
res.status(401).send("Missing body");
return;
}
utils.account.checkSession(req.body.userId, req.body.sessionId, (isValidId) => {
utils.account.isAdmin(req.body.userId, (isAdmin) => {
if (isValidId && isAdmin) {
let c = new Category(req.body);
c.save()
.then(() => {
res.json(c);
})
.catch((e) => {
console.error(e);
res.status(500).send("Error creating category");
});
} else {
res.status(401).send("Invalid permissions to create category.");
return;
}
})
});
});
/*
GET - /category/all
Get all categories
Response: 200 - OK
*/
categoryRoutes.route('/all').get((req, res) => {
Category.find({}, (err, cArr) => {
if (err) {
console.error(err);
res.status(500).send("Error getting categories");
return;
}
res.status(200).send(cArr);
});
});
/*
GET - /category/posts
Get all posts with specified category
Response: 200 - OK
*/
categoryRoutes.route('/posts').get((req, res) => {
Post.find({}, (err, postArr) => {
if (err) {
console.error(err);
res.status(500).send("Error getting posts");
return;
}
postArr = postArr.filter(post => post.category.includes(req.body._id));
res.status(200).send(postArr);
});
});
module.exports = categoryRoutes;

181
routes/post.route.js Normal file
View File

@@ -0,0 +1,181 @@
const utils = require('../utils/utils');
const express = require('express');
const postRoutes = express.Router();
let Post = require('../schema/post.model');
let Category = require('../schema/category.model');
let Session = require('../schema/session.model');
let User = require('../schema/user.model');
/*
POST - /post/create
Create a post
Response: 201 - Created
401 - Unauthorized
*/
postRoutes.route('/create').post((req, res) => {
if (!req.body) {
res.status(401).send("Missing body");
return;
}
utils.account.checkSession(req.body.userId, req.body.sessionId, (isValidId) => {
utils.account.isAdmin(req.body.userId, (isAdmin) => {
if (isValidId && isAdmin) {
let p = new Post(req.body);
p.date = utils.date.dateToEpoch(p.date);
p.save()
.then(() => {
res.json(p);
})
.catch((e) => {
console.error(e);
res.status(500).send("Error creating post");
});
} else {
res.status(401).send("Invalid permissions to create post.");
return;
}
})
});
});
/*
POST - /post/delete
Create a post
Response: 200 - Deleted
401 - Unauthorized
*/
postRoutes.route('/delete').post((req, res) => {
if (!req.body) {
res.status(401).send("Missing body");
return;
}
utils.account.checkSession(req.body.userId, req.body.sessionId, (isValidId) => {
utils.account.isAdmin(req.body.userId, (isAdmin) => {
if (isValidId && isAdmin) {
Post.findByIdAndDelete(req.body._id, (err, r) => {
if (err) {
res.status(500).send("Error deleting post");
return;
}
res.status(200).send("Deleted post");
});
} else {
res.status(401).send("Invalid permissions to delete post.");
return;
}
})
});
});
/*
POST - /post/edit
Create a post
Response: 200 - Edited
401 - Unauthorized
*/
postRoutes.route('/edit').post((req, res) => {
if (!req.body) {
res.status(401).send("Missing body");
return;
}
utils.account.checkSession(req.body.userId, req.body.sessionId, (isValidId) => {
utils.account.isAdmin(req.body.userId, (isAdmin) => {
if (isValidId && isAdmin) {
Post.findById(req.body._id, (err, r) => {
if (err) {
res.status(500).send("Error editing post");
return;
}
r.save()
.then(() => {
res.json(r);
})
.catch((e) => {
console.error(e);
res.status(500).send("Error creating post");
});
res.status(200).send("Edited post");
});
} else {
res.status(401).send("Invalid permissions to delete post.");
return;
}
})
});
});
/*
POST - /post/id
Get post by id
Response: 200 - OK
*/
postRoutes.route('/id').post((req, res) => {
Post.findById(req.body._id, (err, post) => {
if (err) {
console.error(err);
res.status(500).send("Error getting posts");
return;
}
res.status(200).send(post);
});
});
/*
POST - /post/date
Get post by date
Response: 200 - OK
*/
postRoutes.route('/date').post((req, res) => {
let d = new Date(req.body.date);
d = utils.date.dateToEpoch(d);
Post.find({ date: d}, (err, post) => {
if (err) {
console.error(err);
res.status(500).send("Error getting posts");
return;
}
res.status(200).send(post);
});
});
/*
GET - /post/all
Get all posts
Response: 200 - OK
*/
postRoutes.route('/all').get((req, res) => {
Post.find({}, (err, postArr) => {
if (err) {
console.error(err);
res.status(500).send("Error getting posts");
return;
}
res.status(200).send(postArr);
});
});
/*
GET - /posts/today
Get the posts for today
Response: 200 - Removed session
400 - No session exists
*/
postRoutes.route('/today').get((req, res) => {
let date = new Date();
date = utils.date.dateToEpoch(date);
Post.find({ date: date }, (err, postArr) => {
if (err) {
console.error(err);
res.status(500).send("Error getting posts");
return;
}
res.status(200).send(postArr);
});
});
module.exports = postRoutes;

250
routes/user.route.js Normal file
View File

@@ -0,0 +1,250 @@
const utils = require('../utils/utils');
const express = require('express');
const bcrypt = require('bcrypt');
const userRoutes = express.Router();
const saltRounds = 10;
const sessionLength = 25;
let Session = require('../schema/session.model');
let User = require('../schema/user.model');
let Post = require('../schema/post.model');
/*
POST - /user/create
Create a user
Response: 201 - Created
409 - Account already exists
*/
userRoutes.route('/create').post((req, res) => {
if (!req.body) {
res.status(401).send("Missing body");
return;
} else if (!req.body.email || !req.body.password || !req.body.name) {
res.status(401).send("Missing body");
return;
} else if (req.body.email == "" || req.body.password == "" || req.body.name == "") {
res.status(401).send("Empty fields");
return;
}
let u = new User(req.body);
bcrypt.hash(u.password, saltRounds, (err, hash) => {
if (err) {
console.error(err);
res.status(500).send("Error creating user");;
} else {
u.password = hash;
User.find({ email: u.email }, (err, arr) => {
if (err) {
console.error(err);
res.status(500).send("Error creating user");
}
// Account already exists
if (arr.length != 0) {
res.status(409).send("Account already exists");
return;
}
u.permission = 0;
u.save()
.then(() => {
res.status(201).send("Success creating user");
})
.catch(() => {
res.status(500).send("Error creating user");;
});
});
}
});
});
/*
POST - /user/login
Login a user
Response: 201 - Created session
401 - Incorrect
*/
userRoutes.route('/login').post((req, res) => {
if (!req.body) {
res.status(401).send("Missing body");
return;
} else if (!req.body.email || !req.body.password) {
res.status(401).send("Missing body");
return;
} else if (req.body.email == "" || req.body.password == "") {
res.status(401).send("Empty fields");
return;
}
User.findOne({ email: req.body.email }, (err, u) => {
if (err) {
console.error(err);
res.status(500).send("Error logging in user");
return;
}
if (!u) {
res.status(401).send("No user exists with that email");
return;
}
bcrypt.compare(req.body.password, u.password, (err, result) => {
if (err) {
console.error(err);
res.status(500).send("Error logging in user");
return;
}
if (result) {
let s = new Session();
s.sessionId = generateSession();
s.userId = u._id;
s.date = new Date();
s.save()
.then(() => {
res.json(s);
})
.catch(() => {
res.status(500).send("Error logging in user");
});
} else {
res.status(401).send("Incorrect password");
}
});
});
});
/*
POST - /user/logout
Logout a user
Response: 200 - Removed session
400 - No session exists
*/
userRoutes.route('/logout').post((req, res) => {
Session.findOne({ sessionId: req.body.sessionId }, (err, sess) => {
if (err) {
console.error(err);
res.status(500).send("Error logging out");
return;
}
if (!sess) {
res.status(400).send("No session found");
return;
}
sess.delete()
.then(() => {
res.status(201).send("Success deleting session");
})
.catch((e) => {
console.error(e);
res.status(500).send("Error logging out");
});
});
});
// TODO: Add forgotten password route
/*
POST - /user/favorite/add
Add a favorite article
Response: 200 - OK
401 - Unauthorized
*/
userRoutes.route('/favorite/add').post((req, res) => {
utils.account.checkSession(req.body.userId, req.body.sessionId, valid => {
if (valid) {
User.findById(req.body.userId, (err, user) => {
if (err) {
console.error(err);
res.status(500).send("Error adding article");
return;
}
user.favorites.push(req.body.postId);
user.save()
.then(() => {
res.status(201).send("Success saving article");
})
.catch((e) => {
console.error(e);
res.status(500).send("Error saving article");
});
})
} else {
res.status(401).send("Unauthorized");
}
})
})
/*
POST - /user/favorite/remove
Remove a favorite article
Response: 200 - OK
401 - Unauthorized
*/
userRoutes.route('/favorite/remove').post((req, res) => {
utils.account.checkSession(req.body.userId, req.body.sessionId, valid => {
if (valid) {
User.findById(req.body.userId, (err, user) => {
if (err) {
console.error(err);
res.status(500).send("Error removing article");
return;
}
user.favorites = utils.array.removeValue(user.favorites, req.body.articleId);
user.save()
.then(() => {
res.status(201).send("Success removing article");
})
.catch((e) => {
console.error(e);
res.status(500).send("Error removing article");
});
})
} else {
res.status(401).send("Unauthorized");
}
})
})
/*
POST - /user/favorite/get
Get all favorite articles
Response: 200 - OK
401 - Unauthorized
*/
userRoutes.route('/favorite/get').post((req, res) => {
utils.account.checkSession(req.body.userId, req.body.sessionId, valid => {
if (valid) {
User.findById(req.body.userId, (err, user) => {
if (err) {
console.error(err);
res.status(500).send("Error removing article");
return;
}
Post.find({ '_id': { $in: user.favorites } }, (err, postArray) => {
res.json(postArray);
})
})
} else {
res.status(401).send("Unauthorized");
}
})
})
module.exports = userRoutes;
function generateSession() {
var length = sessionLength;
var result = [];
var characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-+!@#$%^&*()';
var charactersLength = characters.length;
for (var i = 0; i < length; i++) {
result.push(characters.charAt(Math.floor(Math.random() * charactersLength)));
}
return result.join('');
}