Server.js script. Server.js is the brain of this application. In this file I manage all routing urls, run queries to be able to access, store, update data from database, manage login/registration system and manage private real time chat system.
let express = require('express');
let session = require('express-session');
let expressFileupload = require('express-fileupload');
let sharp = require('sharp');
let app = express();
let server = require('http').createServer(app);
let io = require('socket.io').listen(server);
let flash = require('connect-flash');
const mysql = require('mysql');
const path = require('path');
const jquery = require('jquery');
const fs = require('fs');
const htmlspecialchars = require('htmlspecialchars');
const htmlspecialchars_decode = require('htmlspecialchars_decode');
// prepare server
// app.use('/api', api); // redirect API calls
app.use('/custom', express.static(__dirname + '/node_modules/custom/')); // redirect root
app.use('/jsjq', express.static(__dirname + '/node_modules/jquery/dist')); // redirect JS jQuery
app.use('/js', express.static(__dirname + '/node_modules/bootstrap/dist/js')); // redirect bootstrap JS
app.use('/css', express.static(__dirname + '/node_modules/bootstrap/dist/css')); // redirect CSS bootstrap
app.use('/font', express.static(__dirname + '/font')); // redirect root
app.use('/img', express.static(__dirname + '/img')); // redirect to img
app.use(flash());
app.use(express.static('public'));
app.use(express.urlencoded({
extended: false
}));
app.use(session({
secret: 'secret',
resave: true,
saveUninitialized: true
}));
app.use(expressFileupload());
const connection = mysql.createConnection({
host: 'localhost',
user: 'root',
password: '',
database: 'chat',
multipleStatements: true
});
yangterhubung = []
app.set('views', './views');
app.set('view engine', 'ejs');
server.listen(3000);
console.log('Server sedang Berjalan...')
function now__() {
function pad(val) {
var str = "" + val;
var pad = "00";
var ans = pad.substring(0, pad.length - str.length) + str;
return ans;
}
let dt = new Date();
let time = pad(dt.getHours()) + ":" + pad(dt.getMinutes()) + " " + dt.getDate() + "/" + pad(dt.getMonth()+1) + "/" + dt.getFullYear();
return time
}
app.get('/', function(req, res) {
let flash_data = req.flash('login');
let flash_ = '';
if (flash_data != '') {
flash_ = flash_data[0];
}else {
flash_ = 'Succ';
}
console.log(flash_);
res.render('login', {flash_login: flash_data, type: flash_.substr(0, 4) == 'Succ' ? 'signup' : 'signin'});
})
app.get('/index/:id_receiver', (req, res) => {
if (req.session.loggedin) {
connection.query(
`SELECT cf.id_group_chat FROM cn_user cu
JOIN cn_friend cf on cf.id_user = cu.id_user or cf.id_friend = cu.id_user
WHERE cu.id_user = ${req.session.id_user}
AND (cf.id_user = ${req.params.id_receiver} or cf.id_friend = ${req.params.id_receiver});
SELECT name, id_user FROM cn_user WHERE id_user = ${req.params.id_receiver};`,
(error, results) => {
if (error) throw error;
connection.query(
`SELECT id_chat, user_id, message, time_chat, img FROM cn_chat WHERE id_group_chat = '${results[0][0].id_group_chat}' AND who = '${req.session.username}' ORDER BY id_chat ASC;
SELECT img_profile, information FROM cn_user WHERE id_user = ${req.params.id_receiver}`,
(error_, return_two) => {
if (error) throw error;
if (results[0][0] == undefined) {
res.redirect('/list');
} else {
// console.log(history_chat);
return_two[0].forEach((item, i) => {
return_two[0][i]['message'] = htmlspecialchars_decode(item.message)
})
res.render('index', {
user_login: req.session,
data_receiver: results[1][0],
group: results[0][0],
history_chat: return_two[0],
img_profile: return_two[1][0].img_profile,
information: htmlspecialchars_decode(return_two[1][0].information)
});
}
});
});
} else {
res.redirect('/');
}
});
app.get('/list', (req, res) => {
if (req.session.loggedin) {
connection.query(
`SELECT cu.*
FROM cn_friend cf, cn_user cu
WHERE cf.id_user = ${req.session.id_user}
AND cf.who = '${req.session.username}'
AND cf.id_friend = cu.id_user;
SELECT cc.*, cu.name, cu.id_user id_friend, cu.username, cu.img_profile
FROM cn_chat cc, cn_user cu
WHERE id_chat IN (SELECT MAX(id_chat)
FROM cn_chat WHERE id_group_chat LIKE '%${req.session.username}%'
AND who = '${req.session.username}'
GROUP BY id_group_chat)
AND cu.username = SUBSTRING_INDEX(cc.id_group_chat, "_", (CASE WHEN SUBSTRING_INDEX(cc.id_group_chat, "_", -1) = '${req.session.username}' THEN 1 ELSE -1 END));
SELECT img_profile FROM cn_user WHERE id_user = ${req.session.id_user}`,
(error, results) => {
results[1].forEach((item, i) => {
results[1][i]['message'] = htmlspecialchars_decode(item.message)
})
res.render('list', {
items: results[0],
chat_list: results[1],
user_login: req.session,
flash: req.flash('login'),
img_profile: results[2][0].img_profile
});
}
);
} else {
res.redirect('/');
}
});
app.post('/check_username', (req, res) => {
connection.query(
`SELECT username FROM cn_user WHERE username = '${htmlspecialchars(req.body.username)}'`,
(error, results) => {
if (results.length > 0) {
res.json(1);
}else {
res.json(0);
}
}
)
})
app.post('/signup', (req, res) => {
connection.query(
'INSERT INTO cn_user (username, name, password) VALUES (?, ?, ?)',
[htmlspecialchars(req.body.username), htmlspecialchars(req.body.name), htmlspecialchars(req.body.password)],
(error, results) => {
req.flash('login', 'Successfully added account, please login...');
res.redirect('/');
}
)
});
app.post('/signin', (req, res) => {
connection.query(
'SELECT * FROM cn_user WHERE username = ? AND password = ?',
[htmlspecialchars(req.body.username), htmlspecialchars(req.body.password)],
(error, results) => {
if (results.length > 0) {
req.session.loggedin = true;
req.session.username = results[0].username;
req.session.name = results[0].name;
req.session.id_user = results[0].id_user;
req.session.information = htmlspecialchars_decode(results[0].information);
req.session.img_profile = results[0].img_profile;
req.flash('login', 1);
res.redirect('/list');
} else {
req.flash('login', 'Wrong Password or Username!');
res.redirect('/');
}
}
);
});
app.post('/find', (req, res) => {
connection.query(
`SELECT * FROM cn_user WHERE username LIKE '%${req.body.username}%' and id_user != ${req.session.id_user}`,
(error, results) => {
let hasil;
if (results.length > 0) {
hasil = results;
}else {
hasil = 0;
}
res.render('result', {
user_login: req.session,
items: hasil,
});
}
)
});
app.post('/cek_friend', (req, res) => {
connection.query(
`SELECT * FROM cn_friend WHERE id_friend = ${req.body.id_friend} AND id_user = ${req.session.id_user} and who = '${req.session.username}'`,
(error, results) => {
// console.log(results);
let hasil;
if (results.length > 0) {
hasil = 0;
}else {
hasil = 1;
}
res.json(hasil);
}
)
});
app.post('/add_friend', (req, res) => {
if (req.session.loggedin) {
gt = req.body.id_group_chat.split('_');
connection.query(
`INSERT INTO cn_friend (id_user, id_friend, id_group_chat, who) VALUES (?, ?, ?, ?), (?, ?, ?, ?);
SELECT img_profile FROM cn_user WHERE id_user = ${req.body.id_friend}`,
[req.session.id_user, req.body.id_friend, req.body.id_group_chat, `${gt[0]}`,
req.session.id_user, req.body.id_friend, req.body.id_group_chat, `${gt[1]}`],
(error, results) => {
if (results[0].affectedRows) {
res.json({success:1,img_profile_friend:`${results[1][0].img_profile}`});
}
}
)
}
});
app.post('/uploadpp', function(req, res) {
let thefile = req.files.img;
if (req.session.loggedin) {
if (!req.files || Object.keys(req.files).length === 0) {
return res.status(400).send('No files were uploaded. Back');
}else if (thefile.mimetype.split('/')[0] != 'image') {
return res.status(400).send('file must be image. Back');
}
let name_file = +new Date() + thefile.md5;
let path = __dirname + '/img/ak47/' + name_file + '.' + thefile.name.split('.')[1];
let for_save = '/img/ak47/' + name_file + '.' + thefile.name.split('.')[1];
sharp(thefile.data)
.resize(400, 400)
.toFile(path, (err, info) => {
if (err == null) {
connection.query(
`SELECT img_profile FROM cn_user WHERE id_user = ${req.session.id_user}`,
(error, img_data) => {
if (img_data.length > 0) {
fs.unlink(__dirname+img_data[0].img_profile, (err) => {
if (err) {
console.error(err)
}
})
}
//UPDATE
connection.query(
`UPDATE cn_user SET img_profile = ? WHERE id_user = ?`,
[for_save, req.session.id_user],
(error, results) => {
if (error) {
return res.status(400).send('Service Unracable, try again later. Back');
}else {
req.session.img_profile = for_save
res.redirect('/list');
}
}
)
}
)
}else {
return res.status(500).send(err);
}
});
}else {
res.redirect('/')
}
});
app.post('/sendimg', function(req, res) {
if (req.session.loggedin) {
let thefile = req.files.img;
let explode = req.body.group_chat.split('_');
let time = now__();
if (!req.files || Object.keys(req.files).length === 0) {
return res.status(400).send('No files were uploaded. Back');
}else if (thefile.mimetype.split('/')[0] != 'image') {
return res.status(400).send('file must be image. Back');
}
let name_file = +new Date() + thefile.md5;
let path = __dirname + '/img/ss2v5/' + name_file + '.' + thefile.name.split('.')[1];
let for_save = '/img/ss2v5/' + name_file + '.' + thefile.name.split('.')[1];
sharp(thefile.data)
.resize(400, 400)
.toFile(path, (err, info) => {
if (err == null) {
connection.query(
`INSERT INTO cn_chat (message, id_group_chat, user_id, time_chat, who, img) VALUES
('${req.session.name} send an image!', '${req.body.group_chat}', ${req.body.id_user}, '${time}', '${explode[0]}', '${for_save}'),
('${req.session.name} send an image!', '${req.body.group_chat}', ${req.body.id_user}, '${time}', '${explode[1]}', '${for_save}')`,
(error, results) => {
console.log(results);
if (results.affectedRows) {
io.sockets.emit(`new_message_${req.body.group_chat}`, {
msg: `${req.session.name} send an image !`,
sender: req.body.username,
time: time,
img: `${for_save}`
});
io.sockets.emit(`notification_${req.body.receiver}`, {
msg: htmlspecialchars_decode(req.body.message),
sender: req.body.username,
receiver: req.body.receiver,
time: time,
id_receiver: req.body.id_user,
name:req.body.name,
img_profile: req.body.img_profile
});
res.redirect('/index/'+req.body.id_receiver);
}else {
return res.status(400).send('Service Unracable, try again later. Back');
}
}
)
}else {
return res.status(500).send(err);
}
});
}else {
res.redirect('/')
}
});
app.post('/delete_chat', (req, res) => {
if (req.session.loggedin) {
connection.query( // get id group
`SELECT cf.id_group_chat FROM cn_user cu
JOIN cn_friend cf on cf.id_user = cu.id_user or cf.id_friend = cu.id_user
WHERE cu.id_user = ${req.session.id_user}
AND (cf.id_user = ${req.body.id_friend} or cf.id_friend = ${req.body.id_friend})`,
(error, result_1) => {
connection.query( //cek if last item to delete
`SELECT DISTINCT who FROM cn_chat WHERE id_group_chat = '${result_1[0].id_group_chat}';
SELECT DISTINCT img FROM cn_chat WHERE id_group_chat = '${result_1[0].id_group_chat}' and img != ''`,
(error, results_2) => {
if (results_2[0].length == 1 && results_2[1].length != 0) {
results_2[1].forEach((val, i)=>{
fs.unlink(__dirname+val.img, (err) => {
if (err) {
console.error(err)
}
})
})
}
connection.query( // delete the chat on database
`DELETE FROM cn_chat WHERE id_group_chat = '${result_1[0].id_group_chat}' AND who = '${req.session.username}'`,
(error, results_3) => {
if (results_3.affectedRows) {
res.json(1);
}else {
res.json(0)
}
})
})
})
}else {
res.redirect('/');
}
});
app.post('/delete_contact', (req, res) => {
if (req.session.loggedin) {
connection.query(
`SELECT cf.id_group_chat FROM cn_user cu
JOIN cn_friend cf on cf.id_user = cu.id_user or cf.id_friend = cu.id_user
WHERE cu.id_user = ${req.session.id_user}
AND (cf.id_user = ${req.body.id_friend} or cf.id_friend = ${req.body.id_friend});`,
(error, result_1) => {
connection.query(
`DELETE FROM cn_chat WHERE id_group_chat = '${result_1[0].id_group_chat}' AND who = '${req.session.username}';
DELETE FROM cn_friend WHERE id_group_chat = '${result_1[0].id_group_chat}' AND who = '${req.session.username}'`,
(error, results) => {
if (results[1].affectedRows) {
res.json(1);
}else {
res.json(0);
}
})
})
}else {
res.redirect('/');
}
});
app.post('/update_stat_info', function(req, res) {
if (req.session.loggedin) {
connection.query(
`UPDATE cn_user SET ${req.body.kind__} = '${htmlspecialchars(req.body.val__)}' WHERE id_user = ${req.session.id_user}`,
(error, results) => {
if (error) {
console.log(error);
return res.status(400).send('Service Unracable, try again later. Back');
}else {
if (results.affectedRows) {
if (req.body.kind__ == 'name') {
req.session.name = req.body.val__;
}else {
req.session.information = req.body.val__;
}
res.redirect('/list');
}
}
}
)
}else {
res.redeirect('/')
}
})
app.get('/logout', function(req, res) {
req.session.destroy((err) => {
if (err) {
return console.log(err);
}
res.redirect('/');
});
});
io.sockets.on('connection', function(socket) {
yangterhubung.push(socket);
console.log('banyak socket yang terhubung : %s', yangterhubung.length);
//disconnect
socket.on('disconnect', function(data) {
yangterhubung.splice(yangterhubung.indexOf(socket), 1)
console.log('Terputus: %s ', yangterhubung.length)
})
// send message
socket.on('send_message', function(data) {
let explode = data.group.split('_');
let time = now__();
connection.query(
`INSERT INTO cn_chat (message, id_group_chat, user_id, time_chat, who) VALUES
('${htmlspecialchars(data.message)}', '${data.group}', ${data.id_me}, '${time}', '${explode[0]}'),
('${htmlspecialchars(data.message)}', '${data.group}', ${data.id_me}, '${time}', '${explode[1]}')`,
(error, results) => {
console.log(error);
console.log(results);
if (results != undefined) {
if (results.affectedRows) {
io.sockets.emit(`new_message_${data.group}`, {
msg: htmlspecialchars_decode(data.message),
sender: data.username,
time: time,
img: ''
});
io.sockets.emit(`notification_${data.receiver}`, {
msg: htmlspecialchars_decode(data.message),
sender: data.username,
receiver: data.receiver,
time: time,
id_receiver: data.id_me,
name:data.name,
img_profile: data.img_profile
});
// console.log('data receiver '+data.receiver);
}else {
io.sockets.emit(`new_message_${data.group}`, {
msg: '~',
sender: '~'
});
}
}else {
io.sockets.emit(`new_message_${data.group}`, {
msg: '~*',
sender: '~*'
});
}
}
)
});
})
Done.
github repository : https://github.com/aldipraddana/nodejs-web-message.git
Thank you for visiting. I hope this article will help you.