1 import * as Sequelize from 'sequelize'
3 import { logger } from '../../helpers'
5 import { addMethodsToModel } from '../utils'
12 } from './oauth-token-interface'
14 let OAuthToken: Sequelize.Model<OAuthTokenInstance, OAuthTokenAttributes>
15 let getByRefreshTokenAndPopulateClient: OAuthTokenMethods.GetByRefreshTokenAndPopulateClient
16 let getByTokenAndPopulateUser: OAuthTokenMethods.GetByTokenAndPopulateUser
17 let getByRefreshTokenAndPopulateUser: OAuthTokenMethods.GetByRefreshTokenAndPopulateUser
18 let removeByUserId: OAuthTokenMethods.RemoveByUserId
20 export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
21 OAuthToken = sequelize.define<OAuthTokenInstance, OAuthTokenAttributes>('OAuthToken',
24 type: DataTypes.STRING,
27 accessTokenExpiresAt: {
32 type: DataTypes.STRING,
35 refreshTokenExpiresAt: {
43 fields: [ 'refreshToken' ],
47 fields: [ 'accessToken' ],
54 fields: [ 'oAuthClientId' ]
60 const classMethods = [
63 getByRefreshTokenAndPopulateClient,
64 getByTokenAndPopulateUser,
65 getByRefreshTokenAndPopulateUser,
68 addMethodsToModel(OAuthToken, classMethods)
73 // ---------------------------------------------------------------------------
75 function associate (models) {
76 OAuthToken.belongsTo(models.User, {
84 OAuthToken.belongsTo(models.OAuthClient, {
86 name: 'oAuthClientId',
93 getByRefreshTokenAndPopulateClient = function (refreshToken: string) {
96 refreshToken: refreshToken
98 include: [ OAuthToken['sequelize'].models.OAuthClient ]
101 return OAuthToken.findOne(query).then(function (token) {
102 if (!token) return null
104 const tokenInfos: OAuthTokenInfo = {
105 refreshToken: token.refreshToken,
106 refreshTokenExpiresAt: token.refreshTokenExpiresAt,
108 id: token['client'].id
116 }).catch(function (err) {
117 logger.info('getRefreshToken error.', err)
121 getByTokenAndPopulateUser = function (bearerToken: string) {
124 accessToken: bearerToken
126 include: [ OAuthToken['sequelize'].models.User ]
129 return OAuthToken.findOne(query).then(function (token) {
130 if (token) token['user'] = token.User
136 getByRefreshTokenAndPopulateUser = function (refreshToken: string) {
139 refreshToken: refreshToken
141 include: [ OAuthToken['sequelize'].models.User ]
144 return OAuthToken.findOne(query).then(function (token) {
145 token['user'] = token.User
151 removeByUserId = function (userId: number) {
158 return OAuthToken.destroy(query)