AttachFolder.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.navigator;
 
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Collection;
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.attach.search.AttachResults;
import org.turro.attach.version.AttachVersionMap;
import org.turro.attach.zul.FolderNameCombobox;
import org.turro.collections.CollectionUtil;
import org.turro.command.Command;
import org.turro.command.Context;
import org.turro.elephant.context.Application;
import org.turro.elephant.impl.util.StringParser;
import org.turro.elephant.util.Messages;
import org.turro.entities.Entities;
import org.turro.i18n.I_;
import org.turro.log.SystemLogType;
import org.turro.log.SystemLogger;
import org.turro.plugin.attach.IAttachFolder;
import org.turro.zkoss.dialog.InputDialog;
import org.turro.zkoss.dialog.InputField;
import org.zkoss.lang.Strings;
import org.zkoss.util.media.Media;
import org.zkoss.zk.ui.HtmlBasedComponent;
import org.zkoss.zk.ui.event.DropEvent;
import org.zkoss.zk.ui.event.Event;
import org.zkoss.zk.ui.event.EventListener;
import org.zkoss.zk.ui.event.Events;
import org.zkoss.zk.ui.ext.AfterCompose;
import org.zkoss.zul.*;
 
public class AttachFolder extends Treeitem implements IAttachFolder, AfterCompose {
 
  private Application app = Application.getApplication();
 
  private Treechildren children;
  private String label, realLabel;
  private String path;
  private boolean loadOnDemand;
 
  public AttachFolder(String label, String path, boolean loadOnDemand) {
    super();
    this.label = label;
    this.realLabel = label;
    this.path = path;
    this.loadOnDemand = loadOnDemand;
    this.setTooltiptext(label);
  }
 
  public AttachFolder(String label, Object value) {
    throw new UnsupportedOperationException("Do not use");
  }
 
  public AttachFolder(String label) {
    throw new UnsupportedOperationException("Do not use");
  }
 
  public AttachFolder() {
    throw new UnsupportedOperationException("Do not use");
  }
 
  @Override
  public AttachTree getTree() {
    return (AttachTree) super.getTree();
  }
 
  public void doUpload() throws InterruptedException, IOException {
    addMedias(Fileupload.get(-1, true));
  }
 
  public void addMedias(Media[] medias) throws InterruptedException, IOException {
    if(medias == null || medias.length == 0) return;
    fillFolder();
    setOpen(true);
    for(Media media : medias) {
      Attachment attachment = new Attachment();
      attachment.setModification(new Date());
      attachment.setPath(path);
      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);
      if(getTree().getAttachListener() != null) {
        getTree().getAttachListener().uploadDone(attachment);
      }
      reloadContents();
    }
  }
 
  public void doAddFolder() throws InterruptedException {
    InputDialog.getInput(
            getPage(),
            I_.get("Folder"),
            new InputField[] {
              new InputField("Name", "", null, 0) {
                @Override
                protected HtmlBasedComponent createEditor() {
                  return new FolderNameCombobox(AttachFolder.this);
                }
              }
            }, new Command() {
      @Override
      public Object execute(Context context) {
        InputField[] fields = (InputField[]) context.get("fields");
        if(fields.length > 0) {
          for(InputField f : fields) {
            if("Name".equals(f.getLabel())) {
              fillFolder();
              AttachFolder af = AttachFolder.this;
              for(String s : ((String) f.getValue()).split("\\/")) {
                if(!Strings.isBlank(s)) {
                  af.setOpen(true);
                  af = af.addFolder(s, af.getPath() + "/" + s);
                }
              }
            }
          }
        }
        return null;
      }
    });
  }
 
  public AttachFolder addFolder(String label, String path) {
    AttachFolder af = new AttachFolder(label, path, loadOnDemand);
    if(children == null) {
      children = new Treechildren();
      appendChild(children);
    }
    children.appendChild(af);
    af.afterCompose();
    af.doProcessors();
    return af;
  }
 
  public void showContents() {
    fillFolder();
    setOpen(true);
  }
 
  public void reloadContents() {
    if(children != null) {
      children.getChildren().clear();
      showContents();
    }
  }
 
  private void addCells() {
    Treerow row = new Treerow();
    row.setDroppable("true");
    row.addEventListener(Events.ON_DROP, new EventListener() {
      @Override
      public void onEvent(Event event) throws Exception {
        final DropEvent dp = (DropEvent) event;
        if(dp.getDragged() instanceof AttachmentRow) {
          Messages.confirmMove().show(() -> {
            fillFolder();
            AttachmentRow row = (AttachmentRow) dp.getDragged();
            for(Attachment a : row.getAttachmentSet()) {
              a.setPath(Strings.isEmpty(getPath()) ? "/" : getPath());
              a = new AttachPU().saveObject(a);
            }
            row.detach();
          });
        }
      }
    });
    appendChild(row);
    Treecell cell = new Treecell(label);
    cell.setImage("/_zul/images/folder.png");
    row.appendChild(cell);
  }
 
  private void addChildrenSpace() {
    if(!isLeaf()) {
      children = new Treechildren();
      appendChild(children);
      if(!loadOnDemand) {
        fillFolder();
        setOpen(true);
      }
    }
  }
 
  public boolean isLoadOnDemand() {
    return loadOnDemand;
  }
 
  public void setLoadOnDemand(boolean loadOnDemand) {
    this.loadOnDemand = loadOnDemand;
  }
 
  public String getPath() {
    return path;
  }
 
  public void setPath(String path) {
    this.path = path;
  }
 
  private void initLoadOnDemand() {
    if(!loadOnDemand) return;
 
    setOpen(false);
 
    addEventListener(Events.ON_OPEN, new EventListener() {
      @Override
      public void onEvent(Event event) throws Exception {
        // load files and add folders
        if(isOpen()) {
          fillFolder();
        }
      }
    });
  }
 
  private void fillFolder() {
    if(children != null && children.getChildren().isEmpty()) {
      for(String s : getFolders()) {
        addFolder(s, path + "/" + s);
      }
    }
  }
 
  @Override
  public IAttachFolder getParentFolder() {
    return (IAttachFolder) getParentItem();
  }
 
  @Override
  public String getFolderPath() {
    return path;
  }
 
  @Override
  public String getFolderLabel() {
    return realLabel;
  }
 
  @Override
  public void setFolderLabel(String label) {
    setLabel(label);
  }
 
  public void doProcessors() {
    String str = Entities.getController(getPath()).getTreeLabel();
    if(str != null) {
      setFolderLabel(StringParser.cutString(str, 50));
      setTooltiptext(str);
    }
  }
 
  public boolean isLeaf() {
    return getFolders().isEmpty();
  }
 
  public Collection<String> getFolders() {
    AttachResults results = getTree().getResults();
    results.setAttachPath(path + "/_%");
    results.setCkExactPath(false);
    return results.getPathList(path);
  }
 
  public Collection<AttachFolder> getAttachFolders() {
    showContents();
    return children != null ? CollectionUtil.from(children.getChildren()).<Collection<AttachFolder>>cast() : new ArrayList<>();
  }
 
  public Collection<Attachment> getAttachments() {
    AttachResults results = getTree().getResults();
    results.setAttachPath(Strings.isEmpty(path) ? "/" : path);
    results.setCkExactPath(true);
    return results.getAttachmentList();
  }
 
  public AttachVersionMap getVersionMap() {
    AttachVersionMap avm = new AttachVersionMap();
    for(Attachment a : getAttachments()) {
      avm.addAttachment(a);
    }
    return avm;
  }
 
  public Collection<Attachment> getAllAttachments() {
    AttachResults results = getTree().getResults();
    results.setAttachPath(Strings.isEmpty(path) ? "/" : path + "/%");
    results.setCkExactPath(false);
    return results.getAttachmentList();
  }
 
  @Override
  public void afterCompose() {
    initLoadOnDemand();
    addCells();
    addChildrenSpace();
  }
 
}