server/models/oauth-token.js

   1 'use strict'
   2
   3 const logger = require('../helpers/logger')
   4
   5 // ---------------------------------------------------------------------------
   6
   7 module.exports = function (sequelize, DataTypes) {
   8   const OAuthToken = sequelize.define('OAuthToken',
   9     {
  10       accessToken: {
  11         type: DataTypes.STRING,
  12         allowNull: false
  13       },
  14       accessTokenExpiresAt: {
  15         type: DataTypes.DATE,
  16         allowNull: false
  17       },
  18       refreshToken: {
  19         type: DataTypes.STRING,
  20         allowNull: false
  21       },
  22       refreshTokenExpiresAt: {
  23         type: DataTypes.DATE,
  24         allowNull: false
  25       }
  26     },
  27     {
  28       classMethods: {
  29         associate,
  30
  31         getByRefreshTokenAndPopulateClient,
  32         getByTokenAndPopulateUser,
  33         getByRefreshTokenAndPopulateUser,
  34         removeByUserId
  35       }
  36     }
  37   )
  38
  39   return OAuthToken
  40 }
  41
  42 // ---------------------------------------------------------------------------
  43
  44 function associate (models) {
  45   this.belongsTo(models.User, {
  46     foreignKey: {
  47       name: 'userId',
  48       allowNull: false
  49     },
  50     onDelete: 'cascade'
  51   })
  52 }
  53
  54 function getByRefreshTokenAndPopulateClient (refreshToken) {
  55   const query = {
  56     where: {
  57       refreshToken: refreshToken
  58     },
  59     include: [ this.associations.OAuthClient ]
  60   }
  61
  62   return this.findOne(query).then(function (token) {
  63     if (!token) return token
  64
  65     const tokenInfos = {
  66       refreshToken: token.refreshToken,
  67       refreshTokenExpiresAt: token.refreshTokenExpiresAt,
  68       client: {
  69         id: token.client.id
  70       },
  71       user: {
  72         id: token.user
  73       }
  74     }
  75
  76     return tokenInfos
  77   }).catch(function (err) {
  78     logger.info('getRefreshToken error.', { error: err })
  79   })
  80 }
  81
  82 function getByTokenAndPopulateUser (bearerToken) {
  83   const query = {
  84     where: {
  85       accessToken: bearerToken
  86     },
  87     include: [ this.sequelize.models.User ]
  88   }
  89
  90   return this.findOne(query).then(function (token) {
  91     if (token) token.user = token.User
  92
  93     return token
  94   })
  95 }
  96
  97 function getByRefreshTokenAndPopulateUser (refreshToken) {
  98   const query = {
  99     where: {
 100       refreshToken: refreshToken
 101     },
 102     include: [ this.sequelize.models.User ]
 103   }
 104
 105   return this.findOne(query).then(function (token) {
 106     token.user = token.User
 107
 108     return token
 109   })
 110 }
 111
 112 function removeByUserId (userId, callback) {
 113   const query = {
 114     where: {
 115       userId: userId
 116     }
 117   }
 118
 119   return this.destroy(query).asCallback(callback)
 120 }