diff options
Diffstat (limited to 'server/models')
-rw-r--r-- | server/models/oauth-token.js | 7 | ||||
-rw-r--r-- | server/models/user.js | 8 |
2 files changed, 14 insertions, 1 deletions
diff --git a/server/models/oauth-token.js b/server/models/oauth-token.js index f6a814c36..d53fdcf31 100644 --- a/server/models/oauth-token.js +++ b/server/models/oauth-token.js | |||
@@ -20,7 +20,8 @@ OAuthTokenSchema.path('user').required(true) | |||
20 | OAuthTokenSchema.statics = { | 20 | OAuthTokenSchema.statics = { |
21 | getByRefreshTokenAndPopulateClient: getByRefreshTokenAndPopulateClient, | 21 | getByRefreshTokenAndPopulateClient: getByRefreshTokenAndPopulateClient, |
22 | getByTokenAndPopulateUser: getByTokenAndPopulateUser, | 22 | getByTokenAndPopulateUser: getByTokenAndPopulateUser, |
23 | getByRefreshToken: getByRefreshToken | 23 | getByRefreshToken: getByRefreshToken, |
24 | removeByUserId: removeByUserId | ||
24 | } | 25 | } |
25 | 26 | ||
26 | mongoose.model('OAuthToken', OAuthTokenSchema) | 27 | mongoose.model('OAuthToken', OAuthTokenSchema) |
@@ -53,3 +54,7 @@ function getByTokenAndPopulateUser (bearerToken) { | |||
53 | function getByRefreshToken (refreshToken) { | 54 | function getByRefreshToken (refreshToken) { |
54 | return this.findOne({ refreshToken: refreshToken }).exec() | 55 | return this.findOne({ refreshToken: refreshToken }).exec() |
55 | } | 56 | } |
57 | |||
58 | function removeByUserId (userId, callback) { | ||
59 | return this.remove({ user: userId }, callback) | ||
60 | } | ||
diff --git a/server/models/user.js b/server/models/user.js index c2c8807f0..91e8aeae1 100644 --- a/server/models/user.js +++ b/server/models/user.js | |||
@@ -4,6 +4,8 @@ const customUsersValidators = require('../helpers/custom-validators').users | |||
4 | const modelUtils = require('./utils') | 4 | const modelUtils = require('./utils') |
5 | const peertubeCrypto = require('../helpers/peertube-crypto') | 5 | const peertubeCrypto = require('../helpers/peertube-crypto') |
6 | 6 | ||
7 | const OAuthToken = mongoose.model('OAuthToken') | ||
8 | |||
7 | // --------------------------------------------------------------------------- | 9 | // --------------------------------------------------------------------------- |
8 | 10 | ||
9 | const UserSchema = mongoose.Schema({ | 11 | const UserSchema = mongoose.Schema({ |
@@ -46,6 +48,12 @@ UserSchema.pre('save', function (next) { | |||
46 | }) | 48 | }) |
47 | }) | 49 | }) |
48 | 50 | ||
51 | UserSchema.pre('remove', function (next) { | ||
52 | const user = this | ||
53 | |||
54 | OAuthToken.removeByUserId(user._id, next) | ||
55 | }) | ||
56 | |||
49 | mongoose.model('User', UserSchema) | 57 | mongoose.model('User', UserSchema) |
50 | 58 | ||
51 | // ------------------------------ METHODS ------------------------------ | 59 | // ------------------------------ METHODS ------------------------------ |