001/*
002 *  Copyright 2023 Anyware Services
003 *
004 *  Licensed under the Apache License, Version 2.0 (the "License");
005 *  you may not use this file except in compliance with the License.
006 *  You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 *  Unless required by applicable law or agreed to in writing, software
011 *  distributed under the License is distributed on an "AS IS" BASIS,
012 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 *  See the License for the specific language governing permissions and
014 *  limitations under the License.
015 */
016package org.ametys.plugins.extrausermgt.proxy;
017
018import java.util.Set;
019
020import org.apache.avalon.framework.activity.Initializable;
021import org.apache.avalon.framework.service.ServiceException;
022import org.apache.avalon.framework.service.ServiceManager;
023import org.apache.avalon.framework.service.Serviceable;
024
025import org.ametys.plugins.extrausermgt.oauth.DefaultOauthProvider;
026import org.ametys.plugins.extrausermgt.oauth.OAuthProvider;
027import org.ametys.plugins.extrausermgt.oauth.OauthProviderExtensionPoint;
028import org.ametys.plugins.site.proxy.SessionAttributeRequestProxy;
029
030/**
031 * Define the list of session attribute to proxy to the front for all the defined Oauth providers
032 */
033public class OauthTokenSessionAttributeRequestProxy extends SessionAttributeRequestProxy implements Serviceable, Initializable
034{
035    private OauthProviderExtensionPoint _oauthEP;
036
037    public void service(ServiceManager manager) throws ServiceException
038    {
039        _oauthEP = (OauthProviderExtensionPoint) manager.lookup(OauthProviderExtensionPoint.ROLE);
040    }
041    
042    public void initialize() throws Exception
043    {
044        Set<String> providerIds = _oauthEP.getExtensionsIds();
045        for (String providerId : providerIds)
046        {
047            OAuthProvider provider = _oauthEP.getExtension(providerId);
048            _attributes.add(DefaultOauthProvider.OAUTH_ACCESS_TOKEN_SESSION_ATTRIBUTE + "$" + providerId);
049            _attributes.add(DefaultOauthProvider.OAUTH_ACCESS_TOKEN_EXPIRATION_DATE_SESSION_ATTRIBUTE + "$" + providerId);
050            _attributes.add(DefaultOauthProvider.OAUTH_REFRESH_TOKEN_SESSION_ATTRIBUTE + "$" + providerId);
051            
052            for (String customParam : provider.getCustomParametersName())
053            {
054                _attributes.add(DefaultOauthProvider.OAUTH_CUSTOM_PARAMETER + "$" + providerId + "#" + customParam);
055            }
056        }
057    }
058}