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;
028
029/**
030 * Resource reader handling hashed resources in site cache as well as "normal" resources. 
031 */
032public class SiteResourceReader extends RuntimeResourceReader
033{
034    
035    @Override
036    public void setup(SourceResolver res, Map cocoonObjectModel, String src, Parameters par) throws ProcessingException, SAXException, IOException
037    {
038        super.setup(res, cocoonObjectModel, src, par);
039        
040        if (this.inputSource instanceof FileSource)
041        {
042            FileSource fileSource = (FileSource) this.inputSource;
043            
044            // If a download is requested, set the "attachment" Content-Disposition header.
045            if ("true".equals(this.request.getParameter("download")))
046            {
047                setAttachmentContentDisposition(fileSource);
048            }
049            
050            if (!SiteCacheHelper.isValid(fileSource))
051            {
052                this.inputSource = SiteCacheHelper.getHashedFileSource(res, fileSource);
053                
054                // Setup the headers another time, as the source was changed.
055                setupHeaders();
056            }
057        }
058    }
059    
060    /**
061     * Sets the Content-Disposition header as "attachment" with the wanted filename.
062     * @param fileSource the file source.
063     */
064    protected void setAttachmentContentDisposition(FileSource fileSource)
065    {
066        String name = fileSource.getName();
067        
068        response.setHeader("Content-Disposition", "attachment; filename=\"" + name + "\";filename*=utf-8''" + name);
069    }
070    
071}