AttachWd.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.wd.entities;
 
import java.io.*;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.List;
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.search.AttachHeadlessResults;
import org.turro.attach.version.AttachVersionMap;
import org.turro.attach.zul.AttachmentUtil;
import org.turro.elephant.context.ElephantContext;
import org.turro.wd.common.entities.ServerFile;
import org.turro.wd.common.entities.WdFormats;
 
public class AttachWd {
 
  public static Attachment saveFromFile(File file, String fileName, String fileDate, String path, 
          String user, String comment, boolean lock, boolean validated) {
    InputStream is = null;
    try {
      Attachment attachment = new Attachment();
      attachment.setFileName(fileName);
      attachment.setModification(WdFormats.parse(fileDate));
      attachment.setPath(path);
      attachment.setOwner(user);
      attachment.setOnlyOwner(false);
      attachment.setComment(comment);
      attachment.setShowKey(null);
      if(lock) {
        attachment.setLocked(true);
        attachment.setLocker(user);
      }
      AttachContent ac = new AttachContent();
      byte[] buffer = new byte[102400];
      is = new FileInputStream(file);
      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.setFileContentType(WdFormats.getContentType(file));
      attachment.setFileSize(ac.getFileContent().length);
      attachment.setValidated(validated);
      return new AttachPU().saveObject(attachment);
    } catch (ParseException ex) {
      Logger.getLogger(AttachWd.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    } catch (IOException ex) {
      Logger.getLogger(AttachWd.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    } finally {
      try {
        if(is != null) is.close();
      } catch (IOException ex) {
        Logger.getLogger(AttachWd.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
      }
    }
    return null;
  }
 
  public static File convertToFile(String path, String fileName, String user, boolean forEditing) {
    try {
      Attachment attachment = queryVersionedAttachment(path, fileName, user);
      if(forEditing) {
        if(attachment.isLocked()) return null;
        attachment.setLocked(true);
        attachment.setLocker(user);
        attachment = new AttachPU().saveObject(attachment);
      }
      attachment = AttachmentUtil.withContent(attachment);
      File tempFile = File.createTempFile(AttachWd.class.getName(), null);
      FileOutputStream outputStream = new FileOutputStream(tempFile);
      try {
        ByteArrayInputStream inputStream = new ByteArrayInputStream(attachment.getAttachContent().getFileContent());
        for(int data = inputStream.read(); data != -1; data = inputStream.read()) {
            outputStream.write((byte)data);
        }
      } finally {
          outputStream.close();
      }
      tempFile.setLastModified(attachment.getModification().getTime());
      return tempFile;
    } catch (IOException ex) {
      Logger.getLogger(AttachWd.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
    return null;
  }
 
  public static void writeToFile(Attachment attachment, File file) {
    try {
      attachment = AttachmentUtil.withContent(attachment);
      FileOutputStream outputStream = new FileOutputStream(file);
      try {
        ByteArrayInputStream inputStream = new ByteArrayInputStream(attachment.getAttachContent().getFileContent());
        for(int data = inputStream.read(); data != -1; data = inputStream.read()) {
            outputStream.write((byte)data);
        }
      } finally {
        outputStream.flush();
        outputStream.close();
      }
      file.setLastModified(attachment.getModification().getTime());
    } catch (IOException ex) {
      Logger.getLogger(AttachWd.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    }
  }
 
  public static List<String> queryFolders(String path, String user) {
    AttachHeadlessResults results = new AttachHeadlessResults();
    results.setAttachValue("*");
    results.setAttachPath(path + "/_%");
    results.setCkExactPath(false);
    return new ArrayList<String>(results.getPathList(path));
  }
 
  public static List<ServerFile> queryAttachments(String path, String user) {
    AttachHeadlessResults results = new AttachHeadlessResults();
    results.setAttachValue("*");
    results.setAttachPath(Strings.isEmpty(path) ? "/" : path);
    results.setCkExactPath(true);
    return new ServerFileAdapter(results.getAttachmentList());
  }
 
  public static List<ServerFile> queryVersionedAttachments(String path, String user) {
    AttachHeadlessResults results = new AttachHeadlessResults();
    results.setAttachValue("*");
    results.setAttachPath(Strings.isEmpty(path) ? "/" : path);
    results.setCkExactPath(true);
    AttachVersionMap avm = new AttachVersionMap();
    for(Attachment a : results.getAttachmentList()) {
      avm.addAttachment(a);
    }
    return new ServerFileAdapter(avm.getVersionedAttachments());
  }
 
  public static Attachment queryVersionedAttachment(String path, String fileName, String user) {
    AttachHeadlessResults results = new AttachHeadlessResults();
    results.setAttachValue(fileName);
    results.setAttachPath(Strings.isEmpty(path) ? "/" : path);
    results.setCkExactPath(true);
    AttachVersionMap avm = new AttachVersionMap();
    for(Attachment a : results.getAttachmentList()) {
      avm.addAttachment(a);
    }
    return avm.isEmpty() ? null : avm.get(fileName).first();
  }
 
  public static Object unlockFile(String path, String fileName, String user) {
    Attachment attachment = queryVersionedAttachment(path, fileName, user);
    if(attachment != null && attachment.isLocked()) {
      attachment.setLocked(false);
      attachment.setLocker(null);
      attachment.setLockComment(null);
      attachment.setLockDate(null);
      attachment.setLockCommit(null);
      attachment = new AttachPU().saveObject(attachment);
    }
    ServerFile sf = new ServerFile();
    if(attachment != null) {
      sf.setFileContentType(attachment.getFileContentType());
      sf.setFileName(attachment.getFileName());
      sf.setFileSize(attachment.getFileSize());
      sf.setLocked(attachment.isLocked());
      sf.setLocker(attachment.getLocker());
      sf.setModification(attachment.getModification());
      sf.setOwner(attachment.getOwner());
      sf.setPath(attachment.getPath());
    }
    return sf;
  }
 
  private AttachWd() {
  }
 
}