server/models/user.js

   1 const mongoose = require('mongoose')
   2
   3 const customUsersValidators = require('../helpers/custom-validators').users
   4 const modelUtils = require('./utils')
   5 const peertubeCrypto = require('../helpers/peertube-crypto')
   6
   7 // ---------------------------------------------------------------------------
   8
   9 const UserSchema = mongoose.Schema({
  10   createdDate: {
  11     type: Date,
  12     default: Date.now
  13   },
  14   password: String,
  15   username: String,
  16   role: String
  17 })
  18
  19 UserSchema.path('password').required(customUsersValidators.isUserPasswordValid)
  20 UserSchema.path('username').required(customUsersValidators.isUserUsernameValid)
  21 UserSchema.path('role').validate(customUsersValidators.isUserRoleValid)
  22
  23 UserSchema.methods = {
  24   isPasswordMatch: isPasswordMatch,
  25   toFormatedJSON: toFormatedJSON
  26 }
  27
  28 UserSchema.statics = {
  29   countTotal: countTotal,
  30   getByUsername: getByUsername,
  31   listForApi: listForApi,
  32   loadById: loadById,
  33   loadByUsername: loadByUsername
  34 }
  35
  36 UserSchema.pre('save', function (next) {
  37   const user = this
  38
  39   peertubeCrypto.cryptPassword(this.password, function (err, hash) {
  40     if (err) return next(err)
  41
  42     user.password = hash
  43
  44     return next()
  45   })
  46 })
  47
  48 mongoose.model('User', UserSchema)
  49
  50 // ------------------------------ METHODS ------------------------------
  51
  52 function isPasswordMatch (password, callback) {
  53   return peertubeCrypto.comparePassword(password, this.password, callback)
  54 }
  55
  56 function toFormatedJSON () {
  57   return {
  58     id: this._id,
  59     username: this.username,
  60     role: this.role,
  61     createdDate: this.createdDate
  62   }
  63 }
  64 // ------------------------------ STATICS ------------------------------
  65
  66 function countTotal (callback) {
  67   return this.count(callback)
  68 }
  69
  70 function getByUsername (username) {
  71   return this.findOne({ username: username })
  72 }
  73
  74 function listForApi (start, count, sort, callback) {
  75   const query = {}
  76   return modelUtils.listForApiWithCount.call(this, query, start, count, sort, callback)
  77 }
  78
  79 function loadById (id, callback) {
  80   return this.findById(id, callback)
  81 }
  82
  83 function loadByUsername (username, callback) {
  84   return this.findOne({ username: username }, callback)
  85 }