AttachmentUtil.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 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;
 
import java.io.IOException;
import java.util.Collection;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.turro.string.Strings;
import org.turro.attach.db.AttachPU;
import org.turro.attach.entity.AttachContent;
import org.turro.attach.entity.Attachment;
import org.turro.attach.version.AttachVersionSet;
import org.turro.elephant.util.Messages;
import org.turro.http.Respond;
import org.turro.jpa.Dao;
import org.turro.jpa.DaoTransaction;
import org.turro.log.SystemLogger;
import org.turro.log.WebLoggers;
import org.turro.zul.frame.Framework;
import org.zkoss.zul.Filedownload;
 
public class AttachmentUtil {
 
  public static boolean download(Attachment attachment) {
    boolean done = false;
//    Dao dao = new AttachPU();
//    dao.startTrans();
    try(DaoTransaction transaction = new DaoTransaction(new AttachPU())) {
//      attachment = dao.saveObjectWithTrans(attachment);
      attachment = transaction.saveObject(attachment);
      Filedownload.save(attachment.getAttachContent().getFileContent(), attachment.getFileContentType(), attachment.getFileName());
      done = true;
//    } finally {
//      dao.endTrans();
      SystemLogger.info().entity(attachment).comment("downloaded").log();
    }
    return done;
  }
 
  public static Attachment withContent(Attachment attachment) {
//    Dao dao = new AttachPU();
//    dao.startTrans();
    try(DaoTransaction transaction = new DaoTransaction(new AttachPU())) {
      attachment = transaction.saveObject(attachment);
      AttachContent content = attachment.getAttachContent();
      content.getFileContent();
//    } finally {
//      dao.endTrans();
    }
    return attachment;
  }
 
  public static Attachment withContent(Long id) {
    try(DaoTransaction transaction = new DaoTransaction(new AttachPU())) {
      Attachment attachment = transaction.find(Attachment.class, id);
      AttachContent content = attachment.getAttachContent();
      content.getFileContent();
      return attachment;
    }
  }
 
  public static boolean respond(Respond respond, Long id) {
    try {
      Attachment attachment = withContent(id);
      respond.name(attachment.getFileName())
              .type(attachment.getFileContentType())
              .lenght(attachment.getFileSize())
              .flush(attachment.getAttachContent().getFileContent());
      return true;
    } catch (IOException ex) {
      WebLoggers.severe(AttachmentUtil.class).exception(ex).log();
      return false;
    }    
  }
  
  public static void paste(String toPath) {
    pasteAttachment(toPath);
    pasteFolder(toPath);
    clearClipboard();
  }
 
  public static Collection<Attachment> delete(String path) {
    Dao dao = new AttachPU();
    Collection<Attachment> list = (Collection<Attachment>) dao.getResultList(
            "select a from Attachment a where a.path like ? or a.path = ?",
            new Object[] {path + "/%", path});
    for(Attachment a : list) {
      dao.deleteObject(a);
    }
    return list;
  }
  
  public static void copyAttachment(Object value) {
    clearClipboard();
    Framework.getClipboard().put("bba_clip_attach_copy", value);
  }
 
  public static void cutAttachment(String name, final Object value) {
    Messages.confirmCutting().add(name).show(() -> {
      clearClipboard();
      Framework.getClipboard().put("bba_clip_attach_cut", value);
    });
  }
 
  private static void pasteAttachment(String toPath) {
    boolean cut = false;
    AttachVersionSet avs = (AttachVersionSet) Framework.getClipboard().get("bba_clip_attach_copy");
    if (avs == null) {
      avs = (AttachVersionSet) Framework.getClipboard().get("bba_clip_attach_cut");
      cut = true;
    }
    if (avs != null) {
      Dao dao = new AttachPU();
      for (Attachment a : avs) {
        if (cut) {
          a.setPath(Strings.isEmpty(toPath) ? "/" : toPath);
          dao.saveObject(a);
        } else {
          Attachment na = a.cloneEntity();
          na.setPath(Strings.isEmpty(toPath) ? "/" : toPath);
          dao.saveObject(na);
        }
      }
    }
  }
 
  public static void copyFolder(Object value) {
    clearClipboard();
    Framework.getClipboard().put("bba_clip_attach_folder_copy", value);
  }
 
  public static void cutFolder(String name, final Object value) {
    Messages.confirmCutting().add(name).show(() -> {
      clearClipboard();
      Framework.getClipboard().put("bba_clip_attach_folder_cut", value);
    });
  }
 
  private static void pasteFolder(String toPath) {
    boolean cut = false;
    String path = (String) Framework.getClipboard().get("bba_clip_attach_folder_copy");
    if(Strings.isBlank(path)) {
      path = (String) Framework.getClipboard().get("bba_clip_attach_folder_cut");
      cut = true;
    }
    if(!Strings.isBlank(path)) {
      Dao dao = new AttachPU();
      for(Attachment a : (Collection<Attachment>) dao.getResultList("select a from Attachment a where a.path like ? or a.path = ?",
              new Object[] {path + "/%", path})) {
        if(cut) {
          a.setPath(convertPath(path, a.getPath(), toPath));
          dao.saveObject(a);
        } else {
          Attachment na = a.cloneEntity();
          na.setPath(convertPath(path, a.getPath(), toPath));
          dao.saveObject(na);
        }
      }
    }
  }
 
  public static boolean hasClipboard() {
    return Framework.getClipboard().get("bba_clip_attach_folder_copy") != null ||
           Framework.getClipboard().get("bba_clip_attach_folder_cut") != null ||
           Framework.getClipboard().get("bba_clip_attach_copy") != null ||
           Framework.getClipboard().get("bba_clip_attach_cut") != null;
  }
 
  public static void clearClipboard() {
    Framework.getClipboard().remove("bba_clip_attach_folder_copy");
    Framework.getClipboard().remove("bba_clip_attach_folder_cut");
    Framework.getClipboard().remove("bba_clip_attach_copy");
    Framework.getClipboard().remove("bba_clip_attach_cut");
  }
 
  private static String convertPath(String root, String path, String toPath) {
    String reminder = path.substring(root.length());
    return (Strings.isEmpty(toPath) ? "/" : toPath) + reminder;
  }
 
  private AttachmentUtil() {
  }
 
}