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.program.actions;
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.Request;
029import org.apache.cocoon.environment.Response;
030import org.apache.cocoon.environment.SourceResolver;
031
032import org.ametys.cms.repository.Content;
033import org.ametys.odf.program.Program;
034import org.ametys.plugins.repository.AmetysObjectResolver;
035
036/**
037 * Prepare request attributes and response header for export
038 *
039 */
040public class SetFileNameAction extends ServiceableAction
041{
042    private AmetysObjectResolver _resolver;
043    
044    @Override
045    public void service(ServiceManager smanager) throws ServiceException
046    {
047        super.service(smanager);
048        _resolver = (AmetysObjectResolver) smanager.lookup(AmetysObjectResolver.ROLE);
049    }
050    
051    public Map act(Redirector redirector, SourceResolver resolver, Map objectModel, String source, Parameters parameters) throws Exception
052    {
053        Request request = ObjectModelHelper.getRequest(objectModel);
054        String programId = request.getParameter("id");
055        String extension = parameters.getParameter("extension", "");
056        
057        Program program = _resolver.resolveById(programId);
058        
059        request.setAttribute(Content.class.getName(), program);
060        
061        Response response = ObjectModelHelper.getResponse(objectModel);
062        String fileName = program.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}