001/*
002 *  Copyright 2013 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.site;
017
018import java.io.IOException;
019import java.util.Map;
020
021import org.apache.avalon.framework.parameters.Parameters;
022import org.apache.cocoon.ProcessingException;
023import org.apache.cocoon.environment.SourceResolver;
024import org.apache.excalibur.source.impl.FileSource;
025import org.xml.sax.SAXException;
026
027import org.ametys.core.cocoon.RuntimeResourceReader;
028import org.ametys.core.util.URIUtils;
029
030/**
031 * Resource reader handling hashed resources in site cache as well as "normal" resources. 
032 */
033public class SiteResourceReader extends RuntimeResourceReader
034{
035    
036    @Override
037    public void setup(SourceResolver res, Map cocoonObjectModel, String src, Parameters par) throws ProcessingException, SAXException, IOException
038    {
039        super.setup(res, cocoonObjectModel, src, par);
040        
041        if (this.inputSource instanceof FileSource)
042        {
043            FileSource fileSource = (FileSource) this.inputSource;
044            
045            // If a download is requested, set the "attachment" Content-Disposition header.
046            if ("true".equals(this.request.getParameter("download")))
047            {
048                setAttachmentContentDisposition(fileSource);
049            }
050            
051            if (!SiteCacheHelper.isValid(fileSource))
052            {
053                this.inputSource = SiteCacheHelper.getHashedFileSource(res, fileSource);
054                
055                // Setup the headers another time, as the source was changed.
056                setupHeaders();
057            }
058        }
059    }
060    
061    /**
062     * Sets the Content-Disposition header as "attachment" with the wanted filename.
063     * @param fileSource the file source.
064     */
065    protected void setAttachmentContentDisposition(FileSource fileSource)
066    {
067        String name = fileSource.getName();
068        String encodedName = URIUtils.encodeHeader(name);
069        
070        response.setHeader("Content-Disposition", "attachment; filename=\"" + encodedName + "\";filename*=UTF-8''" + encodedName);
071    }
072    
073}