]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - server/lib/oauth-model.js
OAuth/User models refractoring -> use mongoose api
[github/Chocobozzz/PeerTube.git] / server / lib / oauth-model.js
1 const mongoose = require('mongoose')
2
3 const logger = require('../helpers/logger')
4
5 const OAuthClient = mongoose.model('OAuthClient')
6 const OAuthToken = mongoose.model('OAuthToken')
7 const User = mongoose.model('User')
8
9 // See https://github.com/oauthjs/node-oauth2-server/wiki/Model-specification for the model specifications
10 const OAuthModel = {
11 getAccessToken: getAccessToken,
12 getClient: getClient,
13 getRefreshToken: getRefreshToken,
14 getUser: getUser,
15 saveToken: saveToken
16 }
17
18 // ---------------------------------------------------------------------------
19
20 function getAccessToken (bearerToken) {
21 logger.debug('Getting access token (bearerToken: ' + bearerToken + ').')
22
23 return OAuthToken.loadByTokenAndPopulateUser(bearerToken)
24 }
25
26 function getClient (clientId, clientSecret) {
27 logger.debug('Getting Client (clientId: ' + clientId + ', clientSecret: ' + clientSecret + ').')
28
29 // TODO req validator
30 const mongoId = new mongoose.mongo.ObjectID(clientId)
31 return OAuthClient.loadByIdAndSecret(mongoId, clientSecret)
32 }
33
34 function getRefreshToken (refreshToken) {
35 logger.debug('Getting RefreshToken (refreshToken: ' + refreshToken + ').')
36
37 return OAuthToken.loadByRefreshToken(refreshToken)
38 }
39
40 function getUser (username, password) {
41 logger.debug('Getting User (username: ' + username + ', password: ' + password + ').')
42
43 return User.loadByUsernameAndPassword(username, password)
44 }
45
46 function saveToken (token, client, user) {
47 logger.debug('Saving token for client ' + client.id + ' and user ' + user.id + '.')
48
49 const tokenObj = new OAuthToken({
50 accessToken: token.accessToken,
51 accessTokenExpiresOn: token.accessTokenExpiresOn,
52 client: client.id,
53 refreshToken: token.refreshToken,
54 refreshTokenExpiresOn: token.refreshTokenExpiresOn,
55 user: user.id
56 })
57
58 return tokenObj.save(function (err, tokenCreated) {
59 if (err) throw err // node-oauth2-server library uses Promise.try
60
61 tokenCreated.client = client
62 tokenCreated.user = user
63
64 return tokenCreated
65 })
66 }
67
68 // ---------------------------------------------------------------------------
69
70 module.exports = OAuthModel