Converted workflow/eventemitter code to async.waterfall

controllers/forgot.js

@@ -3,7 +3,7 @@
 /**
  * Module dependencies.
  */
-
+var async = require('async');
 var bcrypt = require('bcrypt-nodejs');
 var crypto = require('crypto');
 var mongoose = require('mongoose');
@@ -72,120 +72,68 @@ exports.getForgot = function(req, res) {
  */
 
 exports.postForgot = function(req, res) {
+  req.assert('email', 'Please enter a valid email address.').isEmail();
 
-  // Begin a workflow
+  var errors = req.validationErrors();
-  var workflow = new (require('events').EventEmitter)();
 
-  /**
+  if (errors) {
-   * Step 1: Is the email valid?
+    req.flash('errors', errors);
-   */
+    return res.redirect('/forgot');
+  }
 
-  workflow.on('validate', function() {
+  async.waterfall([
-
+    function(done) {
-    // Check for form errors
+      /**
-    req.assert('email', 'Please enter a valid email address.').isEmail();
+       * Generate a one-time token.
-    var errors = req.validationErrors();
+       */
-
+      crypto.randomBytes(32, function(err, buf) {
-    if (errors) {
+        var token = buf.toString('base64');
-      req.flash('errors', errors);
+        done(err, token);
-      return res.redirect('/forgot');
+      });
-    }
+    },
-
+    function(token, done) {
-    // next step
+      /**
-    workflow.emit('generateToken');
+       * Save the token and token expiration.
-  });
+       */
-
+      User.findOne({ email: req.body.email.toLowerCase() }, function(err, user) {
-  /**
+        if (!user) {
-   * Step 2: Generate a one-time (nonce) token
+          req.flash('errors', { msg: 'No account with that email address exists.' });
-   */
-
-  workflow.on('generateToken', function() {
-    // generate token
-    crypto.randomBytes(24, function(err, buf) {
-      if (err) return next(err);
-      var token = buf.toString('base64');
-      console.log(token);
-      workflow.emit('saveToken', token)
-    });
-  });
-
-  /**
-   * Step 3: Save the token and token expiration
-   */
-
-  workflow.on('saveToken', function(token) {
-    // lookup user
-    User.findOne({ email: req.body.email.toLowerCase() }, function(err, user) {
-      if (err) {
-        req.flash('errors', err);
-        return res.redirect('/forgot');
-      }
-      if (!user) {
-        // If we didn't find a user associated with that
-        // email address then just finish the workflow
-        req.flash('info', { msg: 'If you have an account with that email address then we sent you an email with instructions. Check your email!' });
-        return res.redirect('/forgot');
-      }
-
-      user.resetPasswordToken = token;
-      user.resetPasswordExpires = Date.now() + 3600000; // 1 hour
-
-      // update the user's record with the token
-      user.save(function(err) {
-        if (err) {
-          req.flash('errors', err);
           return res.redirect('/forgot');
         }
+
+        user.resetPasswordToken = token;
+        user.resetPasswordExpires = Date.now() + 3600000; // 1 hour
+
+        user.save(function(err) {
+          done(err, token, user);
+        });
       });
-
+    },
-      // next step
+    function(token, user, done) {
-      workflow.emit('sendEmail', token, user);
+      /**
-    });
+       * Send the user an email with a reset link.
-  });
+       */
-
+      var smtpTransport = nodemailer.createTransport('SMTP', {
-  /**
+        service: 'SendGrid',
-   * Step 4: Send the user an email with a reset link
+        auth: {
-   */
+          user: secrets.sendgrid.user,
-
+          pass: secrets.sendgrid.password
-  workflow.on('sendEmail', function(token, user) {
+        }
-    var smtpTransport = nodemailer.createTransport('SMTP', {
+      });
-      service: 'SendGrid',
+      var mailOptions = {
-      auth: {
+        to: user.profile.name + ' <' + user.email + '>',
-        user: secrets.sendgrid.user,
+        from: 'hackathon@starter.com',
-        pass: secrets.sendgrid.password
+        subject: 'Hackathon Starter Password Reset',
-      }
+        text: 'You are receiving this because you (or someone else) have requested the reset of the password for your account.\n\n' +
-    });
+          'Please click on the following link, or paste this into your browser to complete the process:\n\n' +
-
+          'http://' + req.headers.host + '/reset/' + token + '\n\n' +
-    var mailOptions = {
+          'If you did not request this, please ignore this email and your password will remain unchanged.\n'
-      to: user.profile.name + ' <' + user.email + '>',
+      };
-      from: 'hackathon@starter.com',
+      smtpTransport.sendMail(mailOptions, function(err) {
-      subject: 'Hackathon Starter Password Reset',
-      text: 'You are receiving this because you (or someone else) have requested the reset of the password for your account.\n\n' +
-        'Please click on the following link, or paste this into your browser to complete the process:\n\n' +
-        'http://' + req.headers.host + '/reset/' + token + '\n\n' +
-        'If you did not request this, please ignore this email and your password will remain unchanged.\n'
-    };
-
-    smtpTransport.sendMail(mailOptions, function(err) {
-      if (err) {
-        req.flash('errors', { msg: err.message });
-        return res.redirect('/forgot');
-      } else {
-        // Message to user
         req.flash('info', { msg: 'We have sent an email to ' + user.email + ' for further instructions.' });
-        return res.redirect('/forgot');
+        done(err, 'done');
-      }
+        res.redirect('/forgot');
-    });
+      });
-
+    }
-    // shut down the connection pool, no more messages
+  ]);
-    smtpTransport.close();
-
-  });
-
-  /**
-   * Initiate the workflow
-   */
-
-  workflow.emit('validate');
-
 };