]>
Commit | Line | Data |
---|---|---|
cef534ed | 1 | import * as Bluebird from 'bluebird' |
141b177d | 2 | import { AccessDeniedError } from 'oauth2-server' |
da854ddd | 3 | import { logger } from '../helpers/logger' |
3fd3ab2d C |
4 | import { UserModel } from '../models/account/user' |
5 | import { OAuthClientModel } from '../models/oauth/oauth-client' | |
6 | import { OAuthTokenModel } from '../models/oauth/oauth-token' | |
557b13ae | 7 | import { LRU_CACHE } from '../initializers/constants' |
f201a749 | 8 | import { Transaction } from 'sequelize' |
6dd9de95 | 9 | import { CONFIG } from '../initializers/config' |
557b13ae | 10 | import * as LRUCache from 'lru-cache' |
69b0a27c | 11 | |
69818c93 | 12 | type TokenInfo = { accessToken: string, refreshToken: string, accessTokenExpiresAt: Date, refreshTokenExpiresAt: Date } |
557b13ae C |
13 | |
14 | const accessTokenCache = new LRUCache<string, OAuthTokenModel>({ max: LRU_CACHE.USER_TOKENS.MAX_SIZE }) | |
15 | const userHavingToken = new LRUCache<number, string>({ max: LRU_CACHE.USER_TOKENS.MAX_SIZE }) | |
69818c93 | 16 | |
69b0a27c C |
17 | // --------------------------------------------------------------------------- |
18 | ||
f201a749 C |
19 | function deleteUserToken (userId: number, t?: Transaction) { |
20 | clearCacheByUserId(userId) | |
21 | ||
22 | return OAuthTokenModel.deleteUserToken(userId, t) | |
23 | } | |
24 | ||
25 | function clearCacheByUserId (userId: number) { | |
557b13ae C |
26 | const token = userHavingToken.get(userId) |
27 | ||
f201a749 | 28 | if (token !== undefined) { |
557b13ae C |
29 | accessTokenCache.del(token) |
30 | userHavingToken.del(userId) | |
f201a749 C |
31 | } |
32 | } | |
33 | ||
34 | function clearCacheByToken (token: string) { | |
557b13ae C |
35 | const tokenModel = accessTokenCache.get(token) |
36 | ||
f201a749 | 37 | if (tokenModel !== undefined) { |
557b13ae C |
38 | userHavingToken.del(tokenModel.userId) |
39 | accessTokenCache.del(token) | |
f201a749 C |
40 | } |
41 | } | |
42 | ||
69818c93 | 43 | function getAccessToken (bearerToken: string) { |
69b0a27c C |
44 | logger.debug('Getting access token (bearerToken: ' + bearerToken + ').') |
45 | ||
3acc5084 C |
46 | if (!bearerToken) return Bluebird.resolve(undefined) |
47 | ||
557b13ae | 48 | if (accessTokenCache.has(bearerToken)) return Bluebird.resolve(accessTokenCache.get(bearerToken)) |
f201a749 | 49 | |
3fd3ab2d | 50 | return OAuthTokenModel.getByTokenAndPopulateUser(bearerToken) |
f201a749 C |
51 | .then(tokenModel => { |
52 | if (tokenModel) { | |
557b13ae C |
53 | accessTokenCache.set(bearerToken, tokenModel) |
54 | userHavingToken.set(tokenModel.userId, tokenModel.accessToken) | |
f201a749 C |
55 | } |
56 | ||
57 | return tokenModel | |
58 | }) | |
69b0a27c C |
59 | } |
60 | ||
69818c93 | 61 | function getClient (clientId: string, clientSecret: string) { |
69b0a27c C |
62 | logger.debug('Getting Client (clientId: ' + clientId + ', clientSecret: ' + clientSecret + ').') |
63 | ||
3fd3ab2d | 64 | return OAuthClientModel.getByIdAndSecret(clientId, clientSecret) |
69b0a27c C |
65 | } |
66 | ||
69818c93 | 67 | function getRefreshToken (refreshToken: string) { |
69b0a27c C |
68 | logger.debug('Getting RefreshToken (refreshToken: ' + refreshToken + ').') |
69 | ||
3fd3ab2d | 70 | return OAuthTokenModel.getByRefreshTokenAndPopulateClient(refreshToken) |
69b0a27c C |
71 | } |
72 | ||
ba12e8b3 C |
73 | async function getUser (usernameOrEmail: string, password: string) { |
74 | logger.debug('Getting User (username/email: ' + usernameOrEmail + ', password: ******).') | |
69b0a27c | 75 | |
ba12e8b3 | 76 | const user = await UserModel.loadByUsernameOrEmail(usernameOrEmail) |
f5028693 | 77 | if (!user) return null |
26d7d31b | 78 | |
f5028693 C |
79 | const passwordMatch = await user.isPasswordMatch(password) |
80 | if (passwordMatch === false) return null | |
26d7d31b | 81 | |
e6921918 C |
82 | if (user.blocked) throw new AccessDeniedError('User is blocked.') |
83 | ||
d9eaee39 JM |
84 | if (CONFIG.SIGNUP.REQUIRES_EMAIL_VERIFICATION && user.emailVerified === false) { |
85 | throw new AccessDeniedError('User email is not verified.') | |
86 | } | |
87 | ||
f5028693 | 88 | return user |
2f372a86 C |
89 | } |
90 | ||
f5028693 | 91 | async function revokeToken (tokenInfo: TokenInfo) { |
3fd3ab2d | 92 | const token = await OAuthTokenModel.getByRefreshTokenAndPopulateUser(tokenInfo.refreshToken) |
bfcef50d | 93 | if (token) { |
f201a749 C |
94 | clearCacheByToken(token.accessToken) |
95 | ||
bfcef50d C |
96 | token.destroy() |
97 | .catch(err => logger.error('Cannot destroy token when revoking token.', { err })) | |
98 | } | |
f5028693 C |
99 | |
100 | /* | |
101 | * Thanks to https://github.com/manjeshpv/node-oauth2-server-implementation/blob/master/components/oauth/mongo-models.js | |
102 | * "As per the discussion we need set older date | |
103 | * revokeToken will expected return a boolean in future version | |
104 | * https://github.com/oauthjs/node-oauth2-server/pull/274 | |
105 | * https://github.com/oauthjs/node-oauth2-server/issues/290" | |
106 | */ | |
107 | const expiredToken = token | |
108 | expiredToken.refreshTokenExpiresAt = new Date('2015-05-28T06:59:53.000Z') | |
109 | ||
110 | return expiredToken | |
69b0a27c C |
111 | } |
112 | ||
3fd3ab2d | 113 | async function saveToken (token: TokenInfo, client: OAuthClientModel, user: UserModel) { |
32bb4156 | 114 | logger.debug('Saving token ' + token.accessToken + ' for client ' + client.id + ' and user ' + user.id + '.') |
69b0a27c | 115 | |
feb4bdfd | 116 | const tokenToCreate = { |
69b0a27c | 117 | accessToken: token.accessToken, |
2f372a86 | 118 | accessTokenExpiresAt: token.accessTokenExpiresAt, |
69b0a27c | 119 | refreshToken: token.refreshToken, |
2f372a86 | 120 | refreshTokenExpiresAt: token.refreshTokenExpiresAt, |
feb4bdfd C |
121 | oAuthClientId: client.id, |
122 | userId: user.id | |
123 | } | |
69b0a27c | 124 | |
3fd3ab2d | 125 | const tokenCreated = await OAuthTokenModel.create(tokenToCreate) |
e6921918 | 126 | return Object.assign(tokenCreated, { client, user }) |
69b0a27c C |
127 | } |
128 | ||
129 | // --------------------------------------------------------------------------- | |
130 | ||
65fcc311 C |
131 | // See https://github.com/oauthjs/node-oauth2-server/wiki/Model-specification for the model specifications |
132 | export { | |
f201a749 C |
133 | deleteUserToken, |
134 | clearCacheByUserId, | |
135 | clearCacheByToken, | |
65fcc311 C |
136 | getAccessToken, |
137 | getClient, | |
138 | getRefreshToken, | |
139 | getUser, | |
140 | revokeToken, | |
141 | saveToken | |
142 | } |