001/*
002 *  Copyright 2011 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.explorer;
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.plugins.explorer.resources.Resource;
030import org.ametys.plugins.explorer.resources.ResourceCollection;
031import org.ametys.plugins.repository.AmetysObject;
032import org.ametys.plugins.repository.AmetysObjectIterable;
033import org.ametys.plugins.repository.AmetysObjectResolver;
034
035/**
036 * Generates the images' id of a folder
037 *
038 */
039public class GetImagesGenerator extends ServiceableGenerator
040{
041    private AmetysObjectResolver _resolver;
042    
043    @Override
044    public void service(ServiceManager smanager) throws ServiceException
045    {
046        super.service(smanager);
047        
048        _resolver = (AmetysObjectResolver) smanager.lookup(AmetysObjectResolver.ROLE);
049    }
050    
051    @Override
052    public void generate() throws IOException, SAXException, ProcessingException
053    {
054        Request request = ObjectModelHelper.getRequest(objectModel);
055        String id = request.getParameter("id");
056        
057        contentHandler.startDocument();
058        XMLUtils.startElement(contentHandler, "images");
059        
060        ResourceCollection collection = _resolver.resolveById(id);
061        AmetysObjectIterable<AmetysObject> children = collection.getChildren();
062        
063        for (AmetysObject child : children)
064        {
065            if (child instanceof Resource)
066            {
067                Resource resource = (Resource) child;
068                if (_isImage(resource))
069                {
070                    XMLUtils.createElement(contentHandler, "file", resource.getId());
071                }
072            }
073        }
074        
075        XMLUtils.endElement(contentHandler, "images");
076        contentHandler.endDocument();
077    }
078    
079    private boolean _isImage (Resource resource)
080    {
081        String filename = resource.getName().toLowerCase();
082        
083        return filename.endsWith(".jpeg") || filename.endsWith(".jpg") || filename.endsWith(".gif") || filename.endsWith(".png");
084    }
085
086}