import java.io.ByteArrayOutputStream;
import java.io.Closeable;
import java.io.File;
-import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import org.apache.poi.poifs.crypt.EncryptionInfo;
import org.apache.poi.poifs.filesystem.DirectoryNode;
import org.apache.poi.poifs.filesystem.DocumentInputStream;
-import org.apache.poi.poifs.filesystem.DocumentNode;
-import org.apache.poi.poifs.filesystem.NPOIFSDocument;
import org.apache.poi.poifs.filesystem.NPOIFSFileSystem;
import org.apache.poi.poifs.filesystem.OPOIFSFileSystem;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
/**
* Writes out the updated standard Document Information Properties (HPSF)
* into the currently open NPOIFSFileSystem
- * TODO Implement in-place update
*
* @throws IOException if an error when writing to the open
* {@link NPOIFSFileSystem} occurs
- * TODO throws exception if open from stream not file
*/
protected void writeProperties() throws IOException {
- throw new IllegalStateException("In-place write is not yet supported");
+ validateInPlaceWritePossible();
+ writeProperties(directory.getFileSystem(), null);
}
/**
mSet.write(bOut);
byte[] data = bOut.toByteArray();
ByteArrayInputStream bIn = new ByteArrayInputStream(data);
-
- // New or Existing?
- // TODO Use a createOrUpdate method for this to be cleaner!
- try {
- DocumentNode propSetNode = (DocumentNode)outFS.getRoot().getEntry(name);
- NPOIFSDocument propSetDoc = new NPOIFSDocument(propSetNode);
- propSetDoc.replaceContents(bIn);
- } catch (FileNotFoundException e) {
- outFS.createDocument(bIn,name);
- }
+
+ // Create or Update the Property Set stream in the POIFS
+ outFS.createOrUpdateDocument(bIn, name);
logger.log(POILogger.INFO, "Wrote property set " + name + " of size " + data.length);
} catch(org.apache.poi.hpsf.WritingNotSupportedException wnse) {
* @since POI 3.15 beta 3
*
* @throws IOException thrown on errors writing to the file
+ * @throws IllegalStateException if this isn't from a writable File
*/
public abstract void write() throws IOException;
return rval;
}
+ /**
+ * Set the contents of a document, creating if needed,
+ * otherwise updating. Returns the created / updated DocumentEntry
+ *
+ * @param name the name of the new or existing DocumentEntry
+ * @param stream the InputStream from which to populate the DocumentEntry
+ *
+ * @return the new or updated DocumentEntry
+ *
+ * @exception IOException
+ */
+
+ public DocumentEntry createOrUpdateDocument(final String name,
+ final InputStream stream)
+ throws IOException
+ {
+ if (! hasEntry(name)) {
+ return createDocument(name, stream);
+ } else {
+ DocumentNode existing = (DocumentNode)getEntry(name);
+ if (_nfilesystem != null) {
+ NPOIFSDocument nDoc = new NPOIFSDocument(existing);
+ nDoc.replaceContents(stream);
+ return existing;
+ } else {
+ // Do it the hard way for Old POIFS...
+ deleteEntry(existing);
+ return createDocument(name, stream);
+ }
+ }
+ }
+
/**
* Gets the storage clsid of the directory entry
*
return getRoot().createDocument(name, stream);
}
- // TODO Add a createOrUpdateDocument method to simplify code
-
/**
* create a new DocumentEntry in the root entry; the data will be
* provided later
return getRoot().createDirectory(name);
}
+ /**
+ * Set the contents of a document in the root directory,
+ * creating if needed, otherwise updating
+ *
+ * @param stream the InputStream from which the document's data
+ * will be obtained
+ * @param name the name of the new or existing POIFSDocument
+ *
+ * @return the new or updated DocumentEntry
+ *
+ * @exception IOException on error populating the POIFSDocument
+ */
+
+ public DocumentEntry createOrUpdateDocument(final InputStream stream,
+ final String name)
+ throws IOException
+ {
+ return getRoot().createOrUpdateDocument(name, stream);
+ }
+
/**
* Does the filesystem support an in-place write via
* {@link #writeFilesystem()} ? If false, only writing out to