Server: fix remaining milli seconds before the next requests feature

[github/Chocobozzz/PeerTube.git] / server / models / oauth-token.js

diff --git a/server/models/oauth-token.js b/server/models/oauth-token.js
index 23c6987321733bfa70b51f077b5a70dbf175ed42..d53fdcf31fd336a970a722582d7c1577dc11a98b 100644 (file)
--- a/server/models/oauth-token.js
+++ b/server/models/oauth-token.js
@@ -20,7 +20,8 @@ OAuthTokenSchema.path('user').required(true)
 OAuthTokenSchema.statics = {
   getByRefreshTokenAndPopulateClient: getByRefreshTokenAndPopulateClient,
   getByTokenAndPopulateUser: getByTokenAndPopulateUser,
-  getByRefreshToken: getByRefreshToken
+  getByRefreshToken: getByRefreshToken,
+  removeByUserId: removeByUserId
 }
 
 mongoose.model('OAuthToken', OAuthTokenSchema)
@@ -28,7 +29,7 @@ mongoose.model('OAuthToken', OAuthTokenSchema)
 // ---------------------------------------------------------------------------
 
 function getByRefreshTokenAndPopulateClient (refreshToken) {
-  return this.findOne({ refreshToken: refreshToken }).populate('client').then(function (token) {
+  return this.findOne({ refreshToken: refreshToken }).populate('client').exec().then(function (token) {
     if (!token) return token
 
     const tokenInfos = {
@@ -47,9 +48,13 @@ function getByRefreshTokenAndPopulateClient (refreshToken) {
 }
 
 function getByTokenAndPopulateUser (bearerToken) {
-  return this.findOne({ accessToken: bearerToken }).populate('user')
+  return this.findOne({ accessToken: bearerToken }).populate('user').exec()
 }
 
 function getByRefreshToken (refreshToken) {
-  return this.findOne({ refreshToken: refreshToken })
+  return this.findOne({ refreshToken: refreshToken }).exec()
+}
+
+function removeByUserId (userId, callback) {
+  return this.remove({ user: userId }, callback)
 }