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
030public class OauthTokenSessionAttributeRequestProxy extends SessionAttributeRequestProxy implements Serviceable, Initializable
031{
032    private OauthProviderExtensionPoint _oauthEP;
033
034    public void service(ServiceManager manager) throws ServiceException
035    {
036        _oauthEP = (OauthProviderExtensionPoint) manager.lookup(OauthProviderExtensionPoint.ROLE);
037    }
038    
039    public void initialize() throws Exception
040    {
041        Set<String> providerIds = _oauthEP.getExtensionsIds();
042        for (String providerId : providerIds)
043        {
044            OAuthProvider provider = _oauthEP.getExtension(providerId);
045            _attributes.add(DefaultOauthProvider.OAUTH_ACCESS_TOKEN_SESSION_ATTRIBUTE + "$" + providerId);
046            _attributes.add(DefaultOauthProvider.OAUTH_ACCESS_TOKEN_EXPIRATION_DATE_SESSION_ATTRIBUTE + "$" + providerId);
047            _attributes.add(DefaultOauthProvider.OAUTH_REFRESH_TOKEN_SESSION_ATTRIBUTE + "$" + providerId);
048            
049            for (String customParam : provider.getCustomParametersName())
050            {
051                _attributes.add(DefaultOauthProvider.OAUTH_CUSTOM_PARAMETER + "$" + providerId + "#" + customParam);
052            }
053        }
054    }
055}