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", null);
051        String extension = parameters.getParameter("extension", "");
052        
053        Catalog catalog = _catalogsManager.getCatalog(catalogName);
054        
055        if (catalog == null)
056        {
057            throw new IllegalArgumentException("Unknown catalog with name " + catalogName);
058        }
059        
060        Response response = ObjectModelHelper.getResponse(objectModel);
061        
062        // Build filename with context
063        StringBuilder fileNameSB = new StringBuilder();
064        fileNameSB.append(catalog.getTitle());
065        String context = parameters.getParameter("context", "");
066        for (String contextPart : context.split("-"))
067        {
068            if (StringUtils.isNotBlank(contextPart))
069            {
070                fileNameSB
071                    .append(" ")
072                    .append(contextPart);
073            }
074        }
075        fileNameSB.append(extension);
076        String fileName = fileNameSB.toString();
077        
078        String encodedName = URIUtils.encodeHeader(fileName);
079        response.setHeader("Content-Disposition", "attachment; filename=\"" + encodedName + "\";filename*=UTF-8''" + encodedName);
080        
081        return EMPTY_MAP;
082    }
083}