wikijs-fork/server/core/auth.js

84 lines
2.3 KiB
JavaScript
Raw Normal View History

const passport = require('passport')
const fs = require('fs-extra')
2018-05-28 18:46:55 +00:00
const _ = require('lodash')
const path = require('path')
2018-08-04 21:27:55 +00:00
const NodeCache = require('node-cache')
const userCache = new NodeCache({
stdTTL: 10,
checkperiod: 600,
deleteOnExpire: true
})
2018-05-28 18:46:55 +00:00
/* global WIKI */
2017-04-02 23:56:47 +00:00
2017-08-03 03:47:11 +00:00
module.exports = {
strategies: {},
init() {
this.passport = passport
// Serialization user methods
2017-04-02 23:56:47 +00:00
2017-08-03 03:47:11 +00:00
passport.serializeUser(function (user, done) {
2018-01-10 01:41:53 +00:00
done(null, user.id)
2017-08-03 03:47:11 +00:00
})
2017-04-02 23:56:47 +00:00
2017-08-03 03:47:11 +00:00
passport.deserializeUser(function (id, done) {
2018-08-04 21:27:55 +00:00
const usr = userCache.get(id)
if (usr) {
done(null, usr)
} else {
WIKI.models.users.query().findById(id).then((user) => {
if (user) {
userCache.set(id, user)
done(null, user)
} else {
done(new Error(WIKI.lang.t('auth:errors:usernotfound')), null)
}
return true
}).catch((err) => {
done(err, null)
})
}
2017-04-02 23:56:47 +00:00
})
return this
2018-05-28 18:46:55 +00:00
},
async activateStrategies() {
try {
// Unload any active strategies
WIKI.auth.strategies = {}
2018-05-28 18:46:55 +00:00
const currentStrategies = _.keys(passport._strategies)
_.pull(currentStrategies, 'session')
_.forEach(currentStrategies, stg => { passport.unuse(stg) })
// Load enabled strategies
const enabledStrategies = await WIKI.models.authentication.getStrategies()
2018-05-28 18:46:55 +00:00
for (let idx in enabledStrategies) {
const stg = enabledStrategies[idx]
if (!stg.isEnabled) { continue }
const strategy = require(`../modules/authentication/${stg.key}/authentication.js`)
2018-06-11 03:23:09 +00:00
stg.config.callbackURL = `${WIKI.config.host}/login/${stg.key}/callback` // TODO: config.host
2018-05-28 18:46:55 +00:00
strategy.init(passport, stg.config)
fs.readFile(path.join(WIKI.ROOTPATH, `assets/svg/auth-icon-${strategy.key}.svg`), 'utf8').then(iconData => {
strategy.icon = iconData
}).catch(err => {
if (err.code === 'ENOENT') {
strategy.icon = '[missing icon]'
} else {
WIKI.logger.warn(err)
}
})
WIKI.auth.strategies[stg.key] = strategy
WIKI.logger.info(`Authentication Strategy ${stg.title}: [ OK ]`)
}
} catch (err) {
WIKI.logger.error(`Authentication Strategy: [ FAILED ]`)
WIKI.logger.error(err)
}
2017-08-03 03:47:11 +00:00
}
2017-04-02 23:56:47 +00:00
}