AttachAction.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2020 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 java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Date;
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.turro.string.Strings;
import org.apache.commons.fileupload.FileItemIterator;
import org.apache.commons.fileupload.FileItemStream;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.fileupload.util.Streams;
import org.turro.action.Actions;
import org.turro.attach.db.AttachPU;
import org.turro.attach.entity.AttachContent;
import org.turro.attach.entity.Attachment;
import org.turro.auth.Authentication;
import org.turro.collections.KeyValueMap;
import org.turro.elephant.context.Application;
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.http.HttpUtil;
import org.turro.jpa.Dao;
import org.turro.log.SystemLogType;
import org.turro.log.SystemLogger;
import org.turro.marker.ElephantMarker;
import org.turro.plugin.contacts.IContact;
 
@DirectContent(identifier="attach-action")
public class AttachAction implements IDirectContent {
 
  public static String newAttachment(String entityPath, String link, String template) {
    IConstructor constructor = Application.getApplication().getConstructor();
    if(constructor.isInRole("attach:new")) {
      ElephantMarker marker = new ElephantMarker(constructor);
      marker.put("action", DirectContents.createRelativeURL(getIdentifier()));
      marker.put("entityPath", entityPath);
      marker.put("redirect", link);
      return marker.parse("attachment", Strings.isBlank(template) ? "newAttachment" : template);
    }
    return "";
  }
 
  public static String delAttachment(Attachment attachment, String link) {
    IConstructor constructor = Application.getApplication().getConstructor();
    return createURL(constructor, "action=delete;id=" + attachment.getId() + ";redirect=" + HttpUtil.encodeURL(link));
  }
 
  public static String createURL(IConstructor constructor, String values) {
    String exrino = Actions.createRightNowAction(values);
    return ElephantContext.getRootWebPath() +
            DirectContents.DIRECT_CONTENT_PATH + getIdentifier() +
            "?" + exrino;
  }
 
  public static String getIdentifier() {
    return AttachAction.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) {
    if(ServletFileUpload.isMultipartContent(request)) {
      try {
        processUpload(request, response);
      } catch (FileUploadException | IOException ex) {
        Logger.getLogger(AttachAction.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
      }
    } else {
      try {
        processAction(request, response);
      } catch (IOException ex) {
        Logger.getLogger(AttachAction.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
      }
    }
  }
 
  private void processUpload(HttpServletRequest request, HttpServletResponse response) throws FileUploadException, IOException {
    IConstructor constructor = Application.getApplication().getConstructor();
    IContact contact = Authentication.getIContact();
    if(contact == null || !contact.isValid()) {
      return;
    }
    String redirect = null, entityPath = null, comment = null;
    ServletFileUpload upload = new ServletFileUpload();
    upload.setHeaderEncoding(ElephantContext.getEncoding());
    FileItemIterator iter = upload.getItemIterator(request);
    while(iter.hasNext()) {
      FileItemStream item = iter.next();
      String name = item.getFieldName();
      InputStream stream = item.openStream();
      if(item.isFormField()) {
        switch(name) {
          case "redirect":
            redirect = Streams.asString(stream, ElephantContext.getEncoding());
            break;
          case "entityPath":
            entityPath = Streams.asString(stream, ElephantContext.getEncoding());
            break;
          case "comment":
            comment = Streams.asString(stream, ElephantContext.getEncoding());
            break;
        }
      } else if(!Strings.isBlank(entityPath)) {
        Attachment attachment = new Attachment();
        attachment.setModification(new Date());
        attachment.setPath(entityPath);
        attachment.setOwner(contact.getId());
        attachment.setOnlyOwner(false);
        attachment.setComment(comment);
        attachment.setShowKey(null);
        attachment.setPublishable(true);
        AttachContent ac = new AttachContent();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        Streams.copy(item.openStream(), baos, true);
        ac.setFileContent(baos.toByteArray());
        attachment.setAttachContent(ac);
        attachment.setFileName(convertToFileName(item.getName()));
        attachment.setFileContentType(item.getContentType());
        attachment.setFileSize(ac.getFileContent().length);
        attachment.setValidated(Application.getApplication().isInRole("attach:self-validate"));
        if(attachment.getFileSize() > 0) {
          attachment = new AttachPU().saveObject(attachment);
          SystemLogger.getInstance().doLog(SystemLogType.LOG_INFO, attachment, "uploaded", null);
        }
      }
    }
    if(!Strings.isBlank(redirect)) {
      constructor.redirect(redirect);
    }
  }
  
  private String convertToFileName(String fileName) {
    if(fileName.contains("/")) {
      fileName = fileName.substring(fileName.lastIndexOf("/") + 1);
    }
    return fileName;
  }
 
  private void processAction(HttpServletRequest request, HttpServletResponse response) throws IOException {
    IConstructor constructor = ElephantContext.getConstructor(request, response);
    KeyValueMap map = Actions.getRightNowAction(constructor);
    if(map != null) {
      String action = map.get("action");
      String redirect = map.get("redirect");
      if("delete".equals(action)) {
        Dao dao = new AttachPU();
        Attachment attachment = dao.find(Attachment.class, map.get(Long.class, "id"));
        if(attachment != null) {
          dao.deleteObject(attachment);
        }
      }
      if(!Strings.isBlank(redirect)) {
        constructor.redirect(HttpUtil.decodeURL(redirect));
      }
    }
  }
  
}