Converted reset controller from eventemitter to async.waterfall.

This commit is contained in:
Sahat Yalkabov
2014-02-18 03:21:05 -05:00
parent 6d3bdaeaea
commit 4d434aef3c

View File

@ -1,3 +1,4 @@
var async = require('async');
var bcrypt = require('bcrypt-nodejs');
var nodemailer = require('nodemailer');
var User = require('../models/User');
@ -12,194 +13,82 @@ exports.getReset = function(req, res) {
return res.redirect('/');
}
var conditions = {
_id: req.params.id,
resetPasswordExpires: { $gt: Date.now() }
};
// Get the user
User.findOne(conditions, function(err, user) {
if (err) {
req.flash('errors', err);
return res.render('account/reset', {
validToken: false
User
.where('resetPasswordToken', req.params.token)
.where('resetPasswordExpires').gt(Date.now())
.exec(function(err, user) {
if (!user) {
req.flash('errors', { msg: 'Password reset token is invalid or has expired.' });
return res.redirect('/forgot');
}
res.render('account/reset', {
title: 'Password Reset'
});
}
if (!user) {
req.flash('errors', { msg: 'Password reset token is invalid or has expired.' });
return res.redirect('/forgot');
}
// Validate the token
bcrypt.compare(req.params.token, user.resetPasswordToken, function(err, isValid) {
if (err) {
req.flash('errors', err);
return res.render('account/reset', {
validToken: false
});
}
if (!isValid) {
req.flash('errors', { msg: 'Your reset request token is invalid.' });
return res.render('account/reset', {
validToken: false
});
} else {
req.flash('success', { msg: 'Token accepted. Reset your password!' });
return res.render('account/reset', {
validToken: true
});
}
});
});
};
/**
* POST /reset/:id/:token
* Process the POST to reset your password
* POST /reset/:token
* Process the reset password request.
*/
exports.postReset = function(req, res) {
exports.postReset = function(req, res, next) {
req.assert('password', 'Password must be at least 4 characters long.').len(4);
req.assert('confirm', 'Passwords must match.').equals(req.body.password);
// Create a workflow
var workflow = new (require('events').EventEmitter)();
var errors = req.validationErrors();
/**
* Step 1: Validate the password(s) meet complexity requirements and match.
*/
if (errors) {
req.flash('errors', errors);
return res.redirect('back');
}
workflow.on('validate', function() {
async.waterfall([
function(done) {
User
.where('resetPasswordToken', req.params.token)
.where('resetPasswordExpires').gt(Date.now())
.exec(function(err, user) {
if (!user) {
req.flash('errors', { msg: 'Password reset request is invalid. It may have expired.' });
return res.redirect('back');
}
done(err, user);
});
},
function(user, done) {
user.password = req.body.password;
user.resetPasswordToken = undefined;
user.resetPasswordExpires = undefined;
req.assert('password', 'Password must be at least 4 characters long.').len(4);
req.assert('confirm', 'Passwords must match.').equals(req.body.password);
var errors = req.validationErrors();
if (errors) {
req.flash('errors', errors);
return res.render('account/reset', {});
user.save(function(err) {
if (err) return next(err);
req.logIn(user, function(err) {
done(err, user);
});
});
},
function(user, done) {
var smtpTransport = nodemailer.createTransport('SMTP', {
service: 'SendGrid',
auth: {
user: secrets.sendgrid.user,
pass: secrets.sendgrid.password
}
});
var mailOptions = {
to: user.profile.name + ' <' + user.email + '>',
from: 'hackathon@starter.com',
subject: 'Your Hackathon Starter password has been changed',
text: 'Hello,\n\n' +
'This is a confirmation that the password for your account ' + user.email + ' has just been changed.\n'
};
smtpTransport.sendMail(mailOptions, function(err) {
done(err);
});
}
// next step
workflow.emit('findUser');
], function(err) {
if (err) return next(err);
res.redirect('/');
});
/**
* Step 2: Lookup the User
* We are doing this again in case the user changed the URL
*/
workflow.on('findUser', function() {
var conditions = {
_id: req.params.id,
resetPasswordExpires: { $gt: Date.now() }
};
// Get the user
User.findOne(conditions, function(err, user) {
if (err) {
req.flash('errors', err);
return res.render('account/reset', {});
}
if (!user) {
req.flash('errors', { msg: 'Your reset request is invalid. It may have expired.' });
return res.render('account/reset', {});
}
// Validate the token
bcrypt.compare(req.params.token, user.resetPasswordToken, function(err, isValid) {
if (err) {
req.flash('errors', err);
return res.render('account/reset', {});
}
if (!isValid) {
req.flash('errors', { msg: 'Your reset request token is invalid.' });
return res.render('account/reset', {});
}
});
// next step
workflow.emit('updatePassword', user);
});
});
/**
* Step 3: Update the User's Password and clear the
* clear the reset token
*/
workflow.on('updatePassword', function(user) {
user.password = req.body.password;
user.resetPasswordToken = '';
user.resetPasswordExpires = Date.now();
// update the user record
user.save(function(err) {
if (err) {
req.flash('errors', err);
return res.render('account/reset', {});
}
// Log the user in
req.logIn(user, function(err) {
if (err) {
req.flash('errors', err);
return res.render('account/reset', {});
}
// next step
workflow.emit('sendEmail', user);
});
});
});
/**
* Step 4: Send the User an email letting them know thier
* password was changed. This is important in case the
* user did not initiate the reset!
*/
workflow.on('sendEmail', function(user) {
// Create a reusable nodemailer transport method (opens a pool of SMTP connections)
var smtpTransport = nodemailer.createTransport("SMTP", {
service: "Gmail",
auth: {
user: process.env.SMTP_USERNAME || '',
pass: process.env.SMTP_PASSWORD || ''
}
// See nodemailer docs for other transports
// https://github.com/andris9/Nodemailer
});
// create email
var mailOptions = {
to: user.profile.name + ' <' + user.email + '>',
from: 'hackathon@starter.com', // TODO parameterize
subject: 'Password Reset Notice',
text: 'This is a courtesy message from hackathon-starter. Your password was just reset. Cheers!'
};
// send email
smtpTransport.sendMail(mailOptions, function(err) {
if (err) {
req.flash('errors', { msg: err.message });
req.flash('info', { msg: 'You are logged in with your new password!' });
res.redirect('/');
} else {
// Message to user
req.flash('info', { msg: 'You are logged in with your new password!' });
res.redirect('/');
}
});
// shut down the connection pool, no more messages
smtpTransport.close();
});
/**
* Initiate the workflow
*/
workflow.emit('validate');
};