org.digidoc4j.impl.StreamDocument Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of digidoc4j Show documentation

DigiDoc4j is a Java library for digitally signing documents and creating digital signature containers of signed documents

There is a newer version: 6.0.0

Show newest version

/* DigiDoc4J library
 *
 * This software is released under either the GNU Library General Public
 * License (see LICENSE.LGPL).
 *
 * Note that the only valid version of the LGPL license as far as this
 * project is concerned is the original GNU Library General Public License
 * Version 2.1, February 1999
 */

package org.digidoc4j.impl;


import eu.europa.esig.dss.enumerations.MimeType;
import eu.europa.esig.dss.model.CommonDocument;
import eu.europa.esig.dss.model.DSSException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

/**
 * @see eu.europa.esig.dss.model.DSSDocument implementation to handle big files. It writes data to temporary
 * files.
 */
public class StreamDocument extends CommonDocument {
  private static final Logger logger = LoggerFactory.getLogger(StreamDocument.class);

  private static final int MAX_SIZE_IN_MEMORY = 1024 * 5;

  protected final File temporaryFile;

  //TODO if file is small enough you can read it into byte[] and cache it

  /**
   * Add javadoc here
   *
   * @param stream       stream
   * @param documentName document Name
   * @param mimeType     mime type
   */
  public StreamDocument(InputStream stream, String documentName, MimeType mimeType) {
    logger.debug("Document name: " + documentName + ", mime type: " + mimeType);
    this.temporaryFile = createTemporaryFileOfStream(stream);
    super.name = documentName;
    super.mimeType = mimeType;
  }

  private static File createTemporaryFileOfStream(InputStream stream) {
    try {
      File temporaryFile;
      temporaryFile = File.createTempFile("digidoc4j", ".tmp");
      temporaryFile.deleteOnExit();

      try (FileOutputStream out = new FileOutputStream(temporaryFile)) {
        byte[] bytes = new byte[MAX_SIZE_IN_MEMORY];
        int result;

        while ((result = stream.read(bytes)) > 0) {
          out.write(bytes, 0, result);
        }
        out.flush();
      }

      return temporaryFile;
    } catch (IOException e) {
      logger.error(e.getMessage());
      throw new DSSException(e);
    }
  }


  @Override
  public InputStream openStream() throws DSSException {
    try {
      return getTemporaryFileAsStream();
    } catch (FileNotFoundException e) {
      logger.error(e.getMessage());
      throw new DSSException(e);
    }
  }

  @Override
  public void setName(String s) {
  }

  @Override
  public MimeType getMimeType() {
    MimeType mimeType = super.getMimeType();
    logger.debug("Mime type: " + mimeType);
    return mimeType;
  }

  @Override
  public void setMimeType(MimeType mimeType) {
    logger.debug("Mime type: " + mimeType);
    super.setMimeType(mimeType);
  }

  public Long getStreamLengthIfKnown() {
    return temporaryFile.length();
  }

  protected FileInputStream getTemporaryFileAsStream() throws FileNotFoundException {
    return new FileInputStream(this.temporaryFile);
  }

}