Create Models in API

[kbmakevin]

See details in #1

[kbmakevin]

Vital Signs

[kbmakevin]

Motivation

[kbmakevin]

EmergencyAlert

[kbmakevin]

User

const mongoose = require('mongoose');
const crypto = require('crypto');
const jwt = require('jsonwebtoken');

// properties
const JWT_SECRET = "MY_SECRET";
const USER_TYPES = ['nurse', 'patient'];

const userSchema = mongoose.Schema({
    // email set to unique as we'll use it for the login credentials
    email: {
        type: String,
        unique: true,
        required: true
    },
    name: {
        type: String,
        required: true
    },
    // the hash and salt will be used instead of saving a password
    // hash created by combining pw provided by user and the salt + applying one-way encryption
    hash: String,
    // string of chars unique to each user
    salt: String,
    type: {
        type: String,
        enum: USER_TYPES,
        default: 'patient'
    },
    // FOR NURSES ONLY ------------------------------------------------
    authoredMotivation: [{ type: mongoose.Schema.ObjectId, ref: 'Motivation' }],
    receivedEmergencyAlerts: [{ type: mongoose.Schema.ObjectId, ref: 'EmergencyAlert' }],

    // FOR PATIENTS ONLY ----------------------------------------------
    vitalSigns: [{ type: mongoose.Schema.ObjectId, ref: 'VitalSigns' }],
    receivedMotivation: [{ type: mongoose.Schema.ObjectId, ref: 'Motivation' }],
    sentEmergencyAlerts: [{ type: mongoose.Schema.ObjectId, ref: 'EmergencyAlert' }]
});

userSchema.methods.setPassword = function (password) {
    this.salt = crypto.randomBytes(16).toString('hex');
    this.hash = crypto.pbkdf2Sync(password, this.salt, 1000, 64, 'sha512').toString('hex');
}

userSchema.methods.validPassword = function (password) {
    const hash = crypto.pbkdf2Sync(password, this.salt, 1000, 64, 'sha512').toString('hex');
    return this.hash == hash;
}

// whenever user registers/logs in we need to generate a jwt
userSchema.methods.generateJwt = function () {
    let expiry = new Date();
    expiry.setDate(expiry.getDate() + 7);

    return jwt.sign({
        _id: this._id,
        email: this.email,
        name: this.name,
        exp: parseInt(expiry.getTime() / 1000),
    }, JWT_SECRET);
}

module.exports = mongoose.model('User', userSchema);