AttachDownload.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2015 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.attach.www;
 
import com.lowagie.text.DocumentException;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.amic.util.string.Cipher;
import org.turro.attach.db.AttachPU;
import org.turro.attach.entity.Attachment;
import org.turro.auth.Authentication;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.context.IConstructor;
import org.turro.elephant.direct.DirectContent;
import org.turro.elephant.direct.DirectContents;
import org.turro.elephant.direct.IDirectContent;
import org.turro.log.SystemLogType;
import org.turro.log.SystemLogger;
 
@DirectContent(identifier="attachment-show")
public class AttachDownload implements IDirectContent {
 
  public static String createURL(IConstructor constructor, Attachment attachment) {
    Map<String, String> downKeys = (Map<String, String>) constructor.getSessionAttribute("downKeys");
    if(downKeys == null) {
      downKeys = new HashMap<>();
    }
    String key = Cipher.digest(attachment.getId() + "", 40);
    downKeys.put(key, attachment.getId() + "");
    constructor.setSessionAttribute("downKeys", downKeys);
    return ElephantContext.getRootWebPath() + DirectContents.DIRECT_CONTENT_PATH + getIdentifier() + "?id=" + key;
  }
 
  public static String createROURL(IConstructor constructor, Attachment attachment) {
    if("pdf".equalsIgnoreCase(attachment.getFileExtension())) {
      try {
        Map<String, String> downKeys = (Map<String, String>) constructor.getSessionAttribute("downKeys");
        if(downKeys == null) {
          downKeys = new HashMap<>();
        }
        String key = Cipher.digest(attachment.getId() + "", 40);
        downKeys.put(key, attachment.getId() + "");
        constructor.setSessionAttribute("downKeys", downKeys);
        return ElephantContext.getRootWebPath() +
                "/_internal/js/pdfjs/web/roviewer.html?file=" +
                ElephantContext.getRootWebPath() + DirectContents.DIRECT_CONTENT_PATH + getIdentifier() + "?id" + URLEncoder.encode("=", "UTF-8") + key;
      } catch (UnsupportedEncodingException ex) {
        Logger.getLogger(AttachDownload.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
        return null;
      }
    } else {
      return createURL(constructor, attachment);
    }
  }
  
  public static String getIdentifier() {
    return AttachDownload.class.getAnnotation(DirectContent.class).identifier();
  }
 
  @Override
  public boolean itsMe(String id) {
    return getIdentifier().equals(id);
  }
 
  @Override
  public boolean myTurn(HttpServletRequest request) {
    return DirectContents.isYourTurn(request, getIdentifier());
  }
 
  @Override
  public void execute(ServletContext context, HttpServletRequest request, HttpServletResponse response) {
    Map downKeys = (Map) request.getSession().getAttribute("downKeys");
    if(downKeys != null) {
      String value = (String) downKeys.get(request.getParameter("id"));
      if(value != null) {
        Attachment attachment = new AttachPU().find(Attachment.class, Long.parseLong(value));
        if(attachment != null) {
          try {
            String contentType = attachment.getFileContentType();
            String fileName = attachment.getFileName();
            
            response.setHeader("pragma", "no-cache");
            response.setHeader("Cache-control", "no-cache, no-store, must-revalidate");
            response.setHeader("Expires", "01 Apr 1995 01:10:10 GMT");
            
            if(contentType!=null) {
              response.setContentType(contentType);
            }
            
            if(fileName != null) {
              fileName = fileName.substring(fileName.lastIndexOf('\\')+1);
              fileName = fileName.substring(fileName.lastIndexOf('/')+1);
              
              StringBuilder contentDisposition = new StringBuilder();
              
              contentDisposition.append("filename=\"");
              contentDisposition.append(fileName);
              contentDisposition.append("\"");
              
              response.setHeader("Content-Disposition", contentDisposition.toString());
              response.setHeader("Content-Lenght", attachment.getFileSize() + "");
            }
            
            byte[] bytes = attachment.getContentModified(Authentication.getIContact());//LoadContent.getContent(attachment);
            if (bytes != null) {
              response.getOutputStream().write(bytes);
              SystemLogger.getInstance().doLog(SystemLogType.LOG_INFO, "/attachment/" + attachment.getId(), "downloaded", null);
            }
          } catch (IOException | DocumentException ex) {
            Logger.getLogger(AttachDownload.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
          }
        }
      }
    }
  }
 
}