server/models/user.js

   1 'use strict'
   2
   3 const values = require('lodash/values')
   4
   5 const modelUtils = require('./utils')
   6 const constants = require('../initializers/constants')
   7 const peertubeCrypto = require('../helpers/peertube-crypto')
   8 const customUsersValidators = require('../helpers/custom-validators').users
   9
  10 // ---------------------------------------------------------------------------
  11
  12 module.exports = function (sequelize, DataTypes) {
  13   const User = sequelize.define('User',
  14     {
  15       password: {
  16         type: DataTypes.STRING,
  17         allowNull: false,
  18         validate: {
  19           passwordValid: function (value) {
  20             const res = customUsersValidators.isUserPasswordValid(value)
  21             if (res === false) throw new Error('Password not valid.')
  22           }
  23         }
  24       },
  25       username: {
  26         type: DataTypes.STRING,
  27         allowNull: false,
  28         validate: {
  29           usernameValid: function (value) {
  30             const res = customUsersValidators.isUserUsernameValid(value)
  31             if (res === false) throw new Error('Username not valid.')
  32           }
  33         }
  34       },
  35       role: {
  36         type: DataTypes.ENUM(values(constants.USER_ROLES)),
  37         allowNull: false
  38       }
  39     },
  40     {
  41       indexes: [
  42         {
  43           fields: [ 'username' ]
  44         }
  45       ],
  46       classMethods: {
  47         associate,
  48
  49         countTotal,
  50         getByUsername,
  51         list,
  52         listForApi,
  53         loadById,
  54         loadByUsername
  55       },
  56       instanceMethods: {
  57         isPasswordMatch,
  58         toFormatedJSON
  59       },
  60       hooks: {
  61         beforeCreate: beforeCreateOrUpdate,
  62         beforeUpdate: beforeCreateOrUpdate
  63       }
  64     }
  65   )
  66
  67   return User
  68 }
  69
  70 function beforeCreateOrUpdate (user, options, next) {
  71   peertubeCrypto.cryptPassword(user.password, function (err, hash) {
  72     if (err) return next(err)
  73
  74     user.password = hash
  75
  76     return next()
  77   })
  78 }
  79
  80 // ------------------------------ METHODS ------------------------------
  81
  82 function isPasswordMatch (password, callback) {
  83   return peertubeCrypto.comparePassword(password, this.password, callback)
  84 }
  85
  86 function toFormatedJSON () {
  87   return {
  88     id: this.id,
  89     username: this.username,
  90     role: this.role,
  91     createdAt: this.createdAt
  92   }
  93 }
  94 // ------------------------------ STATICS ------------------------------
  95
  96 function associate (models) {
  97   this.hasMany(models.OAuthToken, {
  98     foreignKey: 'userId',
  99     onDelete: 'cascade'
 100   })
 101 }
 102
 103 function countTotal (callback) {
 104   return this.count().asCallback(callback)
 105 }
 106
 107 function getByUsername (username) {
 108   const query = {
 109     where: {
 110       username: username
 111     }
 112   }
 113
 114   return this.findOne(query)
 115 }
 116
 117 function list (callback) {
 118   return this.find().asCallback(callback)
 119 }
 120
 121 function listForApi (start, count, sort, callback) {
 122   const query = {
 123     offset: start,
 124     limit: count,
 125     order: [ modelUtils.getSort(sort) ]
 126   }
 127
 128   return this.findAndCountAll(query).asCallback(function (err, result) {
 129     if (err) return callback(err)
 130
 131     return callback(null, result.rows, result.count)
 132   })
 133 }
 134
 135 function loadById (id, callback) {
 136   return this.findById(id).asCallback(callback)
 137 }
 138
 139 function loadByUsername (username, callback) {
 140   const query = {
 141     where: {
 142       username: username
 143     }
 144   }
 145
 146   return this.findOne(query).asCallback(callback)
 147 }