wikijs-fork/server/modules/authentication/rocketchat/authentication.js

const _ = require('lodash')

/* global WIKI */

// ------------------------------------
// Rocket.chat Account
// ------------------------------------

const OAuth2Strategy = require('passport-oauth2').Strategy

module.exports = {
  init (passport, conf) {
    const siteURL = conf.siteURL.slice(-1) === '/' ? conf.siteURL.slice(0, -1) : conf.siteURL

    OAuth2Strategy.prototype.userProfile = function (accessToken, cb) {
      this._oauth2.get(`${siteURL}/api/v1/me`, accessToken, (err, body, res) => {
        if (err) {
          WIKI.logger.warn('Rocket.chat - Failed to fetch user profile.')
          return cb(err)
        }
        try {
          const usr = JSON.parse(body)
          cb(null, {
            id: usr._id,
            displayName: _.isEmpty(usr.name) ? usr.username : usr.name,
            email: usr.emails[0].address,
            picture: usr.avatarUrl
          })
        } catch (err) {
          WIKI.logger.warn('Rocket.chat - Failed to parse user profile.')
          cb(err)
        }
      })
    }

    passport.use('rocketchat',
      new OAuth2Strategy({
        authorizationURL: `${siteURL}/oauth/authorize`,
        tokenURL: `${siteURL}/oauth/token`,
        clientID: conf.clientId,
        clientSecret: conf.clientSecret,
        callbackURL: conf.callbackURL,
        passReqToCallback: true
      }, async (req, accessToken, refreshToken, profile, cb) => {
        try {
          const user = await WIKI.models.users.processProfile({
            providerKey: req.params.strategy,
            profile
          })
          cb(null, user)
        } catch (err) {
          cb(err, null)
        }
      })
    )
  },
  logout (conf) {
    if (!conf.logoutURL) {
      return '/'
    } else {
      return conf.logoutURL
    }
  }
}