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 const OAuthToken = mongoose.model('OAuthToken')
   8
   9 // ---------------------------------------------------------------------------
  10
  11 const UserSchema = mongoose.Schema({
  12   createdDate: {
  13     type: Date,
  14     default: Date.now
  15   },
  16   password: String,
  17   username: String,
  18   role: String
  19 })
  20
  21 UserSchema.path('password').required(customUsersValidators.isUserPasswordValid)
  22 UserSchema.path('username').required(customUsersValidators.isUserUsernameValid)
  23 UserSchema.path('role').validate(customUsersValidators.isUserRoleValid)
  24
  25 UserSchema.methods = {
  26   isPasswordMatch,
  27   toFormatedJSON
  28 }
  29
  30 UserSchema.statics = {
  31   countTotal,
  32   getByUsername,
  33   list,
  34   listForApi,
  35   loadById,
  36   loadByUsername
  37 }
  38
  39 UserSchema.pre('save', function (next) {
  40   const user = this
  41
  42   peertubeCrypto.cryptPassword(this.password, function (err, hash) {
  43     if (err) return next(err)
  44
  45     user.password = hash
  46
  47     return next()
  48   })
  49 })
  50
  51 UserSchema.pre('remove', function (next) {
  52   const user = this
  53
  54   OAuthToken.removeByUserId(user._id, next)
  55 })
  56
  57 mongoose.model('User', UserSchema)
  58
  59 // ------------------------------ METHODS ------------------------------
  60
  61 function isPasswordMatch (password, callback) {
  62   return peertubeCrypto.comparePassword(password, this.password, callback)
  63 }
  64
  65 function toFormatedJSON () {
  66   return {
  67     id: this._id,
  68     username: this.username,
  69     role: this.role,
  70     createdDate: this.createdDate
  71   }
  72 }
  73 // ------------------------------ STATICS ------------------------------
  74
  75 function countTotal (callback) {
  76   return this.count(callback)
  77 }
  78
  79 function getByUsername (username) {
  80   return this.findOne({ username: username })
  81 }
  82
  83 function list (callback) {
  84   return this.find(callback)
  85 }
  86
  87 function listForApi (start, count, sort, callback) {
  88   const query = {}
  89   return modelUtils.listForApiWithCount.call(this, query, start, count, sort, callback)
  90 }
  91
  92 function loadById (id, callback) {
  93   return this.findById(id, callback)
  94 }
  95
  96 function loadByUsername (username, callback) {
  97   return this.findOne({ username: username }, callback)
  98 }