server/models/oauth/oauth-token.ts

   1 import * as Sequelize from 'sequelize'
   2
   3 import { logger } from '../../helpers'
   4
   5 import { addMethodsToModel } from '../utils'
   6 import {
   7   OAuthTokenInstance,
   8   OAuthTokenAttributes,
   9
  10   OAuthTokenMethods,
  11   OAuthTokenInfo
  12 } from './oauth-token-interface'
  13
  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
  19
  20 export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
  21   OAuthToken = sequelize.define<OAuthTokenInstance, OAuthTokenAttributes>('OAuthToken',
  22     {
  23       accessToken: {
  24         type: DataTypes.STRING,
  25         allowNull: false
  26       },
  27       accessTokenExpiresAt: {
  28         type: DataTypes.DATE,
  29         allowNull: false
  30       },
  31       refreshToken: {
  32         type: DataTypes.STRING,
  33         allowNull: false
  34       },
  35       refreshTokenExpiresAt: {
  36         type: DataTypes.DATE,
  37         allowNull: false
  38       }
  39     },
  40     {
  41       indexes: [
  42         {
  43           fields: [ 'refreshToken' ],
  44           unique: true
  45         },
  46         {
  47           fields: [ 'accessToken' ],
  48           unique: true
  49         },
  50         {
  51           fields: [ 'userId' ]
  52         },
  53         {
  54           fields: [ 'oAuthClientId' ]
  55         }
  56       ]
  57     }
  58   )
  59
  60   const classMethods = [
  61     associate,
  62
  63     getByRefreshTokenAndPopulateClient,
  64     getByTokenAndPopulateUser,
  65     getByRefreshTokenAndPopulateUser,
  66     removeByUserId
  67   ]
  68   addMethodsToModel(OAuthToken, classMethods)
  69
  70   return OAuthToken
  71 }
  72
  73 // ---------------------------------------------------------------------------
  74
  75 function associate (models) {
  76   OAuthToken.belongsTo(models.User, {
  77     foreignKey: {
  78       name: 'userId',
  79       allowNull: false
  80     },
  81     onDelete: 'cascade'
  82   })
  83
  84   OAuthToken.belongsTo(models.OAuthClient, {
  85     foreignKey: {
  86       name: 'oAuthClientId',
  87       allowNull: false
  88     },
  89     onDelete: 'cascade'
  90   })
  91 }
  92
  93 getByRefreshTokenAndPopulateClient = function (refreshToken: string) {
  94   const query = {
  95     where: {
  96       refreshToken: refreshToken
  97     },
  98     include: [ OAuthToken['sequelize'].models.OAuthClient ]
  99   }
 100
 101   return OAuthToken.findOne(query).then(function (token) {
 102     if (!token) return null
 103
 104     const tokenInfos: OAuthTokenInfo = {
 105       refreshToken: token.refreshToken,
 106       refreshTokenExpiresAt: token.refreshTokenExpiresAt,
 107       client: {
 108         id: token['client'].id
 109       },
 110       user: {
 111         id: token['user']
 112       }
 113     }
 114
 115     return tokenInfos
 116   }).catch(function (err) {
 117     logger.info('getRefreshToken error.', err)
 118   })
 119 }
 120
 121 getByTokenAndPopulateUser = function (bearerToken: string) {
 122   const query = {
 123     where: {
 124       accessToken: bearerToken
 125     },
 126     include: [ OAuthToken['sequelize'].models.User ]
 127   }
 128
 129   return OAuthToken.findOne(query).then(function (token) {
 130     if (token) token['user'] = token.User
 131
 132     return token
 133   })
 134 }
 135
 136 getByRefreshTokenAndPopulateUser = function (refreshToken: string) {
 137   const query = {
 138     where: {
 139       refreshToken: refreshToken
 140     },
 141     include: [ OAuthToken['sequelize'].models.User ]
 142   }
 143
 144   return OAuthToken.findOne(query).then(function (token) {
 145     token['user'] = token.User
 146
 147     return token
 148   })
 149 }
 150
 151 removeByUserId = function (userId: number) {
 152   const query = {
 153     where: {
 154       userId: userId
 155     }
 156   }
 157
 158   return OAuthToken.destroy(query)
 159 }