Added /login error handling for unregistered users

This commit is contained in:
Matthew Patrick 2021-11-18 11:58:37 +07:00
parent b59a15440f
commit dc1829cffd

View file

@ -1,242 +1,292 @@
const express = require('express'); const express = require("express");
const bcrypt = require("bcrypt"); const bcrypt = require("bcrypt");
const randtoken = require("rand-token"); const randtoken = require("rand-token");
const router = express.Router(); const router = express.Router();
const db = require('../modules/db'); const db = require("../modules/db");
const mail = require('../modules/mail'); const mail = require("../modules/mail");
const saltRounds = 10; const saltRounds = 10;
let config; let config;
try { try {
config = require('../config'); config = require("../config");
} catch (e) { } catch (e) {
console.log('No config file found'); console.log("No config file found");
process.exit(0); process.exit(0);
} }
router.get('/getproducts', (req, res) => { router.get("/getproducts", (req, res) => {
const conn = db.connect(); const conn = db.connect();
conn.execute('SELECT * FROM `products`', [], function (err, results) { conn.execute("SELECT * FROM `products`", [], function (err, results) {
res.json(results); res.json(results);
}); });
}) });
router.get('/getproduct', (req, res) => { router.get("/getproduct", (req, res) => {
const conn = db.connect(); const conn = db.connect();
conn.execute('SELECT * FROM `products` WHERE `ID` = ?', [req.query.id], function (err, results) { conn.execute(
res.json(results); "SELECT * FROM `products` WHERE `ID` = ?",
}); [req.query.id],
}) function (err, results) {
res.json(results);
router.post('/register', async (req, res) => {
if (!req.body.email || !req.body.password || !req.body.first_name || !req.body.last_name) {
res.status(400);
res.json({
'message': 'Bad Request'
})
} else {
const conn = db.connect();
conn.query("SELECT * FROM users WHERE email = ?", [req.body.email], async function (error, response, fields) {
if (error) {
res.status(400);
res.json({
'message': 'Bad Request'
})
} else {
if (response.length > 0) {
res.status(400);
res.json({
'message': 'Bad Request'
})
} else {
const encryptedPassword = await bcrypt.hash(req.body.password, saltRounds);
var users = {
first_name: req.body.first_name,
last_name: req.body.last_name,
email: req.body.email,
password: encryptedPassword,
session: '[]',
};
conn.query("INSERT INTO users SET ?", users, function (error, response, fields) {
if (error) {
res.status(400);
res.json({
'message': 'Bad Request'
})
} else {
conn.query('SELECT * FROM users WHERE email ="' + req.body.email + '"', function (err, result) {
if (err) {
res.status(400);
res.json({
'message': 'Bad Request'
})
}
if (result.length > 0) {
var token = randtoken.generate(20);
if (result[0].verify == 0) {
let subject = "Account Verification - Nekoya";
let content = `<p>Hello!!! Please click this link <a href="${config.host}/verify-mail?token=${token}">link</a> to verify your account!!! Thanks!!!</p>`;
var sent = mail.send(req.body.email, subject, content);
if (sent != "0") {
var data = {
token: token,
};
conn.query('UPDATE users SET ? WHERE email ="' + req.body.email + '"', data, function (err, result) {
if (err) {
res.status(400);
res.json({
'message': 'Bad Request'
})
} else {
res.status(200);
res.json({
'message': 'Register Verification Sent ~'
})
}
});
} else {
res.status(400);
res.json({
'message': 'Bad Request'
})
}
}
} else {
res.status(400);
res.json({
'message': 'Bad Request'
})
}
});
}
});
}
}
});
} }
}) );
});
router.post('/login', async (req, res) => { router.post("/register", async (req, res) => {
if (!req.body.email || !req.body.password) { if (
res.status(400); !req.body.email ||
res.json({ !req.body.password ||
'message': 'Bad Request' !req.body.first_name ||
}) !req.body.last_name
} else { ) {
const conn = db.connect(); res.status(400);
conn.query( res.json({
"SELECT * FROM users WHERE email = ?", message: "Bad Request",
[req.body.email], });
async function (error, response, fields) { } else {
const passCheck = await bcrypt.compare(req.body.password, response[0].password);
if (error) {
res.status(400);
res.json({
'message': 'Bad Request'
})
} else {
if (response.length > 0) {
if (passCheck) {
if (response[0].verify == 0) {
res.status(204);
res.json({
'message': 'Sorry You havent verified your email'
})
} else {
let token = randtoken.generate(256);
let session = JSON.parse(response[0].session);
session.push({
user_agent: req.body.ua || req.headers['user-agent'],
ip: req.body.ip || req.headers['x-forwarded-for'] || req.socket.remoteAddress,
session: token,
})
conn.query('UPDATE users SET ? WHERE email ="' + req.body.email + '"', {
session: JSON.stringify(session)
}, function (err, result) {
if (err) {
res.status(400);
res.json({
'message': 'Bad Request'
})
} else {
res.status(200);
res.json({
'id': response[0].id,
'first_name': response[0].first_name,
'last_name': response[0].last_name,
'email': response[0].email,
'verify': (response[0].verify == 1) ? true : false,
'session_token': token,
})
}
});
}
} else {
res.status(401);
res.json({
'message': 'Unauthorized'
})
}
} else {
res.status(400);
res.json({
'message': 'Bad Request'
})
}
}
}
);
}
})
router.get('/verify-mail', async (req, res) => {
const conn = db.connect(); const conn = db.connect();
conn.query( conn.query(
'SELECT * FROM users WHERE token ="' + req.query.token + '"', "SELECT * FROM users WHERE email = ?",
function (err, result) { [req.body.email],
if (err) { async function (error, response, fields) {
res.status(400); if (error) {
res.json({ res.status(400);
'message': 'Bad Request' res.json({
}) message: "Bad Request",
} });
if (result.length > 0) { } else {
if (result[0].verify == 0) { if (response.length > 0) {
var data = { res.status(400);
verify: 1, res.json({
}; message: "Bad Request",
db_connect.query( });
'UPDATE users SET ? WHERE email ="' + result[0].email + '"', } else {
data, const encryptedPassword = await bcrypt.hash(
function (err, result) { req.body.password,
if (err) { saltRounds
res.status(400); );
res.json({ var users = {
'message': 'Bad Request' first_name: req.body.first_name,
}) last_name: req.body.last_name,
} else { email: req.body.email,
res.status(200); password: encryptedPassword,
res.json({ session: "[]",
'message': 'Verified ~' };
}) conn.query(
} "INSERT INTO users SET ?",
} users,
); function (error, response, fields) {
if (error) {
res.status(400);
res.json({
message: "Bad Request",
});
} else { } else {
res.status(403); conn.query(
res.json({ 'SELECT * FROM users WHERE email ="' + req.body.email + '"',
'message': 'Forbidden' function (err, result) {
}) if (err) {
res.status(400);
res.json({
message: "Bad Request",
});
}
if (result.length > 0) {
var token = randtoken.generate(20);
if (result[0].verify == 0) {
let subject = "Account Verification - Nekoya";
let content = `<p>Hello!!! Please click this link <a href="${config.host}/verify-mail?token=${token}">link</a> to verify your account!!! Thanks!!!</p>`;
var sent = mail.send(
req.body.email,
subject,
content
);
if (sent != "0") {
var data = {
token: token,
};
conn.query(
'UPDATE users SET ? WHERE email ="' +
req.body.email +
'"',
data,
function (err, result) {
if (err) {
res.status(400);
res.json({
message: "Bad Request",
});
} else {
res.status(200);
res.json({
message: "Register Verification Sent ~",
});
}
}
);
} else {
res.status(400);
res.json({
message: "Bad Request",
});
}
}
} else {
res.status(400);
res.json({
message: "Bad Request",
});
}
}
);
} }
}
);
}
}
}
);
}
});
router.post("/login", async (req, res) => {
if (!req.body.email || !req.body.password) {
res.status(400);
res.json({
message: "Bad Request",
});
} else {
const conn = db.connect();
conn.query(
"SELECT * FROM users WHERE email = ?",
[req.body.email],
async function (error, response, fields) {
if (!response[0]) {
res.status(205);
res.json({
message: "Sorry Your email is not registered in our system",
});
} else {
const passCheck = await bcrypt.compare(
req.body.password,
response[0].password
);
if (error) {
res.status(400);
res.json({
message: "Bad Request",
});
} else {
if (response.length > 0) {
if (passCheck) {
if (response[0].verify == 0) {
res.status(204);
res.json({
message: "Sorry You havent verified your email",
});
} else {
let token = randtoken.generate(256);
let session = JSON.parse(response[0].session);
session.push({
user_agent: req.body.ua || req.headers["user-agent"],
ip:
req.body.ip ||
req.headers["x-forwarded-for"] ||
req.socket.remoteAddress,
session: token,
});
conn.query(
'UPDATE users SET ? WHERE email ="' + req.body.email + '"',
{
session: JSON.stringify(session),
},
function (err, result) {
if (err) {
res.status(400);
res.json({
message: "Bad Request",
});
} else {
res.status(200);
res.json({
id: response[0].id,
first_name: response[0].first_name,
last_name: response[0].last_name,
email: response[0].email,
verify: response[0].verify == 1 ? true : false,
session_token: token,
});
}
}
);
}
} else {
res.status(401);
res.json({
message: "Unauthorized",
});
}
} else { } else {
res.status(400);
res.json({
message: "Bad Request",
});
}
}
}
}
);
}
});
router.get("/verify-mail", async (req, res) => {
const conn = db.connect();
conn.query(
'SELECT * FROM users WHERE token ="' + req.query.token + '"',
function (err, result) {
if (err) {
res.status(400);
res.json({
message: "Bad Request",
});
}
if (result.length > 0) {
if (result[0].verify == 0) {
var data = {
verify: 1,
};
db_connect.query(
'UPDATE users SET ? WHERE email ="' + result[0].email + '"',
data,
function (err, result) {
if (err) {
res.status(400); res.status(400);
res.json({ res.json({
'message': 'Bad Request' message: "Bad Request",
}) });
} else {
res.status(200);
res.json({
message: "Verified ~",
});
}
} }
);
} else {
res.status(403);
res.json({
message: "Forbidden",
});
} }
); } else {
}) res.status(400);
res.json({
message: "Bad Request",
});
}
}
);
});
module.exports = router; module.exports = router;