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.plugins.survey.actions;
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.Request;
027import org.apache.cocoon.environment.Response;
028import org.apache.cocoon.environment.SourceResolver;
029import org.apache.commons.lang.StringUtils;
030
031import org.ametys.plugins.repository.AmetysObjectResolver;
032import org.ametys.plugins.survey.repository.Survey;
033
034/**
035 * Sets the Content-Disposition header as attachment and injects the survey title as the attachment filename.
036 */
037public class SetSurveyHeaderAction extends ServiceableAction
038{
039    
040    /** The ametys object resolver. */
041    protected AmetysObjectResolver _resolver;
042    
043    @Override
044    public void service(ServiceManager serviceManager) throws ServiceException
045    {
046        super.service(serviceManager);
047        _resolver = (AmetysObjectResolver) serviceManager.lookup(AmetysObjectResolver.ROLE);
048    }
049    
050    @Override
051    public Map act(Redirector redirector, SourceResolver resolver, Map objectModel, String source, Parameters parameters) throws Exception
052    {
053        Request request = ObjectModelHelper.getRequest(objectModel);
054        Response response = ObjectModelHelper.getResponse(objectModel);
055        
056        String surveyId = parameters.getParameter("surveyId", request.getParameter("surveyId"));
057        String suffix = parameters.getParameter("suffix", request.getParameter("suffix"));
058        
059        Survey survey = _resolver.resolveById(surveyId);
060        
061        String title = survey.getTitle();
062        if (StringUtils.isEmpty(title))
063        {
064            title = survey.getLabel();
065        }
066        response.setHeader("Content-Disposition", "attachment; filename=\"" + title + suffix + "\"");
067        
068        return EMPTY_MAP;
069    }
070
071}