001/*
002 *  Copyright 2010 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.web.repository.sitemap.generators;
017
018import java.io.IOException;
019import java.util.Map;
020import java.util.Set;
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.generation.ServiceableGenerator;
027import org.apache.cocoon.xml.AttributesImpl;
028import org.apache.cocoon.xml.XMLUtils;
029import org.xml.sax.SAXException;
030
031import org.ametys.core.right.RightManager;
032import org.ametys.core.user.CurrentUserProvider;
033import org.ametys.core.user.UserIdentity;
034import org.ametys.plugins.repository.AmetysObjectResolver;
035import org.ametys.web.repository.sitemap.Sitemap;
036import org.ametys.web.skin.SkinsManager;
037
038/**
039 * SAX the site map information
040 */
041public class SitemapInformationGenerator extends ServiceableGenerator
042{
043    /** Ametys object resovler. */
044    protected AmetysObjectResolver _resolver;
045    /** The {@link SkinsManager} */
046    protected SkinsManager _skinsManager;
047    /** The rights manager */
048    protected RightManager _rightManager;
049    /** The user provider */
050    protected CurrentUserProvider _userProvider;
051    
052    @Override
053    public void service(ServiceManager serviceManager) throws ServiceException
054    {
055        super.service(serviceManager);
056        _resolver = (AmetysObjectResolver) serviceManager.lookup(AmetysObjectResolver.ROLE);
057        _skinsManager = (SkinsManager) serviceManager.lookup(SkinsManager.ROLE);
058        _rightManager = (RightManager) serviceManager.lookup(RightManager.ROLE);
059        _userProvider = (CurrentUserProvider) serviceManager.lookup(CurrentUserProvider.ROLE);
060    }
061    
062    @Override
063    public void generate() throws IOException, SAXException, ProcessingException
064    {
065        @SuppressWarnings("unchecked")
066        Map<String, Object> jsParameters = (Map<String, Object>) objectModel.get(ObjectModelHelper.PARENT_CONTEXT);
067        String id = (String) jsParameters.get("id");
068        
069        contentHandler.startDocument();
070        Sitemap sitemap = _resolver.resolveById(id);
071        
072        AttributesImpl attrs = new AttributesImpl();
073        attrs.addCDATAAttribute("id", sitemap.getId());
074        attrs.addCDATAAttribute("lang", sitemap.getName());
075        attrs.addCDATAAttribute("siteName", sitemap.getSite().getName());
076        XMLUtils.startElement(contentHandler, "sitemap", attrs);
077        
078        // SAX user rights
079        saxUserRights(sitemap);
080        
081        XMLUtils.endElement(contentHandler, "sitemap");
082        contentHandler.endDocument();
083    }
084    
085    /**
086     * SAX the user rights on sitemap
087     * @param sitemap The sitemap to check rights
088     * @throws SAXException if an error occurred while SAXing
089     */
090    protected void saxUserRights (Sitemap sitemap) throws SAXException
091    {
092        UserIdentity user = _userProvider.getUser();
093        Set<String> rights = _rightManager.getUserRights(user, sitemap);
094        
095        XMLUtils.startElement(contentHandler, "user-rights");
096        
097        for (String rightID : rights)
098        {
099            AttributesImpl rightAttr = new AttributesImpl();
100            rightAttr.addCDATAAttribute("id", rightID);
101            XMLUtils.createElement(contentHandler, "right", rightAttr);
102        }
103        XMLUtils.endElement(contentHandler, "user-rights");
104    }
105}