mirror of
https://github.com/status-im/codimd.git
synced 2025-02-05 16:33:23 +00:00
104 lines
3.3 KiB
TypeScript
104 lines
3.3 KiB
TypeScript
import * as fs from "fs";
|
|
import {Router} from "express";
|
|
import * as passport from "passport";
|
|
import {Strategy as SamlStrategy} from "passport-saml";
|
|
|
|
import * as config from "../../config";
|
|
import * as models from "../../models";
|
|
import * as logger from "../../logger";
|
|
import {urlencodedParser} from "../../utils";
|
|
|
|
const intersection = function (array1, array2) {
|
|
return array1.filter((n) => array2.includes(n))
|
|
}
|
|
|
|
const samlAuth = Router()
|
|
export = samlAuth
|
|
|
|
passport.use(new SamlStrategy({
|
|
callbackUrl: config.serverURL + '/auth/saml/callback',
|
|
entryPoint: config.saml.idpSsoUrl,
|
|
issuer: config.saml.issuer || config.serverURL,
|
|
cert: fs.readFileSync(config.saml.idpCert, 'utf-8'),
|
|
identifierFormat: config.saml.identifierFormat,
|
|
disableRequestedAuthnContext: config.saml.disableRequestedAuthnContext
|
|
}, function (user, done) {
|
|
// check authorization if needed
|
|
if (config.saml.externalGroups && config.saml.groupAttribute) {
|
|
const externalGroups = intersection(config.saml.externalGroups, user[config.saml.groupAttribute])
|
|
if (externalGroups.length > 0) {
|
|
logger.error('saml permission denied: ' + externalGroups.join(', '))
|
|
return done('Permission denied', null)
|
|
}
|
|
}
|
|
if (config.saml.requiredGroups && config.saml.groupAttribute) {
|
|
if (intersection(config.saml.requiredGroups, user[config.saml.groupAttribute]).length === 0) {
|
|
logger.error('saml permission denied')
|
|
return done('Permission denied', null)
|
|
}
|
|
}
|
|
// user creation
|
|
const uuid = user[config.saml.attribute.id] || user.nameID
|
|
const profile = {
|
|
provider: 'saml',
|
|
id: 'SAML-' + uuid,
|
|
username: user[config.saml.attribute.username] || user.nameID,
|
|
emails: user[config.saml.attribute.email] ? [user[config.saml.attribute.email]] : []
|
|
}
|
|
if (profile.emails.length === 0 && config.saml.identifierFormat === 'urn:oasis:names:tc:SAML:1.1:nameid-format:emailAddress') {
|
|
profile.emails.push(user.nameID)
|
|
}
|
|
const stringifiedProfile = JSON.stringify(profile)
|
|
models.User.findOrCreate({
|
|
where: {
|
|
profileid: profile.id.toString()
|
|
},
|
|
defaults: {
|
|
profile: stringifiedProfile
|
|
}
|
|
}).spread(function (user, created) {
|
|
if (user) {
|
|
let needSave = false
|
|
if (user.profile !== stringifiedProfile) {
|
|
user.profile = stringifiedProfile
|
|
needSave = true
|
|
}
|
|
if (needSave) {
|
|
user.save().then(function () {
|
|
if (config.debug) {
|
|
logger.debug('user login: ' + user.id)
|
|
}
|
|
return done(null, user)
|
|
})
|
|
} else {
|
|
if (config.debug) {
|
|
logger.debug('user login: ' + user.id)
|
|
}
|
|
return done(null, user)
|
|
}
|
|
}
|
|
}).catch(function (err) {
|
|
logger.error('saml auth failed: ' + err)
|
|
return done(err, null)
|
|
})
|
|
}))
|
|
|
|
samlAuth.get('/auth/saml',
|
|
passport.authenticate('saml', {
|
|
successReturnToOrRedirect: config.serverURL + '/',
|
|
failureRedirect: config.serverURL + '/'
|
|
})
|
|
)
|
|
|
|
samlAuth.post('/auth/saml/callback', urlencodedParser,
|
|
passport.authenticate('saml', {
|
|
successReturnToOrRedirect: config.serverURL + '/',
|
|
failureRedirect: config.serverURL + '/'
|
|
})
|
|
)
|
|
|
|
samlAuth.get('/auth/saml/metadata', function (req, res) {
|
|
res.type('application/xml')
|
|
res.send((passport as any)._strategy('saml').generateServiceProviderMetadata())
|
|
})
|