001/*
002 *  Copyright 2017 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.linkdirectory.dynamic;
017
018import java.io.IOException;
019import java.util.Arrays;
020import java.util.List;
021
022import org.apache.avalon.framework.service.ServiceException;
023import org.apache.avalon.framework.service.ServiceManager;
024import org.apache.cocoon.ProcessingException;
025import org.apache.cocoon.environment.ObjectModelHelper;
026import org.apache.cocoon.environment.Request;
027import org.apache.cocoon.generation.ServiceableGenerator;
028import org.apache.cocoon.xml.XMLUtils;
029import org.apache.commons.lang3.StringUtils;
030import org.xml.sax.SAXException;
031
032import org.ametys.core.user.CurrentUserProvider;
033import org.ametys.plugins.linkdirectory.DirectoryHelper;
034import org.ametys.plugins.linkdirectory.Link.LinkStatus;
035import org.ametys.plugins.linkdirectory.repository.DefaultLink;
036
037/**
038 * This generator sax the dynamic information provided by {@link DynamicInformationProvider}s
039 */
040public class DynamicInformationGenerator extends ServiceableGenerator
041{
042    private DirectoryHelper _directoryHelper;
043    private DynamicInformationProviderExtensionPoint _dynamicInfoExtensionPoint;
044    private CurrentUserProvider _currentUserProvider;
045
046    @Override
047    public void service(ServiceManager serviceManager) throws ServiceException
048    {
049        super.service(serviceManager);
050        _directoryHelper = (DirectoryHelper) serviceManager.lookup(DirectoryHelper.ROLE);
051        _dynamicInfoExtensionPoint = (DynamicInformationProviderExtensionPoint) serviceManager.lookup(DynamicInformationProviderExtensionPoint.ROLE);
052        _currentUserProvider = (CurrentUserProvider) serviceManager.lookup(CurrentUserProvider.ROLE);
053    }
054
055    @Override
056    public void generate() throws IOException, SAXException, ProcessingException
057    {
058        Request request = ObjectModelHelper.getRequest(objectModel);
059        int maxItems = Integer.valueOf(request.getParameter("maxItems"));
060        
061        String[] themeIds = request.getParameter("themes") != null ? request.getParameter("themes").split(",") : new String[0];
062        String siteName = request.getParameter("site");
063        String lang = request.getParameter("lang");
064        
065        List<DefaultLink> links = _directoryHelper.getLinks(Arrays.asList(themeIds), _currentUserProvider.getUser(), siteName, lang);
066
067        contentHandler.startDocument();
068        XMLUtils.startElement(contentHandler, "dynamic-information");
069        for (DefaultLink defaultLink : links)
070        {
071            if (defaultLink.getStatus() != LinkStatus.BROKEN)
072            {
073                String extensionId = defaultLink.getDynamicInformationProvider();
074                if (StringUtils.isNotBlank(extensionId))
075                {
076                    DynamicInformationProvider provider = _dynamicInfoExtensionPoint.getExtension(extensionId);
077                    if (provider != null)
078                    {
079                        String linkId = defaultLink.getId();
080                        try
081                        {
082                            provider.saxDynamicInformation(contentHandler, linkId, siteName, lang, maxItems);
083                        }
084                        catch (Exception e)
085                        {
086                            getLogger().error("Unable to sax dynamic information from provider of id '" + provider.getId() + "'", e);
087                        }
088                    }
089                }
090            }
091        }
092        XMLUtils.endElement(contentHandler, "dynamic-information");
093        contentHandler.endDocument();
094    }
095}