Add RecordingInputStream.asOutputStream()

CHANGES.md

@@ -1,6 +1,10 @@
 2.0.0
 -----
 
+### New features
+
+- Added `RecordingInputStream.asOutputStream()` for direct writing of recorded data without an input stream. [#108](https://github.com/iipc/webarchive-commons/pull/108)
+
 ### Removals
 
 #### Removed Apache HttpClient 3.1

src/main/java/org/archive/io/RecordingInputStream.java

@@ -383,12 +383,12 @@ public synchronized void mark(int readlimit) {
 
     @Override
     public boolean markSupported() {
-        return this.in.markSupported(); 
+        return in != null && this.in.markSupported();
     }
 
     @Override
     public synchronized void reset() throws IOException {
-        this.in.reset();
+        if (in != null) this.in.reset();
         this.recordingOutputStream.reset();
     }
 
@@ -418,4 +418,13 @@ public void chopAtMessageBodyBegin() {
     public void clearForReuse() throws IOException {
         recordingOutputStream.clearForReuse();
     }
+
+    /**
+     * Returns an OutputStream that can be used for recording input data. This is useful if the input comes in some
+     * form other than an InputStream. For example, if the input is provided by a callback periodically called with
+     * a chunk of data.
+     */
+    public RecordingOutputStream asOutputStream() {
+        return this.recordingOutputStream;
+    }
 }

src/test/java/org/archive/io/RecordingInputStreamTest.java

@@ -41,7 +41,6 @@ public class RecordingInputStreamTest {
     @TempDir
     File tempDir;
 
-
     /**
      * Test readFullyOrUntil soft (no exception) and hard (exception) 
      * length cutoffs, timeout, and rate-throttling. 
@@ -128,4 +127,16 @@ public void run() {
         }.start();
 
     }
+
+    @Test
+    public void testAsOutputStream() throws IOException {
+        RecordingInputStream ris = new RecordingInputStream(16384, (new File(
+                tempDir, "testAsOutputStream").getAbsolutePath()));
+        ris.open(null);
+        ris.asOutputStream().write("hello".getBytes());
+        ris.close();
+        ByteArrayOutputStream baos = new ByteArrayOutputStream();
+        ris.getReplayInputStream().readFullyTo(baos);
+        assertEquals("hello", baos.toString());
+    }
 }