234 lines
5.4 KiB
PHP
234 lines
5.4 KiB
PHP
|
<?php
|
||
|
namespace GuzzleHttp\Psr7;
|
||
|
|
||
|
use Psr\Http\Message\StreamInterface;
|
||
|
|
||
|
/**
|
||
|
* Reads from multiple streams, one after the other.
|
||
|
*
|
||
|
* This is a read-only stream decorator.
|
||
|
*/
|
||
|
class AppendStream implements StreamInterface
|
||
|
{
|
||
|
/** @var StreamInterface[] Streams being decorated */
|
||
|
private $streams = [];
|
||
|
|
||
|
private $seekable = true;
|
||
|
private $current = 0;
|
||
|
private $pos = 0;
|
||
|
private $detached = false;
|
||
|
|
||
|
/**
|
||
|
* @param StreamInterface[] $streams Streams to decorate. Each stream must
|
||
|
* be readable.
|
||
|
*/
|
||
|
public function __construct(array $streams = [])
|
||
|
{
|
||
|
foreach ($streams as $stream) {
|
||
|
$this->addStream($stream);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public function __toString()
|
||
|
{
|
||
|
try {
|
||
|
$this->rewind();
|
||
|
return $this->getContents();
|
||
|
} catch (\Exception $e) {
|
||
|
return '';
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Add a stream to the AppendStream
|
||
|
*
|
||
|
* @param StreamInterface $stream Stream to append. Must be readable.
|
||
|
*
|
||
|
* @throws \InvalidArgumentException if the stream is not readable
|
||
|
*/
|
||
|
public function addStream(StreamInterface $stream)
|
||
|
{
|
||
|
if (!$stream->isReadable()) {
|
||
|
throw new \InvalidArgumentException('Each stream must be readable');
|
||
|
}
|
||
|
|
||
|
// The stream is only seekable if all streams are seekable
|
||
|
if (!$stream->isSeekable()) {
|
||
|
$this->seekable = false;
|
||
|
}
|
||
|
|
||
|
$this->streams[] = $stream;
|
||
|
}
|
||
|
|
||
|
public function getContents()
|
||
|
{
|
||
|
return copy_to_string($this);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Closes each attached stream.
|
||
|
*
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
public function close()
|
||
|
{
|
||
|
$this->pos = $this->current = 0;
|
||
|
|
||
|
foreach ($this->streams as $stream) {
|
||
|
$stream->close();
|
||
|
}
|
||
|
|
||
|
$this->streams = [];
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Detaches each attached stream
|
||
|
*
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
public function detach()
|
||
|
{
|
||
|
$this->close();
|
||
|
$this->detached = true;
|
||
|
}
|
||
|
|
||
|
public function tell()
|
||
|
{
|
||
|
return $this->pos;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Tries to calculate the size by adding the size of each stream.
|
||
|
*
|
||
|
* If any of the streams do not return a valid number, then the size of the
|
||
|
* append stream cannot be determined and null is returned.
|
||
|
*
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
public function getSize()
|
||
|
{
|
||
|
$size = 0;
|
||
|
|
||
|
foreach ($this->streams as $stream) {
|
||
|
$s = $stream->getSize();
|
||
|
if ($s === null) {
|
||
|
return null;
|
||
|
}
|
||
|
$size += $s;
|
||
|
}
|
||
|
|
||
|
return $size;
|
||
|
}
|
||
|
|
||
|
public function eof()
|
||
|
{
|
||
|
return !$this->streams ||
|
||
|
($this->current >= count($this->streams) - 1 &&
|
||
|
$this->streams[$this->current]->eof());
|
||
|
}
|
||
|
|
||
|
public function rewind()
|
||
|
{
|
||
|
$this->seek(0);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Attempts to seek to the given position. Only supports SEEK_SET.
|
||
|
*
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
public function seek($offset, $whence = SEEK_SET)
|
||
|
{
|
||
|
if (!$this->seekable) {
|
||
|
throw new \RuntimeException('This AppendStream is not seekable');
|
||
|
} elseif ($whence !== SEEK_SET) {
|
||
|
throw new \RuntimeException('The AppendStream can only seek with SEEK_SET');
|
||
|
}
|
||
|
|
||
|
$this->pos = $this->current = 0;
|
||
|
|
||
|
// Rewind each stream
|
||
|
foreach ($this->streams as $i => $stream) {
|
||
|
try {
|
||
|
$stream->rewind();
|
||
|
} catch (\Exception $e) {
|
||
|
throw new \RuntimeException('Unable to seek stream '
|
||
|
. $i . ' of the AppendStream', 0, $e);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Seek to the actual position by reading from each stream
|
||
|
while ($this->pos < $offset && !$this->eof()) {
|
||
|
$result = $this->read(min(8096, $offset - $this->pos));
|
||
|
if ($result === '') {
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Reads from all of the appended streams until the length is met or EOF.
|
||
|
*
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
public function read($length)
|
||
|
{
|
||
|
$buffer = '';
|
||
|
$total = count($this->streams) - 1;
|
||
|
$remaining = $length;
|
||
|
$progressToNext = false;
|
||
|
|
||
|
while ($remaining > 0) {
|
||
|
|
||
|
// Progress to the next stream if needed.
|
||
|
if ($progressToNext || $this->streams[$this->current]->eof()) {
|
||
|
$progressToNext = false;
|
||
|
if ($this->current === $total) {
|
||
|
break;
|
||
|
}
|
||
|
$this->current++;
|
||
|
}
|
||
|
|
||
|
$result = $this->streams[$this->current]->read($remaining);
|
||
|
|
||
|
// Using a loose comparison here to match on '', false, and null
|
||
|
if ($result == null) {
|
||
|
$progressToNext = true;
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
$buffer .= $result;
|
||
|
$remaining = $length - strlen($buffer);
|
||
|
}
|
||
|
|
||
|
$this->pos += strlen($buffer);
|
||
|
|
||
|
return $buffer;
|
||
|
}
|
||
|
|
||
|
public function isReadable()
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
public function isWritable()
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
public function isSeekable()
|
||
|
{
|
||
|
return $this->seekable;
|
||
|
}
|
||
|
|
||
|
public function write($string)
|
||
|
{
|
||
|
throw new \RuntimeException('Cannot write to an AppendStream');
|
||
|
}
|
||
|
|
||
|
public function getMetadata($key = null)
|
||
|
{
|
||
|
return $key ? null : [];
|
||
|
}
|
||
|
}
|