Server: don't be rude when serving unknown video in watch html file

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

diff --git a/server/models/oauth-token.js b/server/models/oauth-token.js
index f6a814c36fc37989c714c4076ed809b1e6e0048e..aff73bfb131f72fe2899d0d556bfe3e419dd5419 100644 (file)
--- a/server/models/oauth-token.js
+++ b/server/models/oauth-token.js
@@ -18,9 +18,10 @@ OAuthTokenSchema.path('client').required(true)
 OAuthTokenSchema.path('user').required(true)
 
 OAuthTokenSchema.statics = {
-  getByRefreshTokenAndPopulateClient: getByRefreshTokenAndPopulateClient,
-  getByTokenAndPopulateUser: getByTokenAndPopulateUser,
-  getByRefreshToken: getByRefreshToken
+  getByRefreshTokenAndPopulateClient,
+  getByTokenAndPopulateUser,
+  getByRefreshTokenAndPopulateUser,
+  removeByUserId
 }
 
 mongoose.model('OAuthToken', OAuthTokenSchema)
@@ -37,7 +38,9 @@ function getByRefreshTokenAndPopulateClient (refreshToken) {
       client: {
         id: token.client._id.toString()
       },
-      user: token.user
+      user: {
+        id: token.user
+      }
     }
 
     return tokenInfos
@@ -50,6 +53,10 @@ function getByTokenAndPopulateUser (bearerToken) {
   return this.findOne({ accessToken: bearerToken }).populate('user').exec()
 }
 
-function getByRefreshToken (refreshToken) {
-  return this.findOne({ refreshToken: refreshToken }).exec()
+function getByRefreshTokenAndPopulateUser (refreshToken) {
+  return this.findOne({ refreshToken: refreshToken }).populate('user').exec()
+}
+
+function removeByUserId (userId, callback) {
+  return this.remove({ user: userId }, callback)
 }