serverless/lib/models/model_user.js

/**
 * Model: Users
 * - users of your application
 */


// Dependencies
var Config = require('../config/config');
var Utilities = require('../utilities/utilities');
var AppDynamoDB = require('./model_aws').DynamoDB;

var moment = require('moment');
var bcryptjs = require('bcryptjs');
var _ = require('lodash');
var validator = require('validator');



module.exports = new User();



function User() {}



/**
 * SignUp
 */

User.prototype.signUp = function(data, callback) {

    // Defaults
    var _this = this;


    /**
     * Validate
     */

    if (!data.email) return callback({
        status: 400,
        message: 'Bad Request: Email is required'
    }, null);

    if (!data.password) return callback({
        status: 400,
        message: 'Bad Request: Password is required'
    }, null);

    if (data.password && data.password.length < 8) return callback({
        status: 400,
        message: 'Bad Request: Password must be at least 8 characters long'
    }, null);



    // Check if email is already in use
    _this.showByEmail(data.email, function(error, user) {

        if (error) return callback(error, null);

        if (user) return callback({
            status: 409,
            message: 'Email is already in use'
        }, null);



        /**
         * Instantiate
         */

        var user = {
            _id: Utilities.generateID('user'),
            email: data.email ? data.email : null,
            password: data.password ? data.password : null,
            created: moment().unix(),
            updated: moment().unix(),
            plan: 'free',
            sign_in_count: 0
        };


        // Hash Password
        user = _this.hashPassword(user);



        /**
         * Save
         */

        _this.save(user, function(error, user) {

            if (error) return (error, null);

            return callback(null, user);

        });
    });
}





/**
 * ShowByEmail
 */

User.prototype.showByEmail = function(email, callback) {


    /**
     * Validate
     */

    if (!email) return callback({
        status: 400,
        message: 'Bad Request: Email is required'
    }, null);



    /**
     * Find User
     */

    var params = {};
    params.TableName = "app_users";
    params.IndexName = "email-index";
    params.KeyConditions = [
        AppDynamoDB.Condition("email", "EQ", email)
    ];

    AppDynamoDB.query(params, function(error, response) {

        if (error || !response) return callback({
            status: 500,
            message: 'Sorry, something went wrong.'
            raw: error
        }, null);

        return callback(null, response.Items && response.Items[0] ? response.Items[0] : null);

    });
}




/**
 * Save
 * - Updates existing record or creates a record if one does not already exist
 */

User.prototype.save = function(data, callback) {


    /**
     * Validate
     */

    if (!data.email) return callback({
        status: 400,
        message: 'Bad Request: Email is required'
    }, null);

    if (!data.password) return callback({
        status: 400,
        message: 'Bad Request: Password is required'
    }, null);


    /**
     * Perform Save
     */

    var params = {
        TableName: 'app_users',
        ReturnValues: 'ALL_NEW',
        Key: {
            '_id': user._id
        },
        UpdateExpression: 'SET ',
        ExpressionAttributeNames: {},
        ExpressionAttributeValues: {}
    };


    /**
     * Basic Information
     */


    // email
    params.UpdateExpression = params.UpdateExpression + '#a0 = :email_val, ';
    params.ExpressionAttributeNames['#a0'] = 'email';
    params.ExpressionAttributeValues[':email_val'] = user.email;

    // password
    params.UpdateExpression = params.UpdateExpression + '#a1 = :password_val, ';
    params.ExpressionAttributeNames['#a1'] = 'password';
    params.ExpressionAttributeValues[':password_val'] = user.password;

    // salt
    params.UpdateExpression = params.UpdateExpression + '#a2 = :salt_val, ';
    params.ExpressionAttributeNames['#a2'] = 'salt';
    params.ExpressionAttributeValues[':salt_val'] = user.salt;

    // plan
    params.UpdateExpression = params.UpdateExpression + '#a3 = :plan_val, ';
    params.ExpressionAttributeNames['#a3'] = 'plan';
    params.ExpressionAttributeValues[':plan_val'] = user.plan;

    // sign_in_count
    params.UpdateExpression = params.UpdateExpression + '#a4 = :sign_in_count_val, ';
    params.ExpressionAttributeNames['#a4'] = 'sign_in_count';
    params.ExpressionAttributeValues[':sign_in_count_val'] = user.sign_in_count;

    // updated
    params.UpdateExpression = params.UpdateExpression + '#b0 = :updated_val, ';
    params.ExpressionAttributeNames['#b0'] = 'updated';
    params.ExpressionAttributeValues[':updated_val'] = moment().unix();

    // created
    if (isNaN(user.created)) user.created = moment(user.created).unix();
    params.UpdateExpression = params.UpdateExpression + '#b1 = :created_val, ';
    params.ExpressionAttributeNames['#b1'] = 'created';
    params.ExpressionAttributeValues[':created_val'] = user.created;



    /**
     * Save
     */

    // Remove Any Trailing Commas & Space
    params.UpdateExpression = params.UpdateExpression.trim();
    if (params.UpdateExpression[params.UpdateExpression.length - 1] === ',') params.UpdateExpression = params.UpdateExpression.substring(0, params.UpdateExpression.length - 1);


    AppDynamoDB.updateItem(params, function(error, response) {

        if (error || !response) return callback({
            status: 500,
            raw: error
        }, null);

        return callback(null, response.Attributes);

    });
}




/**
 * Hash Password
 */

User.prototype.hashPassword = function(user) {

    user.salt = bcryptjs.genSaltSync(10);
    user.password = bcryptjs.hashSync(user.password, user.salt);

    return user;

};