OAuth/User models refractoring -> use mongoose api
[oweals/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