151 lines
4.2 KiB
JavaScript
151 lines
4.2 KiB
JavaScript
import mongoose from 'mongoose';
|
|
|
|
const bcrypt = require('bcrypt-nodejs');
|
|
|
|
const EmailConfirmationStates = {
|
|
Verified: 'verified',
|
|
Sent: 'sent',
|
|
Resent: 'resent',
|
|
};
|
|
|
|
const { Schema } = mongoose;
|
|
|
|
const apiKeySchema = new Schema({
|
|
label: { type: String, default: 'API Key' },
|
|
lastUsedAt: { type: Date, required: true, default: Date.now },
|
|
hashedKey: { type: String, required: true },
|
|
}, { timestamps: true, _id: true });
|
|
|
|
apiKeySchema.virtual('publicFields').get(function publicFields() {
|
|
return {
|
|
id: this.id, label: this.label, lastUsedAt: this.lastUsedAt, createdAt: this.createdAt
|
|
};
|
|
});
|
|
|
|
apiKeySchema.virtual('id').get(function getApiKeyId() {
|
|
return this._id.toHexString();
|
|
});
|
|
|
|
apiKeySchema.set('toJSON', {
|
|
virtuals: true
|
|
});
|
|
|
|
const userSchema = new Schema({
|
|
name: { type: String, default: '' },
|
|
username: { type: String, required: true, unique: true },
|
|
password: { type: String },
|
|
resetPasswordToken: String,
|
|
resetPasswordExpires: Date,
|
|
verified: { type: String },
|
|
verifiedToken: String,
|
|
verifiedTokenExpires: Date,
|
|
github: { type: String },
|
|
email: { type: String, unique: true },
|
|
tokens: Array,
|
|
apiKeys: { type: [apiKeySchema] },
|
|
preferences: {
|
|
fontSize: { type: Number, default: 18 },
|
|
indentationAmount: { type: Number, default: 2 },
|
|
isTabIndent: { type: Boolean, default: false },
|
|
autosave: { type: Boolean, default: true },
|
|
linewrap: { type: Boolean, default: true },
|
|
lintWarning: { type: Boolean, default: false },
|
|
textOutput: { type: Boolean, default: false },
|
|
gridOutput: { type: Boolean, default: false },
|
|
soundOutput: { type: Boolean, default: false },
|
|
theme: { type: String, default: 'light' },
|
|
autorefresh: { type: Boolean, default: false }
|
|
}
|
|
}, { timestamps: true, usePushEach: true });
|
|
|
|
/**
|
|
* Password hash middleware.
|
|
*/
|
|
userSchema.pre('save', function checkPassword(next) { // eslint-disable-line consistent-return
|
|
const user = this;
|
|
if (!user.isModified('password')) { return next(); }
|
|
bcrypt.genSalt(10, (err, salt) => { // eslint-disable-line consistent-return
|
|
if (err) { return next(err); }
|
|
bcrypt.hash(user.password, salt, null, (innerErr, hash) => {
|
|
if (innerErr) { return next(innerErr); }
|
|
user.password = hash;
|
|
return next();
|
|
});
|
|
});
|
|
});
|
|
|
|
/**
|
|
* API keys hash middleware
|
|
*/
|
|
userSchema.pre('save', function checkApiKey(next) { // eslint-disable-line consistent-return
|
|
const user = this;
|
|
if (!user.isModified('apiKeys')) { return next(); }
|
|
let hasNew = false;
|
|
user.apiKeys.forEach((k) => {
|
|
if (k.isNew) {
|
|
hasNew = true;
|
|
bcrypt.genSalt(10, (err, salt) => { // eslint-disable-line consistent-return
|
|
if (err) { return next(err); }
|
|
bcrypt.hash(k.hashedKey, salt, null, (innerErr, hash) => {
|
|
if (innerErr) { return next(innerErr); }
|
|
k.hashedKey = hash;
|
|
return next();
|
|
});
|
|
});
|
|
}
|
|
});
|
|
if (!hasNew) return next();
|
|
});
|
|
|
|
userSchema.virtual('id').get(function idToString() {
|
|
return this._id.toHexString();
|
|
});
|
|
|
|
userSchema.virtual('publicApiKeys').get(function publicApiKeys() {
|
|
return this.apiKeys.map(apiKey => apiKey.publicFields);
|
|
});
|
|
|
|
|
|
userSchema.set('toJSON', {
|
|
virtuals: true
|
|
});
|
|
|
|
|
|
/**
|
|
* Helper method for validating user's password.
|
|
*/
|
|
userSchema.methods.comparePassword = function comparePassword(candidatePassword, cb) {
|
|
// userSchema.methods.comparePassword = (candidatePassword, cb) => {
|
|
bcrypt.compare(candidatePassword, this.password, (err, isMatch) => {
|
|
cb(err, isMatch);
|
|
});
|
|
};
|
|
|
|
/**
|
|
* Helper method for validating a user's api key
|
|
*/
|
|
userSchema.methods.findMatchingKey = function findMatchingKey(candidateKey, cb) {
|
|
let foundOne = false;
|
|
this.apiKeys.forEach((k) => {
|
|
if (bcrypt.compareSync(candidateKey, k.hashedKey)) {
|
|
foundOne = true;
|
|
cb(null, true, k._id);
|
|
}
|
|
});
|
|
if (!foundOne) cb('Matching API key not found !', false, null);
|
|
};
|
|
|
|
userSchema.statics.findByMailOrName = function findByMailOrName(email) {
|
|
const query = {
|
|
$or: [{
|
|
email,
|
|
}, {
|
|
username: email,
|
|
}],
|
|
};
|
|
return this.findOne(query).exec();
|
|
};
|
|
|
|
userSchema.statics.EmailConfirmation = EmailConfirmationStates;
|
|
|
|
export default mongoose.model('User', userSchema);
|