mirror of
https://github.com/itflow-org/itflow
synced 2026-08-04 22:57:14 +00:00
244 lines
5.8 KiB
PHP
244 lines
5.8 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace GuzzleHttp\Psr7;
|
|
|
|
use Psr\Http\Message\StreamInterface;
|
|
|
|
/**
|
|
* Reads from multiple streams, one after the other.
|
|
*
|
|
* This is a read-only stream decorator.
|
|
*/
|
|
final class AppendStream implements StreamInterface
|
|
{
|
|
use NonSerializableStreamTrait;
|
|
|
|
/** @var StreamInterface[] Streams being decorated */
|
|
private array $streams = [];
|
|
|
|
private bool $seekable = true;
|
|
|
|
private int $current = 0;
|
|
|
|
private int $pos = 0;
|
|
|
|
/**
|
|
* @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(): string
|
|
{
|
|
$this->rewind();
|
|
|
|
return $this->getContents();
|
|
}
|
|
|
|
/**
|
|
* 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): void
|
|
{
|
|
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(): string
|
|
{
|
|
return Utils::copyToString($this);
|
|
}
|
|
|
|
/**
|
|
* Closes each attached stream.
|
|
*/
|
|
public function close(): void
|
|
{
|
|
$this->pos = $this->current = 0;
|
|
$this->seekable = true;
|
|
|
|
foreach ($this->streams as $stream) {
|
|
$stream->close();
|
|
}
|
|
|
|
$this->streams = [];
|
|
}
|
|
|
|
/**
|
|
* Detaches each attached stream.
|
|
*
|
|
* Returns null as it's not clear which underlying stream resource to return.
|
|
*/
|
|
public function detach()
|
|
{
|
|
$this->pos = $this->current = 0;
|
|
$this->seekable = true;
|
|
|
|
foreach ($this->streams as $stream) {
|
|
$stream->detach();
|
|
}
|
|
|
|
$this->streams = [];
|
|
|
|
return null;
|
|
}
|
|
|
|
public function tell(): int
|
|
{
|
|
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.
|
|
*/
|
|
public function getSize(): ?int
|
|
{
|
|
$size = 0;
|
|
|
|
foreach ($this->streams as $stream) {
|
|
$s = $stream->getSize();
|
|
if ($s === null) {
|
|
return null;
|
|
}
|
|
$size = Integers::add($size, $s);
|
|
}
|
|
|
|
return $size;
|
|
}
|
|
|
|
public function eof(): bool
|
|
{
|
|
return !$this->streams
|
|
|| ($this->current >= count($this->streams) - 1
|
|
&& $this->streams[$this->current]->eof());
|
|
}
|
|
|
|
public function rewind(): void
|
|
{
|
|
$this->seek(0);
|
|
}
|
|
|
|
/**
|
|
* Attempts to seek to the given position. Only supports SEEK_SET.
|
|
*/
|
|
public function seek(int $offset, int $whence = SEEK_SET): void
|
|
{
|
|
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.
|
|
*/
|
|
public function read(int $length): string
|
|
{
|
|
if ($length < 0) {
|
|
throw new \RuntimeException('Length parameter cannot be negative');
|
|
}
|
|
|
|
if ($this->streams === []) {
|
|
return '';
|
|
}
|
|
|
|
$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 = StreamTimeout::read($this->streams[$this->current], $remaining, 'Unable to read from stream: timed out');
|
|
|
|
if ($result === '') {
|
|
$progressToNext = true;
|
|
continue;
|
|
}
|
|
|
|
$buffer .= $result;
|
|
$remaining = $length - strlen($buffer);
|
|
}
|
|
|
|
$this->pos = Integers::add($this->pos, strlen($buffer));
|
|
|
|
return $buffer;
|
|
}
|
|
|
|
public function isReadable(): bool
|
|
{
|
|
return true;
|
|
}
|
|
|
|
public function isWritable(): bool
|
|
{
|
|
return false;
|
|
}
|
|
|
|
public function isSeekable(): bool
|
|
{
|
|
return $this->seekable;
|
|
}
|
|
|
|
public function write(string $string): int
|
|
{
|
|
throw new \RuntimeException('Cannot write to an AppendStream');
|
|
}
|
|
|
|
public function getMetadata(?string $key = null): ?array
|
|
{
|
|
return $key === null ? [] : null;
|
|
}
|
|
}
|