001/*
002 *  Copyright 2015 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.odf.export;
017
018import java.util.Map;
019
020import org.apache.avalon.framework.parameters.Parameters;
021import org.apache.avalon.framework.service.ServiceException;
022import org.apache.avalon.framework.service.ServiceManager;
023import org.apache.cocoon.acting.ServiceableAction;
024import org.apache.cocoon.environment.ObjectModelHelper;
025import org.apache.cocoon.environment.Redirector;
026import org.apache.cocoon.environment.Response;
027import org.apache.cocoon.environment.SourceResolver;
028import org.apache.commons.lang3.StringUtils;
029
030import org.ametys.core.util.URIUtils;
031import org.ametys.odf.catalog.Catalog;
032import org.ametys.odf.catalog.CatalogsManager;
033
034/**
035 * Set the title of the catalog in response header
036 */
037public class SetCatalogHeader extends ServiceableAction
038{
039    private CatalogsManager _catalogsManager;
040    
041    @Override
042    public void service(ServiceManager smanager) throws ServiceException
043    {
044        super.service(smanager);
045        _catalogsManager = (CatalogsManager) smanager.lookup(CatalogsManager.ROLE);
046    }
047    
048    public Map act(Redirector redirector, SourceResolver resolver, Map objectModel, String source, Parameters parameters) throws Exception
049    {
050        String catalogName = parameters.getParameter("catalog", "_default");
051        String extension = parameters.getParameter("extension", "");
052        
053        Catalog catalog = catalogName.equals("_default")
054                ? _catalogsManager.getDefaultCatalog()
055                : _catalogsManager.getCatalog(catalogName);
056        
057        if (catalog == null)
058        {
059            throw new IllegalArgumentException("Unknown catalog with name " + catalogName);
060        }
061        
062        Response response = ObjectModelHelper.getResponse(objectModel);
063        
064        // Build filename with context
065        StringBuilder fileNameSB = new StringBuilder();
066        fileNameSB.append(catalog.getTitle());
067        String context = parameters.getParameter("context", "");
068        for (String contextPart : context.split("-"))
069        {
070            if (StringUtils.isNotBlank(contextPart))
071            {
072                fileNameSB
073                    .append(" ")
074                    .append(contextPart);
075            }
076        }
077        fileNameSB.append(extension);
078        String fileName = fileNameSB.toString();
079        
080        String encodedName = URIUtils.encodeHeader(fileName);
081        response.setHeader("Content-Disposition", "attachment; filename=\"" + encodedName + "\";filename*=UTF-8''" + encodedName);
082        
083        return EMPTY_MAP;
084    }
085}