From ecaec1d44aa73024949e56ea9be2a0a753a0d1df Mon Sep 17 00:00:00 2001
From: Matt Nadareski <mnadareski@outlook.com>
Date: Tue, 16 Apr 2024 20:02:54 -0400
Subject: [PATCH] Add single-stream constructor to ROCS

---
 .../ReadOnlyCompositeStreamTests.cs             |  8 ++++++++
 .../Streams/ReadOnlyCompositeStream.cs          | 17 +++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/SabreTools.IO.Test/ReadOnlyCompositeStreamTests.cs b/SabreTools.IO.Test/ReadOnlyCompositeStreamTests.cs
index aa607c9..2454f9b 100644
--- a/SabreTools.IO.Test/ReadOnlyCompositeStreamTests.cs
+++ b/SabreTools.IO.Test/ReadOnlyCompositeStreamTests.cs
@@ -35,6 +35,14 @@ namespace SabreTools.IO.Test
             Assert.Equal(0, stream.Position);
         }
 
+        [Fact]
+        public void SingleStreamConstructorTest()
+        {
+            var stream = new ReadOnlyCompositeStream(new MemoryStream(new byte[1024]));
+            Assert.Equal(1024, stream.Length);
+            Assert.Equal(0, stream.Position);
+        }
+
         [Fact]
         public void FilledArrayConstructorTest()
         {
diff --git a/SabreTools.IO/Streams/ReadOnlyCompositeStream.cs b/SabreTools.IO/Streams/ReadOnlyCompositeStream.cs
index aa5546b..0063975 100644
--- a/SabreTools.IO/Streams/ReadOnlyCompositeStream.cs
+++ b/SabreTools.IO/Streams/ReadOnlyCompositeStream.cs
@@ -69,6 +69,23 @@ namespace SabreTools.IO.Streams
             _position = 0;
         }
 
+        /// <summary>
+        /// Create a new ReadOnlyCompositeStream from a single Stream
+        /// </summary>
+        /// <param name="stream"></param>
+        public ReadOnlyCompositeStream(Stream stream)
+        {
+            _streams = [stream];
+            _length = 0;
+            _position = 0;
+
+            // Verify the stream and add to the length
+            if (!stream.CanRead || !stream.CanSeek)
+                throw new ArgumentException($"{nameof(stream)} needs to be readable and seekable");
+
+            _length += stream.Length;
+        }
+
         /// <summary>
         /// Create a new ReadOnlyCompositeStream from an existing collection of Streams
         /// </summary>