55 lines
2.5 KiB
Ruby
55 lines
2.5 KiB
Ruby
# Enterprise Edition SAML SSO Provider
|
|
# This initializer adds SAML authentication support for Enterprise customers
|
|
|
|
# SAML setup proc for multi-tenant configuration
|
|
SAML_SETUP_PROC = proc do |env|
|
|
request = ActionDispatch::Request.new(env)
|
|
|
|
# Extract account_id from various sources
|
|
account_id = request.params['account_id'] ||
|
|
env['omniauth.params']&.dig('account_id')
|
|
relay_state = request.params['RelayState'] || ''
|
|
for_mobile = relay_state.casecmp('mobile').zero?
|
|
|
|
if account_id
|
|
# Keep SAML request context in OmniAuth env so the callback can be processed
|
|
# without depending on the Rails session cookie.
|
|
env['omniauth.params'] ||= {}
|
|
env['omniauth.params']['account_id'] = account_id
|
|
env['omniauth.params']['RelayState'] = relay_state
|
|
|
|
# Find SAML settings for this account
|
|
settings = AccountSamlSettings.find_by(account_id: account_id)
|
|
|
|
if settings
|
|
# Carry the relay state in the ACS URL (like account_id) only for mobile,
|
|
# so the mobile signal survives the cross-site callback POST without
|
|
# relying on the session cookie or the IdP echoing RelayState back. Web
|
|
# logins keep the registered ACS URL untouched so strict IdPs that
|
|
# validate it exactly are unaffected.
|
|
acs_url = "#{ENV.fetch('FRONTEND_URL', 'http://localhost:3000')}/omniauth/saml/callback?account_id=#{account_id}"
|
|
acs_url += "&RelayState=#{ERB::Util.url_encode(relay_state)}" if for_mobile
|
|
|
|
# Configure the strategy options dynamically
|
|
env['omniauth.strategy'].options[:idp_sso_service_url_runtime_params] = { RelayState: :RelayState }
|
|
env['omniauth.strategy'].options[:assertion_consumer_service_url] = acs_url
|
|
env['omniauth.strategy'].options[:sp_entity_id] = settings.sp_entity_id
|
|
env['omniauth.strategy'].options[:idp_entity_id] = settings.idp_entity_id
|
|
env['omniauth.strategy'].options[:idp_sso_service_url] = settings.sso_url
|
|
env['omniauth.strategy'].options[:idp_cert] = settings.certificate
|
|
env['omniauth.strategy'].options[:name_identifier_format] = 'urn:oasis:names:tc:SAML:1.1:nameid-format:emailAddress'
|
|
else
|
|
# Set a dummy certificate to avoid the error
|
|
env['omniauth.strategy'].options[:idp_cert] = 'DUMMY'
|
|
end
|
|
else
|
|
# Set a dummy certificate to avoid the error
|
|
env['omniauth.strategy'].options[:idp_cert] = 'DUMMY'
|
|
end
|
|
end
|
|
|
|
Rails.application.config.middleware.use OmniAuth::Builder do
|
|
# SAML provider with setup phase for multi-tenant configuration
|
|
provider :saml, setup: SAML_SETUP_PROC
|
|
end
|