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.net.URI;
019import java.net.URISyntaxException;
020import java.util.Map;
021
022import org.apache.avalon.framework.parameters.Parameters;
023import org.apache.avalon.framework.service.ServiceException;
024import org.apache.avalon.framework.service.ServiceManager;
025import org.apache.cocoon.acting.ServiceableAction;
026import org.apache.cocoon.environment.ObjectModelHelper;
027import org.apache.cocoon.environment.Redirector;
028import org.apache.cocoon.environment.Response;
029import org.apache.cocoon.environment.SourceResolver;
030
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        
061        Response response = ObjectModelHelper.getResponse(objectModel);
062        String fileName = catalog.getTitle() + extension;
063        String encodedName = _encodeFileName(fileName);
064        
065        response.setHeader("Content-Disposition", "attachment; filename=\"" + (encodedName != null ? encodedName : fileName) + "\"" + (encodedName != null ? ";filename*=UTF-8''" + encodedName : ""));
066        
067        return EMPTY_MAP;
068    }
069    
070    private String _encodeFileName(String fileName)
071    {
072        String encodedName = null;
073        try
074        {
075            URI uri = new URI(null, null, fileName, null);
076            encodedName = uri.toASCIIString();
077            // EXPLORER-358 : Fix for Chrome which does not support comma in filename
078            return encodedName.replaceAll(",", "%2C");
079        }
080        catch (URISyntaxException e)
081        {
082            // do nothing and send no encoded name
083            return null;
084        }
085    }
086
087}