AttachmentButton.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2019 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.zul.control;
 
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.Date;
import org.apache.commons.io.input.ReaderInputStream;
import org.turro.attach.db.AttachPU;
import org.turro.attach.entity.AttachContent;
import org.turro.attach.entity.Attachment;
import org.turro.elephant.context.Application;
import org.turro.entities.Entities;
import org.turro.log.SystemLogType;
import org.turro.log.SystemLogger;
import org.zkoss.util.media.Media;
import org.zkoss.zk.ui.event.EventListener;
import org.zkoss.zk.ui.event.Events;
import org.zkoss.zk.ui.event.UploadEvent;
import org.zkoss.zul.Button;
 
public class AttachmentButton extends Button implements EventListener<UploadEvent> {
 
  private Object entity;
  private String entityPath;
  
  public AttachmentButton() {
    addEventListener(Events.ON_UPLOAD, this);
  }
  
  public void setEntity(Object entity) {
    this.entity = entity;
    setEntityPath(Entities.getController(entity).getPath());
  }
  
  public void setEntityPath(String entityPath) {
    this.entityPath = entityPath;
  }
  
  @Override
  public void onEvent(UploadEvent event) throws Exception {
    Application app = Application.getApplication();
    Media[] medias = event.getMedias();
    if(medias == null || medias.length == 0) return;
    for(Media media : medias) {
      Attachment attachment = new Attachment();
      attachment.setModification(new Date());
      attachment.setPath(entityPath);
      attachment.setOwner(app.getConstructor().getUser().getId());
      attachment.setOnlyOwner(false);
      attachment.setComment("");
      attachment.setShowKey(null);
      attachment.setPublishable(true);
      AttachContent ac = new AttachContent();
      if(media.inMemory()) {
        ac.setFileContent(media.isBinary() ? media.getByteData() : media.getStringData().getBytes());
      } else {
        byte[] buffer = new byte[102400];
        InputStream is = media.isBinary() ? media.getStreamData() : new ReaderInputStream(media.getReaderData(), Charset.defaultCharset());
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        int r;
        while((r = is.read(buffer)) != -1) {
          baos.write(buffer, 0, r);
        }
        is.close();
        ac.setFileContent(baos.toByteArray());
      }
      attachment.setAttachContent(ac);
      attachment.setFileName(media.getName());
      attachment.setFileContentType(media.getContentType());
      attachment.setFileSize(ac.getFileContent().length);
      attachment.setValidated(app.isInRole("attach:self-validate"));
      attachment = new AttachPU().saveObject(attachment);
      SystemLogger.getInstance().doLog(SystemLogType.LOG_INFO, attachment, "uploaded", null);
    }
  }
 
}