v3.7

Author: fenric

src/Stream.php

@@ -11,6 +11,8 @@
 
 namespace Sunrise\Http\Message;
 
+use App\Exception\ExceptionFactory;
+use OverflowException;
 use Psr\Http\Message\StreamInterface;
 use Sunrise\Http\Message\Exception\InvalidArgumentException;
 use Sunrise\Http\Message\Exception\RuntimeException;
@@ -204,6 +206,24 @@ public function write($string): int
         return $result;
     }
 
+    /**
+     * @since 3.7.0
+     *
+     * @throws OverflowException
+     */
+    public function writeStream(StreamInterface $stream, int $limit = -1): int
+    {
+        $written = 0;
+        while (!$stream->eof()) {
+            $written += $this->write($stream->read(4096));
+            if ($limit > 0 && $written > $limit) {
+                throw new OverflowException('Maximum stream size exceeded.');
+            }
+        }
+
+        return $written;
+    }
+
     /**
      * @inheritDoc
      */

tests/StreamTest.php

@@ -4,12 +4,14 @@
 
 namespace Sunrise\Http\Message\Tests;
 
+use OverflowException;
 use PHPUnit\Framework\TestCase;
 use Psr\Http\Message\StreamInterface;
 use Sunrise\Http\Message\Exception\InvalidArgumentException;
 use Sunrise\Http\Message\Exception\RuntimeException;
 use Sunrise\Http\Message\Stream;
 
+use Sunrise\Http\Message\StreamFactory;
 use function fclose;
 use function fopen;
 use function is_resource;
@@ -345,6 +347,20 @@ public function testWriteToUnwritableResource(): void
         $testStream->write('foo');
     }
 
+    public function testWriteStream(): void
+    {
+        $source = (new StreamFactory())->createStream('foo');
+        self::assertSame(3, $this->testStream->writeStream($source));
+        $this->assertSame('foo', (string) $this->testStream);
+    }
+
+    public function testWriteTooLargeStream(): void
+    {
+        $source = (new StreamFactory())->createStream('foox');
+        $this->expectException(OverflowException::class);
+        $this->testStream->writeStream($source, 3);
+    }
+
     public function testIsReadable(): void
     {
         $this->assertTrue($this->testStream->isReadable());