001/*
002 *  Copyright 2024 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.pagesubscription.generator;
017
018import java.io.IOException;
019
020import org.apache.avalon.framework.service.ServiceException;
021import org.apache.avalon.framework.service.ServiceManager;
022import org.apache.cocoon.ProcessingException;
023import org.apache.cocoon.environment.ObjectModelHelper;
024import org.apache.cocoon.environment.Request;
025import org.apache.cocoon.generation.ServiceableGenerator;
026import org.apache.cocoon.xml.XMLUtils;
027import org.xml.sax.SAXException;
028
029import org.ametys.core.user.CurrentUserProvider;
030import org.ametys.core.user.UserIdentity;
031import org.ametys.plugins.pagesubscription.FrequencyHelper;
032import org.ametys.plugins.pagesubscription.Subscription;
033import org.ametys.plugins.pagesubscription.context.PageSubscriptionContext;
034import org.ametys.plugins.pagesubscription.type.PageSubscriptionType;
035import org.ametys.plugins.pagesubscription.type.SubscriptionTypeExtensionPoint;
036import org.ametys.plugins.repository.AmetysObjectResolver;
037import org.ametys.web.WebHelper;
038import org.ametys.web.repository.site.Site;
039import org.ametys.web.repository.site.SiteManager;
040
041/**
042 * SAX user subscriptions
043 */
044public class UserPageSubscriptionsGenerator extends ServiceableGenerator
045{
046    /** The current user provider */
047    protected CurrentUserProvider _currentUserProvider;
048    
049    /** The site manager */
050    protected SiteManager _siteManager;
051    
052    /** The subscription type extension point */
053    protected SubscriptionTypeExtensionPoint _subscriptionTypeEP;
054    
055    /** The ametys object resolver */
056    protected AmetysObjectResolver _resolver;
057    
058    @Override
059    public void service(ServiceManager smanager) throws ServiceException
060    {
061        super.service(manager);
062        _currentUserProvider = (CurrentUserProvider) smanager.lookup(CurrentUserProvider.ROLE);
063        _subscriptionTypeEP = (SubscriptionTypeExtensionPoint) smanager.lookup(SubscriptionTypeExtensionPoint.ROLE);
064        _siteManager = (SiteManager) smanager.lookup(SiteManager.ROLE);
065        _resolver = (AmetysObjectResolver) smanager.lookup(AmetysObjectResolver.ROLE);
066    }
067    
068    @Override
069    public void generate() throws IOException, SAXException, ProcessingException
070    {
071        contentHandler.startDocument();
072        XMLUtils.startElement(contentHandler, "subscriptions");
073        
074        UserIdentity user = _currentUserProvider.getUser();
075        if (user != null)
076        {
077            FrequencyHelper.saxFrequencies(contentHandler);
078            
079            Request request = ObjectModelHelper.getRequest(objectModel);
080            String siteName = WebHelper.getSiteName(request);
081            PageSubscriptionType subscriptionType = (PageSubscriptionType) _subscriptionTypeEP.getExtension(PageSubscriptionType.ID);
082            
083            Site site = _siteManager.getSite(siteName);
084            for (Subscription subscription : subscriptionType.getUserSubscriptions(site, user, false, PageSubscriptionContext.newInstance()))
085            {
086                subscriptionType.saxSubscription(contentHandler, subscription);
087            }
088        }
089
090        XMLUtils.endElement(contentHandler, "subscriptions");
091        contentHandler.endDocument();
092    }
093    
094}