Server: fix remaining milli seconds before the next requests feature

[github/Chocobozzz/PeerTube.git] / server / models / user.js

diff --git a/server/models/user.js b/server/models/user.js
index c2c8807f0830da9452ea66c97231199115c7af41..91e8aeae1c93fa26aaae2fa4fa65692f55c8def6 100644 (file)
--- a/server/models/user.js
+++ b/server/models/user.js
@@ -4,6 +4,8 @@ const customUsersValidators = require('../helpers/custom-validators').users
 const modelUtils = require('./utils')
 const peertubeCrypto = require('../helpers/peertube-crypto')
 
+const OAuthToken = mongoose.model('OAuthToken')
+
 // ---------------------------------------------------------------------------
 
 const UserSchema = mongoose.Schema({
@@ -46,6 +48,12 @@ UserSchema.pre('save', function (next) {
   })
 })
 
+UserSchema.pre('remove', function (next) {
+  const user = this
+
+  OAuthToken.removeByUserId(user._id, next)
+})
+
 mongoose.model('User', UserSchema)
 
 // ------------------------------ METHODS ------------------------------